How to use register_project method in dbt-osmosis

Best Python code snippet using dbt-osmosis_python

test_admin.py

Source:test_admin.py

...186        self._ac.deregister_user(self._c, 'bob')187        self.assertRaises(exception.UserNotFound,188                          self._ac.describe_user,189                          self._c, 'bob')190    def test_register_project(self):191        self._ac.register_user(self._c, 'bob')192        self.assertEqual('bobs_project',193            self._ac.register_project(self._c,194                                      'bobs_project',195                                      'bob')['projectname'])196    def test_describe_projects(self):197        self._ac.register_user(self._c, 'bob')198        self._ac.register_project(self._c, 'bobs_project', 'bob')199        projects = self._ac.describe_projects(self._c)200        self.assertIn('projectSet', projects)201        self.assertEqual('bobs_project',202                         projects['projectSet'][0]['projectname'])203    def test_deregister_project(self):204        self._ac.register_user(self._c, 'bob')205        self._ac.register_project(self._c, 'bobs_project', 'bob')206        self._ac.deregister_project(self._c, 'bobs_project')207        self.assertRaises(exception.ProjectNotFound,208                          self._ac.describe_project,209                          self._c, 'bobs_project')210    def test_describe_project_members(self):211        self._ac.register_user(self._c, 'bob')212        self._ac.register_project(self._c, 'bobs_project', 'bob')213        members = self._ac.describe_project_members(self._c, 'bobs_project')214        self.assertIn('members', members)215        self.assertEqual('bob', members['members'][0]['member'])216    def test_modify_project(self):217        self._ac.register_user(self._c, 'bob')218        self._ac.register_project(self._c, 'bobs_project', 'bob')219        self._ac.modify_project(self._c, 'bobs_project', 'bob',220                          description='I like cake')221        project = self._ac.describe_project(self._c, 'bobs_project')222        self.assertEqual('I like cake', project['description'])223    def test_modify_project_member_add(self):224        self._ac.register_user(self._c, 'bob')225        self._ac.register_user(self._c, 'mary')226        self._ac.register_project(self._c, 'bobs_project', 'bob')227        self._ac.modify_project_member(self._c, 'mary', 'bobs_project', 'add')228        members = self._ac.describe_project_members(self._c, 'bobs_project')229        self.assertIn('mary', [m['member'] for m in members['members']])230    def test_modify_project_member_remove(self):231        self._ac.register_user(self._c, 'bob')232        self._ac.register_project(self._c, 'bobs_project', 'bob')233        self._ac.modify_project_member(self._c, 'bob', 'bobs_project',234                                       'remove')235        members = self._ac.describe_project_members(self._c, 'bobs_project')236        self.assertNotIn('bob', [m['member'] for m in members['members']])237    def test_modify_project_member_invalid_operation(self):238        self._ac.register_user(self._c, 'bob')239        self._ac.register_project(self._c, 'bobs_project', 'bob')240        self.assertRaises(exception.ApiError,241                          self._ac.modify_project_member,242                          self._c, 'bob', 'bobs_project', 'invalid_operation')243    def test_describe_roles(self):244        self._ac.register_user(self._c, 'bob')245        self._ac.register_project(self._c, 'bobs_project', 'bob')246        roles = self._ac.describe_roles(self._c, 'bobs_project')247        # Default roles ('sysadmin', 'netadmin', 'developer') should be in here248        roles = [r['role'] for r in roles['roles']]249        for role in ('sysadmin', 'netadmin', 'developer'):250            self.assertIn('sysadmin', roles)251    def test_modify_user_role_add(self):252        self._ac.register_user(self._c, 'bob')253        self._ac.register_project(self._c, 'bobs_project', 'bob')254        self._ac.modify_user_role(self._c, 'bob', 'itsec')255        user_roles = self._ac.describe_user_roles(self._c, 'bob')256        self.assertIn('itsec', [r['role'] for r in user_roles['roles']])257    def test_modify_user_role_project_add(self):258        self._ac.register_user(self._c, 'bob')259        self._ac.register_project(self._c, 'bobs_project', 'bob')260        self._ac.modify_user_role(self._c, 'bob', 'developer', 'bobs_project')261        user_roles = self._ac.describe_user_roles(self._c, 'bob',262                                                  'bobs_project')263        self.assertIn('developer', [r['role'] for r in user_roles['roles']])264    def test_modify_user_role_remove(self):265        self._ac.register_user(self._c, 'bob')266        self._ac.register_project(self._c, 'bobs_project', 'bob')267        self._ac.modify_user_role(self._c, 'bob', 'itsec')268        self._ac.modify_user_role(self._c, 'bob', 'itsec', operation='remove')269        user_roles = self._ac.describe_user_roles(self._c, 'bob')270        self.assertNotIn('itsec', [r['role'] for r in user_roles['roles']])271    def test_modify_user_role_project_remove(self):272        self._ac.register_user(self._c, 'bob')273        self._ac.register_project(self._c, 'bobs_project', 'bob')274        self._ac.modify_user_role(self._c, 'bob', 'developer', 'bobs_project')275        self._ac.modify_user_role(self._c, 'bob', 'developer', 'bobs_project',276                                  'remove')277        user_roles = self._ac.describe_user_roles(self._c, 'bob',278                                                  'bobs_project')279        self.assertNotIn('developer', [r['role'] for r in user_roles['roles']])280    def test_modify_user_role_invalid(self):281        self.assertRaises(exception.ApiError,282                          self._ac.modify_user_role,283                          self._c, 'bob', 'itsec',284                          operation='invalid_operation')285    def test_describe_hosts_compute(self):286        db.service_create(self._c, {'host': 'host1',287            'binary': "nova-compute",288            'topic': 'compute',289            'report_count': 0,290            'availability_zone': "zone1"})291        hosts = self._ac.describe_hosts(self._c)['hosts']292        self.assertEqual('host1', hosts[0]['hostname'])293    def test_describe_hosts_volume(self):294        db.service_create(self._c, {'host': 'volume1',295            'binary': "nova-volume",296            'topic': 'volume',297            'report_count': 0,298            'availability_zone': "zone1"})299        hosts = self._ac.describe_hosts(self._c)['hosts']300        self.assertEqual('volume1', hosts[0]['hostname'])301    def test_block_external_addresses(self):302        result = self._ac.block_external_addresses(self._c, '192.168.100.1/24')303        self.assertEqual('OK', result['status'])304        self.assertEqual('Added 3 rules', result['message'])305    def test_block_external_addresses_already_existent_rule(self):306        self._ac.block_external_addresses(self._c, '192.168.100.1/24')307        self.assertRaises(exception.ApiError,308                          self._ac.block_external_addresses,309                          self._c, '192.168.100.1/24')310    def test_describe_external_address_blocks(self):311        self._ac.block_external_addresses(self._c, '192.168.100.1/24')312        self.assertEqual(313                {'externalIpBlockInfo': [{'cidr': u'192.168.100.1/24'}]},314                self._ac.describe_external_address_blocks(self._c))315    def test_remove_external_address_block(self):316        self._ac.block_external_addresses(self._c, '192.168.100.1/24')317        result = self._ac.remove_external_address_block(self._c,318                                                        '192.168.100.1/24')319        self.assertEqual('OK', result['status'])320        self.assertEqual('Deleted 3 rules', result['message'])321        result = self._ac.describe_external_address_blocks(self._c)322        self.assertEqual([], result['externalIpBlockInfo'])323    def test_start_vpn(self):324        def fake_launch_vpn_instance(self, *args):325            pass326        def get_fake_instance_func():327            first_call = [True]328            def fake_instance_get_all_by_project(self, *args):329                if first_call[0]:330                    first_call[0] = False331                    return []332                else:333                    return [{'id': 1,334                            'user_id': 'bob',335                            'image_id': str(flags.FLAGS.vpn_image_id),336                            'project_id': 'bobs_project',337                            'instance_type_id': '1',338                            'os_type': 'linux',339                            'architecture': 'x86-64',340                            'state_description': 'running',341                            'vm_state': vm_states.ACTIVE,342                            'image_ref': '3'}]343            return fake_instance_get_all_by_project344        self.stubs.Set(pipelib.CloudPipe, 'launch_vpn_instance',345                       fake_launch_vpn_instance)346        self.stubs.Set(db, 'instance_get_all_by_project',347                       get_fake_instance_func())348        self._ac.register_user(self._c, 'bob')349        self._ac.register_project(self._c, 'bobs_project', 'bob')350        self.assertEqual('i-00000001',351                 self._ac.start_vpn(self._c, 'bobs_project')['instance_id'])352    def test_describe_vpns(self):353        def fake_instance_get_all_by_project(self, *args):354            now = datetime.datetime.now()355            created_at = now - datetime.timedelta(seconds=70)356            return [{'id': 1,357                    'user_id': 'bob',358                    'image_id': str(flags.FLAGS.vpn_image_id),359                    'project_id': 'bobs_project',360                    'instance_type_id': '1',361                    'os_type': 'linux',362                    'architecture': 'x86-64',363                    'state_description': 'running',364                    'created_at': created_at,365                    'vm_state': vm_states.ACTIVE,366                    'image_ref': '3'}]367        self.stubs.Set(db, 'instance_get_all_by_project',368                       fake_instance_get_all_by_project)369        self._ac.register_user(self._c, 'bob')370        self._ac.register_project(self._c, 'bobs_project', 'bob')371        vpns = self._ac.describe_vpns(self._c)372        self.assertIn('items', vpns)373        item = vpns['items'][0]374        self.assertEqual('i-00000001', item['instance_id'])375        self.assertEqual(None, item['public_port'])376        self.assertEqual(None, item['public_ip'])377        self.assertEqual('down - invalid project vpn config', item['state'])...

