def test_firewall_get(self): exp_firewall = self.firewall_groups_v2.first() ret_dict = {'firewall_group': self.api_firewall_groups_v2.first()} ingress_policy_id = exp_firewall.ingress_firewall_policy_id ingress_policy = [ p for p in self.api_fw_policies_v2.list() if p['id'] == ingress_policy_id ][0] egress_policy_id = exp_firewall.egress_firewall_policy_id egress_policy = [ p for p in self.api_fw_policies_v2.list() if p['id'] == egress_policy_id ][0] neutronclient.show_fwaas_firewall_group( exp_firewall.id).AndReturn(ret_dict) neutronclient.show_fwaas_firewall_policy(ingress_policy_id)\ .AndReturn({'firewall_policy': ingress_policy}) neutronclient.show_fwaas_firewall_policy(egress_policy_id)\ .AndReturn({'firewall_policy': egress_policy}) self.mox.ReplayAll() ret_val = api_fwaas_v2.firewall_get(self.request, exp_firewall.id) self._assert_firewall_return_value(ret_val, exp_firewall)
def test_policy_get_no_rule(self): # 2nd policy is not associated with any rules. exp_policy = self.fw_policies_v2.list()[1] policy_dict = self.api_fw_policies_v2.list()[1] ret_dict = {'firewall_policy': policy_dict} neutronclient.show_fwaas_firewall_policy( exp_policy.id).AndReturn(ret_dict) self.mox.ReplayAll() ret_val = api_fwaas_v2.policy_get(self.request, exp_policy.id) self.assertIsInstance(ret_val, api_fwaas_v2.Policy) self.assertEqual(exp_policy.name, ret_val.name) self.assertTrue(ret_val.id) self.assertFalse(len(ret_val.rules))
def test_policy_get(self): exp_policy = self.fw_policies_v2.first() policy_dict = self.api_fw_policies_v2.first() # The first two rules are associated with the first policy. api_rules = self.api_fw_rules_v2.list()[:2] ret_dict = {'firewall_policy': policy_dict} neutronclient.show_fwaas_firewall_policy( exp_policy.id).AndReturn(ret_dict) filters = {'firewall_policy_id': exp_policy.id} ret_dict = {'firewall_rules': api_rules} neutronclient.list_fwaas_firewall_rules(**filters).AndReturn(ret_dict) self.mox.ReplayAll() ret_val = api_fwaas_v2.policy_get(self.request, exp_policy.id) self._assert_policy_return_value(ret_val, exp_policy)