def test_list_works(self): tenants = ( doubles.make(self.mox, doubles.Tenant, name='t1', id=u'PID1'), doubles.make(self.mox, doubles.Tenant, name='systenant', id=u'S'), doubles.make(self.mox, doubles.Tenant, name='t2', id=u'PID2')) self.fake_client_set.identity_admin.tenants.list().AndReturn(tenants) tcs1 = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'PID1', fallback_to_api=True) \ .AndReturn(tcs1) tcs1.compute.security_groups.list().AndReturn(['SG1', 'SG2']) fw_rule_sets._sg_to_view('SG1', tenants[0].name).AndReturn('REPLY1') fw_rule_sets._sg_to_view('SG2', tenants[0].name).AndReturn('REPLY2') tcs2 = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'PID2', fallback_to_api=True) \ .AndReturn(tcs2) tcs2.compute.security_groups.list().AndReturn(['SG3']) fw_rule_sets._sg_to_view('SG3', tenants[2].name).AndReturn('REPLY3') expected = { 'collection': { 'name': 'fw-rule-sets', 'size': 3 }, 'fw-rule-sets': ['REPLY1', 'REPLY2', 'REPLY3'] } self.mox.ReplayAll() rv = self.client.get('/v1/fw-rule-sets/') data = self.check_and_parse_response(rv) self.assertEquals(data, expected)
def test_create_no_description(self): params = {'project': u'TENANT', 'name': u'Test SG'} tcs = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'TENANT', fallback_to_api=True, eperm_status=404) \ .AndReturn(tcs) tcs.compute.security_groups.create(name=u'Test SG', description='').AndReturn('SG') fw_rule_sets._sg_to_view('SG').AndReturn('REPLY') self.mox.ReplayAll() data = self.interact(params) self.assertEquals(data, 'REPLY')
def test_list_for_my_projects(self): tenant = doubles.make(self.mox, doubles.Tenant, name='t1', id=u'PID1') self.fake_client_set.identity_public.tenants.list().AndReturn([tenant]) tcs1 = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'PID1', fallback_to_api=True) \ .AndReturn(tcs1) tcs1.compute.security_groups.list().AndReturn(['SG1']) fw_rule_sets._sg_to_view('SG1', tenant.name).AndReturn('REPLY1') self.mox.ReplayAll() rv = self.client.get('/v1/fw-rule-sets/?my-projects=true') data = self.check_and_parse_response(rv) self.assertEquals(data.get('fw-rule-sets'), ['REPLY1'])
def test_list_for_my_projects(self): tenant = doubles.make(self.mox, doubles.Tenant, name='t1', id=u'PID1') self.fake_client_set.identity_public.tenants.list().AndReturn([tenant]) tcs1 = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'PID1', fallback_to_api=True) \ .AndReturn(tcs1) tcs1.compute.security_groups.list().AndReturn(['SG1']) fw_rule_sets._sg_to_view('SG1', tenant.name).AndReturn('REPLY1') self.mox.ReplayAll() rv = self.client.get('/v1/fw-rule-sets/?my-projects=true') data = self.check_and_parse_response(rv) self.assertEquals(data.get('fw-rule-sets'), ['REPLY1'])
def test_create_no_description(self): params = { 'project': u'TENANT', 'name': u'Test SG' } tcs = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'TENANT', fallback_to_api=True, eperm_status=404) \ .AndReturn(tcs) tcs.compute.security_groups.create( name=u'Test SG', description='').AndReturn('SG') fw_rule_sets._sg_to_view('SG').AndReturn('REPLY') self.mox.ReplayAll() data = self.interact(params) self.assertEquals(data, 'REPLY')
def test_list_works(self): tenants = (doubles.make(self.mox, doubles.Tenant, name='t1', id=u'PID1'), doubles.make(self.mox, doubles.Tenant, name='systenant', id=u'S'), doubles.make(self.mox, doubles.Tenant, name='t2', id=u'PID2')) self.fake_client_set.identity_admin.tenants.list().AndReturn(tenants) tcs1 = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'PID1', fallback_to_api=True) \ .AndReturn(tcs1) tcs1.compute.security_groups.list().AndReturn(['SG1', 'SG2']) fw_rule_sets._sg_to_view('SG1', tenants[0].name).AndReturn('REPLY1') fw_rule_sets._sg_to_view('SG2', tenants[0].name).AndReturn('REPLY2') tcs2 = mock_client_set(self.mox) fw_rule_sets.client_set_for_tenant(u'PID2', fallback_to_api=True) \ .AndReturn(tcs2) tcs2.compute.security_groups.list().AndReturn(['SG3']) fw_rule_sets._sg_to_view('SG3', tenants[2].name).AndReturn('REPLY3') expected = { 'collection': { 'name': 'fw-rule-sets', 'size': 3 }, 'fw-rule-sets': ['REPLY1', 'REPLY2', 'REPLY3'] } self.mox.ReplayAll() rv = self.client.get('/v1/fw-rule-sets/') data = self.check_and_parse_response(rv) self.assertEquals(data, expected)