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_policy_list_for_tenant(self): tenant_id = self.request.user.project_id 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(tenant_id=tenant_id, shared=False).AndReturn({ 'firewall_policies': [] }) neutronclient.list_fwaas_firewall_policies( shared=True).AndReturn(policies_dict) neutronclient.list_fwaas_firewall_rules().AndReturn(rules_dict) self.mox.ReplayAll() ret_val = api_fwaas_v2.policy_list_for_tenant(self.request, tenant_id) for (v, d) in zip(ret_val, exp_policies): self._assert_policy_return_value(v, d)