def test_get_works(self):
ruleid = u'2'
rules = [{ u'id': 1, u'FAKE': u'fst' },
{ u'id': 2, u'FAKE': u'snd' },
{ u'id': 4, u'FAKE': u'lst' } ]
sg = doubles.make(self.mox, doubles.SecurityGroup,
id=u'42', name='Test SG',
rules=rules, tenant_id='PID')
self.fake_client_set.compute.security_groups.get(sg.id).AndReturn(sg)
fw_rules.auth.assert_admin_or_project_user('PID', eperm_status=404)
fw_rules._fw_rule_dict_to_view(rules[1]).AndReturn('REPLY')
self.mox.ReplayAll()
rv = self.client.get('/v1/fw-rule-sets/%s/rules/%s' % (sg.id, ruleid))
data = self.check_and_parse_response(rv)
self.assertEquals(data, 'REPLY')
def test_list_works(self):
sg = doubles.make(self.mox, doubles.SecurityGroup,
id=u'42', name='Test SG', tenant_id='TENANT',
rules=['RULE1', 'RULE2'])
self.fake_client_set.compute.security_groups.get(sg.id).AndReturn(sg)
fw_rules.auth.assert_admin_or_project_user('TENANT', eperm_status=404)
fw_rules._fw_rule_dict_to_view('RULE1').AndReturn('REPLY1')
fw_rules._fw_rule_dict_to_view('RULE2').AndReturn('REPLY2')
expected = {
'collection': {
'name': 'rules',
'size': 2,
'parent-href': '/v1/fw-rule-sets/%s' % sg.id
},
'rules': [ 'REPLY1', 'REPLY2' ]
}
self.mox.ReplayAll()
rv = self.client.get('/v1/fw-rule-sets/%s/rules/' % sg.id)
data = self.check_and_parse_response(rv)
self.assertEquals(data, expected)
def test_rule_dict_to_view_works(self):
self.mox.ReplayAll()
with self.app.test_request_context():
res = fw_rules._fw_rule_dict_to_view(self.to_view)
self.assertEquals(res, self.expected)
