def test_rule_list(self): exp_rules = self.fw_rules_v2.list() api_rules = {'firewall_rules': self.api_fw_rules_v2.list()} neutronclient.list_fwaas_firewall_rules().AndReturn(api_rules) self.mox.ReplayAll() ret_val = api_fwaas_v2.rule_list(self.request) for (v, d) in zip(ret_val, exp_rules): self._assert_rule_return_value(v, d)
def test_policy_list(self): exp_policies = self.fw_policies_v2.list() policies_dict = {'firewall_policies': self.api_fw_policies_v2.list()} rules_dict = {'firewall_rules': self.api_fw_rules_v2.list()} neutronclient.list_fwaas_firewall_policies().AndReturn(policies_dict) neutronclient.list_fwaas_firewall_rules().AndReturn(rules_dict) self.mox.ReplayAll() ret_val = api_fwaas_v2.policy_list(self.request) for (v, d) in zip(ret_val, exp_policies): self._assert_policy_return_value(v, d)
def test_rule_list_for_tenant(self): tenant_id = self.request.user.project_id exp_rules = self.fw_rules_v2.list() api_rules = {'firewall_rules': self.api_fw_rules_v2.list()} neutronclient.list_fwaas_firewall_rules(tenant_id=tenant_id, shared=False).AndReturn( {'firewall_rules': []}) neutronclient.list_fwaas_firewall_rules(shared=True) \ .AndReturn(api_rules) self.mox.ReplayAll() ret_val = api_fwaas_v2.rule_list_for_tenant(self.request, tenant_id) for (v, d) in zip(ret_val, exp_rules): self._assert_rule_return_value(v, d)
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)