def test_fwg_port_list_for_tenant_no_ports(self): tenant_id = self.request.user.project_id neutronclient.list_ports(tenant_id=tenant_id).AndReturn({'ports': []}) neutronclient.list_fwaas_firewall_groups( tenant_id=tenant_id).AndReturn({'firewall_groups': []}) self.mox.ReplayAll() ports = api_fwaas_v2.fwg_port_list_for_tenant(self.request, tenant_id) self.assertEqual([], ports)
def test_firewall_list(self): exp_firewalls = self.firewall_groups_v2.list() firewalls_dict = { 'firewall_groups': self.api_firewall_groups_v2.list() } neutronclient.list_fwaas_firewall_groups().AndReturn(firewalls_dict) self.mox.ReplayAll() ret_val = api_fwaas_v2.firewall_list(self.request) for (v, d) in zip(ret_val, exp_firewalls): self._assert_firewall_return_value(v, d, expand_policy=False)
def test_firewall_list_for_tenant(self): tenant_id = self.request.user.project_id exp_firewalls = self.firewall_groups_v2.list() firewalls_dict = { 'firewall_groups': self.api_firewall_groups_v2.list() } neutronclient.list_fwaas_firewall_groups(shared=False, tenant_id=tenant_id) \ .AndReturn(firewalls_dict) neutronclient.list_fwaas_firewall_groups(shared=True) \ .AndReturn(firewalls_dict) self.mox.ReplayAll() ret_val = api_fwaas_v2.firewall_list_for_tenant( self.request, tenant_id) for (v, d) in zip(ret_val, exp_firewalls): self._assert_firewall_return_value(v, d, expand_policy=False)
def test_fwg_port_list_for_tenant(self): tenant_id = self.request.user.project_id router_port = { 'id': 'id-1', 'name': 'port-1', 'device_owner': 'network:router_interface' } vm_port1 = { 'id': 'id-vm_port-1', 'name': 'port-2', 'device_owner': 'compute:nova' } vm_port2 = { 'id': 'id-vm_port-2', 'name': 'port-2', 'device_owner': 'compute:nova' } gateway_port = { 'id': 'id-3', 'name': 'port-3', 'device_owner': 'network:router_gateway' } dhcp_port = { 'id': 'id-4', 'name': 'port-4', 'device_owner': 'network:dhcp' } dummy_ports = { 'ports': [ router_port, vm_port1, vm_port2, gateway_port, dhcp_port, ] } neutronclient.list_ports(tenant_id=tenant_id).AndReturn(dummy_ports) neutronclient.list_fwaas_firewall_groups( tenant_id=tenant_id).AndReturn({'firewall_groups': []}) self.mox.ReplayAll() ports = api_fwaas_v2.fwg_port_list_for_tenant(self.request, tenant_id) self.assertEqual(router_port['id'], ports[0]['id']) self.assertEqual(vm_port1['id'], ports[1]['id']) self.assertEqual(vm_port2['id'], ports[2]['id'])
def test_fwg_port_list_for_tenant_no_match(self): tenant_id = self.request.user.project_id dummy_ports = { 'ports': [ { 'name': 'port-3', 'device_owner': 'network:router_gateway' }, { 'name': 'port-4', 'device_owner': 'network:dhcp' }, ] } neutronclient.list_ports(tenant_id=tenant_id).AndReturn(dummy_ports) neutronclient.list_fwaas_firewall_groups( tenant_id=tenant_id).AndReturn({'firewall_groups': []}) self.mox.ReplayAll() ports = api_fwaas_v2.fwg_port_list_for_tenant(self.request, tenant_id) self.assertEqual([], ports)