def test_get_works(self):
self.fake_client_set.compute.security_groups._list(
'/servers/%s/os-security-groups' % self.server.id,
'security_groups').AndReturn([self.sg2, self.sg])
ifwrs.assert_admin_or_project_user(
'PID', eperm_status=404)
ifwrs.link_for_security_group(self.sg).AndReturn('REPLY')
self.mox.ReplayAll()
rv = self.client.get('/v1/instances/%s/fw-rule-sets/%s'
% (self.server.id, u'2'))
data = self.check_and_parse_response(rv)
self.assertEquals(data, 'REPLY')
def test_add_works(self):
compute = self.fake_client_set.compute
tcs = mock_client_set(self.mox)
compute.servers.get(self.server.id).AndReturn(self.server)
compute.security_groups.get(self.sg_id).AndReturn(self.sg)
ifwrs.client_set_for_tenant(self.server.tenant_id,
fallback_to_api=True) \
.AndReturn(tcs)
tcs.compute.servers.add_security_group(self.server, self.sg.name)
ifwrs.link_for_security_group(self.sg).AndReturn('REPLY')
self.mox.ReplayAll()
data = self.interact()
self.assertEquals(data, 'REPLY')
def test_list_works(self):
self.fake_client_set.compute.security_groups._list(
'/servers/%s/os-security-groups' % self.server.id,
'security_groups').AndReturn([self.sg, self.sg2])
ifwrs.assert_admin_or_project_user(
'PID', eperm_status=404)
ifwrs.link_for_security_group(self.sg).AndReturn(u'L1')
ifwrs.link_for_security_group(self.sg2).AndReturn(u'L2')
expected = {
'collection': {
'parent-href': '/v1/instances/%s' % self.server.id,
'name': 'fw-rule-sets',
'size': 2
},
'fw-rule-sets': [u'L1', u'L2']
}
self.mox.ReplayAll()
rv = self.client.get('/v1/instances/%s/fw-rule-sets/' % self.server.id)
data = self.check_and_parse_response(rv)
self.assertEquals(data, expected)