test_plugin_readthedocs.py

Source:test_plugin_readthedocs.py

...10class TestPluginReadTheDocs:11    async def test_start(self, bot, aiohttp_server):12        await aiohttp_server(bot)13        assert isinstance(bot["plugins"]["readthedocs"], RTDPlugin)14    async def test_register_project(self, bot):15        bot["plugins"]["readthedocs"].register_project(16            "test", build_url="https://example.com", jeton="aaaaaa"17        )18        assert "test" in bot["plugins"]["readthedocs"]._projects19    async def test_build_project(self, bot):20        bot["plugins"]["readthedocs"]._session.post = asynctest.CoroutineMock()21        bot["plugins"]["readthedocs"].register_project(22            "test", build_url="https://example.com", jeton="aaaaaa"23        )24        await bot["plugins"]["readthedocs"].build("test")25        assert bot["plugins"]["readthedocs"]._session.post.call_count == 126        bot["plugins"]["readthedocs"]._session.post.assert_called_with(27            "https://example.com", json={"branch": "latest", "token": "aaaaaa"}28        )29    async def test_build_project_branch(self, bot):30        bot["plugins"]["readthedocs"]._session.post = asynctest.CoroutineMock()31        bot["plugins"]["readthedocs"].register_project(32            "test", build_url="https://example.com", jeton="aaaaaa"33        )34        await bot["plugins"]["readthedocs"].build("test", branch="dev")35        assert bot["plugins"]["readthedocs"]._session.post.call_count == 136        bot["plugins"]["readthedocs"]._session.post.assert_called_with(37            "https://example.com", json={"branch": "dev", "token": "aaaaaa"}38        )39    async def test_register_handler(self, bot):40        def handler():41            pass42        bot["plugins"]["readthedocs"].register_handler("test", handler)43        h = []44        for f in bot["plugins"]["readthedocs"].dispatch({"slug": "test"}):45            h.append(f)46        assert len(h) == 147        assert h[0] is handler48    async def test_register_multiple_handlers(self, bot):49        def handler():50            pass51        def handler_bis():52            pass53        bot["plugins"]["readthedocs"].register_handler("test", handler)54        bot["plugins"]["readthedocs"].register_handler("test", handler_bis)55        h = []56        for f in bot["plugins"]["readthedocs"].dispatch({"slug": "test"}):57            h.append(f)58        assert len(h) == 259        assert h[0] is handler60        assert h[1] is handler_bis61    async def test_register_handler_then_project(self, bot):62        def handler():63            pass64        bot["plugins"]["readthedocs"].register_handler("test", handler)65        bot["plugins"]["readthedocs"].register_project(66            "test", build_url="https://example.com", jeton="aaaaaa"67        )68        h = []69        for f in bot["plugins"]["readthedocs"].dispatch({"slug": "test"}):70            h.append(f)71        assert len(h) == 172        assert h[0] is handler73        assert "test" in bot["plugins"]["readthedocs"]._projects74    async def test_register_project_then_handler(self, bot):75        def handler():76            pass77        bot["plugins"]["readthedocs"].register_project(78            "test", build_url="https://example.com", jeton="aaaaaa"79        )80        bot["plugins"]["readthedocs"].register_handler("test", handler)81        h = []82        for f in bot["plugins"]["readthedocs"].dispatch({"slug": "test"}):83            h.append(f)84        assert len(h) == 185        assert h[0] is handler86        assert "test" in bot["plugins"]["readthedocs"]._projects87    async def test_register_project_and_handlers(self, bot):88        def handler():89            pass90        def handler_bis():91            pass92        bot["plugins"]["readthedocs"].register_project(93            "test",94            build_url="https://example.com",95            jeton="aaaaaa",96            handlers=[handler, handler_bis],97        )98        h = []99        for f in bot["plugins"]["readthedocs"].dispatch({"slug": "test"}):100            h.append(f)101        assert len(h) == 2102        assert h[0] is handler103        assert h[1] is handler_bis104        assert "test" in bot["plugins"]["readthedocs"]._projects105    async def test_incoming(self, bot, aiohttp_client):106        async def handler(payload, app):107            assert payload == {108                "build": {109                    "date": "2018-03-02 11:33:05",110                    "id": 6831644,111                    "success": False,112                },113                "name": "Sir Bot-a-lot",114                "slug": "sir-bot-a-lot",115            }116            assert app is bot117        client = await aiohttp_client(bot)118        bot["plugins"]["readthedocs"].register_handler("sir-bot-a-lot", handler=handler)119        r = await client.post(120            "/readthedocs",121            json={122                "build": {123                    "date": "2018-03-02 11:33:05",124                    "id": 6831644,125                    "success": False,126                },127                "name": "Sir Bot-a-lot",128                "slug": "sir-bot-a-lot",129            },130        )131        assert r.status == 200132    async def test_incoming_handler_error(self, bot, aiohttp_client):133        async def handler(payload, app):134            raise RuntimeError()135        client = await aiohttp_client(bot)136        bot["plugins"]["readthedocs"].register_handler("sir-bot-a-lot", handler=handler)137        r = await client.post(138            "/readthedocs",139            json={140                "build": {141                    "date": "2018-03-02 11:33:05",142                    "id": 6831644,143                    "success": False,144                },145                "name": "Sir Bot-a-lot",146                "slug": "sir-bot-a-lot",147            },148        )149        assert r.status == 500150    async def test_incoming_no_project(self, bot, aiohttp_client):151        client = await aiohttp_client(bot)152        r = await client.post(153            "/readthedocs",154            json={155                "build": {156                    "date": "2018-03-02 11:33:05",157                    "id": 6831644,158                    "success": False,159                },160                "name": "Sir Bot-a-lot",161                "slug": "sir-bot-a-lot",162            },163        )164        assert r.status == 400165    async def test_incoming_project_no_handler(self, bot, aiohttp_client):166        client = await aiohttp_client(bot)167        bot["plugins"]["readthedocs"].register_project(168            "sir-bot-a-lot", build_url="https://example.com", jeton="aaaaaa"169        )170        r = await client.post(171            "/readthedocs",172            json={173                "build": {174                    "date": "2018-03-02 11:33:05",175                    "id": 6831644,176                    "success": False,177                },178                "name": "Sir Bot-a-lot",179                "slug": "sir-bot-a-lot",180            },181        )...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.