From b25742499a8e12edf6ad0bfb91119631c43905a3 Mon Sep 17 00:00:00 2001 From: Stephen Finucane <stephenfin@redhat.com> Date: Fri, 25 Aug 2023 13:38:32 +0100 Subject: [PATCH] tests: Use consistent shortcut to fake network client This removes the need for a number of base test case subclasses. We use 'network_client' rather than 'network' for consistency with other services. Change-Id: I61f7c43341acf19ae16d06db34e18970297fbb93 Signed-off-by: Stephen Finucane <stephenfin@redhat.com> --- .../tests/unit/common/test_extension.py | 1 - .../tests/unit/common/test_quota.py | 69 ++-- .../tests/unit/compute/v2/test_server.py | 111 +++--- .../tests/unit/network/test_common.py | 22 +- .../unit/network/v2/test_address_group.py | 70 ++-- .../unit/network/v2/test_address_scope.py | 62 +-- .../network/v2/test_floating_ip_network.py | 166 ++++---- .../v2/test_floating_ip_pool_network.py | 2 +- .../v2/test_floating_ip_port_forwarding.py | 60 +-- .../unit/network/v2/test_ip_availability.py | 20 +- .../network/v2/test_l3_conntrack_helper.py | 36 +- .../tests/unit/network/v2/test_local_ip.py | 68 ++-- .../network/v2/test_local_ip_association.py | 44 ++- .../tests/unit/network/v2/test_ndp_proxy.py | 68 ++-- .../tests/unit/network/v2/test_network.py | 146 ++++--- .../unit/network/v2/test_network_agent.py | 116 +++--- .../test_network_auto_allocated_topology.py | 34 +- .../unit/network/v2/test_network_flavor.py | 72 ++-- .../network/v2/test_network_flavor_profile.py | 56 +-- .../unit/network/v2/test_network_meter.py | 36 +- .../network/v2/test_network_meter_rule.py | 36 +- .../network/v2/test_network_qos_policy.py | 62 +-- .../unit/network/v2/test_network_qos_rule.py | 150 ++++---- .../network/v2/test_network_qos_rule_type.py | 14 +- .../unit/network/v2/test_network_rbac.py | 80 ++-- .../unit/network/v2/test_network_segment.py | 60 +-- .../network/v2/test_network_segment_range.py | 60 +-- .../v2/test_network_service_provider.py | 6 +- .../unit/network/v2/test_network_trunk.py | 108 +++--- .../tests/unit/network/v2/test_port.py | 360 +++++++++++------- .../tests/unit/network/v2/test_router.py | 266 +++++++------ .../network/v2/test_security_group_network.py | 82 ++-- .../v2/test_security_group_rule_network.py | 78 ++-- .../tests/unit/network/v2/test_subnet.py | 194 +++++----- .../tests/unit/network/v2/test_subnet_pool.py | 128 ++++--- 35 files changed, 1640 insertions(+), 1303 deletions(-) diff --git a/openstackclient/tests/unit/common/test_extension.py b/openstackclient/tests/unit/common/test_extension.py index d7cb29a722..267d423919 100644 --- a/openstackclient/tests/unit/common/test_extension.py +++ b/openstackclient/tests/unit/common/test_extension.py @@ -46,7 +46,6 @@ class TestExtension(utils.TestCommand): self.app.client_manager.network = mock.Mock() self.network_client = self.app.client_manager.network - self.network_client.extensions = mock.Mock() class TestExtensionList(TestExtension): diff --git a/openstackclient/tests/unit/common/test_quota.py b/openstackclient/tests/unit/common/test_quota.py index 4e9a0f533f..0e0671d601 100644 --- a/openstackclient/tests/unit/common/test_quota.py +++ b/openstackclient/tests/unit/common/test_quota.py @@ -63,7 +63,7 @@ class TestQuota(compute_fakes.TestComputev2): self.volume_quotas_class_mock.reset_mock() self.app.client_manager.network = mock.Mock() - self.network_mock = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.app.client_manager.auth_ref = mock.Mock() self.app.client_manager.auth_ref.service_catalog = mock.Mock() @@ -158,8 +158,7 @@ class TestQuotaList(TestQuota): network_fakes.FakeQuota.create_one_default_net_quota(), network_fakes.FakeQuota.create_one_default_net_quota(), ] - self.network = self.app.client_manager.network - self.network.get_quota_default = mock.Mock( + self.network_client.get_quota_default = mock.Mock( side_effect=self.network_default_quotas, ) @@ -261,7 +260,7 @@ class TestQuotaList(TestQuota): detailed_quota ) - self.network.get_quota = mock.Mock(return_value=detailed_quota) + self.network_client.get_quota = mock.Mock(return_value=detailed_quota) arglist = [ '--detail', @@ -452,7 +451,7 @@ class TestQuotaList(TestQuota): def test_quota_list_network(self): # Two projects with non-default quotas - self.network.get_quota = mock.Mock( + self.network_client.get_quota = mock.Mock( side_effect=self.network_quotas, ) @@ -473,7 +472,7 @@ class TestQuotaList(TestQuota): def test_quota_list_network_default(self): # Two projects with non-default quotas - self.network.get_quota = mock.Mock( + self.network_client.get_quota = mock.Mock( side_effect=[ self.network_quotas[0], network_fakes.FakeQuota.create_one_default_net_quota(), @@ -497,7 +496,7 @@ class TestQuotaList(TestQuota): def test_quota_list_network_no_project(self): # Two projects with non-default quotas - self.network.get_quota = mock.Mock( + self.network_client.get_quota = mock.Mock( side_effect=[ self.network_quotas[0], exceptions.NotFound("NotFound"), @@ -521,7 +520,7 @@ class TestQuotaList(TestQuota): def test_quota_list_network_by_project(self): # Two projects with non-default quotas - self.network.get_quota = mock.Mock( + self.network_client.get_quota = mock.Mock( side_effect=self.network_quotas, ) @@ -663,7 +662,7 @@ class TestQuotaSet(TestQuota): loaded=True, ) - self.network_mock.update_quota = mock.Mock() + self.network_client.update_quota = mock.Mock() self.cmd = quota.SetQuota(self.app, None) @@ -893,7 +892,7 @@ class TestQuotaSet(TestQuota): 'rbac_policy': network_fakes.QUOTA['rbac_policy'], 'port': network_fakes.QUOTA['port'], } - self.network_mock.update_quota.assert_called_once_with( + self.network_client.update_quota.assert_called_once_with( self.projects[0].id, **kwargs ) self.assertIsNone(result) @@ -980,7 +979,7 @@ class TestQuotaSet(TestQuota): self.volume_quotas_class_mock.update.assert_called_with( self.projects[0].name, **kwargs_volume ) - self.assertNotCalled(self.network_mock.update_quota) + self.assertNotCalled(self.network_client.update_quota) self.assertIsNone(result) def test_quota_set_with_force(self): @@ -1031,7 +1030,7 @@ class TestQuotaSet(TestQuota): self.volume_quotas_mock.update.assert_called_once_with( self.projects[0].id, **kwargs_volume ) - self.network_mock.update_quota.assert_called_once_with( + self.network_client.update_quota.assert_called_once_with( self.projects[0].id, **kwargs_network ) self.assertIsNone(result) @@ -1075,7 +1074,7 @@ class TestQuotaSet(TestQuota): self.volume_quotas_mock.update.assert_called_once_with( self.projects[0].id, **kwargs_volume ) - self.network_mock.update_quota.assert_called_once_with( + self.network_client.update_quota.assert_called_once_with( self.projects[0].id, **kwargs_network ) self.assertIsNone(result) @@ -1121,12 +1120,10 @@ class TestQuotaShow(TestQuota): 'network': fake_network_endpoint } - self.app.client_manager.network = mock.Mock() - self.network = self.app.client_manager.network - self.network.get_quota = mock.Mock( + self.network_client.get_quota = mock.Mock( return_value=network_fakes.QUOTA, ) - self.network.get_quota_default = mock.Mock( + self.network_client.get_quota_default = mock.Mock( return_value=network_fakes.QUOTA, ) @@ -1152,11 +1149,11 @@ class TestQuotaShow(TestQuota): self.projects[0].id, usage=False, ) - self.network.get_quota.assert_called_once_with( + self.network_client.get_quota.assert_called_once_with( self.projects[0].id, details=False, ) - self.assertNotCalled(self.network.get_quota_default) + self.assertNotCalled(self.network_client.get_quota_default) def test_quota_show__with_compute(self): arglist = [ @@ -1176,7 +1173,7 @@ class TestQuotaShow(TestQuota): detail=False, ) self.volume_quotas_mock.get.assert_not_called() - self.network.get_quota.assert_not_called() + self.network_client.get_quota.assert_not_called() def test_quota_show__with_volume(self): arglist = [ @@ -1196,7 +1193,7 @@ class TestQuotaShow(TestQuota): self.projects[0].id, usage=False, ) - self.network.get_quota.assert_not_called() + self.network_client.get_quota.assert_not_called() def test_quota_show__with_network(self): arglist = [ @@ -1213,11 +1210,11 @@ class TestQuotaShow(TestQuota): self.compute_quotas_mock.get.assert_not_called() self.volume_quotas_mock.get.assert_not_called() - self.network.get_quota.assert_called_once_with( + self.network_client.get_quota.assert_called_once_with( self.projects[0].id, details=False, ) - self.assertNotCalled(self.network.get_quota_default) + self.assertNotCalled(self.network_client.get_quota_default) def test_quota_show__with_default(self): arglist = [ @@ -1238,10 +1235,10 @@ class TestQuotaShow(TestQuota): self.volume_quotas_mock.defaults.assert_called_once_with( self.projects[0].id, ) - self.network.get_quota_default.assert_called_once_with( + self.network_client.get_quota_default.assert_called_once_with( self.projects[0].id, ) - self.assertNotCalled(self.network.get_quota) + self.assertNotCalled(self.network_client.get_quota) def test_quota_show__with_class(self): arglist = [ @@ -1259,8 +1256,8 @@ class TestQuotaShow(TestQuota): self.compute_quotas_class_mock.get.assert_called_once_with('default') self.volume_quotas_class_mock.get.assert_called_once_with('default') # neutron doesn't have the concept of quota classes - self.assertNotCalled(self.network.get_quota) - self.assertNotCalled(self.network.get_quota_default) + self.assertNotCalled(self.network_client.get_quota) + self.assertNotCalled(self.network_client.get_quota_default) def test_quota_show__with_usage(self): # update mocks to return detailed quota instead @@ -1268,7 +1265,7 @@ class TestQuotaShow(TestQuota): self.compute_quotas_mock.get.return_value = self.compute_quota self.volume_quota = volume_fakes.create_one_detailed_quota() self.volume_quotas_mock.get.return_value = self.volume_quota - self.network.get_quota.return_value = ( + self.network_client.get_quota.return_value = ( network_fakes.FakeQuota.create_one_net_detailed_quota() ) @@ -1292,7 +1289,7 @@ class TestQuotaShow(TestQuota): self.projects[0].id, usage=True, ) - self.network.get_quota.assert_called_once_with( + self.network_client.get_quota.assert_called_once_with( self.projects[0].id, details=True, ) @@ -1312,10 +1309,10 @@ class TestQuotaShow(TestQuota): self.volume_quotas_mock.get.assert_called_once_with( identity_fakes.project_id, usage=False ) - self.network.get_quota.assert_called_once_with( + self.network_client.get_quota.assert_called_once_with( identity_fakes.project_id, details=False ) - self.assertNotCalled(self.network.get_quota_default) + self.assertNotCalled(self.network_client.get_quota_default) class TestQuotaDelete(TestQuota): @@ -1324,7 +1321,7 @@ class TestQuotaDelete(TestQuota): def setUp(self): super().setUp() - self.network_mock.delete_quota = mock.Mock() + self.network_client.delete_quota = mock.Mock() self.cmd = quota.DeleteQuota(self.app, None) @@ -1350,7 +1347,7 @@ class TestQuotaDelete(TestQuota): self.volume_quotas_mock.delete.assert_called_once_with( self.projects[0].id, ) - self.network_mock.delete_quota.assert_called_once_with( + self.network_client.delete_quota.assert_called_once_with( self.projects[0].id, ) @@ -1375,7 +1372,7 @@ class TestQuotaDelete(TestQuota): self.projects[0].id, ) self.volume_quotas_mock.delete.assert_not_called() - self.network_mock.delete_quota.assert_not_called() + self.network_client.delete_quota.assert_not_called() def test_delete__volume(self): """Delete volume quotas only""" @@ -1398,7 +1395,7 @@ class TestQuotaDelete(TestQuota): self.volume_quotas_mock.delete.assert_called_once_with( self.projects[0].id, ) - self.network_mock.delete_quota.assert_not_called() + self.network_client.delete_quota.assert_not_called() def test_delete__network(self): """Delete network quotas only""" @@ -1419,6 +1416,6 @@ class TestQuotaDelete(TestQuota): self.projects_mock.get.assert_called_once_with(self.projects[0].id) self.compute_quotas_mock.delete.assert_not_called() self.volume_quotas_mock.delete.assert_not_called() - self.network_mock.delete_quota.assert_called_once_with( + self.network_client.delete_quota.assert_called_once_with( self.projects[0].id, ) diff --git a/openstackclient/tests/unit/compute/v2/test_server.py b/openstackclient/tests/unit/compute/v2/test_server.py index 5ed5eee927..59a458d685 100644 --- a/openstackclient/tests/unit/compute/v2/test_server.py +++ b/openstackclient/tests/unit/compute/v2/test_server.py @@ -113,6 +113,9 @@ class TestServer(compute_fakes.TestComputev2): self.snapshots_mock = self.app.client_manager.volume.volume_snapshots self.snapshots_mock.reset_mock() + self.app.client_manager.network = mock.Mock() + self.network_client = self.app.client_manager.network + # Set object attributes to be tested. Could be overwritten in subclass. self.attrs = {} @@ -522,9 +525,7 @@ class TestServerAddFloatingIPNetwork( def setUp(self): super(TestServerAddFloatingIPNetwork, self).setUp() - self.app.client_manager.network = mock.Mock() - self.network = self.app.client_manager.network - self.network.update_ip = mock.Mock(return_value=None) + self.network_client.update_ip = mock.Mock(return_value=None) # Get the command object to test self.cmd = server.AddFloatingIP(self.app, self.namespace) @@ -534,8 +535,8 @@ class TestServerAddFloatingIPNetwork( self.servers_mock.get.return_value = _server _port = network_fakes.create_one_port() _floating_ip = network_fakes.FakeFloatingIP.create_one_floating_ip() - self.network.find_ip = mock.Mock(return_value=_floating_ip) - self.network.ports = mock.Mock(return_value=[_port]) + self.network_client.find_ip = mock.Mock(return_value=_floating_ip) + self.network_client.ports = mock.Mock(return_value=[_port]) arglist = [ _server.id, _floating_ip['floating_ip_address'], @@ -552,22 +553,24 @@ class TestServerAddFloatingIPNetwork( 'port_id': _port.id, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( _floating_ip['floating_ip_address'], ignore_missing=False, ) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( device_id=_server.id, ) - self.network.update_ip.assert_called_once_with(_floating_ip, **attrs) + self.network_client.update_ip.assert_called_once_with( + _floating_ip, **attrs + ) def test_server_add_floating_ip_no_ports(self): server = compute_fakes.create_one_server() floating_ip = network_fakes.FakeFloatingIP.create_one_floating_ip() self.servers_mock.get.return_value = server - self.network.find_ip = mock.Mock(return_value=floating_ip) - self.network.ports = mock.Mock(return_value=[]) + self.network_client.find_ip = mock.Mock(return_value=floating_ip) + self.network_client.ports = mock.Mock(return_value=[]) arglist = [ server.id, @@ -586,11 +589,11 @@ class TestServerAddFloatingIPNetwork( 'No attached ports found to associate floating IP with', str(ex) ) - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( floating_ip['floating_ip_address'], ignore_missing=False, ) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( device_id=server.id, ) @@ -599,17 +602,17 @@ class TestServerAddFloatingIPNetwork( self.servers_mock.get.return_value = _server _port = network_fakes.create_one_port() _floating_ip = network_fakes.FakeFloatingIP.create_one_floating_ip() - self.network.find_ip = mock.Mock(return_value=_floating_ip) + self.network_client.find_ip = mock.Mock(return_value=_floating_ip) return_value = [_port] # In the success case, we'll have two ports, where the first port is # not attached to an external gateway but the second port is. if success: return_value.append(_port) - self.network.ports = mock.Mock(return_value=return_value) + self.network_client.ports = mock.Mock(return_value=return_value) side_effect = [sdk_exceptions.NotFoundException()] if success: side_effect.append(None) - self.network.update_ip = mock.Mock(side_effect=side_effect) + self.network_client.update_ip = mock.Mock(side_effect=side_effect) arglist = [ _server.id, _floating_ip['floating_ip_address'], @@ -633,19 +636,19 @@ class TestServerAddFloatingIPNetwork( 'port_id': _port.id, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( _floating_ip['floating_ip_address'], ignore_missing=False, ) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( device_id=_server.id, ) if success: - self.assertEqual(2, self.network.update_ip.call_count) + self.assertEqual(2, self.network_client.update_ip.call_count) calls = [mock.call(_floating_ip, **attrs)] * 2 - self.network.update_ip.assert_has_calls(calls) + self.network_client.update_ip.assert_has_calls(calls) else: - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( _floating_ip, **attrs ) @@ -657,8 +660,8 @@ class TestServerAddFloatingIPNetwork( self.servers_mock.get.return_value = _server _port = network_fakes.create_one_port() _floating_ip = network_fakes.FakeFloatingIP.create_one_floating_ip() - self.network.find_ip = mock.Mock(return_value=_floating_ip) - self.network.ports = mock.Mock(return_value=[_port]) + self.network_client.find_ip = mock.Mock(return_value=_floating_ip) + self.network_client.ports = mock.Mock(return_value=[_port]) # The user has specified a fixed ip that matches one of the ports # already attached to the instance. arglist = [ @@ -683,22 +686,24 @@ class TestServerAddFloatingIPNetwork( 'fixed_ip_address': _port.fixed_ips[0]['ip_address'], } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( _floating_ip['floating_ip_address'], ignore_missing=False, ) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( device_id=_server.id, ) - self.network.update_ip.assert_called_once_with(_floating_ip, **attrs) + self.network_client.update_ip.assert_called_once_with( + _floating_ip, **attrs + ) def test_server_add_floating_ip_with_fixed_ip_no_port_found(self): _server = compute_fakes.create_one_server() self.servers_mock.get.return_value = _server _port = network_fakes.create_one_port() _floating_ip = network_fakes.FakeFloatingIP.create_one_floating_ip() - self.network.find_ip = mock.Mock(return_value=_floating_ip) - self.network.ports = mock.Mock(return_value=[_port]) + self.network_client.find_ip = mock.Mock(return_value=_floating_ip) + self.network_client.ports = mock.Mock(return_value=[_port]) # The user has specified a fixed ip that does not match any of the # ports already attached to the instance. nonexistent_ip = '10.0.0.9' @@ -719,14 +724,14 @@ class TestServerAddFloatingIPNetwork( exceptions.CommandError, self.cmd.take_action, parsed_args ) - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( _floating_ip['floating_ip_address'], ignore_missing=False, ) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( device_id=_server.id, ) - self.network.update_ip.assert_not_called() + self.network_client.update_ip.assert_not_called() class TestServerAddPort(TestServer): @@ -1699,19 +1704,16 @@ class TestServerCreate(TestServer): get_endpoints ) - find_network = mock.Mock() - find_port = mock.Mock() - network_client = self.app.client_manager.network - network_client.find_network = find_network - network_client.find_port = find_port network_resource = mock.Mock(id='net1_uuid') port1_resource = mock.Mock(id='port1_uuid') port2_resource = mock.Mock(id='port2_uuid') - find_network.return_value = network_resource - find_port.side_effect = lambda port_id, ignore_missing: { - "port1": port1_resource, - "port2": port2_resource, - }[port_id] + self.network_client.find_network.return_value = network_resource + self.network_client.find_port.side_effect = ( + lambda port_id, ignore_missing: { + "port1": port1_resource, + "port2": port2_resource, + }[port_id] + ) # Mock sdk APIs. _network_1 = mock.Mock(id='net1_uuid') @@ -1831,17 +1833,9 @@ class TestServerCreate(TestServer): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - find_network = mock.Mock() - network_client = self.app.client_manager.network - network_client.find_network = find_network - network_resource = mock.Mock(id='net1_uuid') - find_network.return_value = network_resource - # Mock sdk APIs. _network = mock.Mock(id='net1_uuid') - find_network = mock.Mock() - find_network.return_value = _network - self.app.client_manager.network.find_network = find_network + self.network_client.find_network.return_value = _network # In base command class ShowOne in cliff, abstract method take_action() # returns a two-part tuple with a tuple of column names and a tuple of @@ -1881,7 +1875,7 @@ class TestServerCreate(TestServer): self.assertEqual(self.columns, columns) self.assertEqual(self.datalist(), data) - network_client.find_network.assert_called_once() + self.network_client.find_network.assert_called_once() self.app.client_manager.network.find_network.assert_called_once() def test_server_create_with_network_tag_pre_v243(self): @@ -2224,11 +2218,8 @@ class TestServerCreate(TestServer): get_endpoints ) - find_port = mock.Mock() - network_client = self.app.client_manager.network - network_client.find_port = find_port port_resource = mock.Mock(id='port1_uuid') - find_port.return_value = port_resource + self.network_client.find_port.return_value = port_resource self.assertRaises( exceptions.CommandError, self.cmd.take_action, parsed_args @@ -7265,8 +7256,8 @@ class TestServerRemoveFloatingIPNetwork(network_fakes.TestNetworkV2): super(TestServerRemoveFloatingIPNetwork, self).setUp() self.app.client_manager.network = mock.Mock() - self.network = self.app.client_manager.network - self.network.update_ip = mock.Mock(return_value=None) + self.network_client = self.app.client_manager.network + self.network_client.update_ip = mock.Mock(return_value=None) # Get the command object to test self.cmd = server.RemoveFloatingIP(self.app, self.namespace) @@ -7274,7 +7265,7 @@ class TestServerRemoveFloatingIPNetwork(network_fakes.TestNetworkV2): def test_server_remove_floating_ip_default(self): _server = compute_fakes.create_one_server() _floating_ip = network_fakes.FakeFloatingIP.create_one_floating_ip() - self.network.find_ip = mock.Mock(return_value=_floating_ip) + self.network_client.find_ip = mock.Mock(return_value=_floating_ip) arglist = [ _server.id, _floating_ip['ip'], @@ -7291,11 +7282,13 @@ class TestServerRemoveFloatingIPNetwork(network_fakes.TestNetworkV2): 'port_id': None, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( _floating_ip['ip'], ignore_missing=False, ) - self.network.update_ip.assert_called_once_with(_floating_ip, **attrs) + self.network_client.update_ip.assert_called_once_with( + _floating_ip, **attrs + ) class TestServerRemovePort(TestServer): diff --git a/openstackclient/tests/unit/network/test_common.py b/openstackclient/tests/unit/network/test_common.py index dec27b7bd4..501bf96ce0 100644 --- a/openstackclient/tests/unit/network/test_common.py +++ b/openstackclient/tests/unit/network/test_common.py @@ -130,8 +130,8 @@ class TestNetworkAndCompute(utils.TestCommand): # Create network client mocks. self.app.client_manager.network = mock.Mock() - self.network = self.app.client_manager.network - self.network.network_action = mock.Mock( + self.network_client = self.app.client_manager.network + self.network_client.network_action = mock.Mock( return_value='take_action_network' ) @@ -151,7 +151,7 @@ class TestNetworkAndCompute(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.network_action.assert_called_with(parsed_args) + self.network_client.network_action.assert_called_with(parsed_args) self.assertEqual('take_action_network', result) def test_take_action_compute(self): @@ -210,8 +210,8 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): self.namespace = argparse.Namespace() self.app.client_manager.network = mock.Mock() - self.network = self.app.client_manager.network - self.network.test_create_action = mock.Mock() + self.network_client = self.app.client_manager.network + self.network_client.test_create_action = mock.Mock() # Subclasses can override the command object to test. self.cmd = FakeCreateNeutronCommandWithExtraArgs( @@ -235,7 +235,7 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.test_create_action.assert_called_with( + self.network_client.test_create_action.assert_called_with( known_attribute='known-value', extra_name='extra_value' ) @@ -262,7 +262,7 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.test_create_action.assert_called_with( + self.network_client.test_create_action.assert_called_with( known_attribute='known-value', extra_name='extra_value' ) @@ -283,7 +283,7 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.test_create_action.assert_called_with( + self.network_client.test_create_action.assert_called_with( known_attribute='known-value', extra_name=True ) @@ -304,7 +304,7 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.test_create_action.assert_called_with( + self.network_client.test_create_action.assert_called_with( known_attribute='known-value', extra_name=8 ) @@ -325,7 +325,7 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.test_create_action.assert_called_with( + self.network_client.test_create_action.assert_called_with( known_attribute='known-value', extra_name=['v_1', 'v_2'] ) @@ -352,6 +352,6 @@ class TestNeutronCommandWithExtraArgs(utils.TestCommand): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.test_create_action.assert_called_with( + self.network_client.test_create_action.assert_called_with( known_attribute='known-value', extra_name={'n1': 'v1', 'n2': 'v2'} ) diff --git a/openstackclient/tests/unit/network/v2/test_address_group.py b/openstackclient/tests/unit/network/v2/test_address_group.py index fa67347abc..39e2137f63 100644 --- a/openstackclient/tests/unit/network/v2/test_address_group.py +++ b/openstackclient/tests/unit/network/v2/test_address_group.py @@ -27,7 +27,7 @@ class TestAddressGroup(network_fakes.TestNetworkV2): super(TestAddressGroup, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -60,7 +60,7 @@ class TestCreateAddressGroup(TestAddressGroup): def setUp(self): super(TestCreateAddressGroup, self).setUp() - self.network.create_address_group = mock.Mock( + self.network_client.create_address_group = mock.Mock( return_value=self.new_address_group ) @@ -97,7 +97,7 @@ class TestCreateAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_address_group.assert_called_once_with( + self.network_client.create_address_group.assert_called_once_with( **{ 'name': self.new_address_group.name, 'addresses': [], @@ -129,7 +129,7 @@ class TestCreateAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_address_group.assert_called_once_with( + self.network_client.create_address_group.assert_called_once_with( **{ 'addresses': ['10.0.0.1/32'], 'project_id': self.project.id, @@ -147,9 +147,11 @@ class TestDeleteAddressGroup(TestAddressGroup): def setUp(self): super(TestDeleteAddressGroup, self).setUp() - self.network.delete_address_group = mock.Mock(return_value=None) - self.network.find_address_group = network_fakes.get_address_groups( - address_groups=self._address_groups + self.network_client.delete_address_group = mock.Mock(return_value=None) + self.network_client.find_address_group = ( + network_fakes.get_address_groups( + address_groups=self._address_groups + ) ) # Get the command object to test @@ -166,10 +168,10 @@ class TestDeleteAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_address_group.assert_called_once_with( + self.network_client.find_address_group.assert_called_once_with( self._address_groups[0].name, ignore_missing=False ) - self.network.delete_address_group.assert_called_once_with( + self.network_client.delete_address_group.assert_called_once_with( self._address_groups[0] ) self.assertIsNone(result) @@ -189,7 +191,7 @@ class TestDeleteAddressGroup(TestAddressGroup): calls = [] for a in self._address_groups: calls.append(call(a)) - self.network.delete_address_group.assert_has_calls(calls) + self.network_client.delete_address_group.assert_has_calls(calls) self.assertIsNone(result) def test_multi_address_groups_delete_with_exception(self): @@ -206,7 +208,7 @@ class TestDeleteAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._address_groups[0], exceptions.CommandError] - self.network.find_address_group = mock.Mock( + self.network_client.find_address_group = mock.Mock( side_effect=find_mock_result ) @@ -216,13 +218,13 @@ class TestDeleteAddressGroup(TestAddressGroup): except exceptions.CommandError as e: self.assertEqual('1 of 2 address groups failed to delete.', str(e)) - self.network.find_address_group.assert_any_call( + self.network_client.find_address_group.assert_any_call( self._address_groups[0].name, ignore_missing=False ) - self.network.find_address_group.assert_any_call( + self.network_client.find_address_group.assert_any_call( 'unexist_address_group', ignore_missing=False ) - self.network.delete_address_group.assert_called_once_with( + self.network_client.delete_address_group.assert_called_once_with( self._address_groups[0] ) @@ -251,7 +253,7 @@ class TestListAddressGroup(TestAddressGroup): def setUp(self): super(TestListAddressGroup, self).setUp() - self.network.address_groups = mock.Mock( + self.network_client.address_groups = mock.Mock( return_value=self.address_groups ) @@ -265,7 +267,7 @@ class TestListAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_groups.assert_called_once_with(**{}) + self.network_client.address_groups.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -280,7 +282,7 @@ class TestListAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_groups.assert_called_once_with( + self.network_client.address_groups.assert_called_once_with( **{'name': self.address_groups[0].name} ) self.assertEqual(self.columns, columns) @@ -299,7 +301,7 @@ class TestListAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_groups.assert_called_once_with( + self.network_client.address_groups.assert_called_once_with( project_id=project.id ) self.assertEqual(self.columns, columns) @@ -320,7 +322,7 @@ class TestListAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_groups.assert_called_once_with( + self.network_client.address_groups.assert_called_once_with( project_id=project.id ) self.assertEqual(self.columns, columns) @@ -333,11 +335,11 @@ class TestSetAddressGroup(TestAddressGroup): def setUp(self): super(TestSetAddressGroup, self).setUp() - self.network.update_address_group = mock.Mock(return_value=None) - self.network.find_address_group = mock.Mock( + self.network_client.update_address_group = mock.Mock(return_value=None) + self.network_client.find_address_group = mock.Mock( return_value=self._address_group ) - self.network.add_addresses_to_address_group = mock.Mock( + self.network_client.add_addresses_to_address_group = mock.Mock( return_value=self._address_group ) # Get the command object to test @@ -354,8 +356,8 @@ class TestSetAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_address_group.assert_not_called() - self.network.add_addresses_to_address_group.assert_not_called() + self.network_client.update_address_group.assert_not_called() + self.network_client.add_addresses_to_address_group.assert_not_called() self.assertIsNone(result) def test_set_name_and_description(self): @@ -378,7 +380,7 @@ class TestSetAddressGroup(TestAddressGroup): 'name': "new_address_group_name", 'description': 'new_address_group_description', } - self.network.update_address_group.assert_called_with( + self.network_client.update_address_group.assert_called_with( self._address_group, **attrs ) self.assertIsNone(result) @@ -396,7 +398,7 @@ class TestSetAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.add_addresses_to_address_group.assert_called_once_with( + self.network_client.add_addresses_to_address_group.assert_called_once_with( self._address_group, ['10.0.0.2/32'] ) self.assertIsNone(result) @@ -416,7 +418,7 @@ class TestSetAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.add_addresses_to_address_group.assert_called_once_with( + self.network_client.add_addresses_to_address_group.assert_called_once_with( self._address_group, ['10.0.0.2/32', '2001::/16'] ) self.assertIsNone(result) @@ -442,7 +444,7 @@ class TestShowAddressGroup(TestAddressGroup): def setUp(self): super(TestShowAddressGroup, self).setUp() - self.network.find_address_group = mock.Mock( + self.network_client.find_address_group = mock.Mock( return_value=self._address_group ) @@ -473,7 +475,7 @@ class TestShowAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_address_group.assert_called_once_with( + self.network_client.find_address_group.assert_called_once_with( self._address_group.name, ignore_missing=False ) self.assertEqual(self.columns, columns) @@ -486,10 +488,10 @@ class TestUnsetAddressGroup(TestAddressGroup): def setUp(self): super(TestUnsetAddressGroup, self).setUp() - self.network.find_address_group = mock.Mock( + self.network_client.find_address_group = mock.Mock( return_value=self._address_group ) - self.network.remove_addresses_from_address_group = mock.Mock( + self.network_client.remove_addresses_from_address_group = mock.Mock( return_value=self._address_group ) # Get the command object to test @@ -506,7 +508,7 @@ class TestUnsetAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.remove_addresses_from_address_group.assert_not_called() + self.network_client.remove_addresses_from_address_group.assert_not_called() self.assertIsNone(result) def test_unset_one_address(self): @@ -522,7 +524,7 @@ class TestUnsetAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.remove_addresses_from_address_group.assert_called_once_with( # noqa: E501 + self.network_client.remove_addresses_from_address_group.assert_called_once_with( # noqa: E501 self._address_group, ['10.0.0.2/32'] ) self.assertIsNone(result) @@ -542,7 +544,7 @@ class TestUnsetAddressGroup(TestAddressGroup): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.remove_addresses_from_address_group.assert_called_once_with( # noqa: E501 + self.network_client.remove_addresses_from_address_group.assert_called_once_with( # noqa: E501 self._address_group, ['10.0.0.2/32', '2001::/16'] ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_address_scope.py b/openstackclient/tests/unit/network/v2/test_address_scope.py index e645cc6046..d19d496022 100644 --- a/openstackclient/tests/unit/network/v2/test_address_scope.py +++ b/openstackclient/tests/unit/network/v2/test_address_scope.py @@ -27,7 +27,7 @@ class TestAddressScope(network_fakes.TestNetworkV2): super(TestAddressScope, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -54,7 +54,7 @@ class TestCreateAddressScope(TestAddressScope): def setUp(self): super(TestCreateAddressScope, self).setUp() - self.network.create_address_scope = mock.Mock( + self.network_client.create_address_scope = mock.Mock( return_value=self.new_address_scope ) @@ -90,7 +90,7 @@ class TestCreateAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_address_scope.assert_called_once_with( + self.network_client.create_address_scope.assert_called_once_with( **{ 'ip_version': self.new_address_scope.ip_version, 'name': self.new_address_scope.name, @@ -121,7 +121,7 @@ class TestCreateAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_address_scope.assert_called_once_with( + self.network_client.create_address_scope.assert_called_once_with( **{ 'ip_version': self.new_address_scope.ip_version, 'shared': True, @@ -145,7 +145,7 @@ class TestCreateAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_address_scope.assert_called_once_with( + self.network_client.create_address_scope.assert_called_once_with( **{ 'ip_version': self.new_address_scope.ip_version, 'shared': False, @@ -162,9 +162,11 @@ class TestDeleteAddressScope(TestAddressScope): def setUp(self): super(TestDeleteAddressScope, self).setUp() - self.network.delete_address_scope = mock.Mock(return_value=None) - self.network.find_address_scope = network_fakes.get_address_scopes( - address_scopes=self._address_scopes + self.network_client.delete_address_scope = mock.Mock(return_value=None) + self.network_client.find_address_scope = ( + network_fakes.get_address_scopes( + address_scopes=self._address_scopes + ) ) # Get the command object to test @@ -181,10 +183,10 @@ class TestDeleteAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_address_scope.assert_called_once_with( + self.network_client.find_address_scope.assert_called_once_with( self._address_scopes[0].name, ignore_missing=False ) - self.network.delete_address_scope.assert_called_once_with( + self.network_client.delete_address_scope.assert_called_once_with( self._address_scopes[0] ) self.assertIsNone(result) @@ -205,7 +207,7 @@ class TestDeleteAddressScope(TestAddressScope): calls = [] for a in self._address_scopes: calls.append(call(a)) - self.network.delete_address_scope.assert_has_calls(calls) + self.network_client.delete_address_scope.assert_has_calls(calls) self.assertIsNone(result) def test_multi_address_scopes_delete_with_exception(self): @@ -222,7 +224,7 @@ class TestDeleteAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._address_scopes[0], exceptions.CommandError] - self.network.find_address_scope = mock.Mock( + self.network_client.find_address_scope = mock.Mock( side_effect=find_mock_result ) @@ -232,13 +234,13 @@ class TestDeleteAddressScope(TestAddressScope): except exceptions.CommandError as e: self.assertEqual('1 of 2 address scopes failed to delete.', str(e)) - self.network.find_address_scope.assert_any_call( + self.network_client.find_address_scope.assert_any_call( self._address_scopes[0].name, ignore_missing=False ) - self.network.find_address_scope.assert_any_call( + self.network_client.find_address_scope.assert_any_call( 'unexist_address_scope', ignore_missing=False ) - self.network.delete_address_scope.assert_called_once_with( + self.network_client.delete_address_scope.assert_called_once_with( self._address_scopes[0] ) @@ -267,7 +269,7 @@ class TestListAddressScope(TestAddressScope): def setUp(self): super(TestListAddressScope, self).setUp() - self.network.address_scopes = mock.Mock( + self.network_client.address_scopes = mock.Mock( return_value=self.address_scopes ) @@ -281,7 +283,7 @@ class TestListAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_scopes.assert_called_once_with(**{}) + self.network_client.address_scopes.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -296,7 +298,7 @@ class TestListAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_scopes.assert_called_once_with( + self.network_client.address_scopes.assert_called_once_with( **{'name': self.address_scopes[0].name} ) self.assertEqual(self.columns, columns) @@ -313,7 +315,7 @@ class TestListAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_scopes.assert_called_once_with( + self.network_client.address_scopes.assert_called_once_with( **{'ip_version': 4} ) self.assertEqual(self.columns, columns) @@ -332,7 +334,7 @@ class TestListAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_scopes.assert_called_once_with( + self.network_client.address_scopes.assert_called_once_with( **{'project_id': project.id} ) self.assertEqual(self.columns, columns) @@ -355,7 +357,7 @@ class TestListAddressScope(TestAddressScope): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.address_scopes.assert_called_once_with(**filters) + self.network_client.address_scopes.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -369,7 +371,7 @@ class TestListAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_scopes.assert_called_once_with( + self.network_client.address_scopes.assert_called_once_with( **{'is_shared': True} ) self.assertEqual(self.columns, columns) @@ -385,7 +387,7 @@ class TestListAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.address_scopes.assert_called_once_with( + self.network_client.address_scopes.assert_called_once_with( **{'is_shared': False} ) self.assertEqual(self.columns, columns) @@ -398,8 +400,8 @@ class TestSetAddressScope(TestAddressScope): def setUp(self): super(TestSetAddressScope, self).setUp() - self.network.update_address_scope = mock.Mock(return_value=None) - self.network.find_address_scope = mock.Mock( + self.network_client.update_address_scope = mock.Mock(return_value=None) + self.network_client.find_address_scope = mock.Mock( return_value=self._address_scope ) @@ -418,7 +420,7 @@ class TestSetAddressScope(TestAddressScope): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_address_scope.assert_called_with( + self.network_client.update_address_scope.assert_called_with( self._address_scope, **attrs ) self.assertIsNone(result) @@ -442,7 +444,7 @@ class TestSetAddressScope(TestAddressScope): 'name': "new_address_scope", 'shared': True, } - self.network.update_address_scope.assert_called_with( + self.network_client.update_address_scope.assert_called_with( self._address_scope, **attrs ) self.assertIsNone(result) @@ -462,7 +464,7 @@ class TestSetAddressScope(TestAddressScope): attrs = { 'shared': False, } - self.network.update_address_scope.assert_called_with( + self.network_client.update_address_scope.assert_called_with( self._address_scope, **attrs ) self.assertIsNone(result) @@ -488,7 +490,7 @@ class TestShowAddressScope(TestAddressScope): def setUp(self): super(TestShowAddressScope, self).setUp() - self.network.find_address_scope = mock.Mock( + self.network_client.find_address_scope = mock.Mock( return_value=self._address_scope ) @@ -519,7 +521,7 @@ class TestShowAddressScope(TestAddressScope): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_address_scope.assert_called_once_with( + self.network_client.find_address_scope.assert_called_once_with( self._address_scope.name, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_floating_ip_network.py b/openstackclient/tests/unit/network/v2/test_floating_ip_network.py index d1d359d0f9..358a22e152 100644 --- a/openstackclient/tests/unit/network/v2/test_floating_ip_network.py +++ b/openstackclient/tests/unit/network/v2/test_floating_ip_network.py @@ -30,7 +30,7 @@ class TestFloatingIPNetwork(network_fakes.TestNetworkV2): super(TestFloatingIPNetwork, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -88,14 +88,16 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): def setUp(self): super(TestCreateFloatingIPNetwork, self).setUp() - self.network.create_ip = mock.Mock(return_value=self.floating_ip) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.create_ip = mock.Mock( + return_value=self.floating_ip + ) + self.network_client.set_tags = mock.Mock(return_value=None) - self.network.find_network = mock.Mock( + self.network_client.find_network = mock.Mock( return_value=self.floating_network ) - self.network.find_subnet = mock.Mock(return_value=self.subnet) - self.network.find_port = mock.Mock(return_value=self.port) + self.network_client.find_subnet = mock.Mock(return_value=self.subnet) + self.network_client.find_port = mock.Mock(return_value=self.port) # Get the command object to test self.cmd = fip.CreateFloatingIP(self.app, self.namespace) @@ -123,7 +125,7 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_ip.assert_called_once_with( + self.network_client.create_ip.assert_called_once_with( **{ 'floating_network_id': self.floating_ip.floating_network_id, } @@ -163,7 +165,7 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_ip.assert_called_once_with( + self.network_client.create_ip.assert_called_once_with( **{ 'subnet_id': self.subnet.id, 'port_id': self.floating_ip.port_id, @@ -194,7 +196,7 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_ip.assert_called_once_with( + self.network_client.create_ip.assert_called_once_with( **{ 'floating_network_id': self.floating_ip.floating_network_id, 'project_id': project.id, @@ -224,7 +226,7 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_ip.assert_called_once_with( + self.network_client.create_ip.assert_called_once_with( **{ 'floating_network_id': self.floating_ip.floating_network_id, 'project_id': project.id, @@ -235,7 +237,9 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): def test_create_floating_ip_with_qos(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) arglist = [ '--qos-policy', qos_policy.id, @@ -249,7 +253,7 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_ip.assert_called_once_with( + self.network_client.create_ip.assert_called_once_with( **{ 'floating_network_id': self.floating_ip.floating_network_id, 'qos_policy_id': qos_policy.id, @@ -276,17 +280,17 @@ class TestCreateFloatingIPNetwork(TestFloatingIPNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_ip.assert_called_once_with( + self.network_client.create_ip.assert_called_once_with( **{ 'floating_network_id': self.floating_ip.floating_network_id, } ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self.floating_ip, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertEqual(self.data, data) @@ -304,14 +308,14 @@ class TestDeleteFloatingIPNetwork(TestFloatingIPNetwork): def setUp(self): super(TestDeleteFloatingIPNetwork, self).setUp() - self.network.find_ip = mock.Mock() - self.network.delete_ip = mock.Mock(return_value=None) + self.network_client.find_ip = mock.Mock() + self.network_client.delete_ip = mock.Mock(return_value=None) # Get the command object to test self.cmd = fip.DeleteFloatingIP(self.app, self.namespace) def test_floating_ip_delete(self): - self.network.find_ip.side_effect = [ + self.network_client.find_ip.side_effect = [ self.floating_ips[0], self.floating_ips[1], ] @@ -325,15 +329,17 @@ class TestDeleteFloatingIPNetwork(TestFloatingIPNetwork): result = self.cmd.take_action(parsed_args) - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ips[0].id, ignore_missing=False, ) - self.network.delete_ip.assert_called_once_with(self.floating_ips[0]) + self.network_client.delete_ip.assert_called_once_with( + self.floating_ips[0] + ) self.assertIsNone(result) def test_floating_ip_delete_multi(self): - self.network.find_ip.side_effect = [ + self.network_client.find_ip.side_effect = [ self.floating_ips[0], self.floating_ips[1], ] @@ -359,16 +365,16 @@ class TestDeleteFloatingIPNetwork(TestFloatingIPNetwork): ignore_missing=False, ), ] - self.network.find_ip.assert_has_calls(calls) + self.network_client.find_ip.assert_has_calls(calls) calls = [] for f in self.floating_ips: calls.append(call(f)) - self.network.delete_ip.assert_has_calls(calls) + self.network_client.delete_ip.assert_has_calls(calls) self.assertIsNone(result) def test_floating_ip_delete_multi_exception(self): - self.network.find_ip.side_effect = [ + self.network_client.find_ip.side_effect = [ self.floating_ips[0], exceptions.CommandError, ] @@ -387,15 +393,17 @@ class TestDeleteFloatingIPNetwork(TestFloatingIPNetwork): except exceptions.CommandError as e: self.assertEqual('1 of 2 floating_ips failed to delete.', str(e)) - self.network.find_ip.assert_any_call( + self.network_client.find_ip.assert_any_call( self.floating_ips[0].id, ignore_missing=False, ) - self.network.find_ip.assert_any_call( + self.network_client.find_ip.assert_any_call( 'unexist_floating_ip', ignore_missing=False, ) - self.network.delete_ip.assert_called_once_with(self.floating_ips[0]) + self.network_client.delete_ip.assert_called_once_with( + self.floating_ips[0] + ) class TestListFloatingIPNetwork(TestFloatingIPNetwork): @@ -467,10 +475,14 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): def setUp(self): super(TestListFloatingIPNetwork, self).setUp() - self.network.ips = mock.Mock(return_value=self.floating_ips) - self.network.find_network = mock.Mock(return_value=self.fake_network) - self.network.find_port = mock.Mock(return_value=self.fake_port) - self.network.find_router = mock.Mock(return_value=self.fake_router) + self.network_client.ips = mock.Mock(return_value=self.floating_ips) + self.network_client.find_network = mock.Mock( + return_value=self.fake_network + ) + self.network_client.find_port = mock.Mock(return_value=self.fake_port) + self.network_client.find_router = mock.Mock( + return_value=self.fake_router + ) # Get the command object to test self.cmd = fip.ListFloatingIP(self.app, self.namespace) @@ -482,7 +494,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with() + self.network_client.ips.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -498,7 +510,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'floating_network_id': 'fake_network_id', } @@ -518,7 +530,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'port_id': 'fake_port_id', } @@ -538,7 +550,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'fixed_ip_address': self.floating_ips[0].fixed_ip_address, } @@ -558,7 +570,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'floating_ip_address': self.floating_ips[ 0 @@ -580,7 +592,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with() + self.network_client.ips.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertEqual(self.data_long, list(data)) @@ -597,7 +609,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'status': 'ACTIVE', } @@ -620,7 +632,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.ips.assert_called_once_with(**filters) + self.network_client.ips.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -642,7 +654,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.ips.assert_called_once_with(**filters) + self.network_client.ips.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -659,7 +671,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'router_id': 'fake_router_id', } @@ -687,7 +699,7 @@ class TestListFloatingIPNetwork(TestFloatingIPNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ips.assert_called_once_with( + self.network_client.ips.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -738,7 +750,7 @@ class TestShowFloatingIPNetwork(TestFloatingIPNetwork): def setUp(self): super(TestShowFloatingIPNetwork, self).setUp() - self.network.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) # Get the command object to test self.cmd = fip.ShowFloatingIP(self.app, self.namespace) @@ -754,7 +766,7 @@ class TestShowFloatingIPNetwork(TestFloatingIPNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) @@ -779,10 +791,10 @@ class TestSetFloatingIP(TestFloatingIPNetwork): def setUp(self): super(TestSetFloatingIP, self).setUp() - self.network.find_ip = mock.Mock(return_value=self.floating_ip) - self.network.find_port = mock.Mock(return_value=self.port) - self.network.update_ip = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_port = mock.Mock(return_value=self.port) + self.network_client.update_ip = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = fip.SetFloatingIP(self.app, self.namespace) @@ -805,12 +817,12 @@ class TestSetFloatingIP(TestFloatingIPNetwork): 'port_id': self.floating_ip.port_id, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -835,11 +847,11 @@ class TestSetFloatingIP(TestFloatingIPNetwork): 'port_id': self.floating_ip.port_id, 'fixed_ip_address': self.floating_ip.fixed_ip_address, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -864,17 +876,19 @@ class TestSetFloatingIP(TestFloatingIPNetwork): 'port_id': self.floating_ip.port_id, 'description': self.floating_ip.description, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) def test_qos_policy_option(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) arglist = [ "--qos-policy", qos_policy.id, @@ -891,17 +905,19 @@ class TestSetFloatingIP(TestFloatingIPNetwork): attrs = { 'qos_policy_id': qos_policy.id, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) def test_port_and_qos_policy_option(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) arglist = [ "--qos-policy", qos_policy.id, @@ -922,11 +938,11 @@ class TestSetFloatingIP(TestFloatingIPNetwork): 'qos_policy_id': qos_policy.id, 'port_id': self.floating_ip.port_id, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -946,11 +962,11 @@ class TestSetFloatingIP(TestFloatingIPNetwork): attrs = { 'qos_policy_id': None, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -974,11 +990,11 @@ class TestSetFloatingIP(TestFloatingIPNetwork): 'qos_policy_id': None, 'port_id': self.floating_ip.port_id, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -997,8 +1013,8 @@ class TestSetFloatingIP(TestFloatingIPNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_ip.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_ip.called) + self.network_client.set_tags.assert_called_once_with( self.floating_ip, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -1026,9 +1042,9 @@ class TestUnsetFloatingIP(TestFloatingIPNetwork): def setUp(self): super(TestUnsetFloatingIP, self).setUp() - self.network.find_ip = mock.Mock(return_value=self.floating_ip) - self.network.update_ip = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.update_ip = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = fip.UnsetFloatingIP(self.app, self.namespace) @@ -1049,11 +1065,11 @@ class TestUnsetFloatingIP(TestFloatingIPNetwork): attrs = { 'port_id': None, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -1075,11 +1091,11 @@ class TestUnsetFloatingIP(TestFloatingIPNetwork): attrs = { 'qos_policy_id': None, } - self.network.find_ip.assert_called_once_with( + self.network_client.find_ip.assert_called_once_with( self.floating_ip.id, ignore_missing=False, ) - self.network.update_ip.assert_called_once_with( + self.network_client.update_ip.assert_called_once_with( self.floating_ip, **attrs ) @@ -1100,8 +1116,8 @@ class TestUnsetFloatingIP(TestFloatingIPNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_ip.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_ip.called) + self.network_client.set_tags.assert_called_once_with( self.floating_ip, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_floating_ip_pool_network.py b/openstackclient/tests/unit/network/v2/test_floating_ip_pool_network.py index fd0a208b2a..63c4594874 100644 --- a/openstackclient/tests/unit/network/v2/test_floating_ip_pool_network.py +++ b/openstackclient/tests/unit/network/v2/test_floating_ip_pool_network.py @@ -25,7 +25,7 @@ class TestFloatingIPPoolNetwork(network_fakes.TestNetworkV2): super(TestFloatingIPPoolNetwork, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network class TestListFloatingIPPoolNetwork(TestFloatingIPPoolNetwork): diff --git a/openstackclient/tests/unit/network/v2/test_floating_ip_port_forwarding.py b/openstackclient/tests/unit/network/v2/test_floating_ip_port_forwarding.py index fb30a1a713..74197cb5fd 100644 --- a/openstackclient/tests/unit/network/v2/test_floating_ip_port_forwarding.py +++ b/openstackclient/tests/unit/network/v2/test_floating_ip_port_forwarding.py @@ -27,13 +27,13 @@ from openstackclient.tests.unit import utils as tests_utils class TestFloatingIPPortForwarding(network_fakes.TestNetworkV2): def setUp(self): super(TestFloatingIPPortForwarding, self).setUp() - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.floating_ip = ( network_fakes.FakeFloatingIP.create_one_floating_ip() ) self.port = network_fakes.create_one_port() self.project = identity_fakes_v2.FakeProject.create_one_project() - self.network.find_port = mock.Mock(return_value=self.port) + self.network_client.find_port = mock.Mock(return_value=self.port) class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): @@ -54,11 +54,11 @@ class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): }, ) - self.network.create_floating_ip_port_forwarding = mock.Mock( + self.network_client.create_floating_ip_port_forwarding = mock.Mock( return_value=self.new_port_forwarding ) - self.network.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) # Get the command object to test self.cmd = floating_ip_port_forwarding.CreateFloatingIPPortForwarding( @@ -144,7 +144,7 @@ class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_floating_ip_port_forwarding.assert_called_once_with( # noqa: E501 + self.network_client.create_floating_ip_port_forwarding.assert_called_once_with( # noqa: E501 self.new_port_forwarding.floatingip_id, **{ 'external_port_range': self.new_port_forwarding_with_ranges.external_port_range, # noqa: E501 @@ -197,7 +197,7 @@ class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): self.fail('CommandError should be raised.') except exceptions.CommandError as e: self.assertEqual(msg, str(e)) - self.network.create_floating_ip_port_forwarding.assert_not_called() + self.network_client.create_floating_ip_port_forwarding.assert_not_called() def test_create_all_options_with_invalid_range_exception(self): invalid_port_range = '80:70' @@ -241,7 +241,7 @@ class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): self.fail('CommandError should be raised.') except exceptions.CommandError as e: self.assertEqual(msg, str(e)) - self.network.create_floating_ip_port_forwarding.assert_not_called() + self.network_client.create_floating_ip_port_forwarding.assert_not_called() def test_create_all_options_with_unmatch_ranges_exception(self): internal_range = '80:90' @@ -286,7 +286,7 @@ class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): self.fail('CommandError should be raised.') except exceptions.CommandError as e: self.assertEqual(msg, str(e)) - self.network.create_floating_ip_port_forwarding.assert_not_called() + self.network_client.create_floating_ip_port_forwarding.assert_not_called() def test_create_all_options(self): arglist = [ @@ -325,7 +325,7 @@ class TestCreateFloatingIPPortForwarding(TestFloatingIPPortForwarding): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_floating_ip_port_forwarding.assert_called_once_with( # noqa: E501 + self.network_client.create_floating_ip_port_forwarding.assert_called_once_with( # noqa: E501 self.new_port_forwarding.floatingip_id, **{ 'external_port': self.new_port_forwarding.external_port, @@ -351,11 +351,11 @@ class TestDeleteFloatingIPPortForwarding(TestFloatingIPPortForwarding): }, ) ) - self.network.delete_floating_ip_port_forwarding = mock.Mock( + self.network_client.delete_floating_ip_port_forwarding = mock.Mock( return_value=None ) - self.network.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) # Get the command object to test self.cmd = floating_ip_port_forwarding.DeleteFloatingIPPortForwarding( self.app, self.namespace @@ -375,7 +375,7 @@ class TestDeleteFloatingIPPortForwarding(TestFloatingIPPortForwarding): result = self.cmd.take_action(parsed_args) - self.network.delete_floating_ip_port_forwarding.assert_called_once_with( # noqa: E501 + self.network_client.delete_floating_ip_port_forwarding.assert_called_once_with( # noqa: E501 self.floating_ip.id, self._port_forwarding[0].id, ignore_missing=False, @@ -405,7 +405,9 @@ class TestDeleteFloatingIPPortForwarding(TestFloatingIPPortForwarding): for a in self._port_forwarding: calls.append(call(a.floatingip_id, a.id, ignore_missing=False)) - self.network.delete_floating_ip_port_forwarding.assert_has_calls(calls) + self.network_client.delete_floating_ip_port_forwarding.assert_has_calls( + calls + ) self.assertIsNone(result) def test_multi_port_forwarding_delete_with_exception(self): @@ -425,8 +427,8 @@ class TestDeleteFloatingIPPortForwarding(TestFloatingIPPortForwarding): delete_mock_result = [None, exceptions.CommandError] - self.network.delete_floating_ip_port_forwarding = mock.MagicMock( - side_effect=delete_mock_result + self.network_client.delete_floating_ip_port_forwarding = ( + mock.MagicMock(side_effect=delete_mock_result) ) try: @@ -437,12 +439,12 @@ class TestDeleteFloatingIPPortForwarding(TestFloatingIPPortForwarding): '1 of 2 Port forwarding failed to delete.', str(e) ) - self.network.delete_floating_ip_port_forwarding.assert_any_call( + self.network_client.delete_floating_ip_port_forwarding.assert_any_call( self.floating_ip.id, 'unexist_port_forwarding_id', ignore_missing=False, ) - self.network.delete_floating_ip_port_forwarding.assert_any_call( + self.network_client.delete_floating_ip_port_forwarding.assert_any_call( self.floating_ip.id, self._port_forwarding[0].id, ignore_missing=False, @@ -488,10 +490,10 @@ class TestListFloatingIPPortForwarding(TestFloatingIPPortForwarding): port_forwarding.description, ) ) - self.network.floating_ip_port_forwardings = mock.Mock( + self.network_client.floating_ip_port_forwardings = mock.Mock( return_value=self.port_forwardings ) - self.network.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) # Get the command object to test self.cmd = floating_ip_port_forwarding.ListFloatingIPPortForwarding( self.app, self.namespace @@ -504,7 +506,7 @@ class TestListFloatingIPPortForwarding(TestFloatingIPPortForwarding): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.floating_ip_port_forwardings.assert_called_once_with( + self.network_client.floating_ip_port_forwardings.assert_called_once_with( self.floating_ip, **{} ) self.assertEqual(self.columns, columns) @@ -539,7 +541,7 @@ class TestListFloatingIPPortForwarding(TestFloatingIPPortForwarding): 'protocol': self.port_forwardings[0].protocol, } - self.network.floating_ip_port_forwardings.assert_called_once_with( + self.network_client.floating_ip_port_forwardings.assert_called_once_with( self.floating_ip, **query ) self.assertEqual(self.columns, columns) @@ -555,14 +557,14 @@ class TestSetFloatingIPPortForwarding(TestFloatingIPPortForwarding): 'floatingip_id': self.floating_ip.id, } ) - self.network.update_floating_ip_port_forwarding = mock.Mock( + self.network_client.update_floating_ip_port_forwarding = mock.Mock( return_value=None ) - self.network.find_floating_ip_port_forwarding = mock.Mock( + self.network_client.find_floating_ip_port_forwarding = mock.Mock( return_value=self._port_forwarding ) - self.network.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) # Get the command object to test self.cmd = floating_ip_port_forwarding.SetFloatingIPPortForwarding( self.app, self.namespace @@ -582,7 +584,7 @@ class TestSetFloatingIPPortForwarding(TestFloatingIPPortForwarding): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_floating_ip_port_forwarding.assert_called_with( + self.network_client.update_floating_ip_port_forwarding.assert_called_with( self._port_forwarding.floatingip_id, self._port_forwarding.id, **attrs @@ -643,7 +645,7 @@ class TestSetFloatingIPPortForwarding(TestFloatingIPPortForwarding): result = self.cmd.take_action(parsed_args) - self.network.update_floating_ip_port_forwarding.assert_called_with( + self.network_client.update_floating_ip_port_forwarding.assert_called_with( self._port_forwarding.floatingip_id, self._port_forwarding.id, **attrs @@ -688,10 +690,10 @@ class TestShowFloatingIPPortForwarding(TestFloatingIPPortForwarding): self._port_forwarding.internal_port_range, self._port_forwarding.protocol, ) - self.network.find_floating_ip_port_forwarding = mock.Mock( + self.network_client.find_floating_ip_port_forwarding = mock.Mock( return_value=self._port_forwarding ) - self.network.find_ip = mock.Mock(return_value=self.floating_ip) + self.network_client.find_ip = mock.Mock(return_value=self.floating_ip) # Get the command object to test self.cmd = floating_ip_port_forwarding.ShowFloatingIPPortForwarding( self.app, self.namespace @@ -723,7 +725,7 @@ class TestShowFloatingIPPortForwarding(TestFloatingIPPortForwarding): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_floating_ip_port_forwarding.assert_called_once_with( + self.network_client.find_floating_ip_port_forwarding.assert_called_once_with( self.floating_ip, self._port_forwarding.id, ignore_missing=False ) diff --git a/openstackclient/tests/unit/network/v2/test_ip_availability.py b/openstackclient/tests/unit/network/v2/test_ip_availability.py index 50ead91325..543a89efd3 100644 --- a/openstackclient/tests/unit/network/v2/test_ip_availability.py +++ b/openstackclient/tests/unit/network/v2/test_ip_availability.py @@ -26,7 +26,7 @@ class TestIPAvailability(network_fakes.TestNetworkV2): super(TestIPAvailability, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects @@ -58,7 +58,7 @@ class TestListIPAvailability(TestIPAvailability): super(TestListIPAvailability, self).setUp() self.cmd = ip_availability.ListIPAvailability(self.app, self.namespace) - self.network.network_ip_availabilities = mock.Mock( + self.network_client.network_ip_availabilities = mock.Mock( return_value=self._ip_availability ) @@ -71,7 +71,7 @@ class TestListIPAvailability(TestIPAvailability): columns, data = self.cmd.take_action(parsed_args) filters = {'ip_version': 4} - self.network.network_ip_availabilities.assert_called_once_with( + self.network_client.network_ip_availabilities.assert_called_once_with( **filters ) self.assertEqual(self.columns, columns) @@ -89,7 +89,7 @@ class TestListIPAvailability(TestIPAvailability): columns, data = self.cmd.take_action(parsed_args) filters = {'ip_version': 4} - self.network.network_ip_availabilities.assert_called_once_with( + self.network_client.network_ip_availabilities.assert_called_once_with( **filters ) self.assertEqual(self.columns, columns) @@ -104,7 +104,7 @@ class TestListIPAvailability(TestIPAvailability): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': self.project.id, 'ip_version': 4} - self.network.network_ip_availabilities.assert_called_once_with( + self.network_client.network_ip_availabilities.assert_called_once_with( **filters ) self.assertEqual(self.columns, columns) @@ -137,10 +137,12 @@ class TestShowIPAvailability(TestIPAvailability): def setUp(self): super(TestShowIPAvailability, self).setUp() - self.network.find_network_ip_availability = mock.Mock( + self.network_client.find_network_ip_availability = mock.Mock( return_value=self._ip_availability ) - self.network.find_network = mock.Mock(return_value=self._network) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) # Get the command object to test self.cmd = ip_availability.ShowIPAvailability(self.app, self.namespace) @@ -164,10 +166,10 @@ class TestShowIPAvailability(TestIPAvailability): verifylist = [('network', self._ip_availability.network_name)] parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_network_ip_availability.assert_called_once_with( + self.network_client.find_network_ip_availability.assert_called_once_with( self._ip_availability.network_id, ignore_missing=False ) - self.network.find_network.assert_called_once_with( + self.network_client.find_network.assert_called_once_with( self._ip_availability.network_name, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_l3_conntrack_helper.py b/openstackclient/tests/unit/network/v2/test_l3_conntrack_helper.py index df1d8854b0..00629e7dac 100644 --- a/openstackclient/tests/unit/network/v2/test_l3_conntrack_helper.py +++ b/openstackclient/tests/unit/network/v2/test_l3_conntrack_helper.py @@ -24,9 +24,9 @@ class TestConntrackHelper(network_fakes.TestNetworkV2): def setUp(self): super(TestConntrackHelper, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.router = network_fakes.FakeRouter.create_one_router() - self.network.find_router = mock.Mock(return_value=self.router) + self.network_client.find_router = mock.Mock(return_value=self.router) class TestCreateL3ConntrackHelper(TestConntrackHelper): @@ -47,7 +47,7 @@ class TestCreateL3ConntrackHelper(TestConntrackHelper): self.ct_helper.protocol, self.ct_helper.router_id, ) - self.network.create_conntrack_helper = mock.Mock( + self.network_client.create_conntrack_helper = mock.Mock( return_value=self.ct_helper ) @@ -89,7 +89,7 @@ class TestCreateL3ConntrackHelper(TestConntrackHelper): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_conntrack_helper.assert_called_once_with( + self.network_client.create_conntrack_helper.assert_called_once_with( self.router.id, **{'helper': 'tftp', 'protocol': 'udp', 'port': 69} ) self.assertEqual(self.columns, columns) @@ -122,7 +122,9 @@ class TestDeleteL3ConntrackHelper(TestConntrackHelper): attrs ) ) - self.network.delete_conntrack_helper = mock.Mock(return_value=None) + self.network_client.delete_conntrack_helper = mock.Mock( + return_value=None + ) # Get the command object to test self.cmd = l3_conntrack_helper.DeleteConntrackHelper( @@ -137,7 +139,7 @@ class TestDeleteL3ConntrackHelper(TestConntrackHelper): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_conntrack_helper.assert_called_once_with( + self.network_client.delete_conntrack_helper.assert_called_once_with( self.ct_helper.id, self.router.id, ignore_missing=False ) self.assertIsNone(result) @@ -148,7 +150,7 @@ class TestDeleteL3ConntrackHelper(TestConntrackHelper): ('conntrack_helper_id', [self.ct_helper.id]), ('router', self.router.id), ] - self.network.delete_conntrack_helper.side_effect = Exception( + self.network_client.delete_conntrack_helper.side_effect = Exception( 'Error message' ) parsed_args = self.check_parser(self.cmd, arglist, verifylist) @@ -184,7 +186,9 @@ class TestListL3ConntrackHelper(TestConntrackHelper): ct_helper.port, ) ) - self.network.conntrack_helpers = mock.Mock(return_value=ct_helpers) + self.network_client.conntrack_helpers = mock.Mock( + return_value=ct_helpers + ) # Get the command object to test self.cmd = l3_conntrack_helper.ListConntrackHelper( @@ -200,7 +204,9 @@ class TestListL3ConntrackHelper(TestConntrackHelper): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.conntrack_helpers.assert_called_once_with(self.router.id) + self.network_client.conntrack_helpers.assert_called_once_with( + self.router.id + ) self.assertEqual(self.columns, columns) list_data = list(data) self.assertEqual(len(self.data), len(list_data)) @@ -217,7 +223,9 @@ class TestSetL3ConntrackHelper(TestConntrackHelper): attrs ) ) - self.network.update_conntrack_helper = mock.Mock(return_value=None) + self.network_client.update_conntrack_helper = mock.Mock( + return_value=None + ) # Get the command object to test self.cmd = l3_conntrack_helper.SetConntrackHelper( @@ -237,7 +245,7 @@ class TestSetL3ConntrackHelper(TestConntrackHelper): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_conntrack_helper.assert_called_once_with( + self.network_client.update_conntrack_helper.assert_called_once_with( self.ct_helper.id, self.router.id ) self.assertIsNone(result) @@ -258,7 +266,7 @@ class TestSetL3ConntrackHelper(TestConntrackHelper): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_conntrack_helper.assert_called_once_with( + self.network_client.update_conntrack_helper.assert_called_once_with( self.ct_helper.id, self.router.id, port=124 ) self.assertIsNone(result) @@ -282,7 +290,7 @@ class TestShowL3ConntrackHelper(TestConntrackHelper): self.ct_helper.protocol, self.ct_helper.router_id, ) - self.network.get_conntrack_helper = mock.Mock( + self.network_client.get_conntrack_helper = mock.Mock( return_value=self.ct_helper ) @@ -317,7 +325,7 @@ class TestShowL3ConntrackHelper(TestConntrackHelper): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_conntrack_helper.assert_called_once_with( + self.network_client.get_conntrack_helper.assert_called_once_with( self.ct_helper.id, self.router.id ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_local_ip.py b/openstackclient/tests/unit/network/v2/test_local_ip.py index d5a57545d2..d42595c575 100644 --- a/openstackclient/tests/unit/network/v2/test_local_ip.py +++ b/openstackclient/tests/unit/network/v2/test_local_ip.py @@ -29,7 +29,7 @@ class TestLocalIP(network_fakes.TestNetworkV2): super().setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -79,13 +79,13 @@ class TestCreateLocalIP(TestLocalIP): def setUp(self): super().setUp() - self.network.create_local_ip = mock.Mock( + self.network_client.create_local_ip = mock.Mock( return_value=self.new_local_ip ) - self.network.find_network = mock.Mock( + self.network_client.find_network = mock.Mock( return_value=self.local_ip_network ) - self.network.find_port = mock.Mock(return_value=self.port) + self.network_client.find_port = mock.Mock(return_value=self.port) # Get the command object to test self.cmd = local_ip.CreateLocalIP(self.app, self.namespace) @@ -97,7 +97,7 @@ class TestCreateLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, [], []) columns, data = self.cmd.take_action(parsed_args) - self.network.create_local_ip.assert_called_once_with(**{}) + self.network_client.create_local_ip.assert_called_once_with(**{}) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -131,7 +131,7 @@ class TestCreateLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_local_ip.assert_called_once_with( + self.network_client.create_local_ip.assert_called_once_with( **{ 'name': self.new_local_ip.name, 'description': self.new_local_ip.description, @@ -151,8 +151,8 @@ class TestDeleteLocalIP(TestLocalIP): def setUp(self): super().setUp() - self.network.delete_local_ip = mock.Mock(return_value=None) - self.network.find_local_ip = network_fakes.get_local_ips( + self.network_client.delete_local_ip = mock.Mock(return_value=None) + self.network_client.find_local_ip = network_fakes.get_local_ips( local_ips=self._local_ips ) @@ -170,10 +170,10 @@ class TestDeleteLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_local_ip.assert_called_once_with( + self.network_client.find_local_ip.assert_called_once_with( self._local_ips[0].name, ignore_missing=False ) - self.network.delete_local_ip.assert_called_once_with( + self.network_client.delete_local_ip.assert_called_once_with( self._local_ips[0] ) self.assertIsNone(result) @@ -193,7 +193,7 @@ class TestDeleteLocalIP(TestLocalIP): calls = [] for a in self._local_ips: calls.append(call(a)) - self.network.delete_local_ip.assert_has_calls(calls) + self.network_client.delete_local_ip.assert_has_calls(calls) self.assertIsNone(result) def test_multi_local_ips_delete_with_exception(self): @@ -207,7 +207,9 @@ class TestDeleteLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._local_ips[0], exceptions.CommandError] - self.network.find_local_ip = mock.Mock(side_effect=find_mock_result) + self.network_client.find_local_ip = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -215,13 +217,13 @@ class TestDeleteLocalIP(TestLocalIP): except exceptions.CommandError as e: self.assertEqual('1 of 2 local IPs failed to delete.', str(e)) - self.network.find_local_ip.assert_any_call( + self.network_client.find_local_ip.assert_any_call( self._local_ips[0].name, ignore_missing=False ) - self.network.find_local_ip.assert_any_call( + self.network_client.find_local_ip.assert_any_call( 'unexist_local_ip', ignore_missing=False ) - self.network.delete_local_ip.assert_called_once_with( + self.network_client.delete_local_ip.assert_called_once_with( self._local_ips[0] ) @@ -258,8 +260,10 @@ class TestListLocalIP(TestLocalIP): def setUp(self): super().setUp() - self.network.local_ips = mock.Mock(return_value=self.local_ips) - self.network.find_network = mock.Mock(return_value=self.fake_network) + self.network_client.local_ips = mock.Mock(return_value=self.local_ips) + self.network_client.find_network = mock.Mock( + return_value=self.fake_network + ) # Get the command object to test self.cmd = local_ip.ListLocalIP(self.app, self.namespace) @@ -271,7 +275,7 @@ class TestListLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.local_ips.assert_called_once_with(**{}) + self.network_client.local_ips.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -286,7 +290,7 @@ class TestListLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.local_ips.assert_called_once_with( + self.network_client.local_ips.assert_called_once_with( **{'name': self.local_ips[0].name} ) self.assertEqual(self.columns, columns) @@ -305,7 +309,7 @@ class TestListLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.local_ips.assert_called_once_with( + self.network_client.local_ips.assert_called_once_with( **{'project_id': project.id} ) self.assertEqual(self.columns, columns) @@ -328,7 +332,7 @@ class TestListLocalIP(TestLocalIP): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.local_ips.assert_called_once_with(**filters) + self.network_client.local_ips.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -344,7 +348,7 @@ class TestListLocalIP(TestLocalIP): columns, data = self.cmd.take_action(parsed_args) - self.network.local_ips.assert_called_once_with( + self.network_client.local_ips.assert_called_once_with( **{ 'network_id': 'fake_network_id', } @@ -365,7 +369,7 @@ class TestListLocalIP(TestLocalIP): columns, data = self.cmd.take_action(parsed_args) - self.network.local_ips.assert_called_once_with( + self.network_client.local_ips.assert_called_once_with( **{ 'local_ip_address': self.local_ips[0].local_ip_address, } @@ -385,7 +389,7 @@ class TestListLocalIP(TestLocalIP): columns, data = self.cmd.take_action(parsed_args) - self.network.local_ips.assert_called_once_with( + self.network_client.local_ips.assert_called_once_with( **{ 'ip_mode': self.local_ips[0].ip_mode, } @@ -400,8 +404,10 @@ class TestSetLocalIP(TestLocalIP): def setUp(self): super().setUp() - self.network.update_local_ip = mock.Mock(return_value=None) - self.network.find_local_ip = mock.Mock(return_value=self._local_ip) + self.network_client.update_local_ip = mock.Mock(return_value=None) + self.network_client.find_local_ip = mock.Mock( + return_value=self._local_ip + ) # Get the command object to test self.cmd = local_ip.SetLocalIP(self.app, self.namespace) @@ -417,7 +423,7 @@ class TestSetLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_local_ip.assert_not_called() + self.network_client.update_local_ip.assert_not_called() self.assertIsNone(result) def test_set_name_and_description(self): @@ -440,7 +446,7 @@ class TestSetLocalIP(TestLocalIP): 'name': "new_local_ip_name", 'description': 'new_local_ip_description', } - self.network.update_local_ip.assert_called_with( + self.network_client.update_local_ip.assert_called_with( self._local_ip, **attrs ) self.assertIsNone(result) @@ -478,7 +484,9 @@ class TestShowLocalIP(TestLocalIP): def setUp(self): super().setUp() - self.network.find_local_ip = mock.Mock(return_value=self._local_ip) + self.network_client.find_local_ip = mock.Mock( + return_value=self._local_ip + ) # Get the command object to test self.cmd = local_ip.ShowLocalIP(self.app, self.namespace) @@ -507,7 +515,7 @@ class TestShowLocalIP(TestLocalIP): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_local_ip.assert_called_once_with( + self.network_client.find_local_ip.assert_called_once_with( self._local_ip.name, ignore_missing=False ) self.assertEqual(set(self.columns), set(columns)) diff --git a/openstackclient/tests/unit/network/v2/test_local_ip_association.py b/openstackclient/tests/unit/network/v2/test_local_ip_association.py index ab7341d2e7..8e59c2d61f 100644 --- a/openstackclient/tests/unit/network/v2/test_local_ip_association.py +++ b/openstackclient/tests/unit/network/v2/test_local_ip_association.py @@ -26,11 +26,11 @@ from openstackclient.tests.unit.network.v2 import fakes as network_fakes class TestLocalIPAssociation(network_fakes.TestNetworkV2): def setUp(self): super().setUp() - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.local_ip = network_fakes.create_one_local_ip() self.fixed_port = network_fakes.create_one_port() self.project = identity_fakes_v2.FakeProject.create_one_project() - self.network.find_port = mock.Mock(return_value=self.fixed_port) + self.network_client.find_port = mock.Mock(return_value=self.fixed_port) class TestCreateLocalIPAssociation(TestLocalIPAssociation): @@ -44,11 +44,13 @@ class TestCreateLocalIPAssociation(TestLocalIPAssociation): } ) ) - self.network.create_local_ip_association = mock.Mock( + self.network_client.create_local_ip_association = mock.Mock( return_value=self.new_local_ip_association ) - self.network.find_local_ip = mock.Mock(return_value=self.local_ip) + self.network_client.find_local_ip = mock.Mock( + return_value=self.local_ip + ) # Get the command object to test self.cmd = local_ip_association.CreateLocalIPAssociation( @@ -81,7 +83,7 @@ class TestCreateLocalIPAssociation(TestLocalIPAssociation): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_local_ip_association.assert_called_once_with( + self.network_client.create_local_ip_association.assert_called_once_with( self.new_local_ip_association.local_ip_id, **{ 'fixed_port_id': self.new_local_ip_association.fixed_port_id, @@ -105,7 +107,7 @@ class TestCreateLocalIPAssociation(TestLocalIPAssociation): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_local_ip_association.assert_called_once_with( + self.network_client.create_local_ip_association.assert_called_once_with( self.new_local_ip_association.local_ip_id, **{ 'fixed_port_id': self.new_local_ip_association.fixed_port_id, @@ -127,9 +129,13 @@ class TestDeleteLocalIPAssociation(TestLocalIPAssociation): }, ) ) - self.network.delete_local_ip_association = mock.Mock(return_value=None) + self.network_client.delete_local_ip_association = mock.Mock( + return_value=None + ) - self.network.find_local_ip = mock.Mock(return_value=self.local_ip) + self.network_client.find_local_ip = mock.Mock( + return_value=self.local_ip + ) # Get the command object to test self.cmd = local_ip_association.DeleteLocalIPAssociation( self.app, self.namespace @@ -149,7 +155,7 @@ class TestDeleteLocalIPAssociation(TestLocalIPAssociation): result = self.cmd.take_action(parsed_args) - self.network.delete_local_ip_association.assert_called_once_with( + self.network_client.delete_local_ip_association.assert_called_once_with( self.local_ip.id, self._local_ip_association[0].fixed_port_id, ignore_missing=False, @@ -181,7 +187,7 @@ class TestDeleteLocalIPAssociation(TestLocalIPAssociation): call(a.local_ip_id, a.fixed_port_id, ignore_missing=False) ) - self.network.delete_local_ip_association.assert_has_calls(calls) + self.network_client.delete_local_ip_association.assert_has_calls(calls) self.assertIsNone(result) def test_multi_local_ip_association_delete_with_exception(self): @@ -204,7 +210,7 @@ class TestDeleteLocalIPAssociation(TestLocalIPAssociation): delete_mock_result = [None, exceptions.CommandError] - self.network.delete_local_ip_association = mock.MagicMock( + self.network_client.delete_local_ip_association = mock.MagicMock( side_effect=delete_mock_result ) @@ -216,10 +222,10 @@ class TestDeleteLocalIPAssociation(TestLocalIPAssociation): '1 of 2 Local IP Associations failed to delete.', str(e) ) - self.network.delete_local_ip_association.assert_any_call( + self.network_client.delete_local_ip_association.assert_any_call( self.local_ip.id, 'unexist_fixed_port_id', ignore_missing=False ) - self.network.delete_local_ip_association.assert_any_call( + self.network_client.delete_local_ip_association.assert_any_call( self.local_ip.id, self._local_ip_association[0].fixed_port_id, ignore_missing=False, @@ -257,11 +263,13 @@ class TestListLocalIPAssociation(TestLocalIPAssociation): lip_assoc.host, ) ) - self.network.local_ip_associations = mock.Mock( + self.network_client.local_ip_associations = mock.Mock( return_value=self.local_ip_associations ) - self.network.find_local_ip = mock.Mock(return_value=self.local_ip) - self.network.find_port = mock.Mock(return_value=self.fixed_port) + self.network_client.find_local_ip = mock.Mock( + return_value=self.local_ip + ) + self.network_client.find_port = mock.Mock(return_value=self.fixed_port) # Get the command object to test self.cmd = local_ip_association.ListLocalIPAssociation( self.app, self.namespace @@ -274,7 +282,7 @@ class TestListLocalIPAssociation(TestLocalIPAssociation): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.local_ip_associations.assert_called_once_with( + self.network_client.local_ip_associations.assert_called_once_with( self.local_ip, **{} ) self.assertEqual(set(self.columns), set(columns)) @@ -307,7 +315,7 @@ class TestListLocalIPAssociation(TestLocalIPAssociation): 'host': self.local_ip_associations[0].host, } - self.network.local_ip_associations.assert_called_once_with( + self.network_client.local_ip_associations.assert_called_once_with( self.local_ip, **attrs ) self.assertEqual(set(self.columns), set(columns)) diff --git a/openstackclient/tests/unit/network/v2/test_ndp_proxy.py b/openstackclient/tests/unit/network/v2/test_ndp_proxy.py index 04a9eb61ce..8e79d760d6 100644 --- a/openstackclient/tests/unit/network/v2/test_ndp_proxy.py +++ b/openstackclient/tests/unit/network/v2/test_ndp_proxy.py @@ -30,13 +30,13 @@ class TestNDPProxy(network_fakes.TestNetworkV2): # Get a shortcut to the DomainManager Mock self.domains_mock = self.app.client_manager.identity.domains # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.router = network_fakes.FakeRouter.create_one_router( {'id': 'fake-router-id'} ) - self.network.find_router = mock.Mock(return_value=self.router) + self.network_client.find_router = mock.Mock(return_value=self.router) self.port = network_fakes.create_one_port() - self.network.find_port = mock.Mock(return_value=self.port) + self.network_client.find_port = mock.Mock(return_value=self.port) class TestCreateNDPProxy(TestNDPProxy): @@ -69,7 +69,9 @@ class TestCreateNDPProxy(TestNDPProxy): self.ndp_proxy.router_id, self.ndp_proxy.updated_at, ) - self.network.create_ndp_proxy = mock.Mock(return_value=self.ndp_proxy) + self.network_client.create_ndp_proxy = mock.Mock( + return_value=self.ndp_proxy + ) # Get the command object to test self.cmd = ndp_proxy.CreateNDPProxy(self.app, self.namespace) @@ -109,7 +111,7 @@ class TestCreateNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_ndp_proxy.assert_called_once_with( + self.network_client.create_ndp_proxy.assert_called_once_with( **{ 'name': self.ndp_proxy.name, 'router_id': self.ndp_proxy.router_id, @@ -128,8 +130,10 @@ class TestDeleteNDPProxy(TestNDPProxy): attrs = {'router_id': self.router.id, 'port_id': self.port.id} self.ndp_proxies = network_fakes.create_ndp_proxies(attrs) self.ndp_proxy = self.ndp_proxies[0] - self.network.delete_ndp_proxy = mock.Mock(return_value=None) - self.network.find_ndp_proxy = mock.Mock(return_value=self.ndp_proxy) + self.network_client.delete_ndp_proxy = mock.Mock(return_value=None) + self.network_client.find_ndp_proxy = mock.Mock( + return_value=self.ndp_proxy + ) # Get the command object to test self.cmd = ndp_proxy.DeleteNDPProxy(self.app, self.namespace) @@ -139,7 +143,9 @@ class TestDeleteNDPProxy(TestNDPProxy): verifylist = [('ndp_proxy', [self.ndp_proxy.id])] parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_ndp_proxy.assert_called_once_with(self.ndp_proxy) + self.network_client.delete_ndp_proxy.assert_called_once_with( + self.ndp_proxy + ) self.assertIsNone(result) def test_delete_error(self): @@ -147,7 +153,9 @@ class TestDeleteNDPProxy(TestNDPProxy): self.ndp_proxy.id, ] verifylist = [('ndp_proxy', [self.ndp_proxy.id])] - self.network.delete_ndp_proxy.side_effect = Exception('Error message') + self.network_client.delete_ndp_proxy.side_effect = Exception( + 'Error message' + ) parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.assertRaises( exceptions.CommandError, self.cmd.take_action, parsed_args @@ -168,7 +176,7 @@ class TestDeleteNDPProxy(TestNDPProxy): result = self.cmd.take_action(parsed_args) - self.network.delete_ndp_proxy.assert_has_calls( + self.network_client.delete_ndp_proxy.assert_has_calls( [call(self.ndp_proxy), call(self.ndp_proxy)] ) self.assertIsNone(result) @@ -198,7 +206,7 @@ class TestListNDPProxy(TestNDPProxy): ) ) - self.network.ndp_proxies = mock.Mock(return_value=ndp_proxies) + self.network_client.ndp_proxies = mock.Mock(return_value=ndp_proxies) # Get the command object to test self.cmd = ndp_proxy.ListNDPProxy(self.app, self.namespace) @@ -210,7 +218,7 @@ class TestListNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ndp_proxies.assert_called_once_with() + self.network_client.ndp_proxies.assert_called_once_with() self.assertEqual(self.columns, columns) list_data = list(data) self.assertEqual(len(self.data), len(list_data)) @@ -229,7 +237,7 @@ class TestListNDPProxy(TestNDPProxy): columns, data = self.cmd.take_action(parsed_args) - self.network.ndp_proxies.assert_called_once_with( + self.network_client.ndp_proxies.assert_called_once_with( **{'router_id': 'fake-router-id'} ) self.assertEqual(self.columns, columns) @@ -247,7 +255,7 @@ class TestListNDPProxy(TestNDPProxy): columns, data = self.cmd.take_action(parsed_args) - self.network.ndp_proxies.assert_called_once_with( + self.network_client.ndp_proxies.assert_called_once_with( **{'port_id': self.port.id} ) self.assertEqual(self.columns, columns) @@ -265,7 +273,7 @@ class TestListNDPProxy(TestNDPProxy): columns, data = self.cmd.take_action(parsed_args) - self.network.ndp_proxies.assert_called_once_with( + self.network_client.ndp_proxies.assert_called_once_with( **{'name': 'fake-ndp-proxy-name'} ) self.assertEqual(self.columns, columns) @@ -283,7 +291,7 @@ class TestListNDPProxy(TestNDPProxy): columns, data = self.cmd.take_action(parsed_args) - self.network.ndp_proxies.assert_called_once_with( + self.network_client.ndp_proxies.assert_called_once_with( **{'ip_address': '2001::1:2'} ) self.assertEqual(self.columns, columns) @@ -302,7 +310,7 @@ class TestListNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ndp_proxies.assert_called_once_with( + self.network_client.ndp_proxies.assert_called_once_with( **{'project_id': project.id} ) self.assertEqual(self.columns, columns) @@ -325,7 +333,7 @@ class TestListNDPProxy(TestNDPProxy): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.ndp_proxies.assert_called_once_with(**filters) + self.network_client.ndp_proxies.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertItemsEqual(self.data, list(data)) @@ -335,8 +343,10 @@ class TestSetNDPProxy(TestNDPProxy): super(TestSetNDPProxy, self).setUp() attrs = {'router_id': self.router.id, 'port_id': self.port.id} self.ndp_proxy = network_fakes.create_one_ndp_proxy(attrs) - self.network.update_ndp_proxy = mock.Mock(return_value=None) - self.network.find_ndp_proxy = mock.Mock(return_value=self.ndp_proxy) + self.network_client.update_ndp_proxy = mock.Mock(return_value=None) + self.network_client.find_ndp_proxy = mock.Mock( + return_value=self.ndp_proxy + ) # Get the command object to test self.cmd = ndp_proxy.SetNDPProxy(self.app, self.namespace) @@ -352,7 +362,9 @@ class TestSetNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_ndp_proxy.assert_called_once_with(self.ndp_proxy) + self.network_client.update_ndp_proxy.assert_called_once_with( + self.ndp_proxy + ) self.assertIsNone(result) def test_set_name(self): @@ -369,7 +381,7 @@ class TestSetNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_ndp_proxy.assert_called_once_with( + self.network_client.update_ndp_proxy.assert_called_once_with( self.ndp_proxy, name='fake-name' ) self.assertIsNone(result) @@ -388,7 +400,7 @@ class TestSetNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_ndp_proxy.assert_called_once_with( + self.network_client.update_ndp_proxy.assert_called_once_with( self.ndp_proxy, description='balala' ) self.assertIsNone(result) @@ -425,8 +437,12 @@ class TestShowNDPProxy(TestNDPProxy): self.ndp_proxy.router_id, self.ndp_proxy.updated_at, ) - self.network.get_ndp_proxy = mock.Mock(return_value=self.ndp_proxy) - self.network.find_ndp_proxy = mock.Mock(return_value=self.ndp_proxy) + self.network_client.get_ndp_proxy = mock.Mock( + return_value=self.ndp_proxy + ) + self.network_client.find_ndp_proxy = mock.Mock( + return_value=self.ndp_proxy + ) # Get the command object to test self.cmd = ndp_proxy.ShowNDPProxy(self.app, self.namespace) @@ -455,7 +471,7 @@ class TestShowNDPProxy(TestNDPProxy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_ndp_proxy.assert_called_once_with( + self.network_client.find_ndp_proxy.assert_called_once_with( self.ndp_proxy.id, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network.py b/openstackclient/tests/unit/network/v2/test_network.py index 76b33a556b..60a7f7e996 100644 --- a/openstackclient/tests/unit/network/v2/test_network.py +++ b/openstackclient/tests/unit/network/v2/test_network.py @@ -33,7 +33,7 @@ class TestNetwork(network_fakes.TestNetworkV2): super(TestNetwork, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -117,15 +117,19 @@ class TestCreateNetworkIdentityV3(TestNetwork): def setUp(self): super(TestCreateNetworkIdentityV3, self).setUp() - self.network.create_network = mock.Mock(return_value=self._network) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.create_network = mock.Mock( + return_value=self._network + ) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = network.CreateNetwork(self.app, self.namespace) self.projects_mock.get.return_value = self.project self.domains_mock.get.return_value = self.domain - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) def test_create_no_options(self): arglist = [] @@ -154,13 +158,13 @@ class TestCreateNetworkIdentityV3(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network.assert_called_once_with( + self.network_client.create_network.assert_called_once_with( **{ 'admin_state_up': True, 'name': self._network.name, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -217,7 +221,7 @@ class TestCreateNetworkIdentityV3(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network.assert_called_once_with( + self.network_client.create_network.assert_called_once_with( **{ 'admin_state_up': False, 'availability_zone_hints': ["nova"], @@ -258,7 +262,7 @@ class TestCreateNetworkIdentityV3(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network.assert_called_once_with( + self.network_client.create_network.assert_called_once_with( **{ 'admin_state_up': True, 'name': self._network.name, @@ -290,15 +294,15 @@ class TestCreateNetworkIdentityV3(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network.assert_called_once_with( + self.network_client.create_network.assert_called_once_with( name=self._network.name, admin_state_up=True ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self._network, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -378,8 +382,10 @@ class TestCreateNetworkIdentityV2(TestNetwork): def setUp(self): super(TestCreateNetworkIdentityV2, self).setUp() - self.network.create_network = mock.Mock(return_value=self._network) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.create_network = mock.Mock( + return_value=self._network + ) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = network.CreateNetwork(self.app, self.namespace) @@ -415,14 +421,14 @@ class TestCreateNetworkIdentityV2(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network.assert_called_once_with( + self.network_client.create_network.assert_called_once_with( **{ 'admin_state_up': True, 'name': self._network.name, 'project_id': self.project.id, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -459,9 +465,9 @@ class TestDeleteNetwork(TestNetwork): # The networks to delete self._networks = network_fakes.create_networks(count=3) - self.network.delete_network = mock.Mock(return_value=None) + self.network_client.delete_network = mock.Mock(return_value=None) - self.network.find_network = network_fakes.get_networks( + self.network_client.find_network = network_fakes.get_networks( networks=self._networks ) @@ -479,7 +485,9 @@ class TestDeleteNetwork(TestNetwork): result = self.cmd.take_action(parsed_args) - self.network.delete_network.assert_called_once_with(self._networks[0]) + self.network_client.delete_network.assert_called_once_with( + self._networks[0] + ) self.assertIsNone(result) def test_delete_multiple_networks(self): @@ -496,7 +504,7 @@ class TestDeleteNetwork(TestNetwork): calls = [] for n in self._networks: calls.append(call(n)) - self.network.delete_network.assert_has_calls(calls) + self.network_client.delete_network.assert_has_calls(calls) self.assertIsNone(result) def test_delete_multiple_networks_exception(self): @@ -516,14 +524,14 @@ class TestDeleteNetwork(TestNetwork): exceptions.NotFound('404'), self._networks[1], ] - self.network.find_network = mock.Mock(side_effect=ret_find) + self.network_client.find_network = mock.Mock(side_effect=ret_find) # Fake exception in delete_network() ret_delete = [ None, exceptions.NotFound('404'), ] - self.network.delete_network = mock.Mock(side_effect=ret_delete) + self.network_client.delete_network = mock.Mock(side_effect=ret_delete) self.assertRaises( exceptions.CommandError, self.cmd.take_action, parsed_args @@ -535,7 +543,7 @@ class TestDeleteNetwork(TestNetwork): call(self._networks[0]), call(self._networks[1]), ] - self.network.delete_network.assert_has_calls(calls) + self.network_client.delete_network.assert_has_calls(calls) class TestListNetwork(TestNetwork): @@ -595,17 +603,17 @@ class TestListNetwork(TestNetwork): # Get the command object to test self.cmd = network.ListNetwork(self.app, self.namespace) - self.network.networks = mock.Mock(return_value=self._network) + self.network_client.networks = mock.Mock(return_value=self._network) self._agent = network_fakes.create_one_network_agent() - self.network.get_agent = mock.Mock(return_value=self._agent) + self.network_client.get_agent = mock.Mock(return_value=self._agent) - self.network.dhcp_agent_hosting_networks = mock.Mock( + self.network_client.dhcp_agent_hosting_networks = mock.Mock( return_value=self._network ) # TestListTagMixin - self._tag_list_resource_mock = self.network.networks + self._tag_list_resource_mock = self.network_client.networks def test_network_list_no_options(self): arglist = [] @@ -620,7 +628,7 @@ class TestListNetwork(TestNetwork): # containing the data to be listed. columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with() + self.network_client.networks.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -639,7 +647,7 @@ class TestListNetwork(TestNetwork): # containing the data to be listed. columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'router:external': True, 'is_router_external': True} ) self.assertEqual(self.columns, columns) @@ -656,7 +664,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'router:external': False, 'is_router_external': False} ) self.assertEqual(self.columns, columns) @@ -677,7 +685,7 @@ class TestListNetwork(TestNetwork): # containing the data to be listed. columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with() + self.network_client.networks.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertCountEqual(self.data_long, list(data)) @@ -695,7 +703,9 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with(**{'name': test_name}) + self.network_client.networks.assert_called_once_with( + **{'name': test_name} + ) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -711,7 +721,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'admin_state_up': True, 'is_admin_state_up': True} ) self.assertEqual(self.columns, columns) @@ -725,7 +735,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'admin_state_up': False, 'is_admin_state_up': False} ) self.assertEqual(self.columns, columns) @@ -744,7 +754,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'project_id': project.id} ) @@ -768,7 +778,7 @@ class TestListNetwork(TestNetwork): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.networks.assert_called_once_with(**filters) + self.network_client.networks.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -783,7 +793,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'shared': True, 'is_shared': True} ) self.assertEqual(self.columns, columns) @@ -800,7 +810,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'shared': False, 'is_shared': False} ) self.assertEqual(self.columns, columns) @@ -820,7 +830,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{'status': test_status} ) self.assertEqual(self.columns, columns) @@ -838,7 +848,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{ 'provider:network_type': network_type, 'provider_network_type': network_type, @@ -859,7 +869,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{ 'provider:physical_network': physical_network, 'provider_physical_network': physical_network, @@ -880,7 +890,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{ 'provider:segmentation_id': segmentation_id, 'provider_segmentation_id': segmentation_id, @@ -899,7 +909,7 @@ class TestListNetwork(TestNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.dhcp_agent_hosting_networks.assert_called_once_with( + self.network_client.dhcp_agent_hosting_networks.assert_called_once_with( self._agent ) @@ -926,7 +936,7 @@ class TestListNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.networks.assert_called_once_with( + self.network_client.networks.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -948,11 +958,15 @@ class TestSetNetwork(TestNetwork): def setUp(self): super(TestSetNetwork, self).setUp() - self.network.update_network = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.update_network = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) - self.network.find_network = mock.Mock(return_value=self._network) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) # Get the command object to test self.cmd = network.SetNetwork(self.app, self.namespace) @@ -1013,7 +1027,7 @@ class TestSetNetwork(TestNetwork): 'qos_policy_id': self.qos_policy.id, 'dns_domain': 'example.org.', } - self.network.update_network.assert_called_once_with( + self.network_client.update_network.assert_called_once_with( self._network, **attrs ) self.assertIsNone(result) @@ -1046,7 +1060,7 @@ class TestSetNetwork(TestNetwork): 'port_security_enabled': False, 'qos_policy_id': None, } - self.network.update_network.assert_called_once_with( + self.network_client.update_network.assert_called_once_with( self._network, **attrs ) self.assertIsNone(result) @@ -1062,8 +1076,8 @@ class TestSetNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_network.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_network.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def _test_set_tags(self, with_tags=True): @@ -1081,8 +1095,8 @@ class TestSetNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_network.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_network.called) + self.network_client.set_tags.assert_called_once_with( self._network, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -1160,7 +1174,9 @@ class TestShowNetwork(TestNetwork): def setUp(self): super(TestShowNetwork, self).setUp() - self.network.find_network = mock.Mock(return_value=self._network) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) # Get the command object to test self.cmd = network.ShowNetwork(self.app, self.namespace) @@ -1188,7 +1204,7 @@ class TestShowNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_network.assert_called_once_with( + self.network_client.find_network.assert_called_once_with( self._network.name, ignore_missing=False ) @@ -1206,11 +1222,15 @@ class TestUnsetNetwork(TestNetwork): def setUp(self): super(TestUnsetNetwork, self).setUp() - self.network.update_network = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.update_network = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) - self.network.find_network = mock.Mock(return_value=self._network) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) # Get the command object to test self.cmd = network.UnsetNetwork(self.app, self.namespace) @@ -1226,8 +1246,8 @@ class TestUnsetNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_network.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_network.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def _test_unset_tags(self, with_tags=True): @@ -1245,8 +1265,8 @@ class TestUnsetNetwork(TestNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_network.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_network.called) + self.network_client.set_tags.assert_called_once_with( self._network, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_network_agent.py b/openstackclient/tests/unit/network/v2/test_network_agent.py index e3f1269c22..cdfacd1067 100644 --- a/openstackclient/tests/unit/network/v2/test_network_agent.py +++ b/openstackclient/tests/unit/network/v2/test_network_agent.py @@ -27,7 +27,7 @@ class TestNetworkAgent(network_fakes.TestNetworkV2): super(TestNetworkAgent, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network class TestAddNetworkToAgent(TestNetworkAgent): @@ -37,10 +37,10 @@ class TestAddNetworkToAgent(TestNetworkAgent): def setUp(self): super(TestAddNetworkToAgent, self).setUp() - self.network.get_agent = mock.Mock(return_value=self.agent) - self.network.find_network = mock.Mock(return_value=self.net) - self.network.name = self.network.find_network.name - self.network.add_dhcp_agent_to_network = mock.Mock() + self.network_client.get_agent = mock.Mock(return_value=self.agent) + self.network_client.find_network = mock.Mock(return_value=self.net) + self.network_client.name = self.network_client.find_network.name + self.network_client.add_dhcp_agent_to_network = mock.Mock() self.cmd = network_agent.AddNetworkToAgent(self.app, self.namespace) def test_show_no_options(self): @@ -67,7 +67,7 @@ class TestAddNetworkToAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.add_dhcp_agent_to_network.assert_called_once_with( + self.network_client.add_dhcp_agent_to_network.assert_called_once_with( self.agent, self.net ) @@ -78,10 +78,10 @@ class TestAddRouterAgent(TestNetworkAgent): def setUp(self): super(TestAddRouterAgent, self).setUp() - self.network.add_router_to_agent = mock.Mock() + self.network_client.add_router_to_agent = mock.Mock() self.cmd = network_agent.AddRouterToAgent(self.app, self.namespace) - self.network.get_agent = mock.Mock(return_value=self._agent) - self.network.find_router = mock.Mock(return_value=self._router) + self.network_client.get_agent = mock.Mock(return_value=self._agent) + self.network_client.find_router = mock.Mock(return_value=self._router) def test_add_no_options(self): arglist = [] @@ -111,7 +111,7 @@ class TestAddRouterAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.add_router_to_agent.assert_called_with( + self.network_client.add_router_to_agent.assert_called_with( self._agent, self._router ) self.assertIsNone(result) @@ -122,7 +122,7 @@ class TestDeleteNetworkAgent(TestNetworkAgent): def setUp(self): super(TestDeleteNetworkAgent, self).setUp() - self.network.delete_agent = mock.Mock(return_value=None) + self.network_client.delete_agent = mock.Mock(return_value=None) # Get the command object to test self.cmd = network_agent.DeleteNetworkAgent(self.app, self.namespace) @@ -138,7 +138,7 @@ class TestDeleteNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_agent.assert_called_once_with( + self.network_client.delete_agent.assert_called_once_with( self.network_agents[0].id, ignore_missing=False ) self.assertIsNone(result) @@ -158,7 +158,7 @@ class TestDeleteNetworkAgent(TestNetworkAgent): calls = [] for n in self.network_agents: calls.append(call(n.id, ignore_missing=False)) - self.network.delete_agent.assert_has_calls(calls) + self.network_client.delete_agent.assert_has_calls(calls) self.assertIsNone(result) def test_multi_network_agents_delete_with_exception(self): @@ -175,7 +175,9 @@ class TestDeleteNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) delete_mock_result = [True, exceptions.CommandError] - self.network.delete_agent = mock.Mock(side_effect=delete_mock_result) + self.network_client.delete_agent = mock.Mock( + side_effect=delete_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -183,10 +185,10 @@ class TestDeleteNetworkAgent(TestNetworkAgent): except exceptions.CommandError as e: self.assertEqual('1 of 2 network agents failed to delete.', str(e)) - self.network.delete_agent.assert_any_call( + self.network_client.delete_agent.assert_any_call( self.network_agents[0].id, ignore_missing=False ) - self.network.delete_agent.assert_any_call( + self.network_client.delete_agent.assert_any_call( 'unexist_network_agent', ignore_missing=False ) @@ -219,22 +221,28 @@ class TestListNetworkAgent(TestNetworkAgent): def setUp(self): super(TestListNetworkAgent, self).setUp() - self.network.agents = mock.Mock(return_value=self.network_agents) - - _testagent = network_fakes.create_one_network_agent() - self.network.get_agent = mock.Mock(return_value=_testagent) - - self._testnetwork = network_fakes.create_one_network() - self.network.find_network = mock.Mock(return_value=self._testnetwork) - self.network.network_hosting_dhcp_agents = mock.Mock( + self.network_client.agents = mock.Mock( return_value=self.network_agents ) - self.network.get_agent = mock.Mock(return_value=_testagent) + _testagent = network_fakes.create_one_network_agent() + self.network_client.get_agent = mock.Mock(return_value=_testagent) + + self._testnetwork = network_fakes.create_one_network() + self.network_client.find_network = mock.Mock( + return_value=self._testnetwork + ) + self.network_client.network_hosting_dhcp_agents = mock.Mock( + return_value=self.network_agents + ) + + self.network_client.get_agent = mock.Mock(return_value=_testagent) self._testrouter = network_fakes.FakeRouter.create_one_router() - self.network.find_router = mock.Mock(return_value=self._testrouter) - self.network.routers_hosting_l3_agents = mock.Mock( + self.network_client.find_router = mock.Mock( + return_value=self._testrouter + ) + self.network_client.routers_hosting_l3_agents = mock.Mock( return_value=self.network_agents ) @@ -248,7 +256,7 @@ class TestListNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.agents.assert_called_once_with(**{}) + self.network_client.agents.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -264,7 +272,7 @@ class TestListNetworkAgent(TestNetworkAgent): columns, data = self.cmd.take_action(parsed_args) - self.network.agents.assert_called_once_with( + self.network_client.agents.assert_called_once_with( **{ 'agent_type': 'DHCP agent', } @@ -284,7 +292,7 @@ class TestListNetworkAgent(TestNetworkAgent): columns, data = self.cmd.take_action(parsed_args) - self.network.agents.assert_called_once_with( + self.network_client.agents.assert_called_once_with( **{ 'host': self.network_agents[0].host, } @@ -304,7 +312,7 @@ class TestListNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.network_hosting_dhcp_agents.assert_called_once_with( + self.network_client.network_hosting_dhcp_agents.assert_called_once_with( self._testnetwork ) self.assertEqual(self.columns, columns) @@ -324,7 +332,9 @@ class TestListNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.routers_hosting_l3_agents.assert_called_once_with(*attrs) + self.network_client.routers_hosting_l3_agents.assert_called_once_with( + *attrs + ) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -344,7 +354,9 @@ class TestListNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.routers_hosting_l3_agents.assert_called_once_with(*attrs) + self.network_client.routers_hosting_l3_agents.assert_called_once_with( + *attrs + ) # Add a column 'HA State' and corresponding data. router_agent_columns = self.columns + ('HA State',) @@ -361,10 +373,10 @@ class TestRemoveNetworkFromAgent(TestNetworkAgent): def setUp(self): super(TestRemoveNetworkFromAgent, self).setUp() - self.network.get_agent = mock.Mock(return_value=self.agent) - self.network.find_network = mock.Mock(return_value=self.net) - self.network.name = self.network.find_network.name - self.network.remove_dhcp_agent_from_network = mock.Mock() + self.network_client.get_agent = mock.Mock(return_value=self.agent) + self.network_client.find_network = mock.Mock(return_value=self.net) + self.network_client.name = self.network_client.find_network.name + self.network_client.remove_dhcp_agent_from_network = mock.Mock() self.cmd = network_agent.RemoveNetworkFromAgent( self.app, self.namespace ) @@ -408,7 +420,7 @@ class TestRemoveNetworkFromAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.remove_dhcp_agent_from_network.assert_called_once_with( + self.network_client.remove_dhcp_agent_from_network.assert_called_once_with( self.agent, self.net ) @@ -419,12 +431,12 @@ class TestRemoveRouterAgent(TestNetworkAgent): def setUp(self): super(TestRemoveRouterAgent, self).setUp() - self.network.remove_router_from_agent = mock.Mock() + self.network_client.remove_router_from_agent = mock.Mock() self.cmd = network_agent.RemoveRouterFromAgent( self.app, self.namespace ) - self.network.get_agent = mock.Mock(return_value=self._agent) - self.network.find_router = mock.Mock(return_value=self._router) + self.network_client.get_agent = mock.Mock(return_value=self._agent) + self.network_client.find_router = mock.Mock(return_value=self._router) def test_remove_no_options(self): arglist = [] @@ -454,7 +466,7 @@ class TestRemoveRouterAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.remove_router_from_agent.assert_called_with( + self.network_client.remove_router_from_agent.assert_called_with( self._agent, self._router ) self.assertIsNone(result) @@ -465,8 +477,10 @@ class TestSetNetworkAgent(TestNetworkAgent): def setUp(self): super(TestSetNetworkAgent, self).setUp() - self.network.update_agent = mock.Mock(return_value=None) - self.network.get_agent = mock.Mock(return_value=self._network_agent) + self.network_client.update_agent = mock.Mock(return_value=None) + self.network_client.get_agent = mock.Mock( + return_value=self._network_agent + ) # Get the command object to test self.cmd = network_agent.SetNetworkAgent(self.app, self.namespace) @@ -483,7 +497,7 @@ class TestSetNetworkAgent(TestNetworkAgent): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_agent.assert_called_once_with( + self.network_client.update_agent.assert_called_once_with( self._network_agent, **attrs ) self.assertIsNone(result) @@ -510,7 +524,7 @@ class TestSetNetworkAgent(TestNetworkAgent): 'admin_state_up': True, 'is_admin_state_up': True, } - self.network.update_agent.assert_called_once_with( + self.network_client.update_agent.assert_called_once_with( self._network_agent, **attrs ) self.assertIsNone(result) @@ -533,7 +547,7 @@ class TestSetNetworkAgent(TestNetworkAgent): 'admin_state_up': False, 'is_admin_state_up': False, } - self.network.update_agent.assert_called_once_with( + self.network_client.update_agent.assert_called_once_with( self._network_agent, **attrs ) self.assertIsNone(result) @@ -579,7 +593,9 @@ class TestShowNetworkAgent(TestNetworkAgent): def setUp(self): super(TestShowNetworkAgent, self).setUp() - self.network.get_agent = mock.Mock(return_value=self._network_agent) + self.network_client.get_agent = mock.Mock( + return_value=self._network_agent + ) # Get the command object to test self.cmd = network_agent.ShowNetworkAgent(self.app, self.namespace) @@ -608,6 +624,8 @@ class TestShowNetworkAgent(TestNetworkAgent): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_agent.assert_called_once_with(self._network_agent.id) + self.network_client.get_agent.assert_called_once_with( + self._network_agent.id + ) self.assertEqual(set(self.columns), set(columns)) self.assertEqual(len(list(self.data)), len(list(data))) diff --git a/openstackclient/tests/unit/network/v2/test_network_auto_allocated_topology.py b/openstackclient/tests/unit/network/v2/test_network_auto_allocated_topology.py index fb0c4f8963..9b55d41767 100644 --- a/openstackclient/tests/unit/network/v2/test_network_auto_allocated_topology.py +++ b/openstackclient/tests/unit/network/v2/test_network_auto_allocated_topology.py @@ -23,7 +23,7 @@ from openstackclient.tests.unit.network.v2 import fakes as network_fakes class TestAutoAllocatedTopology(network_fakes.TestNetworkV2): def setUp(self): super(TestAutoAllocatedTopology, self).setUp() - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.projects_mock = self.app.client_manager.identity.projects @@ -51,7 +51,7 @@ class TestCreateAutoAllocatedTopology(TestAutoAllocatedTopology): self.cmd = network_auto_allocated_topology.CreateAutoAllocatedTopology( self.app, self.namespace ) - self.network.get_auto_allocated_topology = mock.Mock( + self.network_client.get_auto_allocated_topology = mock.Mock( return_value=self.topology ) @@ -62,7 +62,9 @@ class TestCreateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_auto_allocated_topology.assert_called_with(None) + self.network_client.get_auto_allocated_topology.assert_called_with( + None + ) self.assertEqual(self.columns, columns) self.assertEqual(self.data, data) @@ -80,7 +82,7 @@ class TestCreateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_auto_allocated_topology.assert_called_with( + self.network_client.get_auto_allocated_topology.assert_called_with( self.project.id ) @@ -103,7 +105,7 @@ class TestCreateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_auto_allocated_topology.assert_called_with( + self.network_client.get_auto_allocated_topology.assert_called_with( self.project.id ) @@ -122,7 +124,9 @@ class TestCreateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_auto_allocated_topology.assert_called_with(None) + self.network_client.get_auto_allocated_topology.assert_called_with( + None + ) self.assertEqual(self.columns, columns) self.assertEqual(self.data, data) @@ -152,7 +156,7 @@ class TestValidateAutoAllocatedTopology(TestAutoAllocatedTopology): self.cmd = network_auto_allocated_topology.CreateAutoAllocatedTopology( self.app, self.namespace ) - self.network.validate_auto_allocated_topology = mock.Mock( + self.network_client.validate_auto_allocated_topology = mock.Mock( return_value=self.topology ) @@ -167,7 +171,9 @@ class TestValidateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.validate_auto_allocated_topology.assert_called_with(None) + self.network_client.validate_auto_allocated_topology.assert_called_with( + None + ) def test_show_dry_run_project_option(self): arglist = [ @@ -183,7 +189,7 @@ class TestValidateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.validate_auto_allocated_topology.assert_called_with( + self.network_client.validate_auto_allocated_topology.assert_called_with( self.project.id ) @@ -204,7 +210,7 @@ class TestValidateAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.validate_auto_allocated_topology.assert_called_with( + self.network_client.validate_auto_allocated_topology.assert_called_with( self.project.id ) @@ -223,7 +229,7 @@ class TestDeleteAutoAllocatedTopology(TestAutoAllocatedTopology): self.cmd = network_auto_allocated_topology.DeleteAutoAllocatedTopology( self.app, self.namespace ) - self.network.delete_auto_allocated_topology = mock.Mock( + self.network_client.delete_auto_allocated_topology = mock.Mock( return_value=None ) @@ -234,7 +240,7 @@ class TestDeleteAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_auto_allocated_topology.assert_called_once_with( + self.network_client.delete_auto_allocated_topology.assert_called_once_with( None ) @@ -252,7 +258,7 @@ class TestDeleteAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_auto_allocated_topology.assert_called_once_with( + self.network_client.delete_auto_allocated_topology.assert_called_once_with( self.project.id ) @@ -273,7 +279,7 @@ class TestDeleteAutoAllocatedTopology(TestAutoAllocatedTopology): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_auto_allocated_topology.assert_called_once_with( + self.network_client.delete_auto_allocated_topology.assert_called_once_with( self.project.id ) diff --git a/openstackclient/tests/unit/network/v2/test_network_flavor.py b/openstackclient/tests/unit/network/v2/test_network_flavor.py index 35df3b77f3..e3345fd4f9 100644 --- a/openstackclient/tests/unit/network/v2/test_network_flavor.py +++ b/openstackclient/tests/unit/network/v2/test_network_flavor.py @@ -29,7 +29,7 @@ class TestNetworkFlavor(network_fakes.TestNetworkV2): super(TestNetworkFlavor, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -42,11 +42,13 @@ class TestAddNetworkFlavorToProfile(TestNetworkFlavor): def setUp(self): super(TestAddNetworkFlavorToProfile, self).setUp() - self.network.find_flavor = mock.Mock(return_value=self.network_flavor) - self.network.find_service_profile = mock.Mock( + self.network_client.find_flavor = mock.Mock( + return_value=self.network_flavor + ) + self.network_client.find_service_profile = mock.Mock( return_value=self.service_profile ) - self.network.associate_flavor_with_service_profile = mock.Mock() + self.network_client.associate_flavor_with_service_profile = mock.Mock() self.cmd = network_flavor.AddNetworkFlavorToProfile( self.app, self.namespace @@ -75,7 +77,7 @@ class TestAddNetworkFlavorToProfile(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.associate_flavor_with_service_profile.assert_called_once_with( # noqa: E501 + self.network_client.associate_flavor_with_service_profile.assert_called_once_with( # noqa: E501 self.network_flavor, self.service_profile ) @@ -104,7 +106,7 @@ class TestCreateNetworkFlavor(TestNetworkFlavor): def setUp(self): super(TestCreateNetworkFlavor, self).setUp() - self.network.create_flavor = mock.Mock( + self.network_client.create_flavor = mock.Mock( return_value=self.new_network_flavor ) @@ -141,7 +143,7 @@ class TestCreateNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_flavor.assert_called_once_with( + self.network_client.create_flavor.assert_called_once_with( **{ 'service_type': self.new_network_flavor.service_type, 'name': self.new_network_flavor.name, @@ -175,7 +177,7 @@ class TestCreateNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_flavor.assert_called_once_with( + self.network_client.create_flavor.assert_called_once_with( **{ 'description': self.new_network_flavor.description, 'enabled': True, @@ -203,7 +205,7 @@ class TestCreateNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_flavor.assert_called_once_with( + self.network_client.create_flavor.assert_called_once_with( **{ 'enabled': False, 'service_type': self.new_network_flavor.service_type, @@ -220,8 +222,8 @@ class TestDeleteNetworkFlavor(TestNetworkFlavor): def setUp(self): super(TestDeleteNetworkFlavor, self).setUp() - self.network.delete_flavor = mock.Mock(return_value=None) - self.network.find_flavor = network_fakes.get_flavor( + self.network_client.delete_flavor = mock.Mock(return_value=None) + self.network_client.find_flavor = network_fakes.get_flavor( network_flavors=self._network_flavors ) @@ -239,10 +241,10 @@ class TestDeleteNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_flavor.assert_called_once_with( + self.network_client.find_flavor.assert_called_once_with( self._network_flavors[0].name, ignore_missing=False ) - self.network.delete_flavor.assert_called_once_with( + self.network_client.delete_flavor.assert_called_once_with( self._network_flavors[0] ) self.assertIsNone(result) @@ -263,7 +265,7 @@ class TestDeleteNetworkFlavor(TestNetworkFlavor): calls = [] for a in self._network_flavors: calls.append(mock.call(a)) - self.network.delete_flavor.assert_has_calls(calls) + self.network_client.delete_flavor.assert_has_calls(calls) self.assertIsNone(result) def test_multi_network_flavors_delete_with_exception(self): @@ -280,7 +282,9 @@ class TestDeleteNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._network_flavors[0], exceptions.CommandError] - self.network.find_flavor = mock.Mock(side_effect=find_mock_result) + self.network_client.find_flavor = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -288,13 +292,13 @@ class TestDeleteNetworkFlavor(TestNetworkFlavor): except exceptions.CommandError as e: self.assertEqual('1 of 2 flavors failed to delete.', str(e)) - self.network.find_flavor.assert_any_call( + self.network_client.find_flavor.assert_any_call( self._network_flavors[0].name, ignore_missing=False ) - self.network.find_flavor.assert_any_call( + self.network_client.find_flavor.assert_any_call( 'unexist_network_flavor', ignore_missing=False ) - self.network.delete_flavor.assert_called_once_with( + self.network_client.delete_flavor.assert_called_once_with( self._network_flavors[0] ) @@ -323,7 +327,9 @@ class TestListNetworkFlavor(TestNetworkFlavor): def setUp(self): super(TestListNetworkFlavor, self).setUp() - self.network.flavors = mock.Mock(return_value=self._network_flavors) + self.network_client.flavors = mock.Mock( + return_value=self._network_flavors + ) # Get the command object to test self.cmd = network_flavor.ListNetworkFlavor(self.app, self.namespace) @@ -335,7 +341,7 @@ class TestListNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.flavors.assert_called_once_with(**{}) + self.network_client.flavors.assert_called_once_with(**{}) self.assertEqual(set(self.columns), set(columns)) self.assertEqual(self.data, list(data)) @@ -346,11 +352,15 @@ class TestRemoveNetworkFlavorFromProfile(TestNetworkFlavor): def setUp(self): super(TestRemoveNetworkFlavorFromProfile, self).setUp() - self.network.find_flavor = mock.Mock(return_value=self.network_flavor) - self.network.find_service_profile = mock.Mock( + self.network_client.find_flavor = mock.Mock( + return_value=self.network_flavor + ) + self.network_client.find_service_profile = mock.Mock( return_value=self.service_profile ) - self.network.disassociate_flavor_from_service_profile = mock.Mock() + self.network_client.disassociate_flavor_from_service_profile = ( + mock.Mock() + ) self.cmd = network_flavor.RemoveNetworkFlavorFromProfile( self.app, self.namespace @@ -379,7 +389,7 @@ class TestRemoveNetworkFlavorFromProfile(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) self.cmd.take_action(parsed_args) - self.network.disassociate_flavor_from_service_profile.assert_called_once_with( # noqa: E501 + self.network_client.disassociate_flavor_from_service_profile.assert_called_once_with( # noqa: E501 self.network_flavor, self.service_profile ) @@ -406,7 +416,7 @@ class TestShowNetworkFlavor(TestNetworkFlavor): def setUp(self): super(TestShowNetworkFlavor, self).setUp() - self.network.find_flavor = mock.Mock( + self.network_client.find_flavor = mock.Mock( return_value=self.new_network_flavor ) @@ -437,7 +447,7 @@ class TestShowNetworkFlavor(TestNetworkFlavor): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_flavor.assert_called_once_with( + self.network_client.find_flavor.assert_called_once_with( self.new_network_flavor.name, ignore_missing=False ) self.assertEqual(set(self.columns), set(columns)) @@ -450,8 +460,8 @@ class TestSetNetworkFlavor(TestNetworkFlavor): def setUp(self): super(TestSetNetworkFlavor, self).setUp() - self.network.update_flavor = mock.Mock(return_value=None) - self.network.find_flavor = mock.Mock( + self.network_client.update_flavor = mock.Mock(return_value=None) + self.network_client.find_flavor = mock.Mock( return_value=self.new_network_flavor ) @@ -470,7 +480,7 @@ class TestSetNetworkFlavor(TestNetworkFlavor): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_flavor.assert_called_with( + self.network_client.update_flavor.assert_called_with( self.new_network_flavor, **attrs ) self.assertIsNone(result) @@ -494,7 +504,7 @@ class TestSetNetworkFlavor(TestNetworkFlavor): 'name': "new_network_flavor", 'enabled': True, } - self.network.update_flavor.assert_called_with( + self.network_client.update_flavor.assert_called_with( self.new_network_flavor, **attrs ) self.assertIsNone(result) @@ -514,7 +524,7 @@ class TestSetNetworkFlavor(TestNetworkFlavor): attrs = { 'enabled': False, } - self.network.update_flavor.assert_called_with( + self.network_client.update_flavor.assert_called_with( self.new_network_flavor, **attrs ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_network_flavor_profile.py b/openstackclient/tests/unit/network/v2/test_network_flavor_profile.py index 8a2084d030..bcb1b6e94e 100644 --- a/openstackclient/tests/unit/network/v2/test_network_flavor_profile.py +++ b/openstackclient/tests/unit/network/v2/test_network_flavor_profile.py @@ -23,7 +23,7 @@ class TestFlavorProfile(network_fakes.TestNetworkV2): def setUp(self): super(TestFlavorProfile, self).setUp() # Get the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get the DomainManager Mock @@ -55,7 +55,7 @@ class TestCreateFlavorProfile(TestFlavorProfile): def setUp(self): super(TestCreateFlavorProfile, self).setUp() - self.network.create_service_profile = mock.Mock( + self.network_client.create_service_profile = mock.Mock( return_value=self.new_flavor_profile ) self.projects_mock.get.return_value = self.project @@ -91,7 +91,7 @@ class TestCreateFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_service_profile.assert_called_once_with( + self.network_client.create_service_profile.assert_called_once_with( **{ 'description': self.new_flavor_profile.description, 'project_id': self.project.id, @@ -127,7 +127,7 @@ class TestCreateFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_service_profile.assert_called_once_with( + self.network_client.create_service_profile.assert_called_once_with( **{ 'description': self.new_flavor_profile.description, 'project_id': self.project.id, @@ -162,7 +162,7 @@ class TestCreateFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_service_profile.assert_called_once_with( + self.network_client.create_service_profile.assert_called_once_with( **{ 'description': self.new_flavor_profile.description, 'project_id': self.project.id, @@ -213,7 +213,7 @@ class TestCreateFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_service_profile.assert_called_once_with( + self.network_client.create_service_profile.assert_called_once_with( **{ 'enabled': False, 'driver': self.new_flavor_profile.driver, @@ -229,9 +229,13 @@ class TestDeleteFlavorProfile(TestFlavorProfile): def setUp(self): super(TestDeleteFlavorProfile, self).setUp() - self.network.delete_service_profile = mock.Mock(return_value=None) - self.network.find_service_profile = network_fakes.get_service_profile( - flavor_profile=self._network_flavor_profiles + self.network_client.delete_service_profile = mock.Mock( + return_value=None + ) + self.network_client.find_service_profile = ( + network_fakes.get_service_profile( + flavor_profile=self._network_flavor_profiles + ) ) # Get the command object to test @@ -250,10 +254,10 @@ class TestDeleteFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_service_profile.assert_called_once_with( + self.network_client.find_service_profile.assert_called_once_with( self._network_flavor_profiles[0].id, ignore_missing=False ) - self.network.delete_service_profile.assert_called_once_with( + self.network_client.delete_service_profile.assert_called_once_with( self._network_flavor_profiles[0] ) self.assertIsNone(result) @@ -273,7 +277,7 @@ class TestDeleteFlavorProfile(TestFlavorProfile): calls = [] for a in self._network_flavor_profiles: calls.append(mock.call(a)) - self.network.delete_service_profile.assert_has_calls(calls) + self.network_client.delete_service_profile.assert_has_calls(calls) self.assertIsNone(result) def test_multi_network_flavor_profiles_delete_with_exception(self): @@ -296,7 +300,7 @@ class TestDeleteFlavorProfile(TestFlavorProfile): self._network_flavor_profiles[0], exceptions.CommandError, ] - self.network.find_service_profile = mock.Mock( + self.network_client.find_service_profile = mock.Mock( side_effect=find_mock_result ) @@ -308,13 +312,13 @@ class TestDeleteFlavorProfile(TestFlavorProfile): '1 of 2 flavor_profiles failed to delete.', str(e) ) - self.network.find_service_profile.assert_any_call( + self.network_client.find_service_profile.assert_any_call( self._network_flavor_profiles[0].id, ignore_missing=False ) - self.network.find_service_profile.assert_any_call( + self.network_client.find_service_profile.assert_any_call( 'unexist_network_flavor_profile', ignore_missing=False ) - self.network.delete_service_profile.assert_called_once_with( + self.network_client.delete_service_profile.assert_called_once_with( self._network_flavor_profiles[0] ) @@ -344,7 +348,7 @@ class TestListFlavorProfile(TestFlavorProfile): def setUp(self): super(TestListFlavorProfile, self).setUp() - self.network.service_profiles = mock.Mock( + self.network_client.service_profiles = mock.Mock( return_value=self._network_flavor_profiles ) @@ -360,7 +364,7 @@ class TestListFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.service_profiles.assert_called_once_with(**{}) + self.network_client.service_profiles.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -387,7 +391,7 @@ class TestShowFlavorProfile(TestFlavorProfile): def setUp(self): super(TestShowFlavorProfile, self).setUp() - self.network.find_service_profile = mock.Mock( + self.network_client.find_service_profile = mock.Mock( return_value=self.network_flavor_profile ) @@ -407,7 +411,7 @@ class TestShowFlavorProfile(TestFlavorProfile): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_service_profile.assert_called_once_with( + self.network_client.find_service_profile.assert_called_once_with( self.network_flavor_profile.id, ignore_missing=False ) self.assertEqual(self.columns, columns) @@ -420,8 +424,10 @@ class TestSetFlavorProfile(TestFlavorProfile): def setUp(self): super(TestSetFlavorProfile, self).setUp() - self.network.update_service_profile = mock.Mock(return_value=None) - self.network.find_service_profile = mock.Mock( + self.network_client.update_service_profile = mock.Mock( + return_value=None + ) + self.network_client.find_service_profile = mock.Mock( return_value=self.network_flavor_profile ) @@ -440,7 +446,7 @@ class TestSetFlavorProfile(TestFlavorProfile): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_service_profile.assert_called_with( + self.network_client.update_service_profile.assert_called_with( self.network_flavor_profile, **attrs ) self.assertIsNone(result) @@ -460,7 +466,7 @@ class TestSetFlavorProfile(TestFlavorProfile): attrs = { 'enabled': True, } - self.network.update_service_profile.assert_called_with( + self.network_client.update_service_profile.assert_called_with( self.network_flavor_profile, **attrs ) self.assertIsNone(result) @@ -480,7 +486,7 @@ class TestSetFlavorProfile(TestFlavorProfile): attrs = { 'enabled': False, } - self.network.update_service_profile.assert_called_with( + self.network_client.update_service_profile.assert_called_with( self.network_flavor_profile, **attrs ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_network_meter.py b/openstackclient/tests/unit/network/v2/test_network_meter.py index f49d0167e3..e000c612e9 100644 --- a/openstackclient/tests/unit/network/v2/test_network_meter.py +++ b/openstackclient/tests/unit/network/v2/test_network_meter.py @@ -27,7 +27,7 @@ from openstackclient.tests.unit import utils as tests_utils class TestMeter(network_fakes.TestNetworkV2): def setUp(self): super(TestMeter, self).setUp() - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.projects_mock = self.app.client_manager.identity.projects self.domains_mock = self.app.client_manager.identity.domains @@ -55,7 +55,7 @@ class TestCreateMeter(TestMeter): def setUp(self): super(TestCreateMeter, self).setUp() - self.network.create_metering_label = mock.Mock( + self.network_client.create_metering_label = mock.Mock( return_value=self.new_meter ) self.projects_mock.get.return_value = self.project @@ -85,7 +85,7 @@ class TestCreateMeter(TestMeter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_metering_label.assert_called_once_with( + self.network_client.create_metering_label.assert_called_once_with( **{'name': self.new_meter.name} ) self.assertEqual(self.columns, columns) @@ -114,7 +114,7 @@ class TestCreateMeter(TestMeter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_metering_label.assert_called_once_with( + self.network_client.create_metering_label.assert_called_once_with( **{ 'description': self.new_meter.description, 'name': self.new_meter.name, @@ -132,9 +132,11 @@ class TestDeleteMeter(TestMeter): self.meter_list = network_fakes.FakeNetworkMeter.create_meter(count=2) - self.network.delete_metering_label = mock.Mock(return_value=None) + self.network_client.delete_metering_label = mock.Mock( + return_value=None + ) - self.network.find_metering_label = ( + self.network_client.find_metering_label = ( network_fakes.FakeNetworkMeter.get_meter(meter=self.meter_list) ) @@ -152,7 +154,7 @@ class TestDeleteMeter(TestMeter): result = self.cmd.take_action(parsed_args) - self.network.delete_metering_label.assert_called_once_with( + self.network_client.delete_metering_label.assert_called_once_with( self.meter_list[0] ) self.assertIsNone(result) @@ -172,7 +174,7 @@ class TestDeleteMeter(TestMeter): calls = [] for n in self.meter_list: calls.append(call(n)) - self.network.delete_metering_label.assert_has_calls(calls) + self.network_client.delete_metering_label.assert_has_calls(calls) self.assertIsNone(result) def test_delete_multiple_meter_exception(self): @@ -192,13 +194,15 @@ class TestDeleteMeter(TestMeter): exceptions.NotFound('404'), self.meter_list[1], ] - self.network.find_meter = mock.Mock(side_effect=return_find) + self.network_client.find_meter = mock.Mock(side_effect=return_find) ret_delete = [ None, exceptions.NotFound('404'), ] - self.network.delete_metering_label = mock.Mock(side_effect=ret_delete) + self.network_client.delete_metering_label = mock.Mock( + side_effect=ret_delete + ) self.assertRaises( exceptions.CommandError, self.cmd.take_action, parsed_args @@ -208,7 +212,7 @@ class TestDeleteMeter(TestMeter): call(self.meter_list[0]), call(self.meter_list[1]), ] - self.network.delete_metering_label.assert_has_calls(calls) + self.network_client.delete_metering_label.assert_has_calls(calls) class TestListMeter(TestMeter): @@ -236,7 +240,9 @@ class TestListMeter(TestMeter): def setUp(self): super(TestListMeter, self).setUp() - self.network.metering_labels = mock.Mock(return_value=self.meter_list) + self.network_client.metering_labels = mock.Mock( + return_value=self.meter_list + ) self.cmd = network_meter.ListMeter(self.app, self.namespace) @@ -248,7 +254,7 @@ class TestListMeter(TestMeter): columns, data = self.cmd.take_action(parsed_args) - self.network.metering_labels.assert_called_with() + self.network_client.metering_labels.assert_called_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -276,7 +282,7 @@ class TestShowMeter(TestMeter): self.cmd = network_meter.ShowMeter(self.app, self.namespace) - self.network.find_metering_label = mock.Mock( + self.network_client.find_metering_label = mock.Mock( return_value=self.new_meter ) @@ -303,7 +309,7 @@ class TestShowMeter(TestMeter): columns, data = self.cmd.take_action(parsed_args) - self.network.find_metering_label.assert_called_with( + self.network_client.find_metering_label.assert_called_with( self.new_meter.name, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network_meter_rule.py b/openstackclient/tests/unit/network/v2/test_network_meter_rule.py index 9109334550..661ee72e4a 100644 --- a/openstackclient/tests/unit/network/v2/test_network_meter_rule.py +++ b/openstackclient/tests/unit/network/v2/test_network_meter_rule.py @@ -27,7 +27,7 @@ from openstackclient.tests.unit import utils as tests_utils class TestMeterRule(network_fakes.TestNetworkV2): def setUp(self): super(TestMeterRule, self).setUp() - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.projects_mock = self.app.client_manager.identity.projects self.domains_mock = self.app.client_manager.identity.domains @@ -65,12 +65,14 @@ class TestCreateMeterRule(TestMeterRule): {'id': self.new_rule.metering_label_id} ) - self.network.create_metering_label_rule = mock.Mock( + self.network_client.create_metering_label_rule = mock.Mock( return_value=self.new_rule ) self.projects_mock.get.return_value = self.project self.cmd = network_meter_rule.CreateMeterRule(self.app, self.namespace) - self.network.find_metering_label = mock.Mock(return_value=fake_meter) + self.network_client.find_metering_label = mock.Mock( + return_value=fake_meter + ) def test_create_no_options(self): arglist = [] @@ -98,7 +100,7 @@ class TestCreateMeterRule(TestMeterRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_metering_label_rule.assert_called_once_with( + self.network_client.create_metering_label_rule.assert_called_once_with( **{ 'direction': 'ingress', 'metering_label_id': self.new_rule.metering_label_id, @@ -126,7 +128,7 @@ class TestCreateMeterRule(TestMeterRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_metering_label_rule.assert_called_once_with( + self.network_client.create_metering_label_rule.assert_called_once_with( **{ 'direction': self.new_rule.direction, 'excluded': self.new_rule.excluded, @@ -144,9 +146,11 @@ class TestDeleteMeterRule(TestMeterRule): self.rule_list = network_fakes.FakeNetworkMeterRule.create_meter_rule( count=2 ) - self.network.delete_metering_label_rule = mock.Mock(return_value=None) + self.network_client.delete_metering_label_rule = mock.Mock( + return_value=None + ) - self.network.find_metering_label_rule = ( + self.network_client.find_metering_label_rule = ( network_fakes.FakeNetworkMeterRule.get_meter_rule( meter_rule=self.rule_list ) @@ -166,7 +170,7 @@ class TestDeleteMeterRule(TestMeterRule): result = self.cmd.take_action(parsed_args) - self.network.delete_metering_label_rule.assert_called_once_with( + self.network_client.delete_metering_label_rule.assert_called_once_with( self.rule_list[0] ) self.assertIsNone(result) @@ -186,7 +190,7 @@ class TestDeleteMeterRule(TestMeterRule): calls = [] for rule in self.rule_list: calls.append(call(rule)) - self.network.delete_metering_label_rule.assert_has_calls(calls) + self.network_client.delete_metering_label_rule.assert_has_calls(calls) self.assertIsNone(result) def test_delete_multiple_rules_exception(self): @@ -206,7 +210,7 @@ class TestDeleteMeterRule(TestMeterRule): exceptions.NotFound('404'), self.rule_list[1], ] - self.network.find_metering_label_rule = mock.Mock( + self.network_client.find_metering_label_rule = mock.Mock( side_effect=return_find ) @@ -214,7 +218,7 @@ class TestDeleteMeterRule(TestMeterRule): None, exceptions.NotFound('404'), ] - self.network.delete_metering_label_rule = mock.Mock( + self.network_client.delete_metering_label_rule = mock.Mock( side_effect=ret_delete ) @@ -226,7 +230,7 @@ class TestDeleteMeterRule(TestMeterRule): call(self.rule_list[0]), call(self.rule_list[1]), ] - self.network.delete_metering_label_rule.assert_has_calls(calls) + self.network_client.delete_metering_label_rule.assert_has_calls(calls) class TestListMeterRule(TestMeterRule): @@ -258,7 +262,7 @@ class TestListMeterRule(TestMeterRule): def setUp(self): super(TestListMeterRule, self).setUp() - self.network.metering_label_rules = mock.Mock( + self.network_client.metering_label_rules = mock.Mock( return_value=self.rule_list ) @@ -272,7 +276,7 @@ class TestListMeterRule(TestMeterRule): columns, data = self.cmd.take_action(parsed_args) - self.network.metering_label_rules.assert_called_with() + self.network_client.metering_label_rules.assert_called_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -307,7 +311,7 @@ class TestShowMeterRule(TestMeterRule): self.cmd = network_meter_rule.ShowMeterRule(self.app, self.namespace) - self.network.find_metering_label_rule = mock.Mock( + self.network_client.find_metering_label_rule = mock.Mock( return_value=self.new_rule ) @@ -334,7 +338,7 @@ class TestShowMeterRule(TestMeterRule): columns, data = self.cmd.take_action(parsed_args) - self.network.find_metering_label_rule.assert_called_with( + self.network_client.find_metering_label_rule.assert_called_with( self.new_rule.id, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network_qos_policy.py b/openstackclient/tests/unit/network/v2/test_network_qos_policy.py index 213fc4c092..2def6bf20b 100644 --- a/openstackclient/tests/unit/network/v2/test_network_qos_policy.py +++ b/openstackclient/tests/unit/network/v2/test_network_qos_policy.py @@ -28,7 +28,7 @@ class TestQosPolicy(network_fakes.TestNetworkV2): def setUp(self): super(TestQosPolicy, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects @@ -64,7 +64,7 @@ class TestCreateNetworkQosPolicy(TestQosPolicy): def setUp(self): super(TestCreateNetworkQosPolicy, self).setUp() - self.network.create_qos_policy = mock.Mock( + self.network_client.create_qos_policy = mock.Mock( return_value=self.new_qos_policy ) @@ -100,7 +100,7 @@ class TestCreateNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_policy.assert_called_once_with( + self.network_client.create_qos_policy.assert_called_once_with( **{'name': self.new_qos_policy.name} ) self.assertEqual(self.columns, columns) @@ -127,7 +127,7 @@ class TestCreateNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_policy.assert_called_once_with( + self.network_client.create_qos_policy.assert_called_once_with( **{ 'shared': True, 'project_id': self.project.id, @@ -150,7 +150,7 @@ class TestCreateNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_policy.assert_called_once_with( + self.network_client.create_qos_policy.assert_called_once_with( **{ 'name': self.new_qos_policy.name, 'is_default': False, @@ -168,8 +168,8 @@ class TestDeleteNetworkQosPolicy(TestQosPolicy): def setUp(self): super(TestDeleteNetworkQosPolicy, self).setUp() - self.network.delete_qos_policy = mock.Mock(return_value=None) - self.network.find_qos_policy = ( + self.network_client.delete_qos_policy = mock.Mock(return_value=None) + self.network_client.find_qos_policy = ( network_fakes.FakeNetworkQosPolicy.get_qos_policies( qos_policies=self._qos_policies ) @@ -191,10 +191,10 @@ class TestDeleteNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self._qos_policies[0].name, ignore_missing=False ) - self.network.delete_qos_policy.assert_called_once_with( + self.network_client.delete_qos_policy.assert_called_once_with( self._qos_policies[0] ) self.assertIsNone(result) @@ -214,7 +214,7 @@ class TestDeleteNetworkQosPolicy(TestQosPolicy): calls = [] for a in self._qos_policies: calls.append(call(a)) - self.network.delete_qos_policy.assert_has_calls(calls) + self.network_client.delete_qos_policy.assert_has_calls(calls) self.assertIsNone(result) def test_multi_qos_policies_delete_with_exception(self): @@ -228,7 +228,7 @@ class TestDeleteNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._qos_policies[0], exceptions.CommandError] - self.network.find_qos_policy = mock.MagicMock( + self.network_client.find_qos_policy = mock.MagicMock( side_effect=find_mock_result ) @@ -238,13 +238,13 @@ class TestDeleteNetworkQosPolicy(TestQosPolicy): except exceptions.CommandError as e: self.assertEqual('1 of 2 QoS policies failed to delete.', str(e)) - self.network.find_qos_policy.assert_any_call( + self.network_client.find_qos_policy.assert_any_call( self._qos_policies[0].name, ignore_missing=False ) - self.network.find_qos_policy.assert_any_call( + self.network_client.find_qos_policy.assert_any_call( 'unexist_qos_policy', ignore_missing=False ) - self.network.delete_qos_policy.assert_called_once_with( + self.network_client.delete_qos_policy.assert_called_once_with( self._qos_policies[0] ) @@ -275,7 +275,9 @@ class TestListNetworkQosPolicy(TestQosPolicy): def setUp(self): super(TestListNetworkQosPolicy, self).setUp() - self.network.qos_policies = mock.Mock(return_value=self.qos_policies) + self.network_client.qos_policies = mock.Mock( + return_value=self.qos_policies + ) # Get the command object to test self.cmd = network_qos_policy.ListNetworkQosPolicy( @@ -289,7 +291,7 @@ class TestListNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_policies.assert_called_once_with(**{}) + self.network_client.qos_policies.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -304,7 +306,9 @@ class TestListNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_policies.assert_called_once_with(**{'shared': True}) + self.network_client.qos_policies.assert_called_once_with( + **{'shared': True} + ) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -318,7 +322,9 @@ class TestListNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_policies.assert_called_once_with(**{'shared': False}) + self.network_client.qos_policies.assert_called_once_with( + **{'shared': False} + ) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -337,7 +343,7 @@ class TestListNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_policies.assert_called_once_with( + self.network_client.qos_policies.assert_called_once_with( **{'project_id': project.id} ) @@ -351,8 +357,10 @@ class TestSetNetworkQosPolicy(TestQosPolicy): def setUp(self): super(TestSetNetworkQosPolicy, self).setUp() - self.network.update_qos_policy = mock.Mock(return_value=None) - self.network.find_qos_policy = mock.Mock(return_value=self._qos_policy) + self.network_client.update_qos_policy = mock.Mock(return_value=None) + self.network_client.find_qos_policy = mock.Mock( + return_value=self._qos_policy + ) # Get the command object to test self.cmd = network_qos_policy.SetNetworkQosPolicy( @@ -371,7 +379,7 @@ class TestSetNetworkQosPolicy(TestQosPolicy): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_qos_policy.assert_called_with( + self.network_client.update_qos_policy.assert_called_with( self._qos_policy, **attrs ) self.assertIsNone(result) @@ -402,7 +410,7 @@ class TestSetNetworkQosPolicy(TestQosPolicy): 'shared': True, 'is_default': True, } - self.network.update_qos_policy.assert_called_with( + self.network_client.update_qos_policy.assert_called_with( self._qos_policy, **attrs ) self.assertIsNone(result) @@ -422,7 +430,7 @@ class TestSetNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) attrs = {'shared': False, 'is_default': False} - self.network.update_qos_policy.assert_called_with( + self.network_client.update_qos_policy.assert_called_with( self._qos_policy, **attrs ) self.assertIsNone(result) @@ -452,7 +460,9 @@ class TestShowNetworkQosPolicy(TestQosPolicy): def setUp(self): super(TestShowNetworkQosPolicy, self).setUp() - self.network.find_qos_policy = mock.Mock(return_value=self._qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self._qos_policy + ) # Get the command object to test self.cmd = network_qos_policy.ShowNetworkQosPolicy( @@ -483,7 +493,7 @@ class TestShowNetworkQosPolicy(TestQosPolicy): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self._qos_policy.name, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network_qos_rule.py b/openstackclient/tests/unit/network/v2/test_network_qos_rule.py index 0e7170526b..f779985afe 100644 --- a/openstackclient/tests/unit/network/v2/test_network_qos_rule.py +++ b/openstackclient/tests/unit/network/v2/test_network_qos_rule.py @@ -55,11 +55,13 @@ class TestNetworkQosRule(network_fakes.TestNetworkV2): def setUp(self): super(TestNetworkQosRule, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.qos_policy = ( network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() ) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) class TestCreateNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): @@ -92,7 +94,7 @@ class TestCreateNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): self.new_rule.qos_policy_id, self.new_rule.type, ) - self.network.create_qos_minimum_bandwidth_rule = mock.Mock( + self.network_client.create_qos_minimum_bandwidth_rule = mock.Mock( return_value=self.new_rule ) @@ -134,7 +136,7 @@ class TestCreateNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_minimum_bandwidth_rule.assert_called_once_with( + self.network_client.create_qos_minimum_bandwidth_rule.assert_called_once_with( self.qos_policy.id, **{ 'min_kbps': self.new_rule.min_kbps, @@ -201,7 +203,7 @@ class TestCreateNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): self.new_rule.qos_policy_id, self.new_rule.type, ) - self.network.create_qos_minimum_packet_rate_rule = mock.Mock( + self.network_client.create_qos_minimum_packet_rate_rule = mock.Mock( return_value=self.new_rule ) @@ -243,7 +245,7 @@ class TestCreateNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_minimum_packet_rate_rule.assert_called_once_with( # noqa: E501 + self.network_client.create_qos_minimum_packet_rate_rule.assert_called_once_with( # noqa: E501 self.qos_policy.id, **{ 'min_kpps': self.new_rule.min_kpps, @@ -308,7 +310,7 @@ class TestCreateNetworkQosRuleDSCPMarking(TestNetworkQosRule): self.new_rule.qos_policy_id, self.new_rule.type, ) - self.network.create_qos_dscp_marking_rule = mock.Mock( + self.network_client.create_qos_dscp_marking_rule = mock.Mock( return_value=self.new_rule ) @@ -348,7 +350,7 @@ class TestCreateNetworkQosRuleDSCPMarking(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_dscp_marking_rule.assert_called_once_with( + self.network_client.create_qos_dscp_marking_rule.assert_called_once_with( self.qos_policy.id, **{'dscp_mark': self.new_rule.dscp_mark} ) self.assertEqual(self.columns, columns) @@ -412,7 +414,7 @@ class TestCreateNetworkQosRuleBandwidtLimit(TestNetworkQosRule): self.new_rule.qos_policy_id, self.new_rule.type, ) - self.network.create_qos_bandwidth_limit_rule = mock.Mock( + self.network_client.create_qos_bandwidth_limit_rule = mock.Mock( return_value=self.new_rule ) @@ -469,7 +471,9 @@ class TestCreateNetworkQosRuleBandwidtLimit(TestNetworkQosRule): ) with mock.patch.object( - self.network, "create_qos_bandwidth_limit_rule", return_value=rule + self.network_client, + "create_qos_bandwidth_limit_rule", + return_value=rule, ) as create_qos_bandwidth_limit_rule: parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) @@ -507,7 +511,7 @@ class TestCreateNetworkQosRuleBandwidtLimit(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_qos_bandwidth_limit_rule.assert_called_once_with( + self.network_client.create_qos_bandwidth_limit_rule.assert_called_once_with( self.qos_policy.id, **{ 'max_kbps': self.new_rule.max_kbps, @@ -555,10 +559,10 @@ class TestDeleteNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): attrs ) self.qos_policy.rules = [self.new_rule] - self.network.delete_qos_minimum_bandwidth_rule = mock.Mock( + self.network_client.delete_qos_minimum_bandwidth_rule = mock.Mock( return_value=None ) - self.network.find_qos_minimum_bandwidth_rule = ( + self.network_client.find_qos_minimum_bandwidth_rule = ( network_fakes.FakeNetworkQosRule.get_qos_rules( qos_rules=self.new_rule ) @@ -581,10 +585,10 @@ class TestDeleteNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self.qos_policy.id, ignore_missing=False ) - self.network.delete_qos_minimum_bandwidth_rule.assert_called_once_with( + self.network_client.delete_qos_minimum_bandwidth_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertIsNone(result) @@ -599,8 +603,8 @@ class TestDeleteNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): ('id', self.new_rule.id), ] - self.network.delete_qos_minimum_bandwidth_rule.side_effect = Exception( - 'Error message' + self.network_client.delete_qos_minimum_bandwidth_rule.side_effect = ( + Exception('Error message') ) try: parsed_args = self.check_parser(self.cmd, arglist, verifylist) @@ -624,10 +628,10 @@ class TestDeleteNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): attrs ) self.qos_policy.rules = [self.new_rule] - self.network.delete_qos_minimum_packet_rate_rule = mock.Mock( + self.network_client.delete_qos_minimum_packet_rate_rule = mock.Mock( return_value=None ) - self.network.find_qos_minimum_packet_rate_rule = ( + self.network_client.find_qos_minimum_packet_rate_rule = ( network_fakes.FakeNetworkQosRule.get_qos_rules( qos_rules=self.new_rule ) @@ -650,10 +654,10 @@ class TestDeleteNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self.qos_policy.id, ignore_missing=False ) - self.network.delete_qos_minimum_packet_rate_rule.assert_called_once_with( # noqa: E501 + self.network_client.delete_qos_minimum_packet_rate_rule.assert_called_once_with( # noqa: E501 self.new_rule.id, self.qos_policy.id ) self.assertIsNone(result) @@ -668,7 +672,7 @@ class TestDeleteNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): ('id', self.new_rule.id), ] - self.network.delete_qos_minimum_packet_rate_rule.side_effect = ( + self.network_client.delete_qos_minimum_packet_rate_rule.side_effect = ( Exception('Error message') ) try: @@ -693,10 +697,10 @@ class TestDeleteNetworkQosRuleDSCPMarking(TestNetworkQosRule): attrs ) self.qos_policy.rules = [self.new_rule] - self.network.delete_qos_dscp_marking_rule = mock.Mock( + self.network_client.delete_qos_dscp_marking_rule = mock.Mock( return_value=None ) - self.network.find_qos_dscp_marking_rule = ( + self.network_client.find_qos_dscp_marking_rule = ( network_fakes.FakeNetworkQosRule.get_qos_rules( qos_rules=self.new_rule ) @@ -719,10 +723,10 @@ class TestDeleteNetworkQosRuleDSCPMarking(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self.qos_policy.id, ignore_missing=False ) - self.network.delete_qos_dscp_marking_rule.assert_called_once_with( + self.network_client.delete_qos_dscp_marking_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertIsNone(result) @@ -737,8 +741,8 @@ class TestDeleteNetworkQosRuleDSCPMarking(TestNetworkQosRule): ('id', self.new_rule.id), ] - self.network.delete_qos_dscp_marking_rule.side_effect = Exception( - 'Error message' + self.network_client.delete_qos_dscp_marking_rule.side_effect = ( + Exception('Error message') ) try: parsed_args = self.check_parser(self.cmd, arglist, verifylist) @@ -762,10 +766,10 @@ class TestDeleteNetworkQosRuleBandwidthLimit(TestNetworkQosRule): attrs ) self.qos_policy.rules = [self.new_rule] - self.network.delete_qos_bandwidth_limit_rule = mock.Mock( + self.network_client.delete_qos_bandwidth_limit_rule = mock.Mock( return_value=None ) - self.network.find_qos_bandwidth_limit_rule = ( + self.network_client.find_qos_bandwidth_limit_rule = ( network_fakes.FakeNetworkQosRule.get_qos_rules( qos_rules=self.new_rule ) @@ -788,10 +792,10 @@ class TestDeleteNetworkQosRuleBandwidthLimit(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self.qos_policy.id, ignore_missing=False ) - self.network.delete_qos_bandwidth_limit_rule.assert_called_once_with( + self.network_client.delete_qos_bandwidth_limit_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertIsNone(result) @@ -806,8 +810,8 @@ class TestDeleteNetworkQosRuleBandwidthLimit(TestNetworkQosRule): ('id', self.new_rule.id), ] - self.network.delete_qos_bandwidth_limit_rule.side_effect = Exception( - 'Error message' + self.network_client.delete_qos_bandwidth_limit_rule.side_effect = ( + Exception('Error message') ) try: parsed_args = self.check_parser(self.cmd, arglist, verifylist) @@ -831,13 +835,15 @@ class TestSetNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): attrs=attrs ) self.qos_policy.rules = [self.new_rule] - self.network.update_qos_minimum_bandwidth_rule = mock.Mock( + self.network_client.update_qos_minimum_bandwidth_rule = mock.Mock( return_value=None ) - self.network.find_qos_minimum_bandwidth_rule = mock.Mock( + self.network_client.find_qos_minimum_bandwidth_rule = mock.Mock( return_value=self.new_rule ) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) # Get the command object to test self.cmd = network_qos_rule.SetNetworkQosRule(self.app, self.namespace) @@ -855,7 +861,7 @@ class TestSetNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_qos_minimum_bandwidth_rule.assert_called_with( + self.network_client.update_qos_minimum_bandwidth_rule.assert_called_with( self.new_rule, self.qos_policy.id ) self.assertIsNone(result) @@ -889,7 +895,7 @@ class TestSetNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): attrs = { 'min_kbps': self.new_rule.min_kbps, } - self.network.update_qos_minimum_bandwidth_rule.assert_called_with( + self.network_client.update_qos_minimum_bandwidth_rule.assert_called_with( self.new_rule, self.qos_policy.id, **attrs ) self.assertIsNone(result) @@ -933,13 +939,15 @@ class TestSetNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): attrs=attrs ) self.qos_policy.rules = [self.new_rule] - self.network.update_qos_minimum_packet_rate_rule = mock.Mock( + self.network_client.update_qos_minimum_packet_rate_rule = mock.Mock( return_value=None ) - self.network.find_qos_minimum_packet_rate_rule = mock.Mock( + self.network_client.find_qos_minimum_packet_rate_rule = mock.Mock( return_value=self.new_rule ) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) # Get the command object to test self.cmd = network_qos_rule.SetNetworkQosRule(self.app, self.namespace) @@ -957,7 +965,7 @@ class TestSetNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_qos_minimum_packet_rate_rule.assert_called_with( + self.network_client.update_qos_minimum_packet_rate_rule.assert_called_with( self.new_rule, self.qos_policy.id ) self.assertIsNone(result) @@ -991,7 +999,7 @@ class TestSetNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): attrs = { 'min_kpps': self.new_rule.min_kpps, } - self.network.update_qos_minimum_packet_rate_rule.assert_called_with( + self.network_client.update_qos_minimum_packet_rate_rule.assert_called_with( self.new_rule, self.qos_policy.id, **attrs ) self.assertIsNone(result) @@ -1035,13 +1043,15 @@ class TestSetNetworkQosRuleDSCPMarking(TestNetworkQosRule): attrs=attrs ) self.qos_policy.rules = [self.new_rule] - self.network.update_qos_dscp_marking_rule = mock.Mock( + self.network_client.update_qos_dscp_marking_rule = mock.Mock( return_value=None ) - self.network.find_qos_dscp_marking_rule = mock.Mock( + self.network_client.find_qos_dscp_marking_rule = mock.Mock( return_value=self.new_rule ) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) # Get the command object to test self.cmd = network_qos_rule.SetNetworkQosRule(self.app, self.namespace) @@ -1059,7 +1069,7 @@ class TestSetNetworkQosRuleDSCPMarking(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_qos_dscp_marking_rule.assert_called_with( + self.network_client.update_qos_dscp_marking_rule.assert_called_with( self.new_rule, self.qos_policy.id ) self.assertIsNone(result) @@ -1093,7 +1103,7 @@ class TestSetNetworkQosRuleDSCPMarking(TestNetworkQosRule): attrs = { 'dscp_mark': self.new_rule.dscp_mark, } - self.network.update_qos_dscp_marking_rule.assert_called_with( + self.network_client.update_qos_dscp_marking_rule.assert_called_with( self.new_rule, self.qos_policy.id, **attrs ) self.assertIsNone(result) @@ -1137,13 +1147,15 @@ class TestSetNetworkQosRuleBandwidthLimit(TestNetworkQosRule): attrs=attrs ) self.qos_policy.rules = [self.new_rule] - self.network.update_qos_bandwidth_limit_rule = mock.Mock( + self.network_client.update_qos_bandwidth_limit_rule = mock.Mock( return_value=None ) - self.network.find_qos_bandwidth_limit_rule = mock.Mock( + self.network_client.find_qos_bandwidth_limit_rule = mock.Mock( return_value=self.new_rule ) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_policy + ) # Get the command object to test self.cmd = network_qos_rule.SetNetworkQosRule(self.app, self.namespace) @@ -1161,7 +1173,7 @@ class TestSetNetworkQosRuleBandwidthLimit(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_qos_bandwidth_limit_rule.assert_called_with( + self.network_client.update_qos_bandwidth_limit_rule.assert_called_with( self.new_rule, self.qos_policy.id ) self.assertIsNone(result) @@ -1198,7 +1210,7 @@ class TestSetNetworkQosRuleBandwidthLimit(TestNetworkQosRule): attrs = { 'max_kbps': self.new_rule.max_kbps, } - self.network.update_qos_bandwidth_limit_rule.assert_called_with( + self.network_client.update_qos_bandwidth_limit_rule.assert_called_with( self.new_rule, self.qos_policy.id, **attrs ) self.assertIsNone(result) @@ -1237,7 +1249,7 @@ class TestSetNetworkQosRuleBandwidthLimit(TestNetworkQosRule): attrs = { 'max_burst_kbps': self.new_rule.max_burst_kbits, } - self.network.update_qos_bandwidth_limit_rule.assert_called_with( + self.network_client.update_qos_bandwidth_limit_rule.assert_called_with( self.new_rule, self.qos_policy.id, **attrs ) self.assertIsNone(result) @@ -1271,7 +1283,7 @@ class TestSetNetworkQosRuleBandwidthLimit(TestNetworkQosRule): attrs = { 'direction': direction, } - self.network.update_qos_bandwidth_limit_rule.assert_called_with( + self.network_client.update_qos_bandwidth_limit_rule.assert_called_with( self.new_rule, self.qos_policy.id, **attrs ) self.assertIsNone(result) @@ -1329,16 +1341,16 @@ class TestListNetworkQosRule(TestNetworkQosRule): self.new_rule_dscp_mark, self.new_rule_max_bw, ] - self.network.find_qos_minimum_bandwidth_rule = mock.Mock( + self.network_client.find_qos_minimum_bandwidth_rule = mock.Mock( return_value=self.new_rule_min_bw ) - self.network.find_qos_minimum_packet_rate_rule = mock.Mock( + self.network_client.find_qos_minimum_packet_rate_rule = mock.Mock( return_value=self.new_rule_min_pps ) - self.network.find_qos_dscp_marking_rule = mock.Mock( + self.network_client.find_qos_dscp_marking_rule = mock.Mock( return_value=self.new_rule_dscp_mark ) - self.network.find_qos_bandwidth_limit_rule = mock.Mock( + self.network_client.find_qos_bandwidth_limit_rule = mock.Mock( return_value=self.new_rule_max_bw ) self.columns = ( @@ -1383,7 +1395,7 @@ class TestListNetworkQosRule(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_qos_policy.assert_called_once_with( + self.network_client.find_qos_policy.assert_called_once_with( self.qos_policy.id, ignore_missing=False ) self.assertEqual(self.columns, columns) @@ -1421,7 +1433,7 @@ class TestShowNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): self.new_rule.type, ) - self.network.get_qos_minimum_bandwidth_rule = mock.Mock( + self.network_client.get_qos_minimum_bandwidth_rule = mock.Mock( return_value=self.new_rule ) @@ -1456,7 +1468,7 @@ class TestShowNetworkQosRuleMinimumBandwidth(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_qos_minimum_bandwidth_rule.assert_called_once_with( + self.network_client.get_qos_minimum_bandwidth_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertEqual(self.columns, columns) @@ -1491,7 +1503,7 @@ class TestShowNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): self.new_rule.type, ) - self.network.get_qos_minimum_packet_rate_rule = mock.Mock( + self.network_client.get_qos_minimum_packet_rate_rule = mock.Mock( return_value=self.new_rule ) @@ -1526,7 +1538,7 @@ class TestShowNetworkQosRuleMinimumPacketRate(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_qos_minimum_packet_rate_rule.assert_called_once_with( + self.network_client.get_qos_minimum_packet_rate_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertEqual(self.columns, columns) @@ -1559,7 +1571,7 @@ class TestShowNetworkQosDSCPMarking(TestNetworkQosRule): self.new_rule.type, ) - self.network.get_qos_dscp_marking_rule = mock.Mock( + self.network_client.get_qos_dscp_marking_rule = mock.Mock( return_value=self.new_rule ) @@ -1594,7 +1606,7 @@ class TestShowNetworkQosDSCPMarking(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_qos_dscp_marking_rule.assert_called_once_with( + self.network_client.get_qos_dscp_marking_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertEqual(self.columns, columns) @@ -1631,7 +1643,7 @@ class TestShowNetworkQosBandwidthLimit(TestNetworkQosRule): self.new_rule.type, ) - self.network.get_qos_bandwidth_limit_rule = mock.Mock( + self.network_client.get_qos_bandwidth_limit_rule = mock.Mock( return_value=self.new_rule ) @@ -1666,7 +1678,7 @@ class TestShowNetworkQosBandwidthLimit(TestNetworkQosRule): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_qos_bandwidth_limit_rule.assert_called_once_with( + self.network_client.get_qos_bandwidth_limit_rule.assert_called_once_with( self.new_rule.id, self.qos_policy.id ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network_qos_rule_type.py b/openstackclient/tests/unit/network/v2/test_network_qos_rule_type.py index d468cafc41..4f2bb46b8f 100644 --- a/openstackclient/tests/unit/network/v2/test_network_qos_rule_type.py +++ b/openstackclient/tests/unit/network/v2/test_network_qos_rule_type.py @@ -24,7 +24,7 @@ class TestNetworkQosRuleType(network_fakes.TestNetworkV2): def setUp(self): super(TestNetworkQosRuleType, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network class TestShowNetworkQosRuleType(TestNetworkQosRuleType): @@ -38,7 +38,7 @@ class TestShowNetworkQosRuleType(TestNetworkQosRuleType): def setUp(self): super(TestShowNetworkQosRuleType, self).setUp() - self.network.get_qos_rule_type = mock.Mock( + self.network_client.get_qos_rule_type = mock.Mock( return_value=self.qos_rule_type ) @@ -71,7 +71,7 @@ class TestShowNetworkQosRuleType(TestNetworkQosRuleType): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_qos_rule_type.assert_called_once_with( + self.network_client.get_qos_rule_type.assert_called_once_with( self.qos_rule_type.type ) self.assertEqual(self.columns, columns) @@ -90,7 +90,7 @@ class TestListNetworkQosRuleType(TestNetworkQosRuleType): def setUp(self): super(TestListNetworkQosRuleType, self).setUp() - self.network.qos_rule_types = mock.Mock( + self.network_client.qos_rule_types = mock.Mock( return_value=self.qos_rule_types ) @@ -106,7 +106,7 @@ class TestListNetworkQosRuleType(TestNetworkQosRuleType): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_rule_types.assert_called_once_with(**{}) + self.network_client.qos_rule_types.assert_called_once_with(**{}) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -119,7 +119,7 @@ class TestListNetworkQosRuleType(TestNetworkQosRuleType): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_rule_types.assert_called_once_with( + self.network_client.qos_rule_types.assert_called_once_with( **{'all_supported': True} ) self.assertEqual(self.columns, columns) @@ -134,7 +134,7 @@ class TestListNetworkQosRuleType(TestNetworkQosRuleType): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.qos_rule_types.assert_called_once_with( + self.network_client.qos_rule_types.assert_called_once_with( **{'all_rules': True} ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network_rbac.py b/openstackclient/tests/unit/network/v2/test_network_rbac.py index 49e444970f..fee7413703 100644 --- a/openstackclient/tests/unit/network/v2/test_network_rbac.py +++ b/openstackclient/tests/unit/network/v2/test_network_rbac.py @@ -28,7 +28,7 @@ class TestNetworkRBAC(network_fakes.TestNetworkV2): super(TestNetworkRBAC, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects @@ -74,19 +74,25 @@ class TestCreateNetworkRBAC(TestNetworkRBAC): # Get the command object to test self.cmd = network_rbac.CreateNetworkRBAC(self.app, self.namespace) - self.network.create_rbac_policy = mock.Mock( + self.network_client.create_rbac_policy = mock.Mock( return_value=self.rbac_policy ) - self.network.find_network = mock.Mock(return_value=self.network_object) - self.network.find_qos_policy = mock.Mock(return_value=self.qos_object) - self.network.find_security_group = mock.Mock( + self.network_client.find_network = mock.Mock( + return_value=self.network_object + ) + self.network_client.find_qos_policy = mock.Mock( + return_value=self.qos_object + ) + self.network_client.find_security_group = mock.Mock( return_value=self.sg_object ) - self.network.find_address_scope = mock.Mock( + self.network_client.find_address_scope = mock.Mock( return_value=self.as_object ) - self.network.find_subnet_pool = mock.Mock(return_value=self.snp_object) - self.network.find_address_group = mock.Mock( + self.network_client.find_subnet_pool = mock.Mock( + return_value=self.snp_object + ) + self.network_client.find_address_group = mock.Mock( return_value=self.ag_object ) self.projects_mock.get.return_value = self.project @@ -200,7 +206,7 @@ class TestCreateNetworkRBAC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.create_rbac_policy.assert_called_with( + self.network_client.create_rbac_policy.assert_called_with( **{ 'object_id': self.rbac_policy.object_id, 'object_type': self.rbac_policy.object_type, @@ -230,7 +236,7 @@ class TestCreateNetworkRBAC(TestNetworkRBAC): columns, data = self.cmd.take_action(parsed_args) - self.network.create_rbac_policy.assert_called_with( + self.network_client.create_rbac_policy.assert_called_with( **{ 'object_id': self.rbac_policy.object_id, 'object_type': self.rbac_policy.object_type, @@ -269,7 +275,7 @@ class TestCreateNetworkRBAC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.create_rbac_policy.assert_called_with( + self.network_client.create_rbac_policy.assert_called_with( **{ 'object_id': self.rbac_policy.object_id, 'object_type': self.rbac_policy.object_type, @@ -314,7 +320,7 @@ class TestCreateNetworkRBAC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.create_rbac_policy.assert_called_with( + self.network_client.create_rbac_policy.assert_called_with( **{ 'object_id': obj_fake.id, 'object_type': obj_type, @@ -339,8 +345,8 @@ class TestDeleteNetworkRBAC(TestNetworkRBAC): def setUp(self): super(TestDeleteNetworkRBAC, self).setUp() - self.network.delete_rbac_policy = mock.Mock(return_value=None) - self.network.find_rbac_policy = network_fakes.get_network_rbacs( + self.network_client.delete_rbac_policy = mock.Mock(return_value=None) + self.network_client.find_rbac_policy = network_fakes.get_network_rbacs( rbac_policies=self.rbac_policies ) @@ -358,10 +364,10 @@ class TestDeleteNetworkRBAC(TestNetworkRBAC): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_rbac_policy.assert_called_once_with( + self.network_client.find_rbac_policy.assert_called_once_with( self.rbac_policies[0].id, ignore_missing=False ) - self.network.delete_rbac_policy.assert_called_once_with( + self.network_client.delete_rbac_policy.assert_called_once_with( self.rbac_policies[0] ) self.assertIsNone(result) @@ -382,7 +388,7 @@ class TestDeleteNetworkRBAC(TestNetworkRBAC): calls = [] for r in self.rbac_policies: calls.append(call(r)) - self.network.delete_rbac_policy.assert_has_calls(calls) + self.network_client.delete_rbac_policy.assert_has_calls(calls) self.assertIsNone(result) def test_multi_network_policies_delete_with_exception(self): @@ -396,7 +402,9 @@ class TestDeleteNetworkRBAC(TestNetworkRBAC): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self.rbac_policies[0], exceptions.CommandError] - self.network.find_rbac_policy = mock.Mock(side_effect=find_mock_result) + self.network_client.find_rbac_policy = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -404,13 +412,13 @@ class TestDeleteNetworkRBAC(TestNetworkRBAC): except exceptions.CommandError as e: self.assertEqual('1 of 2 RBAC policies failed to delete.', str(e)) - self.network.find_rbac_policy.assert_any_call( + self.network_client.find_rbac_policy.assert_any_call( self.rbac_policies[0].id, ignore_missing=False ) - self.network.find_rbac_policy.assert_any_call( + self.network_client.find_rbac_policy.assert_any_call( 'unexist_rbac_policy', ignore_missing=False ) - self.network.delete_rbac_policy.assert_called_once_with( + self.network_client.delete_rbac_policy.assert_called_once_with( self.rbac_policies[0] ) @@ -456,7 +464,9 @@ class TestListNetworkRABC(TestNetworkRBAC): # Get the command object to test self.cmd = network_rbac.ListNetworkRBAC(self.app, self.namespace) - self.network.rbac_policies = mock.Mock(return_value=self.rbac_policies) + self.network_client.rbac_policies = mock.Mock( + return_value=self.rbac_policies + ) self.project = identity_fakes_v3.FakeProject.create_one_project() self.projects_mock.get.return_value = self.project @@ -469,7 +479,7 @@ class TestListNetworkRABC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.rbac_policies.assert_called_with() + self.network_client.rbac_policies.assert_called_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -484,7 +494,7 @@ class TestListNetworkRABC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.rbac_policies.assert_called_with( + self.network_client.rbac_policies.assert_called_with( **{'object_type': self.rbac_policies[0].object_type} ) self.assertEqual(self.columns, columns) @@ -501,7 +511,7 @@ class TestListNetworkRABC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.rbac_policies.assert_called_with( + self.network_client.rbac_policies.assert_called_with( **{'action': self.rbac_policies[0].action} ) self.assertEqual(self.columns, columns) @@ -520,7 +530,7 @@ class TestListNetworkRABC(TestNetworkRBAC): columns, data = self.cmd.take_action(parsed_args) - self.network.rbac_policies.assert_called_with() + self.network_client.rbac_policies.assert_called_with() self.assertEqual(self.columns_long, columns) self.assertEqual(self.data_long, list(data)) @@ -537,7 +547,7 @@ class TestListNetworkRABC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.rbac_policies.assert_called_with( + self.network_client.rbac_policies.assert_called_with( **{'target_project_id': self.project.id} ) self.assertEqual(self.columns, columns) @@ -556,10 +566,10 @@ class TestSetNetworkRBAC(TestNetworkRBAC): # Get the command object to test self.cmd = network_rbac.SetNetworkRBAC(self.app, self.namespace) - self.network.find_rbac_policy = mock.Mock( + self.network_client.find_rbac_policy = mock.Mock( return_value=self.rbac_policy ) - self.network.update_rbac_policy = mock.Mock(return_value=None) + self.network_client.update_rbac_policy = mock.Mock(return_value=None) self.projects_mock.get.return_value = self.project def test_network_rbac_set_nothing(self): @@ -572,11 +582,11 @@ class TestSetNetworkRBAC(TestNetworkRBAC): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_rbac_policy.assert_called_once_with( + self.network_client.find_rbac_policy.assert_called_once_with( self.rbac_policy.id, ignore_missing=False ) attrs = {} - self.network.update_rbac_policy.assert_called_once_with( + self.network_client.update_rbac_policy.assert_called_once_with( self.rbac_policy, **attrs ) self.assertIsNone(result) @@ -594,11 +604,11 @@ class TestSetNetworkRBAC(TestNetworkRBAC): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_rbac_policy.assert_called_once_with( + self.network_client.find_rbac_policy.assert_called_once_with( self.rbac_policy.id, ignore_missing=False ) attrs = {'target_tenant': self.project.id} - self.network.update_rbac_policy.assert_called_once_with( + self.network_client.update_rbac_policy.assert_called_once_with( self.rbac_policy, **attrs ) self.assertIsNone(result) @@ -631,7 +641,7 @@ class TestShowNetworkRBAC(TestNetworkRBAC): # Get the command object to test self.cmd = network_rbac.ShowNetworkRBAC(self.app, self.namespace) - self.network.find_rbac_policy = mock.Mock( + self.network_client.find_rbac_policy = mock.Mock( return_value=self.rbac_policy ) @@ -659,7 +669,7 @@ class TestShowNetworkRBAC(TestNetworkRBAC): # DisplayCommandBase.take_action() returns two tuples columns, data = self.cmd.take_action(parsed_args) - self.network.find_rbac_policy.assert_called_with( + self.network_client.find_rbac_policy.assert_called_with( self.rbac_policy.id, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_network_segment.py b/openstackclient/tests/unit/network/v2/test_network_segment.py index 1f74048efc..4a2830d679 100644 --- a/openstackclient/tests/unit/network/v2/test_network_segment.py +++ b/openstackclient/tests/unit/network/v2/test_network_segment.py @@ -26,7 +26,7 @@ class TestNetworkSegment(network_fakes.TestNetworkV2): super(TestNetworkSegment, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network class TestCreateNetworkSegment(TestNetworkSegment): @@ -61,10 +61,12 @@ class TestCreateNetworkSegment(TestNetworkSegment): def setUp(self): super(TestCreateNetworkSegment, self).setUp() - self.network.create_segment = mock.Mock( + self.network_client.create_segment = mock.Mock( return_value=self._network_segment ) - self.network.find_network = mock.Mock(return_value=self._network) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) # Get the command object to test self.cmd = network_segment.CreateNetworkSegment( @@ -110,10 +112,10 @@ class TestCreateNetworkSegment(TestNetworkSegment): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_network.assert_called_once_with( + self.network_client.find_network.assert_called_once_with( self._network_segment.network_id, ignore_missing=False ) - self.network.create_segment.assert_called_once_with( + self.network_client.create_segment.assert_called_once_with( **{ 'network_id': self._network_segment.network_id, 'network_type': self._network_segment.network_type, @@ -150,10 +152,10 @@ class TestCreateNetworkSegment(TestNetworkSegment): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_network.assert_called_once_with( + self.network_client.find_network.assert_called_once_with( self._network_segment.network_id, ignore_missing=False ) - self.network.create_segment.assert_called_once_with( + self.network_client.create_segment.assert_called_once_with( **{ 'description': self._network_segment.description, 'network_id': self._network_segment.network_id, @@ -175,8 +177,8 @@ class TestDeleteNetworkSegment(TestNetworkSegment): def setUp(self): super(TestDeleteNetworkSegment, self).setUp() - self.network.delete_segment = mock.Mock(return_value=None) - self.network.find_segment = mock.Mock( + self.network_client.delete_segment = mock.Mock(return_value=None) + self.network_client.find_segment = mock.Mock( side_effect=self._network_segments ) @@ -196,7 +198,7 @@ class TestDeleteNetworkSegment(TestNetworkSegment): result = self.cmd.take_action(parsed_args) - self.network.delete_segment.assert_called_once_with( + self.network_client.delete_segment.assert_called_once_with( self._network_segments[0] ) self.assertIsNone(result) @@ -215,7 +217,7 @@ class TestDeleteNetworkSegment(TestNetworkSegment): calls = [] for _network_segment in self._network_segments: calls.append(call(_network_segment)) - self.network.delete_segment.assert_has_calls(calls) + self.network_client.delete_segment.assert_has_calls(calls) self.assertIsNone(result) def test_delete_multiple_with_exception(self): @@ -229,7 +231,9 @@ class TestDeleteNetworkSegment(TestNetworkSegment): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._network_segments[0], exceptions.CommandError] - self.network.find_segment = mock.Mock(side_effect=find_mock_result) + self.network_client.find_segment = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -239,13 +243,13 @@ class TestDeleteNetworkSegment(TestNetworkSegment): '1 of 2 network segments failed to delete.', str(e) ) - self.network.find_segment.assert_any_call( + self.network_client.find_segment.assert_any_call( self._network_segments[0].id, ignore_missing=False ) - self.network.find_segment.assert_any_call( + self.network_client.find_segment.assert_any_call( 'doesnotexist', ignore_missing=False ) - self.network.delete_segment.assert_called_once_with( + self.network_client.delete_segment.assert_called_once_with( self._network_segments[0] ) @@ -294,8 +298,12 @@ class TestListNetworkSegment(TestNetworkSegment): # Get the command object to test self.cmd = network_segment.ListNetworkSegment(self.app, self.namespace) - self.network.find_network = mock.Mock(return_value=self._network) - self.network.segments = mock.Mock(return_value=self._network_segments) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) + self.network_client.segments = mock.Mock( + return_value=self._network_segments + ) def test_list_no_option(self): arglist = [] @@ -307,7 +315,7 @@ class TestListNetworkSegment(TestNetworkSegment): columns, data = self.cmd.take_action(parsed_args) - self.network.segments.assert_called_once_with() + self.network_client.segments.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -323,7 +331,7 @@ class TestListNetworkSegment(TestNetworkSegment): columns, data = self.cmd.take_action(parsed_args) - self.network.segments.assert_called_once_with() + self.network_client.segments.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertEqual(self.data_long, list(data)) @@ -337,7 +345,7 @@ class TestListNetworkSegment(TestNetworkSegment): columns, data = self.cmd.take_action(parsed_args) - self.network.segments.assert_called_once_with( + self.network_client.segments.assert_called_once_with( **{'network_id': self._network.id} ) self.assertEqual(self.columns, columns) @@ -351,10 +359,10 @@ class TestSetNetworkSegment(TestNetworkSegment): def setUp(self): super(TestSetNetworkSegment, self).setUp() - self.network.find_segment = mock.Mock( + self.network_client.find_segment = mock.Mock( return_value=self._network_segment ) - self.network.update_segment = mock.Mock( + self.network_client.update_segment = mock.Mock( return_value=self._network_segment ) @@ -372,7 +380,7 @@ class TestSetNetworkSegment(TestNetworkSegment): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_segment.assert_called_once_with( + self.network_client.update_segment.assert_called_once_with( self._network_segment, **{} ) self.assertIsNone(result) @@ -398,7 +406,7 @@ class TestSetNetworkSegment(TestNetworkSegment): 'description': 'new description', 'name': 'new name', } - self.network.update_segment.assert_called_once_with( + self.network_client.update_segment.assert_called_once_with( self._network_segment, **attrs ) self.assertIsNone(result) @@ -431,7 +439,7 @@ class TestShowNetworkSegment(TestNetworkSegment): def setUp(self): super(TestShowNetworkSegment, self).setUp() - self.network.find_segment = mock.Mock( + self.network_client.find_segment = mock.Mock( return_value=self._network_segment ) @@ -455,7 +463,7 @@ class TestShowNetworkSegment(TestNetworkSegment): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_segment.assert_called_once_with( + self.network_client.find_segment.assert_called_once_with( self._network_segment.id, ignore_missing=False ) diff --git a/openstackclient/tests/unit/network/v2/test_network_segment_range.py b/openstackclient/tests/unit/network/v2/test_network_segment_range.py index 582a957f05..c94a9f889a 100644 --- a/openstackclient/tests/unit/network/v2/test_network_segment_range.py +++ b/openstackclient/tests/unit/network/v2/test_network_segment_range.py @@ -43,7 +43,7 @@ class TestNetworkSegmentRange(network_fakes.TestNetworkV2): super(TestNetworkSegmentRange, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network class TestCreateNetworkSegmentRange(TestNetworkSegmentRange): @@ -81,8 +81,8 @@ class TestCreateNetworkSegmentRange(TestNetworkSegmentRange): def setUp(self): super(TestCreateNetworkSegmentRange, self).setUp() - self.network.find_extension = mock.Mock() - self.network.create_network_segment_range = mock.Mock( + self.network_client.find_extension = mock.Mock() + self.network_client.create_network_segment_range = mock.Mock( return_value=self._network_segment_range ) @@ -217,7 +217,7 @@ class TestCreateNetworkSegmentRange(TestNetworkSegmentRange): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network_segment_range.assert_called_once_with( + self.network_client.create_network_segment_range.assert_called_once_with( **{ 'shared': True, 'network_type': 'vxlan', @@ -255,7 +255,7 @@ class TestCreateNetworkSegmentRange(TestNetworkSegmentRange): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network_segment_range.assert_called_once_with( + self.network_client.create_network_segment_range.assert_called_once_with( **{ 'shared': False, 'project_id': mock.ANY, @@ -291,7 +291,7 @@ class TestCreateNetworkSegmentRange(TestNetworkSegmentRange): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network_segment_range.assert_called_once_with( + self.network_client.create_network_segment_range.assert_called_once_with( **{ 'shared': True, 'network_type': 'vxlan', @@ -332,7 +332,7 @@ class TestCreateNetworkSegmentRange(TestNetworkSegmentRange): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_network_segment_range.assert_called_once_with( + self.network_client.create_network_segment_range.assert_called_once_with( **{ 'shared': self._network_segment_range.shared, 'project_id': mock.ANY, @@ -355,11 +355,11 @@ class TestDeleteNetworkSegmentRange(TestNetworkSegmentRange): def setUp(self): super(TestDeleteNetworkSegmentRange, self).setUp() - self.network.find_extension = mock.Mock() - self.network.delete_network_segment_range = mock.Mock( + self.network_client.find_extension = mock.Mock() + self.network_client.delete_network_segment_range = mock.Mock( return_value=None ) - self.network.find_network_segment_range = mock.Mock( + self.network_client.find_network_segment_range = mock.Mock( side_effect=self._network_segment_ranges ) @@ -379,7 +379,7 @@ class TestDeleteNetworkSegmentRange(TestNetworkSegmentRange): result = self.cmd.take_action(parsed_args) - self.network.delete_network_segment_range.assert_called_once_with( + self.network_client.delete_network_segment_range.assert_called_once_with( self._network_segment_ranges[0] ) self.assertIsNone(result) @@ -398,7 +398,9 @@ class TestDeleteNetworkSegmentRange(TestNetworkSegmentRange): calls = [] for _network_segment_range in self._network_segment_ranges: calls.append(call(_network_segment_range)) - self.network.delete_network_segment_range.assert_has_calls(calls) + self.network_client.delete_network_segment_range.assert_has_calls( + calls + ) self.assertIsNone(result) def test_delete_multiple_with_exception(self): @@ -415,7 +417,7 @@ class TestDeleteNetworkSegmentRange(TestNetworkSegmentRange): self._network_segment_ranges[0], exceptions.CommandError, ] - self.network.find_network_segment_range = mock.Mock( + self.network_client.find_network_segment_range = mock.Mock( side_effect=find_mock_result ) @@ -427,13 +429,13 @@ class TestDeleteNetworkSegmentRange(TestNetworkSegmentRange): '1 of 2 network segment ranges failed to delete.', str(e) ) - self.network.find_network_segment_range.assert_any_call( + self.network_client.find_network_segment_range.assert_any_call( self._network_segment_ranges[0].id, ignore_missing=False ) - self.network.find_network_segment_range.assert_any_call( + self.network_client.find_network_segment_range.assert_any_call( 'doesnotexist', ignore_missing=False ) - self.network.delete_network_segment_range.assert_called_once_with( + self.network_client.delete_network_segment_range.assert_called_once_with( self._network_segment_ranges[0] ) @@ -501,8 +503,8 @@ class TestListNetworkSegmentRange(TestNetworkSegmentRange): self.app, self.namespace ) - self.network.find_extension = mock.Mock() - self.network.network_segment_ranges = mock.Mock( + self.network_client.find_extension = mock.Mock() + self.network_client.network_segment_ranges = mock.Mock( return_value=self._network_segment_ranges ) @@ -519,7 +521,7 @@ class TestListNetworkSegmentRange(TestNetworkSegmentRange): columns, data = self.cmd.take_action(parsed_args) - self.network.network_segment_ranges.assert_called_once_with() + self.network_client.network_segment_ranges.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -538,7 +540,7 @@ class TestListNetworkSegmentRange(TestNetworkSegmentRange): columns, data = self.cmd.take_action(parsed_args) - self.network.network_segment_ranges.assert_called_once_with() + self.network_client.network_segment_ranges.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertEqual(self.data_long, list(data)) @@ -571,8 +573,8 @@ class TestSetNetworkSegmentRange(TestNetworkSegmentRange): def setUp(self): super(TestSetNetworkSegmentRange, self).setUp() - self.network.find_extension = mock.Mock() - self.network.find_network_segment_range = mock.Mock( + self.network_client.find_extension = mock.Mock() + self.network_client.find_network_segment_range = mock.Mock( return_value=self._network_segment_range ) @@ -590,12 +592,12 @@ class TestSetNetworkSegmentRange(TestNetworkSegmentRange): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - self.network.update_network_segment_range = mock.Mock( + self.network_client.update_network_segment_range = mock.Mock( return_value=self._network_segment_range ) result = self.cmd.take_action(parsed_args) - self.network.update_network_segment_range.assert_called_once_with( + self.network_client.update_network_segment_range.assert_called_once_with( self._network_segment_range, **{} ) self.assertIsNone(result) @@ -618,7 +620,7 @@ class TestSetNetworkSegmentRange(TestNetworkSegmentRange): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - self.network.update_network_segment_range = mock.Mock( + self.network_client.update_network_segment_range = mock.Mock( return_value=self._network_segment_range_updated ) result = self.cmd.take_action(parsed_args) @@ -628,7 +630,7 @@ class TestSetNetworkSegmentRange(TestNetworkSegmentRange): 'minimum': self.minimum_updated, 'maximum': self.maximum_updated, } - self.network.update_network_segment_range.assert_called_once_with( + self.network_client.update_network_segment_range.assert_called_once_with( self._network_segment_range, **attrs ) self.assertIsNone(result) @@ -669,8 +671,8 @@ class TestShowNetworkSegmentRange(TestNetworkSegmentRange): def setUp(self): super(TestShowNetworkSegmentRange, self).setUp() - self.network.find_extension = mock.Mock() - self.network.find_network_segment_range = mock.Mock( + self.network_client.find_extension = mock.Mock() + self.network_client.find_network_segment_range = mock.Mock( return_value=self._network_segment_range ) @@ -696,7 +698,7 @@ class TestShowNetworkSegmentRange(TestNetworkSegmentRange): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_network_segment_range.assert_called_once_with( + self.network_client.find_network_segment_range.assert_called_once_with( self._network_segment_range.id, ignore_missing=False ) diff --git a/openstackclient/tests/unit/network/v2/test_network_service_provider.py b/openstackclient/tests/unit/network/v2/test_network_service_provider.py index 944e693672..01f8cf1d81 100644 --- a/openstackclient/tests/unit/network/v2/test_network_service_provider.py +++ b/openstackclient/tests/unit/network/v2/test_network_service_provider.py @@ -24,7 +24,7 @@ from openstackclient.tests.unit.network.v2 import fakes class TestNetworkServiceProvider(fakes.TestNetworkV2): def setUp(self): super(TestNetworkServiceProvider, self).setUp() - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network class TestListNetworkServiceProvider(TestNetworkServiceProvider): @@ -53,7 +53,7 @@ class TestListNetworkServiceProvider(TestNetworkServiceProvider): def setUp(self): super(TestListNetworkServiceProvider, self).setUp() - self.network.service_providers = mock.Mock( + self.network_client.service_providers = mock.Mock( return_value=self.provider_list ) @@ -69,6 +69,6 @@ class TestListNetworkServiceProvider(TestNetworkServiceProvider): columns, data = self.cmd.take_action(parsed_args) - self.network.service_providers.assert_called_with() + self.network_client.service_providers.assert_called_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) diff --git a/openstackclient/tests/unit/network/v2/test_network_trunk.py b/openstackclient/tests/unit/network/v2/test_network_trunk.py index 19a4f49745..33d6b9b798 100644 --- a/openstackclient/tests/unit/network/v2/test_network_trunk.py +++ b/openstackclient/tests/unit/network/v2/test_network_trunk.py @@ -33,7 +33,7 @@ class TestNetworkTrunk(network_fakes.TestNetworkV2): super().setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -88,8 +88,10 @@ class TestCreateNetworkTrunk(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.create_trunk = mock.Mock(return_value=self.new_trunk) - self.network.find_port = mock.Mock( + self.network_client.create_trunk = mock.Mock( + return_value=self.new_trunk + ) + self.network_client.find_port = mock.Mock( side_effect=[self.parent_port, self.sub_port] ) @@ -125,7 +127,7 @@ class TestCreateNetworkTrunk(TestNetworkTrunk): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_trunk.assert_called_once_with( + self.network_client.create_trunk.assert_called_once_with( **{ 'name': self.new_trunk['name'], 'admin_state_up': self.new_trunk['admin_state_up'], @@ -175,7 +177,7 @@ class TestCreateNetworkTrunk(TestNetworkTrunk): columns, data = self.cmd.take_action(parsed_args) - self.network.create_trunk.assert_called_once_with( + self.network_client.create_trunk.assert_called_once_with( **{ 'name': self.new_trunk.name, 'description': self.new_trunk.description, @@ -247,7 +249,7 @@ class TestCreateNetworkTrunk(TestNetworkTrunk): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_trunk.assert_called_once_with( + self.network_client.create_trunk.assert_called_once_with( **{ 'name': self.new_trunk.name, 'admin_state_up': True, @@ -319,11 +321,11 @@ class TestDeleteNetworkTrunk(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.find_trunk = mock.Mock( + self.network_client.find_trunk = mock.Mock( side_effect=[self.new_trunks[0], self.new_trunks[1]] ) - self.network.delete_trunk = mock.Mock(return_value=None) - self.network.find_port = mock.Mock( + self.network_client.delete_trunk = mock.Mock(return_value=None) + self.network_client.find_port = mock.Mock( side_effect=[self.parent_port, self.sub_port] ) @@ -343,7 +345,7 @@ class TestDeleteNetworkTrunk(TestNetworkTrunk): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_trunk.assert_called_once_with( + self.network_client.delete_trunk.assert_called_once_with( self.new_trunks[0].id ) self.assertIsNone(result) @@ -364,7 +366,7 @@ class TestDeleteNetworkTrunk(TestNetworkTrunk): calls = [] for t in self.new_trunks: calls.append(call(t.id)) - self.network.delete_trunk.assert_has_calls(calls) + self.network_client.delete_trunk.assert_has_calls(calls) self.assertIsNone(result) def test_delete_trunk_multiple_with_exception(self): @@ -377,13 +379,13 @@ class TestDeleteNetworkTrunk(TestNetworkTrunk): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - self.network.find_trunk = mock.Mock( + self.network_client.find_trunk = mock.Mock( side_effect=[self.new_trunks[0], exceptions.CommandError] ) with testtools.ExpectedException(exceptions.CommandError) as e: self.cmd.take_action(parsed_args) self.assertEqual('1 of 2 trunks failed to delete.', str(e)) - self.network.delete_trunk.assert_called_once_with( + self.network_client.delete_trunk.assert_called_once_with( self.new_trunks[0].id ) @@ -418,8 +420,8 @@ class TestShowNetworkTrunk(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.find_trunk = mock.Mock(return_value=self.new_trunk) - self.network.get_trunk = mock.Mock(return_value=self.new_trunk) + self.network_client.find_trunk = mock.Mock(return_value=self.new_trunk) + self.network_client.get_trunk = mock.Mock(return_value=self.new_trunk) self.projects_mock.get.return_value = self.project self.domains_mock.get.return_value = self.domain @@ -450,7 +452,9 @@ class TestShowNetworkTrunk(TestNetworkTrunk): columns, data = self.cmd.take_action(parsed_args) - self.network.get_trunk.assert_called_once_with(self.new_trunk.id) + self.network_client.get_trunk.assert_called_once_with( + self.new_trunk.id + ) self.assertEqual(self.columns, columns) self.assertEqual(self.data, data) @@ -489,7 +493,7 @@ class TestListNetworkTrunk(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.trunks = mock.Mock(return_value=self.new_trunks) + self.network_client.trunks = mock.Mock(return_value=self.new_trunks) self.projects_mock.get.return_value = self.project self.domains_mock.get.return_value = self.domain @@ -504,7 +508,7 @@ class TestListNetworkTrunk(TestNetworkTrunk): columns, data = self.cmd.take_action(parsed_args) - self.network.trunks.assert_called_once_with() + self.network_client.trunks.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -519,7 +523,7 @@ class TestListNetworkTrunk(TestNetworkTrunk): columns, data = self.cmd.take_action(parsed_args) - self.network.trunks.assert_called_once_with() + self.network_client.trunks.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertEqual(self.data_long, list(data)) @@ -568,10 +572,12 @@ class TestSetNetworkTrunk(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.update_trunk = mock.Mock(return_value=self._trunk) - self.network.add_trunk_subports = mock.Mock(return_value=self._trunk) - self.network.find_trunk = mock.Mock(return_value=self._trunk) - self.network.find_port = mock.Mock( + self.network_client.update_trunk = mock.Mock(return_value=self._trunk) + self.network_client.add_trunk_subports = mock.Mock( + return_value=self._trunk + ) + self.network_client.find_trunk = mock.Mock(return_value=self._trunk) + self.network_client.find_port = mock.Mock( side_effect=[self.sub_port, self.sub_port] ) @@ -598,7 +604,9 @@ class TestSetNetworkTrunk(TestNetworkTrunk): attrs = { attr: value, } - self.network.update_trunk.assert_called_once_with(self._trunk, **attrs) + self.network_client.update_trunk.assert_called_once_with( + self._trunk, **attrs + ) self.assertIsNone(result) def test_set_network_trunk_name(self): @@ -623,7 +631,9 @@ class TestSetNetworkTrunk(TestNetworkTrunk): attrs = { 'admin_state_up': False, } - self.network.update_trunk.assert_called_once_with(self._trunk, **attrs) + self.network_client.update_trunk.assert_called_once_with( + self._trunk, **attrs + ) self.assertIsNone(result) def test_set_network_trunk_admin_state_up_enable(self): @@ -642,7 +652,9 @@ class TestSetNetworkTrunk(TestNetworkTrunk): attrs = { 'admin_state_up': True, } - self.network.update_trunk.assert_called_once_with(self._trunk, **attrs) + self.network_client.update_trunk.assert_called_once_with( + self._trunk, **attrs + ) self.assertIsNone(result) def test_set_network_trunk_nothing(self): @@ -657,7 +669,9 @@ class TestSetNetworkTrunk(TestNetworkTrunk): result = self.cmd.take_action(parsed_args) attrs = {} - self.network.update_trunk.assert_called_once_with(self._trunk, **attrs) + self.network_client.update_trunk.assert_called_once_with( + self._trunk, **attrs + ) self.assertIsNone(result) def test_set_network_trunk_subports(self): @@ -690,7 +704,7 @@ class TestSetNetworkTrunk(TestNetworkTrunk): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.add_trunk_subports.assert_called_once_with( + self.network_client.add_trunk_subports.assert_called_once_with( self._trunk, [subport] ) self.assertIsNone(result) @@ -713,7 +727,7 @@ class TestSetNetworkTrunk(TestNetworkTrunk): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.add_trunk_subports.assert_called_once_with( + self.network_client.add_trunk_subports.assert_called_once_with( self._trunk, [subport] ) self.assertIsNone(result) @@ -746,7 +760,7 @@ class TestSetNetworkTrunk(TestNetworkTrunk): with testtools.ExpectedException(argparse.ArgumentTypeError): self.check_parser(self.cmd, arglist, verifylist) - self.network.add_trunk_subports.assert_not_called() + self.network_client.add_trunk_subports.assert_not_called() def test_set_trunk_attrs_with_exception(self): arglist = [ @@ -760,7 +774,7 @@ class TestSetNetworkTrunk(TestNetworkTrunk): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - self.network.update_trunk = mock.Mock( + self.network_client.update_trunk = mock.Mock( side_effect=exceptions.CommandError ) with testtools.ExpectedException(exceptions.CommandError) as e: @@ -769,8 +783,10 @@ class TestSetNetworkTrunk(TestNetworkTrunk): "Failed to set trunk '%s': " % self._trunk['name'], str(e) ) attrs = {'name': 'reallylongname'} - self.network.update_trunk.assert_called_once_with(self._trunk, **attrs) - self.network.add_trunk_subports.assert_not_called() + self.network_client.update_trunk.assert_called_once_with( + self._trunk, **attrs + ) + self.network_client.add_trunk_subports.assert_not_called() def test_set_trunk_add_subport_with_exception(self): arglist = [ @@ -784,10 +800,10 @@ class TestSetNetworkTrunk(TestNetworkTrunk): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - self.network.add_trunk_subports = mock.Mock( + self.network_client.add_trunk_subports = mock.Mock( side_effect=exceptions.CommandError ) - self.network.find_port = mock.Mock( + self.network_client.find_port = mock.Mock( return_value={'id': 'invalid_subport'} ) with testtools.ExpectedException(exceptions.CommandError) as e: @@ -796,8 +812,8 @@ class TestSetNetworkTrunk(TestNetworkTrunk): "Failed to add subports to trunk '%s': " % self._trunk['name'], str(e), ) - self.network.update_trunk.assert_called_once_with(self._trunk) - self.network.add_trunk_subports.assert_called_once_with( + self.network_client.update_trunk.assert_called_once_with(self._trunk) + self.network_client.add_trunk_subports.assert_called_once_with( self._trunk, [{'port_id': 'invalid_subport'}] ) @@ -824,8 +840,8 @@ class TestListNetworkSubport(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.find_trunk = mock.Mock(return_value=self._trunk) - self.network.get_trunk_subports = mock.Mock( + self.network_client.find_trunk = mock.Mock(return_value=self._trunk) + self.network_client.get_trunk_subports = mock.Mock( return_value={network_trunk.SUB_PORTS: self._subports} ) @@ -843,7 +859,9 @@ class TestListNetworkSubport(TestNetworkTrunk): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.get_trunk_subports.assert_called_once_with(self._trunk) + self.network_client.get_trunk_subports.assert_called_once_with( + self._trunk + ) self.assertEqual(self.columns, columns) self.assertEqual(self.data, list(data)) @@ -892,11 +910,13 @@ class TestUnsetNetworkTrunk(TestNetworkTrunk): def setUp(self): super().setUp() - self.network.find_trunk = mock.Mock(return_value=self._trunk) - self.network.find_port = mock.Mock( + self.network_client.find_trunk = mock.Mock(return_value=self._trunk) + self.network_client.find_port = mock.Mock( side_effect=[self.sub_port, self.sub_port] ) - self.network.delete_trunk_subports = mock.Mock(return_value=None) + self.network_client.delete_trunk_subports = mock.Mock( + return_value=None + ) # Get the command object to test self.cmd = network_trunk.UnsetNetworkTrunk(self.app, self.namespace) @@ -918,7 +938,7 @@ class TestUnsetNetworkTrunk(TestNetworkTrunk): result = self.cmd.take_action(parsed_args) - self.network.delete_trunk_subports.assert_called_once_with( + self.network_client.delete_trunk_subports.assert_called_once_with( self._trunk, [{'port_id': subport['port_id']}] ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_port.py b/openstackclient/tests/unit/network/v2/test_port.py index b38d1783c3..893f221afa 100644 --- a/openstackclient/tests/unit/network/v2/test_port.py +++ b/openstackclient/tests/unit/network/v2/test_port.py @@ -35,7 +35,7 @@ class TestPort(network_fakes.TestNetworkV2): super(TestPort, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects @@ -131,17 +131,19 @@ class TestCreatePort(TestPort): def setUp(self): super(TestCreatePort, self).setUp() - self.network.create_port = mock.Mock(return_value=self._port) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.create_port = mock.Mock(return_value=self._port) + self.network_client.set_tags = mock.Mock(return_value=None) fake_net = network_fakes.create_one_network( { 'id': self._port.network_id, } ) - self.network.find_network = mock.Mock(return_value=fake_net) + self.network_client.find_network = mock.Mock(return_value=fake_net) self.fake_subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_subnet = mock.Mock(return_value=self.fake_subnet) - self.network.find_extension = mock.Mock(return_value=[]) + self.network_client.find_subnet = mock.Mock( + return_value=self.fake_subnet + ) + self.network_client.find_extension = mock.Mock(return_value=[]) # Get the command object to test self.cmd = port.CreatePort(self.app, self.namespace) @@ -163,14 +165,14 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, 'name': 'test-port', } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -223,7 +225,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'mac_address': 'aa:aa:aa:aa:aa:aa', 'fixed_ips': [ @@ -300,7 +302,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -314,7 +316,9 @@ class TestCreatePort(TestPort): def test_create_with_security_group(self): secgroup = network_fakes.FakeSecurityGroup.create_one_security_group() - self.network.find_security_group = mock.Mock(return_value=secgroup) + self.network_client.find_security_group = mock.Mock( + return_value=secgroup + ) arglist = [ '--network', self._port.network_id, @@ -336,7 +340,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -369,7 +373,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -384,7 +388,9 @@ class TestCreatePort(TestPort): def test_create_with_security_groups(self): sg_1 = network_fakes.FakeSecurityGroup.create_one_security_group() sg_2 = network_fakes.FakeSecurityGroup.create_one_security_group() - self.network.find_security_group = mock.Mock(side_effect=[sg_1, sg_2]) + self.network_client.find_security_group = mock.Mock( + side_effect=[sg_1, sg_2] + ) arglist = [ '--network', self._port.network_id, @@ -407,7 +413,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -436,7 +442,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -465,7 +471,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -507,7 +513,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -558,7 +564,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -572,7 +578,9 @@ class TestCreatePort(TestPort): def test_create_port_with_qos(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) arglist = [ '--network', self._port.network_id, @@ -593,7 +601,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -626,7 +634,7 @@ class TestCreatePort(TestPort): self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -656,7 +664,7 @@ class TestCreatePort(TestPort): self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -688,7 +696,9 @@ class TestCreatePort(TestPort): else: verifylist.append(('no_tag', True)) - self.network.find_extension = mock.Mock(return_value=add_tags_in_post) + self.network_client.find_extension = mock.Mock( + return_value=add_tags_in_post + ) parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) @@ -703,30 +713,32 @@ class TestCreatePort(TestPort): args['tags'] = sorted(['red', 'blue']) else: args['tags'] = [] - self.network.create_port.assert_called_once() + self.network_client.create_port.assert_called_once() # Now we need to verify if arguments to call create_port are as # expected, # But we can't simply use assert_called_once_with() method because # duplicates from 'tags' are removed with # list(set(parsed_args.tags)) and that don't quarantee order of # tags list which is used to call create_port(). - create_port_call_kwargs = self.network.create_port.call_args[1] + create_port_call_kwargs = ( + self.network_client.create_port.call_args[1] + ) create_port_call_kwargs['tags'] = sorted( create_port_call_kwargs['tags'] ) self.assertDictEqual(args, create_port_call_kwargs) else: - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( admin_state_up=True, network_id=self._port.network_id, name='test-port', ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self._port, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -768,7 +780,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -841,7 +853,7 @@ class TestCreatePort(TestPort): self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -881,7 +893,7 @@ class TestCreatePort(TestPort): } if numa_affinity_policy: create_args['numa_affinity_policy'] = numa_affinity_policy - self.network.create_port.assert_called_once_with(**create_args) + self.network_client.create_port.assert_called_once_with(**create_args) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -929,7 +941,7 @@ class TestCreatePort(TestPort): 'name': 'test-port', 'device_profile': 'cyborg_device_profile_1', } - self.network.create_port.assert_called_once_with(**create_args) + self.network_client.create_port.assert_called_once_with(**create_args) self.assertEqual(set(self.columns), set(columns)) self.assertCountEqual(self.data, data) @@ -1011,7 +1023,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -1046,7 +1058,7 @@ class TestCreatePort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.create_port.assert_called_once_with( + self.network_client.create_port.assert_called_once_with( **{ 'admin_state_up': True, 'network_id': self._port.network_id, @@ -1068,8 +1080,10 @@ class TestDeletePort(TestPort): def setUp(self): super(TestDeletePort, self).setUp() - self.network.delete_port = mock.Mock(return_value=None) - self.network.find_port = network_fakes.get_ports(ports=self._ports) + self.network_client.delete_port = mock.Mock(return_value=None) + self.network_client.find_port = network_fakes.get_ports( + ports=self._ports + ) # Get the command object to test self.cmd = port.DeletePort(self.app, self.namespace) @@ -1083,10 +1097,10 @@ class TestDeletePort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.find_port.assert_called_once_with( + self.network_client.find_port.assert_called_once_with( self._ports[0].name, ignore_missing=False ) - self.network.delete_port.assert_called_once_with(self._ports[0]) + self.network_client.delete_port.assert_called_once_with(self._ports[0]) self.assertIsNone(result) def test_multi_ports_delete(self): @@ -1105,7 +1119,7 @@ class TestDeletePort(TestPort): calls = [] for p in self._ports: calls.append(call(p)) - self.network.delete_port.assert_has_calls(calls) + self.network_client.delete_port.assert_has_calls(calls) self.assertIsNone(result) def test_multi_ports_delete_with_exception(self): @@ -1119,7 +1133,7 @@ class TestDeletePort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._ports[0], exceptions.CommandError] - self.network.find_port = mock.Mock(side_effect=find_mock_result) + self.network_client.find_port = mock.Mock(side_effect=find_mock_result) try: self.cmd.take_action(parsed_args) @@ -1127,13 +1141,13 @@ class TestDeletePort(TestPort): except exceptions.CommandError as e: self.assertEqual('1 of 2 ports failed to delete.', str(e)) - self.network.find_port.assert_any_call( + self.network_client.find_port.assert_any_call( self._ports[0].name, ignore_missing=False ) - self.network.find_port.assert_any_call( + self.network_client.find_port.assert_any_call( 'unexist_port', ignore_missing=False ) - self.network.delete_port.assert_called_once_with(self._ports[0]) + self.network_client.delete_port.assert_called_once_with(self._ports[0]) class TestListPort(TestPort): @@ -1190,7 +1204,7 @@ class TestListPort(TestPort): # Get the command object to test self.cmd = port.ListPort(self.app, self.namespace) - self.network.ports = mock.Mock(return_value=self._ports) + self.network_client.ports = mock.Mock(return_value=self._ports) fake_router = network_fakes.FakeRouter.create_one_router( { 'id': 'fake-router-id', @@ -1201,8 +1215,8 @@ class TestListPort(TestPort): 'id': 'fake-network-id', } ) - self.network.find_router = mock.Mock(return_value=fake_router) - self.network.find_network = mock.Mock(return_value=fake_network) + self.network_client.find_router = mock.Mock(return_value=fake_router) + self.network_client.find_network = mock.Mock(return_value=fake_network) self.app.client_manager.compute = mock.Mock() def test_port_list_no_options(self): @@ -1213,7 +1227,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( fields=LIST_FIELDS_TO_RETRIEVE ) self.assertEqual(self.columns, columns) @@ -1231,7 +1245,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'device_id': 'fake-router-id', 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1255,7 +1269,7 @@ class TestListPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( device_id=fake_server.id, fields=LIST_FIELDS_TO_RETRIEVE ) mock_find.assert_called_once_with(mock.ANY, 'fake-server-name') @@ -1274,7 +1288,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'device_id': self._ports[0].device_id, 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1295,7 +1309,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'device_owner': self._ports[0].device_owner, 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1327,7 +1341,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'device_owner': self._ports[0].device_owner, 'device_id': 'fake-router-id', @@ -1351,7 +1365,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'mac_address': self._ports[0].mac_address, 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1372,7 +1386,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'fixed_ips': ['ip_address=%s' % ip_address], 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1393,7 +1407,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'fixed_ips': ['ip_address_substr=%s' % ip_address_ss], 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1413,11 +1427,13 @@ class TestListPort(TestPort): self.fake_subnet = network_fakes.FakeSubnet.create_one_subnet( {'id': subnet_id} ) - self.network.find_subnet = mock.Mock(return_value=self.fake_subnet) + self.network_client.find_subnet = mock.Mock( + return_value=self.fake_subnet + ) parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'fixed_ips': ['subnet_id=%s' % subnet_id], 'fields': LIST_FIELDS_TO_RETRIEVE, @@ -1440,11 +1456,13 @@ class TestListPort(TestPort): self.fake_subnet = network_fakes.FakeSubnet.create_one_subnet( {'id': subnet_id} ) - self.network.find_subnet = mock.Mock(return_value=self.fake_subnet) + self.network_client.find_subnet = mock.Mock( + return_value=self.fake_subnet + ) parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'fixed_ips': [ 'subnet_id=%s' % subnet_id, @@ -1475,11 +1493,13 @@ class TestListPort(TestPort): 'fields': LIST_FIELDS_TO_RETRIEVE, } ) - self.network.find_subnet = mock.Mock(return_value=self.fake_subnet) + self.network_client.find_subnet = mock.Mock( + return_value=self.fake_subnet + ) parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'fixed_ips': [ 'subnet_id=%s' % subnet_id, @@ -1504,7 +1524,7 @@ class TestListPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( fields=LIST_FIELDS_TO_RETRIEVE + LIST_FIELDS_TO_RETRIEVE_LONG ) self.assertEqual(self.columns_long, columns) @@ -1526,7 +1546,7 @@ class TestListPort(TestPort): 'fields': LIST_FIELDS_TO_RETRIEVE, } - self.network.ports.assert_called_once_with(**filters) + self.network_client.ports.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1548,7 +1568,7 @@ class TestListPort(TestPort): 'fields': LIST_FIELDS_TO_RETRIEVE, } - self.network.ports.assert_called_once_with(**filters) + self.network_client.ports.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1573,7 +1593,7 @@ class TestListPort(TestPort): 'fields': LIST_FIELDS_TO_RETRIEVE, } - self.network.ports.assert_called_once_with(**filters) + self.network_client.ports.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1594,7 +1614,7 @@ class TestListPort(TestPort): 'fields': LIST_FIELDS_TO_RETRIEVE, } - self.network.ports.assert_called_once_with(**filters) + self.network_client.ports.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1618,7 +1638,7 @@ class TestListPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.ports.assert_called_once_with( + self.network_client.ports.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -1648,7 +1668,7 @@ class TestListPort(TestPort): 'fields': LIST_FIELDS_TO_RETRIEVE, } - self.network.ports.assert_called_once_with(**filters) + self.network_client.ports.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1659,10 +1679,12 @@ class TestSetPort(TestPort): def setUp(self): super(TestSetPort, self).setUp() self.fake_subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_subnet = mock.Mock(return_value=self.fake_subnet) - self.network.find_port = mock.Mock(return_value=self._port) - self.network.update_port = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.find_subnet = mock.Mock( + return_value=self.fake_subnet + ) + self.network_client.find_port = mock.Mock(return_value=self._port) + self.network_client.update_port = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = port.SetPort(self.app, self.namespace) @@ -1677,15 +1699,15 @@ class TestSetPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_port.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_port.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def test_set_port_fixed_ip(self): _testport = network_fakes.create_one_port( {'fixed_ips': [{'ip_address': '0.0.0.1'}]} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--fixed-ip', 'ip-address=10.0.0.12', @@ -1704,14 +1726,16 @@ class TestSetPort(TestPort): {'ip_address': '10.0.0.12'}, ], } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_fixed_ip_clear(self): _testport = network_fakes.create_one_port( {'fixed_ips': [{'ip_address': '0.0.0.1'}]} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--fixed-ip', 'ip-address=10.0.0.12', @@ -1730,7 +1754,9 @@ class TestSetPort(TestPort): {'ip_address': '10.0.0.12'}, ], } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_dns_name(self): @@ -1750,14 +1776,16 @@ class TestSetPort(TestPort): attrs = { 'dns_name': '8.8.8.8', } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_overwrite_binding_profile(self): _testport = network_fakes.create_one_port( {'binding_profile': {'lok_i': 'visi_on'}} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--binding-profile', 'lok_i=than_os', @@ -1773,14 +1801,16 @@ class TestSetPort(TestPort): attrs = { 'binding:profile': {'lok_i': 'than_os'}, } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_overwrite_mac_address(self): _testport = network_fakes.create_one_port( {'mac_address': '11:22:33:44:55:66'} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--mac-address', '66:55:44:33:22:11', @@ -1794,7 +1824,9 @@ class TestSetPort(TestPort): attrs = { 'mac_address': '66:55:44:33:22:11', } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_this(self): @@ -1819,7 +1851,9 @@ class TestSetPort(TestPort): 'binding:profile': {}, 'fixed_ips': [], } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_that(self): @@ -1858,7 +1892,9 @@ class TestSetPort(TestPort): 'description': 'newDescription', 'name': 'newName', } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_invalid_json_binding_profile(self): @@ -1908,12 +1944,14 @@ class TestSetPort(TestPort): attrs = { 'binding:profile': {'foo': 'bar', 'foo2': 'bar2'}, } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_security_group(self): sg = network_fakes.FakeSecurityGroup.create_one_security_group() - self.network.find_security_group = mock.Mock(return_value=sg) + self.network_client.find_security_group = mock.Mock(return_value=sg) arglist = [ '--security-group', sg.id, @@ -1929,18 +1967,22 @@ class TestSetPort(TestPort): attrs = { 'security_group_ids': [sg.id], } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_security_group_append(self): sg_1 = network_fakes.FakeSecurityGroup.create_one_security_group() sg_2 = network_fakes.FakeSecurityGroup.create_one_security_group() sg_3 = network_fakes.FakeSecurityGroup.create_one_security_group() - self.network.find_security_group = mock.Mock(side_effect=[sg_2, sg_3]) + self.network_client.find_security_group = mock.Mock( + side_effect=[sg_2, sg_3] + ) _testport = network_fakes.create_one_port( {'security_group_ids': [sg_1.id]} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--security-group', sg_2.id, @@ -1958,7 +2000,9 @@ class TestSetPort(TestPort): attrs = { 'security_group_ids': [sg_1.id, sg_2.id, sg_3.id], } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_security_group_clear(self): @@ -1976,7 +2020,9 @@ class TestSetPort(TestPort): attrs = { 'security_group_ids': [], } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_security_group_replace(self): @@ -1985,8 +2031,8 @@ class TestSetPort(TestPort): _testport = network_fakes.create_one_port( {'security_group_ids': [sg1.id]} ) - self.network.find_port = mock.Mock(return_value=_testport) - self.network.find_security_group = mock.Mock(return_value=sg2) + self.network_client.find_port = mock.Mock(return_value=_testport) + self.network_client.find_security_group = mock.Mock(return_value=sg2) arglist = [ '--security-group', sg2.id, @@ -2003,7 +2049,9 @@ class TestSetPort(TestPort): attrs = { 'security_group_ids': [sg2.id], } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_allowed_address_pair(self): @@ -2023,14 +2071,16 @@ class TestSetPort(TestPort): attrs = { 'allowed_address_pairs': [{'ip_address': '192.168.1.123'}], } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_append_allowed_address_pair(self): _testport = network_fakes.create_one_port( {'allowed_address_pairs': [{'ip_address': '192.168.1.123'}]} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--allowed-address', 'ip-address=192.168.1.45', @@ -2050,14 +2100,16 @@ class TestSetPort(TestPort): {'ip_address': '192.168.1.45'}, ], } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_overwrite_allowed_address_pair(self): _testport = network_fakes.create_one_port( {'allowed_address_pairs': [{'ip_address': '192.168.1.123'}]} ) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--allowed-address', 'ip-address=192.168.1.45', @@ -2076,7 +2128,9 @@ class TestSetPort(TestPort): attrs = { 'allowed_address_pairs': [{'ip_address': '192.168.1.45'}], } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_no_allowed_address_pairs(self): @@ -2095,7 +2149,9 @@ class TestSetPort(TestPort): attrs = { 'allowed_address_pairs': [], } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_extra_dhcp_option(self): @@ -2115,7 +2171,9 @@ class TestSetPort(TestPort): attrs = { 'extra_dhcp_opts': [{'opt_name': 'foo', 'opt_value': 'bar'}], } - self.network.update_port.assert_called_once_with(self._port, **attrs) + self.network_client.update_port.assert_called_once_with( + self._port, **attrs + ) self.assertIsNone(result) def test_set_port_security_enabled(self): @@ -2135,7 +2193,7 @@ class TestSetPort(TestPort): self.cmd.take_action(parsed_args) - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( self._port, **{ 'port_security_enabled': True, @@ -2159,7 +2217,7 @@ class TestSetPort(TestPort): self.cmd.take_action(parsed_args) - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( self._port, **{ 'port_security_enabled': False, @@ -2168,9 +2226,11 @@ class TestSetPort(TestPort): def test_set_port_with_qos(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) _testport = network_fakes.create_one_port({'qos_policy_id': None}) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--qos-policy', qos_policy.id, @@ -2187,12 +2247,14 @@ class TestSetPort(TestPort): attrs = { 'qos_policy_id': qos_policy.id, } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_data_plane_status(self): _testport = network_fakes.create_one_port({'data_plane_status': None}) - self.network.find_port = mock.Mock(return_value=_testport) + self.network_client.find_port = mock.Mock(return_value=_testport) arglist = [ '--data-plane-status', 'ACTIVE', @@ -2210,7 +2272,9 @@ class TestSetPort(TestPort): 'data_plane_status': 'ACTIVE', } - self.network.update_port.assert_called_once_with(_testport, **attrs) + self.network_client.update_port.assert_called_once_with( + _testport, **attrs + ) self.assertIsNone(result) def test_set_port_invalid_data_plane_status_value(self): @@ -2242,8 +2306,8 @@ class TestSetPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_port.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_port.called) + self.network_client.set_tags.assert_called_once_with( self._port, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -2271,7 +2335,7 @@ class TestSetPort(TestPort): self.cmd.take_action(parsed_args) - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( self._port, **{'numa_affinity_policy': policy} ) @@ -2336,8 +2400,8 @@ class TestSetPort(TestPort): def test_set_hints_valid_alias_value(self): testport = network_fakes.create_one_port() - self.network.find_port = mock.Mock(return_value=testport) - self.network.find_extension = mock.Mock( + self.network_client.find_port = mock.Mock(return_value=testport) + self.network_client.find_extension = mock.Mock( return_value=['port-hints', 'port-hint-ovs-tx-steering'] ) arglist = [ @@ -2352,7 +2416,7 @@ class TestSetPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( testport, **{ 'hints': { @@ -2364,8 +2428,8 @@ class TestSetPort(TestPort): def test_set_hints_valid_json(self): testport = network_fakes.create_one_port() - self.network.find_port = mock.Mock(return_value=testport) - self.network.find_extension = mock.Mock( + self.network_client.find_port = mock.Mock(return_value=testport) + self.network_client.find_extension = mock.Mock( return_value=['port-hints', 'port-hint-ovs-tx-steering'] ) arglist = [ @@ -2383,7 +2447,7 @@ class TestSetPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( testport, **{ 'hints': { @@ -2402,7 +2466,7 @@ class TestShowPort(TestPort): def setUp(self): super(TestShowPort, self).setUp() - self.network.find_port = mock.Mock(return_value=self._port) + self.network_client.find_port = mock.Mock(return_value=self._port) # Get the command object to test self.cmd = port.ShowPort(self.app, self.namespace) @@ -2430,7 +2494,7 @@ class TestShowPort(TestPort): columns, data = self.cmd.take_action(parsed_args) - self.network.find_port.assert_called_once_with( + self.network_client.find_port.assert_called_once_with( self._port.name, ignore_missing=False ) @@ -2463,10 +2527,12 @@ class TestUnsetPort(TestPort): self.fake_subnet = network_fakes.FakeSubnet.create_one_subnet( {'id': '042eb10a-3a18-4658-ab-cf47c8d03152'} ) - self.network.find_subnet = mock.Mock(return_value=self.fake_subnet) - self.network.find_port = mock.Mock(return_value=self._testport) - self.network.update_port = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.find_subnet = mock.Mock( + return_value=self.fake_subnet + ) + self.network_client.find_port = mock.Mock(return_value=self._testport) + self.network_client.update_port = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = port.UnsetPort(self.app, self.namespace) @@ -2509,7 +2575,7 @@ class TestUnsetPort(TestPort): 'qos_policy_id': None, 'binding:host_id': None, } - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( self._testport, **attrs ) self.assertIsNone(result) @@ -2556,8 +2622,10 @@ class TestUnsetPort(TestPort): _fake_port = network_fakes.create_one_port( {'security_group_ids': [_fake_sg1.id, _fake_sg2.id]} ) - self.network.find_port = mock.Mock(return_value=_fake_port) - self.network.find_security_group = mock.Mock(return_value=_fake_sg2) + self.network_client.find_port = mock.Mock(return_value=_fake_port) + self.network_client.find_security_group = mock.Mock( + return_value=_fake_sg2 + ) arglist = [ '--security-group', _fake_sg2.id, @@ -2571,7 +2639,9 @@ class TestUnsetPort(TestPort): result = self.cmd.take_action(parsed_args) attrs = {'security_group_ids': [_fake_sg1.id]} - self.network.update_port.assert_called_once_with(_fake_port, **attrs) + self.network_client.update_port.assert_called_once_with( + _fake_port, **attrs + ) self.assertIsNone(result) def test_unset_port_security_group_not_existent(self): @@ -2580,7 +2650,9 @@ class TestUnsetPort(TestPort): _fake_port = network_fakes.create_one_port( {'security_group_ids': [_fake_sg1.id]} ) - self.network.find_security_group = mock.Mock(return_value=_fake_sg2) + self.network_client.find_security_group = mock.Mock( + return_value=_fake_sg2 + ) arglist = [ '--security-group', _fake_sg2.id, @@ -2599,7 +2671,7 @@ class TestUnsetPort(TestPort): _fake_port = network_fakes.create_one_port( {'allowed_address_pairs': [{'ip_address': '192.168.1.123'}]} ) - self.network.find_port = mock.Mock(return_value=_fake_port) + self.network_client.find_port = mock.Mock(return_value=_fake_port) arglist = [ '--allowed-address', 'ip-address=192.168.1.123', @@ -2616,14 +2688,16 @@ class TestUnsetPort(TestPort): 'allowed_address_pairs': [], } - self.network.update_port.assert_called_once_with(_fake_port, **attrs) + self.network_client.update_port.assert_called_once_with( + _fake_port, **attrs + ) self.assertIsNone(result) def test_unset_port_allowed_address_pair_not_existent(self): _fake_port = network_fakes.create_one_port( {'allowed_address_pairs': [{'ip_address': '192.168.1.123'}]} ) - self.network.find_port = mock.Mock(return_value=_fake_port) + self.network_client.find_port = mock.Mock(return_value=_fake_port) arglist = [ '--allowed-address', 'ip-address=192.168.1.45', @@ -2642,7 +2716,7 @@ class TestUnsetPort(TestPort): _fake_port = network_fakes.create_one_port( {'data_plane_status': 'ACTIVE'} ) - self.network.find_port = mock.Mock(return_value=_fake_port) + self.network_client.find_port = mock.Mock(return_value=_fake_port) arglist = [ '--data-plane-status', _fake_port.name, @@ -2659,7 +2733,9 @@ class TestUnsetPort(TestPort): 'data_plane_status': None, } - self.network.update_port.assert_called_once_with(_fake_port, **attrs) + self.network_client.update_port.assert_called_once_with( + _fake_port, **attrs + ) self.assertIsNone(result) def _test_unset_tags(self, with_tags=True): @@ -2677,8 +2753,8 @@ class TestUnsetPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_port.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_port.called) + self.network_client.set_tags.assert_called_once_with( self._testport, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -2693,7 +2769,7 @@ class TestUnsetPort(TestPort): _fake_port = network_fakes.create_one_port( {'numa_affinity_policy': 'required'} ) - self.network.find_port = mock.Mock(return_value=_fake_port) + self.network_client.find_port = mock.Mock(return_value=_fake_port) arglist = [ '--numa-policy', _fake_port.name, @@ -2710,12 +2786,14 @@ class TestUnsetPort(TestPort): 'numa_affinity_policy': None, } - self.network.update_port.assert_called_once_with(_fake_port, **attrs) + self.network_client.update_port.assert_called_once_with( + _fake_port, **attrs + ) self.assertIsNone(result) def test_unset_hints(self): testport = network_fakes.create_one_port() - self.network.find_port = mock.Mock(return_value=testport) + self.network_client.find_port = mock.Mock(return_value=testport) arglist = [ '--hints', testport.name, @@ -2727,7 +2805,7 @@ class TestUnsetPort(TestPort): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_port.assert_called_once_with( + self.network_client.update_port.assert_called_once_with( testport, **{'hints': None}, ) diff --git a/openstackclient/tests/unit/network/v2/test_router.py b/openstackclient/tests/unit/network/v2/test_router.py index 060d2c6557..599df74ffa 100644 --- a/openstackclient/tests/unit/network/v2/test_router.py +++ b/openstackclient/tests/unit/network/v2/test_router.py @@ -28,7 +28,7 @@ class TestRouter(network_fakes.TestNetworkV2): super(TestRouter, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network self.projects_mock = self.app.client_manager.identity.projects @@ -42,10 +42,10 @@ class TestAddPortToRouter(TestRouter): def setUp(self): super(TestAddPortToRouter, self).setUp() - self.network.add_interface_to_router = mock.Mock() + self.network_client.add_interface_to_router = mock.Mock() self.cmd = router.AddPortToRouter(self.app, self.namespace) - self.network.find_router = mock.Mock(return_value=self._router) - self.network.find_port = mock.Mock(return_value=self._port) + self.network_client.find_router = mock.Mock(return_value=self._router) + self.network_client.find_port = mock.Mock(return_value=self._port) def test_add_port_no_option(self): arglist = [] @@ -72,7 +72,7 @@ class TestAddPortToRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.add_interface_to_router.assert_called_with( + self.network_client.add_interface_to_router.assert_called_with( self._router, **{ 'port_id': self._router.port, @@ -91,10 +91,10 @@ class TestAddSubnetToRouter(TestRouter): def setUp(self): super(TestAddSubnetToRouter, self).setUp() - self.network.add_interface_to_router = mock.Mock() + self.network_client.add_interface_to_router = mock.Mock() self.cmd = router.AddSubnetToRouter(self.app, self.namespace) - self.network.find_router = mock.Mock(return_value=self._router) - self.network.find_subnet = mock.Mock(return_value=self._subnet) + self.network_client.find_router = mock.Mock(return_value=self._router) + self.network_client.find_subnet = mock.Mock(return_value=self._subnet) def test_add_subnet_no_option(self): arglist = [] @@ -120,7 +120,7 @@ class TestAddSubnetToRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.add_interface_to_router.assert_called_with( + self.network_client.add_interface_to_router.assert_called_with( self._router, **{'subnet_id': self._router.subnet} ) @@ -165,8 +165,10 @@ class TestCreateRouter(TestRouter): def setUp(self): super(TestCreateRouter, self).setUp() - self.network.create_router = mock.Mock(return_value=self.new_router) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.create_router = mock.Mock( + return_value=self.new_router + ) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = router.CreateRouter(self.app, self.namespace) @@ -182,7 +184,7 @@ class TestCreateRouter(TestRouter): arglist, verifylist, ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) def test_create_default_options(self): arglist = [ @@ -198,21 +200,21 @@ class TestCreateRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( **{ 'admin_state_up': True, 'name': self.new_router.name, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) def test_create_with_gateway(self): _network = network_fakes.create_one_network() _subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_network = mock.Mock(return_value=_network) - self.network.find_subnet = mock.Mock(return_value=_subnet) + self.network_client.find_network = mock.Mock(return_value=_network) + self.network_client.find_subnet = mock.Mock(return_value=_subnet) arglist = [ self.new_router.name, '--external-gateway', @@ -234,7 +236,7 @@ class TestCreateRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( **{ 'admin_state_up': True, 'name': self.new_router.name, @@ -245,7 +247,7 @@ class TestCreateRouter(TestRouter): }, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -265,7 +267,7 @@ class TestCreateRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( **{ 'admin_state_up': True, 'name': self.new_router.name, @@ -296,7 +298,7 @@ class TestCreateRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( **{ 'admin_state_up': True, 'name': self.new_router.name, @@ -330,7 +332,7 @@ class TestCreateRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( **{ 'admin_state_up': True, 'name': self.new_router.name, @@ -361,15 +363,15 @@ class TestCreateRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( name=self.new_router.name, admin_state_up=True ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self.new_router, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -395,7 +397,7 @@ class TestCreateRouter(TestRouter): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_router.assert_called_once_with( + self.network_client.create_router.assert_called_once_with( **{ 'admin_state_up': True, 'name': self.new_router.name, @@ -413,9 +415,9 @@ class TestDeleteRouter(TestRouter): def setUp(self): super(TestDeleteRouter, self).setUp() - self.network.delete_router = mock.Mock(return_value=None) + self.network_client.delete_router = mock.Mock(return_value=None) - self.network.find_router = network_fakes.FakeRouter.get_routers( + self.network_client.find_router = network_fakes.FakeRouter.get_routers( self._routers ) @@ -432,7 +434,9 @@ class TestDeleteRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_router.assert_called_once_with(self._routers[0]) + self.network_client.delete_router.assert_called_once_with( + self._routers[0] + ) self.assertIsNone(result) def test_multi_routers_delete(self): @@ -451,7 +455,7 @@ class TestDeleteRouter(TestRouter): calls = [] for r in self._routers: calls.append(call(r)) - self.network.delete_router.assert_has_calls(calls) + self.network_client.delete_router.assert_has_calls(calls) self.assertIsNone(result) def test_multi_routers_delete_with_exception(self): @@ -465,7 +469,9 @@ class TestDeleteRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._routers[0], exceptions.CommandError] - self.network.find_router = mock.Mock(side_effect=find_mock_result) + self.network_client.find_router = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -473,13 +479,15 @@ class TestDeleteRouter(TestRouter): except exceptions.CommandError as e: self.assertEqual('1 of 2 routers failed to delete.', str(e)) - self.network.find_router.assert_any_call( + self.network_client.find_router.assert_any_call( self._routers[0].name, ignore_missing=False ) - self.network.find_router.assert_any_call( + self.network_client.find_router.assert_any_call( 'unexist_router', ignore_missing=False ) - self.network.delete_router.assert_called_once_with(self._routers[0]) + self.network_client.delete_router.assert_called_once_with( + self._routers[0] + ) class TestListRouter(TestRouter): @@ -568,15 +576,21 @@ class TestListRouter(TestRouter): # Get the command object to test self.cmd = router.ListRouter(self.app, self.namespace) - self.network.agent_hosted_routers = mock.Mock( + self.network_client.agent_hosted_routers = mock.Mock( return_value=self.routers ) - self.network.routers = mock.Mock(return_value=self.routers) - self.network.find_extension = mock.Mock(return_value=self.extensions) - self.network.find_router = mock.Mock(return_value=self.routers[0]) + self.network_client.routers = mock.Mock(return_value=self.routers) + self.network_client.find_extension = mock.Mock( + return_value=self.extensions + ) + self.network_client.find_router = mock.Mock( + return_value=self.routers[0] + ) self._testagent = network_fakes.create_one_network_agent() - self.network.get_agent = mock.Mock(return_value=self._testagent) - self.network.get_router = mock.Mock(return_value=self.routers[0]) + self.network_client.get_agent = mock.Mock(return_value=self._testagent) + self.network_client.get_router = mock.Mock( + return_value=self.routers[0] + ) def test_router_list_no_options(self): arglist = [] @@ -590,7 +604,7 @@ class TestListRouter(TestRouter): # containing the data to be listed. columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with() + self.network_client.routers.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -605,7 +619,9 @@ class TestListRouter(TestRouter): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) - with mock.patch.object(self.network, "routers", return_value=_routers): + with mock.patch.object( + self.network_client, "routers", return_value=_routers + ): columns, data = self.cmd.take_action(parsed_args) self.assertNotIn("is_distributed", columns) @@ -625,7 +641,7 @@ class TestListRouter(TestRouter): # containing the data to be listed. columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with() + self.network_client.routers.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertCountEqual(self.data_long, list(data)) @@ -639,14 +655,14 @@ class TestListRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) # to mock, that no availability zone - self.network.find_extension = mock.Mock(return_value=None) + self.network_client.find_extension = mock.Mock(return_value=None) # In base command class Lister in cliff, abstract method take_action() # returns a tuple containing the column names and an iterable # containing the data to be listed. columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with() + self.network_client.routers.assert_called_once_with() self.assertEqual(self.columns_long_no_az, columns) self.assertCountEqual(self.data_long_no_az, list(data)) @@ -663,7 +679,9 @@ class TestListRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with(**{'name': test_name}) + self.network_client.routers.assert_called_once_with( + **{'name': test_name} + ) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -678,7 +696,7 @@ class TestListRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with( + self.network_client.routers.assert_called_once_with( **{'admin_state_up': True, 'is_admin_state_up': True} ) self.assertEqual(self.columns, columns) @@ -692,7 +710,7 @@ class TestListRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with( + self.network_client.routers.assert_called_once_with( **{'admin_state_up': False, 'is_admin_state_up': False} ) @@ -714,7 +732,7 @@ class TestListRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.routers.assert_called_once_with(**filters) + self.network_client.routers.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -736,7 +754,7 @@ class TestListRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.routers.assert_called_once_with(**filters) + self.network_client.routers.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -771,7 +789,7 @@ class TestListRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.agent_hosted_routers(*attrs) + self.network_client.agent_hosted_routers(*attrs) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -795,7 +813,7 @@ class TestListRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.routers.assert_called_once_with( + self.network_client.routers.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -817,10 +835,10 @@ class TestRemovePortFromRouter(TestRouter): def setUp(self): super(TestRemovePortFromRouter, self).setUp() - self.network.remove_interface_from_router = mock.Mock() + self.network_client.remove_interface_from_router = mock.Mock() self.cmd = router.RemovePortFromRouter(self.app, self.namespace) - self.network.find_router = mock.Mock(return_value=self._router) - self.network.find_port = mock.Mock(return_value=self._port) + self.network_client.find_router = mock.Mock(return_value=self._router) + self.network_client.find_port = mock.Mock(return_value=self._port) def test_remove_port_no_option(self): arglist = [] @@ -847,7 +865,7 @@ class TestRemovePortFromRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.remove_interface_from_router.assert_called_with( + self.network_client.remove_interface_from_router.assert_called_with( self._router, **{'port_id': self._router.port} ) self.assertIsNone(result) @@ -863,10 +881,10 @@ class TestRemoveSubnetFromRouter(TestRouter): def setUp(self): super(TestRemoveSubnetFromRouter, self).setUp() - self.network.remove_interface_from_router = mock.Mock() + self.network_client.remove_interface_from_router = mock.Mock() self.cmd = router.RemoveSubnetFromRouter(self.app, self.namespace) - self.network.find_router = mock.Mock(return_value=self._router) - self.network.find_subnet = mock.Mock(return_value=self._subnet) + self.network_client.find_router = mock.Mock(return_value=self._router) + self.network_client.find_subnet = mock.Mock(return_value=self._subnet) def test_remove_subnet_no_option(self): arglist = [] @@ -892,7 +910,7 @@ class TestRemoveSubnetFromRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.remove_interface_from_router.assert_called_with( + self.network_client.remove_interface_from_router.assert_called_with( self._router, **{'subnet_id': self._router.subnet} ) self.assertIsNone(result) @@ -903,11 +921,11 @@ class TestAddExtraRoutesToRouter(TestRouter): def setUp(self): super(TestAddExtraRoutesToRouter, self).setUp() - self.network.add_extra_routes_to_router = mock.Mock( + self.network_client.add_extra_routes_to_router = mock.Mock( return_value=self._router ) self.cmd = router.AddExtraRoutesToRouter(self.app, self.namespace) - self.network.find_router = mock.Mock(return_value=self._router) + self.network_client.find_router = mock.Mock(return_value=self._router) def test_add_no_extra_route(self): arglist = [ @@ -920,7 +938,7 @@ class TestAddExtraRoutesToRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.add_extra_routes_to_router.assert_called_with( + self.network_client.add_extra_routes_to_router.assert_called_with( self._router, body={'router': {'routes': []}} ) self.assertEqual(2, len(result)) @@ -939,7 +957,7 @@ class TestAddExtraRoutesToRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.add_extra_routes_to_router.assert_called_with( + self.network_client.add_extra_routes_to_router.assert_called_with( self._router, body={ 'router': { @@ -973,7 +991,7 @@ class TestAddExtraRoutesToRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.add_extra_routes_to_router.assert_called_with( + self.network_client.add_extra_routes_to_router.assert_called_with( self._router, body={ 'router': { @@ -992,11 +1010,11 @@ class TestRemoveExtraRoutesFromRouter(TestRouter): def setUp(self): super(TestRemoveExtraRoutesFromRouter, self).setUp() - self.network.remove_extra_routes_from_router = mock.Mock( + self.network_client.remove_extra_routes_from_router = mock.Mock( return_value=self._router ) self.cmd = router.RemoveExtraRoutesFromRouter(self.app, self.namespace) - self.network.find_router = mock.Mock(return_value=self._router) + self.network_client.find_router = mock.Mock(return_value=self._router) def test_remove_no_extra_route(self): arglist = [ @@ -1009,7 +1027,7 @@ class TestRemoveExtraRoutesFromRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.remove_extra_routes_from_router.assert_called_with( + self.network_client.remove_extra_routes_from_router.assert_called_with( self._router, body={'router': {'routes': []}} ) self.assertEqual(2, len(result)) @@ -1028,7 +1046,7 @@ class TestRemoveExtraRoutesFromRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.remove_extra_routes_from_router.assert_called_with( + self.network_client.remove_extra_routes_from_router.assert_called_with( self._router, body={ 'router': { @@ -1062,7 +1080,7 @@ class TestRemoveExtraRoutesFromRouter(TestRouter): result = self.cmd.take_action(parsed_args) - self.network.remove_extra_routes_from_router.assert_called_with( + self.network_client.remove_extra_routes_from_router.assert_called_with( self._router, body={ 'router': { @@ -1087,12 +1105,14 @@ class TestSetRouter(TestRouter): def setUp(self): super(TestSetRouter, self).setUp() - self.network.router_add_gateway = mock.Mock() - self.network.update_router = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) - self.network.find_router = mock.Mock(return_value=self._router) - self.network.find_network = mock.Mock(return_value=self._network) - self.network.find_subnet = mock.Mock(return_value=self._subnet) + self.network_client.router_add_gateway = mock.Mock() + self.network_client.update_router = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) + self.network_client.find_router = mock.Mock(return_value=self._router) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) + self.network_client.find_subnet = mock.Mock(return_value=self._subnet) # Get the command object to test self.cmd = router.SetRouter(self.app, self.namespace) @@ -1126,7 +1146,7 @@ class TestSetRouter(TestRouter): 'ha': False, 'description': 'router', } - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._router, **attrs ) self.assertIsNone(result) @@ -1153,7 +1173,7 @@ class TestSetRouter(TestRouter): 'distributed': False, 'ha': True, } - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._router, **attrs ) self.assertIsNone(result) @@ -1196,7 +1216,7 @@ class TestSetRouter(TestRouter): result = self.cmd.take_action(parsed_args) routes = [{'destination': '10.20.30.0/24', 'nexthop': '10.20.30.1'}] attrs = {'routes': routes + self._router.routes} - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._router, **attrs ) self.assertIsNone(result) @@ -1217,7 +1237,7 @@ class TestSetRouter(TestRouter): attrs = { 'routes': [], } - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._router, **attrs ) self.assertIsNone(result) @@ -1226,7 +1246,7 @@ class TestSetRouter(TestRouter): _testrouter = network_fakes.FakeRouter.create_one_router( {'routes': [{"destination": "10.0.0.2", "nexthop": "1.1.1.1"}]} ) - self.network.find_router = mock.Mock(return_value=_testrouter) + self.network_client.find_router = mock.Mock(return_value=_testrouter) arglist = [ _testrouter.name, '--route', @@ -1248,7 +1268,7 @@ class TestSetRouter(TestRouter): {'destination': '10.20.30.0/24', 'nexthop': '10.20.30.1'} ] } - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( _testrouter, **attrs ) self.assertIsNone(result) @@ -1264,8 +1284,8 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_router.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_router.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def test_wrong_gateway_params(self): @@ -1296,7 +1316,7 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_router.assert_called_with( + self.network_client.update_router.assert_called_with( self._router, **{'external_gateway_info': {'network_id': self._network.id}} ) @@ -1320,7 +1340,7 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_router.assert_called_with( + self.network_client.update_router.assert_called_with( self._router, **{ 'external_gateway_info': { @@ -1354,7 +1374,7 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_router.assert_called_with( + self.network_client.update_router.assert_called_with( self._router, **{ 'external_gateway_info': { @@ -1388,7 +1408,7 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_router.assert_called_with( + self.network_client.update_router.assert_called_with( self._router, **{ 'external_gateway_info': { @@ -1420,8 +1440,8 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_router.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_router.called) + self.network_client.set_tags.assert_called_once_with( self._router, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -1434,7 +1454,9 @@ class TestSetRouter(TestRouter): def test_set_gateway_ip_qos(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) arglist = [ "--external-gateway", self._network.id, @@ -1450,7 +1472,7 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_router.assert_called_with( + self.network_client.update_router.assert_called_with( self._router, **{ 'external_gateway_info': { @@ -1476,7 +1498,7 @@ class TestSetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_router.assert_called_with( + self.network_client.update_router.assert_called_with( self._router, **{ 'external_gateway_info': { @@ -1489,7 +1511,9 @@ class TestSetRouter(TestRouter): def test_set_unset_gateway_ip_qos(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) arglist = [ "--external-gateway", self._network.id, @@ -1515,9 +1539,11 @@ class TestSetRouter(TestRouter): def test_set_gateway_ip_qos_no_gateway(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) router = network_fakes.FakeRouter.create_one_router() - self.network.find_router = mock.Mock(return_value=router) + self.network_client.find_router = mock.Mock(return_value=router) arglist = [ "--qos-policy", qos_policy.id, @@ -1535,9 +1561,11 @@ class TestSetRouter(TestRouter): def test_unset_gateway_ip_qos_no_gateway(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) router = network_fakes.FakeRouter.create_one_router() - self.network.find_router = mock.Mock(return_value=router) + self.network_client.find_router = mock.Mock(return_value=router) arglist = [ "--no-qos-policy", router.id, @@ -1606,8 +1634,8 @@ class TestShowRouter(TestRouter): def setUp(self): super(TestShowRouter, self).setUp() - self.network.find_router = mock.Mock(return_value=self._router) - self.network.ports = mock.Mock(return_value=[self._port]) + self.network_client.find_router = mock.Mock(return_value=self._router) + self.network_client.ports = mock.Mock(return_value=[self._port]) # Get the command object to test self.cmd = router.ShowRouter(self.app, self.namespace) @@ -1635,10 +1663,12 @@ class TestShowRouter(TestRouter): columns, data = self.cmd.take_action(parsed_args) - self.network.find_router.assert_called_once_with( + self.network_client.find_router.assert_called_once_with( self._router.name, ignore_missing=False ) - self.network.ports.assert_called_with(**{'device_id': self._router.id}) + self.network_client.ports.assert_called_with( + **{'device_id': self._router.id} + ) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -1656,7 +1686,7 @@ class TestShowRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) with mock.patch.object( - self.network, "find_router", return_value=_router + self.network_client, "find_router", return_value=_router ): columns, data = self.cmd.take_action(parsed_args) @@ -1675,7 +1705,7 @@ class TestShowRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) with mock.patch.object( - self.network, "find_router", return_value=_router + self.network_client, "find_router", return_value=_router ): columns, data = self.cmd.take_action(parsed_args) @@ -1710,9 +1740,11 @@ class TestUnsetRouter(TestRouter): } ) self.fake_subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_router = mock.Mock(return_value=self._testrouter) - self.network.update_router = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.find_router = mock.Mock( + return_value=self._testrouter + ) + self.network_client.update_router = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = router.UnsetRouter(self.app, self.namespace) @@ -1737,7 +1769,7 @@ class TestUnsetRouter(TestRouter): {"destination": "192.168.101.2/24", "nexthop": "172.24.4.3"} ], } - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._testrouter, **attrs ) self.assertIsNone(result) @@ -1769,7 +1801,7 @@ class TestUnsetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) attrs = {'external_gateway_info': {}} - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._testrouter, **attrs ) self.assertIsNone(result) @@ -1789,8 +1821,8 @@ class TestUnsetRouter(TestRouter): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_router.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_router.called) + self.network_client.set_tags.assert_called_once_with( self._testrouter, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -1815,16 +1847,18 @@ class TestUnsetRouter(TestRouter): "qos_policy_id": None, } } - self.network.update_router.assert_called_once_with( + self.network_client.update_router.assert_called_once_with( self._testrouter, **attrs ) self.assertIsNone(result) def test_unset_gateway_ip_qos_no_network(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) router = network_fakes.FakeRouter.create_one_router() - self.network.find_router = mock.Mock(return_value=router) + self.network_client.find_router = mock.Mock(return_value=router) arglist = [ "--qos-policy", router.id, @@ -1840,11 +1874,13 @@ class TestUnsetRouter(TestRouter): def test_unset_gateway_ip_qos_no_qos(self): qos_policy = network_fakes.FakeNetworkQosPolicy.create_one_qos_policy() - self.network.find_qos_policy = mock.Mock(return_value=qos_policy) + self.network_client.find_qos_policy = mock.Mock( + return_value=qos_policy + ) router = network_fakes.FakeRouter.create_one_router( {"external_gateway_info": {"network_id": "fake-id"}} ) - self.network.find_router = mock.Mock(return_value=router) + self.network_client.find_router = mock.Mock(return_value=router) arglist = [ "--qos-policy", router.id, diff --git a/openstackclient/tests/unit/network/v2/test_security_group_network.py b/openstackclient/tests/unit/network/v2/test_security_group_network.py index f55a39988a..973fc726ec 100644 --- a/openstackclient/tests/unit/network/v2/test_security_group_network.py +++ b/openstackclient/tests/unit/network/v2/test_security_group_network.py @@ -27,7 +27,7 @@ class TestSecurityGroupNetwork(network_fakes.TestNetworkV2): super(TestSecurityGroupNetwork, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -65,13 +65,13 @@ class TestCreateSecurityGroupNetwork(TestSecurityGroupNetwork): def setUp(self): super(TestCreateSecurityGroupNetwork, self).setUp() - self.network.create_security_group = mock.Mock( + self.network_client.create_security_group = mock.Mock( return_value=self._security_group ) self.projects_mock.get.return_value = self.project self.domains_mock.get.return_value = self.domain - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = security_group.CreateSecurityGroup(self.app, self.namespace) @@ -92,7 +92,7 @@ class TestCreateSecurityGroupNetwork(TestSecurityGroupNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group.assert_called_once_with( + self.network_client.create_security_group.assert_called_once_with( **{ 'description': self._security_group.name, 'name': self._security_group.name, @@ -123,7 +123,7 @@ class TestCreateSecurityGroupNetwork(TestSecurityGroupNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group.assert_called_once_with( + self.network_client.create_security_group.assert_called_once_with( **{ 'description': self._security_group.description, 'stateful': self._security_group.stateful, @@ -152,18 +152,18 @@ class TestCreateSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group.assert_called_once_with( + self.network_client.create_security_group.assert_called_once_with( **{ 'description': self._security_group.name, 'name': self._security_group.name, } ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self._security_group, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -181,9 +181,11 @@ class TestDeleteSecurityGroupNetwork(TestSecurityGroupNetwork): def setUp(self): super(TestDeleteSecurityGroupNetwork, self).setUp() - self.network.delete_security_group = mock.Mock(return_value=None) + self.network_client.delete_security_group = mock.Mock( + return_value=None + ) - self.network.find_security_group = ( + self.network_client.find_security_group = ( network_fakes.FakeSecurityGroup.get_security_groups( self._security_groups ) @@ -203,7 +205,7 @@ class TestDeleteSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_security_group.assert_called_once_with( + self.network_client.delete_security_group.assert_called_once_with( self._security_groups[0] ) self.assertIsNone(result) @@ -224,7 +226,7 @@ class TestDeleteSecurityGroupNetwork(TestSecurityGroupNetwork): calls = [] for s in self._security_groups: calls.append(call(s)) - self.network.delete_security_group.assert_has_calls(calls) + self.network_client.delete_security_group.assert_has_calls(calls) self.assertIsNone(result) def test_multi_security_groups_delete_with_exception(self): @@ -241,7 +243,7 @@ class TestDeleteSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._security_groups[0], exceptions.CommandError] - self.network.find_security_group = mock.Mock( + self.network_client.find_security_group = mock.Mock( side_effect=find_mock_result ) @@ -251,13 +253,13 @@ class TestDeleteSecurityGroupNetwork(TestSecurityGroupNetwork): except exceptions.CommandError as e: self.assertEqual('1 of 2 groups failed to delete.', str(e)) - self.network.find_security_group.assert_any_call( + self.network_client.find_security_group.assert_any_call( self._security_groups[0].name, ignore_missing=False ) - self.network.find_security_group.assert_any_call( + self.network_client.find_security_group.assert_any_call( 'unexist_security_group', ignore_missing=False ) - self.network.delete_security_group.assert_called_once_with( + self.network_client.delete_security_group.assert_called_once_with( self._security_groups[0] ) @@ -291,7 +293,7 @@ class TestListSecurityGroupNetwork(TestSecurityGroupNetwork): def setUp(self): super(TestListSecurityGroupNetwork, self).setUp() - self.network.security_groups = mock.Mock( + self.network_client.security_groups = mock.Mock( return_value=self._security_groups ) @@ -307,7 +309,7 @@ class TestListSecurityGroupNetwork(TestSecurityGroupNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_groups.assert_called_once_with( + self.network_client.security_groups.assert_called_once_with( fields=security_group.ListSecurityGroup.FIELDS_TO_RETRIEVE ) self.assertEqual(self.columns, columns) @@ -324,7 +326,7 @@ class TestListSecurityGroupNetwork(TestSecurityGroupNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_groups.assert_called_once_with( + self.network_client.security_groups.assert_called_once_with( fields=security_group.ListSecurityGroup.FIELDS_TO_RETRIEVE ) self.assertEqual(self.columns, columns) @@ -348,7 +350,7 @@ class TestListSecurityGroupNetwork(TestSecurityGroupNetwork): 'fields': security_group.ListSecurityGroup.FIELDS_TO_RETRIEVE, } - self.network.security_groups.assert_called_once_with(**filters) + self.network_client.security_groups.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -373,7 +375,7 @@ class TestListSecurityGroupNetwork(TestSecurityGroupNetwork): 'fields': security_group.ListSecurityGroup.FIELDS_TO_RETRIEVE, } - self.network.security_groups.assert_called_once_with(**filters) + self.network_client.security_groups.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -397,7 +399,7 @@ class TestListSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.security_groups.assert_called_once_with( + self.network_client.security_groups.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -421,12 +423,14 @@ class TestSetSecurityGroupNetwork(TestSecurityGroupNetwork): def setUp(self): super(TestSetSecurityGroupNetwork, self).setUp() - self.network.update_security_group = mock.Mock(return_value=None) + self.network_client.update_security_group = mock.Mock( + return_value=None + ) - self.network.find_security_group = mock.Mock( + self.network_client.find_security_group = mock.Mock( return_value=self._security_group ) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = security_group.SetSecurityGroup(self.app, self.namespace) @@ -447,7 +451,7 @@ class TestSetSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_security_group.assert_called_once_with( + self.network_client.update_security_group.assert_called_once_with( self._security_group, **{} ) self.assertIsNone(result) @@ -478,7 +482,7 @@ class TestSetSecurityGroupNetwork(TestSecurityGroupNetwork): 'name': new_name, 'stateful': True, } - self.network.update_security_group.assert_called_once_with( + self.network_client.update_security_group.assert_called_once_with( self._security_group, **attrs ) self.assertIsNone(result) @@ -498,8 +502,8 @@ class TestSetSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertTrue(self.network.update_security_group.called) - self.network.set_tags.assert_called_once_with( + self.assertTrue(self.network_client.update_security_group.called) + self.network_client.set_tags.assert_called_once_with( self._security_group, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -549,7 +553,7 @@ class TestShowSecurityGroupNetwork(TestSecurityGroupNetwork): def setUp(self): super(TestShowSecurityGroupNetwork, self).setUp() - self.network.find_security_group = mock.Mock( + self.network_client.find_security_group = mock.Mock( return_value=self._security_group ) @@ -572,7 +576,7 @@ class TestShowSecurityGroupNetwork(TestSecurityGroupNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.find_security_group.assert_called_once_with( + self.network_client.find_security_group.assert_called_once_with( self._security_group.id, ignore_missing=False ) self.assertEqual(self.columns, columns) @@ -590,12 +594,14 @@ class TestUnsetSecurityGroupNetwork(TestSecurityGroupNetwork): def setUp(self): super(TestUnsetSecurityGroupNetwork, self).setUp() - self.network.update_security_group = mock.Mock(return_value=None) + self.network_client.update_security_group = mock.Mock( + return_value=None + ) - self.network.find_security_group = mock.Mock( + self.network_client.find_security_group = mock.Mock( return_value=self._security_group ) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = security_group.UnsetSecurityGroup(self.app, self.namespace) @@ -616,8 +622,8 @@ class TestUnsetSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_security_group.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_security_group.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def _test_unset_tags(self, with_tags=True): @@ -635,8 +641,8 @@ class TestUnsetSecurityGroupNetwork(TestSecurityGroupNetwork): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_security_group.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_security_group.called) + self.network_client.set_tags.assert_called_once_with( self._security_group, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_security_group_rule_network.py b/openstackclient/tests/unit/network/v2/test_security_group_rule_network.py index 070851f56c..232ea56bbf 100644 --- a/openstackclient/tests/unit/network/v2/test_security_group_rule_network.py +++ b/openstackclient/tests/unit/network/v2/test_security_group_rule_network.py @@ -27,7 +27,7 @@ class TestSecurityGroupRuleNetwork(network_fakes.TestNetworkV2): super(TestSecurityGroupRuleNetwork, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -71,7 +71,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): attrs ) ) - self.network.create_security_group_rule = mock.Mock( + self.network_client.create_security_group_rule = mock.Mock( return_value=self._security_group_rule ) self.expected_data = ( @@ -92,11 +92,11 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): def setUp(self): super(TestCreateSecurityGroupRuleNetwork, self).setUp() - self.network.find_security_group = mock.Mock( + self.network_client.find_security_group = mock.Mock( return_value=self._security_group ) - self.network.find_address_group = mock.Mock( + self.network_client.find_address_group = mock.Mock( return_value=self._address_group ) @@ -238,7 +238,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -276,7 +276,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -312,7 +312,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -346,7 +346,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -392,7 +392,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -427,7 +427,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -462,7 +462,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -497,7 +497,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -554,7 +554,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -718,7 +718,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -757,7 +757,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -796,7 +796,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -835,7 +835,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -877,7 +877,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -918,7 +918,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'direction': self._security_group_rule.direction, 'ethertype': self._security_group_rule.ether_type, @@ -950,7 +950,7 @@ class TestCreateSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.create_security_group_rule.assert_called_once_with( + self.network_client.create_security_group_rule.assert_called_once_with( **{ 'description': self._security_group_rule.description, 'direction': self._security_group_rule.direction, @@ -975,9 +975,11 @@ class TestDeleteSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): def setUp(self): super(TestDeleteSecurityGroupRuleNetwork, self).setUp() - self.network.delete_security_group_rule = mock.Mock(return_value=None) + self.network_client.delete_security_group_rule = mock.Mock( + return_value=None + ) - self.network.find_security_group_rule = ( + self.network_client.find_security_group_rule = ( network_fakes.FakeSecurityGroupRule.get_security_group_rules( self._security_group_rules ) @@ -999,7 +1001,7 @@ class TestDeleteSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): result = self.cmd.take_action(parsed_args) - self.network.delete_security_group_rule.assert_called_once_with( + self.network_client.delete_security_group_rule.assert_called_once_with( self._security_group_rules[0] ) self.assertIsNone(result) @@ -1020,7 +1022,7 @@ class TestDeleteSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): calls = [] for s in self._security_group_rules: calls.append(call(s)) - self.network.delete_security_group_rule.assert_has_calls(calls) + self.network_client.delete_security_group_rule.assert_has_calls(calls) self.assertIsNone(result) def test_multi_security_group_rules_delete_with_exception(self): @@ -1037,7 +1039,7 @@ class TestDeleteSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): self._security_group_rules[0], exceptions.CommandError, ] - self.network.find_security_group_rule = mock.Mock( + self.network_client.find_security_group_rule = mock.Mock( side_effect=find_mock_result ) @@ -1047,13 +1049,13 @@ class TestDeleteSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): except exceptions.CommandError as e: self.assertEqual('1 of 2 rules failed to delete.', str(e)) - self.network.find_security_group_rule.assert_any_call( + self.network_client.find_security_group_rule.assert_any_call( self._security_group_rules[0].id, ignore_missing=False ) - self.network.find_security_group_rule.assert_any_call( + self.network_client.find_security_group_rule.assert_any_call( 'unexist_rule', ignore_missing=False ) - self.network.delete_security_group_rule.assert_called_once_with( + self.network_client.delete_security_group_rule.assert_called_once_with( self._security_group_rules[0] ) @@ -1151,10 +1153,10 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): def setUp(self): super(TestListSecurityGroupRuleNetwork, self).setUp() - self.network.find_security_group = mock.Mock( + self.network_client.find_security_group = mock.Mock( return_value=self._security_group ) - self.network.security_group_rules = mock.Mock( + self.network_client.security_group_rules = mock.Mock( return_value=self._security_group_rules ) @@ -1169,7 +1171,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_group_rules.assert_called_once_with(**{}) + self.network_client.security_group_rules.assert_called_once_with(**{}) self.assertEqual(self.expected_columns_no_group, columns) self.assertEqual(self.expected_data_no_group, list(data)) @@ -1185,7 +1187,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_group_rules.assert_called_once_with( + self.network_client.security_group_rules.assert_called_once_with( **{ 'security_group_id': self._security_group.id, } @@ -1205,7 +1207,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_group_rules.assert_called_once_with(**{}) + self.network_client.security_group_rules.assert_called_once_with(**{}) self.assertEqual(self.expected_columns_no_group, columns) self.assertEqual(self.expected_data_no_group, list(data)) @@ -1222,7 +1224,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_group_rules.assert_called_once_with( + self.network_client.security_group_rules.assert_called_once_with( **{ 'protocol': 'tcp', } @@ -1242,7 +1244,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_group_rules.assert_called_once_with( + self.network_client.security_group_rules.assert_called_once_with( **{ 'direction': 'ingress', } @@ -1262,7 +1264,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.security_group_rules.assert_called_once_with( + self.network_client.security_group_rules.assert_called_once_with( **{ 'direction': 'egress', } @@ -1310,7 +1312,7 @@ class TestShowSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): def setUp(self): super(TestShowSecurityGroupRuleNetwork, self).setUp() - self.network.find_security_group_rule = mock.Mock( + self.network_client.find_security_group_rule = mock.Mock( return_value=self._security_group_rule ) @@ -1335,7 +1337,7 @@ class TestShowSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork): columns, data = self.cmd.take_action(parsed_args) - self.network.find_security_group_rule.assert_called_once_with( + self.network_client.find_security_group_rule.assert_called_once_with( self._security_group_rule.id, ignore_missing=False ) self.assertEqual(self.columns, columns) diff --git a/openstackclient/tests/unit/network/v2/test_subnet.py b/openstackclient/tests/unit/network/v2/test_subnet.py index 45677f8121..504a7ee56e 100644 --- a/openstackclient/tests/unit/network/v2/test_subnet.py +++ b/openstackclient/tests/unit/network/v2/test_subnet.py @@ -28,7 +28,7 @@ class TestSubnet(network_fakes.TestNetworkV2): super(TestSubnet, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -223,13 +223,17 @@ class TestCreateSubnet(TestSubnet): self.domains_mock.get.return_value = self.domain # Mock SDK calls for all tests. - self.network.create_subnet = mock.Mock(return_value=self._subnet) - self.network.set_tags = mock.Mock(return_value=None) - self.network.find_network = mock.Mock(return_value=self._network) - self.network.find_segment = mock.Mock( + self.network_client.create_subnet = mock.Mock( + return_value=self._subnet + ) + self.network_client.set_tags = mock.Mock(return_value=None) + self.network_client.find_network = mock.Mock( + return_value=self._network + ) + self.network_client.find_segment = mock.Mock( return_value=self._network_segment ) - self.network.find_subnet_pool = mock.Mock( + self.network_client.find_subnet_pool = mock.Mock( return_value=self._subnet_pool ) @@ -246,8 +250,8 @@ class TestCreateSubnet(TestSubnet): arglist, verifylist, ) - self.assertFalse(self.network.create_subnet.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.create_subnet.called) + self.assertFalse(self.network_client.set_tags.called) def test_create_default_options(self): # Mock SDK calls for this test. @@ -271,7 +275,7 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( **{ 'cidr': self._subnet.cidr, 'ip_version': self._subnet.ip_version, @@ -279,14 +283,14 @@ class TestCreateSubnet(TestSubnet): 'network_id': self._subnet.network_id, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) def test_create_from_subnet_pool_options(self): # Mock SDK calls for this test. - self.network.create_subnet.return_value = self._subnet_from_pool - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.create_subnet.return_value = self._subnet_from_pool + self.network_client.set_tags = mock.Mock(return_value=None) self._network.id = self._subnet_from_pool.network_id arglist = [ @@ -343,7 +347,7 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( **{ 'dns_nameservers': self._subnet_from_pool.dns_nameservers, 'enable_dhcp': self._subnet_from_pool.enable_dhcp, @@ -362,7 +366,7 @@ class TestCreateSubnet(TestSubnet): def test_create_options_subnet_range_ipv6(self): # Mock SDK calls for this test. - self.network.create_subnet.return_value = self._subnet_ipv6 + self.network_client.create_subnet.return_value = self._subnet_ipv6 self._network.id = self._subnet_ipv6.network_id arglist = [ @@ -433,7 +437,7 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( **{ 'cidr': self._subnet_ipv6.cidr, 'dns_nameservers': self._subnet_ipv6.dns_nameservers, @@ -449,7 +453,7 @@ class TestCreateSubnet(TestSubnet): 'service_types': self._subnet_ipv6.service_types, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data_ipv6, data) @@ -478,7 +482,7 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( **{ 'cidr': self._subnet.cidr, 'ip_version': self._subnet.ip_version, @@ -487,7 +491,7 @@ class TestCreateSubnet(TestSubnet): 'segment_id': self._network_segment.id, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -516,7 +520,7 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( **{ 'cidr': self._subnet.cidr, 'ip_version': self._subnet.ip_version, @@ -525,7 +529,7 @@ class TestCreateSubnet(TestSubnet): 'description': self._subnet.description, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -553,7 +557,7 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( cidr=self._subnet.cidr, ip_version=self._subnet.ip_version, name=self._subnet.name, @@ -596,18 +600,18 @@ class TestCreateSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet.assert_called_once_with( + self.network_client.create_subnet.assert_called_once_with( cidr=self._subnet.cidr, ip_version=self._subnet.ip_version, name=self._subnet.name, network_id=self._subnet.network_id, ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self._subnet, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -625,9 +629,9 @@ class TestDeleteSubnet(TestSubnet): def setUp(self): super(TestDeleteSubnet, self).setUp() - self.network.delete_subnet = mock.Mock(return_value=None) + self.network_client.delete_subnet = mock.Mock(return_value=None) - self.network.find_subnet = network_fakes.FakeSubnet.get_subnets( + self.network_client.find_subnet = network_fakes.FakeSubnet.get_subnets( self._subnets ) @@ -644,7 +648,9 @@ class TestDeleteSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.delete_subnet.assert_called_once_with(self._subnets[0]) + self.network_client.delete_subnet.assert_called_once_with( + self._subnets[0] + ) self.assertIsNone(result) def test_multi_subnets_delete(self): @@ -663,7 +669,7 @@ class TestDeleteSubnet(TestSubnet): calls = [] for s in self._subnets: calls.append(call(s)) - self.network.delete_subnet.assert_has_calls(calls) + self.network_client.delete_subnet.assert_has_calls(calls) self.assertIsNone(result) def test_multi_subnets_delete_with_exception(self): @@ -677,7 +683,9 @@ class TestDeleteSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._subnets[0], exceptions.CommandError] - self.network.find_subnet = mock.Mock(side_effect=find_mock_result) + self.network_client.find_subnet = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -685,13 +693,15 @@ class TestDeleteSubnet(TestSubnet): except exceptions.CommandError as e: self.assertEqual('1 of 2 subnets failed to delete.', str(e)) - self.network.find_subnet.assert_any_call( + self.network_client.find_subnet.assert_any_call( self._subnets[0].name, ignore_missing=False ) - self.network.find_subnet.assert_any_call( + self.network_client.find_subnet.assert_any_call( 'unexist_subnet', ignore_missing=False ) - self.network.delete_subnet.assert_called_once_with(self._subnets[0]) + self.network_client.delete_subnet.assert_called_once_with( + self._subnets[0] + ) class TestListSubnet(TestSubnet): @@ -753,7 +763,7 @@ class TestListSubnet(TestSubnet): # Get the command object to test self.cmd = subnet_v2.ListSubnet(self.app, self.namespace) - self.network.subnets = mock.Mock(return_value=self._subnet) + self.network_client.subnets = mock.Mock(return_value=self._subnet) def test_subnet_list_no_options(self): arglist = [] @@ -764,7 +774,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) - self.network.subnets.assert_called_once_with() + self.network_client.subnets.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -779,7 +789,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) - self.network.subnets.assert_called_once_with() + self.network_client.subnets.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertCountEqual(self.data_long, list(data)) @@ -796,7 +806,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'ip_version': 4} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -812,7 +822,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'enable_dhcp': True, 'is_dhcp_enabled': True} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -828,7 +838,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'enable_dhcp': False, 'is_dhcp_enabled': False} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -844,7 +854,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'service_types': ['network:router_gateway']} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -863,7 +873,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -889,7 +899,7 @@ class TestListSubnet(TestSubnet): 'network:floatingip_agent_gateway', ] } - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -911,13 +921,13 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) def test_subnet_list_network(self): network = network_fakes.create_one_network() - self.network.find_network = mock.Mock(return_value=network) + self.network_client.find_network = mock.Mock(return_value=network) arglist = [ '--network', network.id, @@ -930,13 +940,13 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'network_id': network.id} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) def test_subnet_list_gateway(self): subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_network = mock.Mock(return_value=subnet) + self.network_client.find_network = mock.Mock(return_value=subnet) arglist = [ '--gateway', subnet.gateway_ip, @@ -949,13 +959,13 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'gateway_ip': subnet.gateway_ip} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) def test_subnet_list_name(self): subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_network = mock.Mock(return_value=subnet) + self.network_client.find_network = mock.Mock(return_value=subnet) arglist = [ '--name', subnet.name, @@ -968,13 +978,13 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'name': subnet.name} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) def test_subnet_list_subnet_range(self): subnet = network_fakes.FakeSubnet.create_one_subnet() - self.network.find_network = mock.Mock(return_value=subnet) + self.network_client.find_network = mock.Mock(return_value=subnet) arglist = [ '--subnet-range', subnet.cidr, @@ -987,7 +997,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'cidr': subnet.cidr} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -996,8 +1006,10 @@ class TestListSubnet(TestSubnet): subnet = network_fakes.FakeSubnet.create_one_subnet( {'subnetpool_id': subnet_pool.id} ) - self.network.find_network = mock.Mock(return_value=subnet) - self.network.find_subnet_pool = mock.Mock(return_value=subnet_pool) + self.network_client.find_network = mock.Mock(return_value=subnet) + self.network_client.find_subnet_pool = mock.Mock( + return_value=subnet_pool + ) arglist = [ '--subnet-pool', subnet_pool.name, @@ -1010,7 +1022,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'subnetpool_id': subnet_pool.id} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1019,8 +1031,10 @@ class TestListSubnet(TestSubnet): subnet = network_fakes.FakeSubnet.create_one_subnet( {'subnetpool_id': subnet_pool.id} ) - self.network.find_network = mock.Mock(return_value=subnet) - self.network.find_subnet_pool = mock.Mock(return_value=subnet_pool) + self.network_client.find_network = mock.Mock(return_value=subnet) + self.network_client.find_subnet_pool = mock.Mock( + return_value=subnet_pool + ) arglist = [ '--subnet-pool', subnet_pool.id, @@ -1033,7 +1047,7 @@ class TestListSubnet(TestSubnet): columns, data = self.cmd.take_action(parsed_args) filters = {'subnetpool_id': subnet_pool.id} - self.network.subnets.assert_called_once_with(**filters) + self.network_client.subnets.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -1057,7 +1071,7 @@ class TestListSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.subnets.assert_called_once_with( + self.network_client.subnets.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -1076,9 +1090,9 @@ class TestSetSubnet(TestSubnet): def setUp(self): super(TestSetSubnet, self).setUp() - self.network.update_subnet = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) - self.network.find_subnet = mock.Mock(return_value=self._subnet) + self.network_client.update_subnet = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) + self.network_client.find_subnet = mock.Mock(return_value=self._subnet) self.cmd = subnet_v2.SetSubnet(self.app, self.namespace) def test_set_this(self): @@ -1104,7 +1118,9 @@ class TestSetSubnet(TestSubnet): 'gateway_ip': self._subnet.gateway_ip, 'name': "new_subnet", } - self.network.update_subnet.assert_called_with(self._subnet, **attrs) + self.network_client.update_subnet.assert_called_with( + self._subnet, **attrs + ) self.assertIsNone(result) def test_set_that(self): @@ -1130,7 +1146,9 @@ class TestSetSubnet(TestSubnet): 'gateway_ip': None, 'name': "new_subnet", } - self.network.update_subnet.assert_called_with(self._subnet, **attrs) + self.network_client.update_subnet.assert_called_with( + self._subnet, **attrs + ) self.assertIsNone(result) def test_set_nothing(self): @@ -1142,8 +1160,8 @@ class TestSetSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_subnet.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_subnet.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def test_append_options(self): @@ -1153,7 +1171,7 @@ class TestSetSubnet(TestSubnet): 'service_types': ["network:router_gateway"], } ) - self.network.find_subnet = mock.Mock(return_value=_testsubnet) + self.network_client.find_subnet = mock.Mock(return_value=_testsubnet) arglist = [ '--dns-nameserver', '10.0.0.2', @@ -1174,7 +1192,7 @@ class TestSetSubnet(TestSubnet): 'network:router_gateway', ], } - self.network.update_subnet.assert_called_once_with( + self.network_client.update_subnet.assert_called_once_with( _testsubnet, **attrs ) self.assertIsNone(result) @@ -1202,7 +1220,9 @@ class TestSetSubnet(TestSubnet): 'gateway_ip': self._subnet.gateway_ip, 'description': "new_description", } - self.network.update_subnet.assert_called_with(self._subnet, **attrs) + self.network_client.update_subnet.assert_called_with( + self._subnet, **attrs + ) self.assertIsNone(result) def test_overwrite_options(self): @@ -1217,7 +1237,7 @@ class TestSetSubnet(TestSubnet): 'dns_nameservers': ["10.0.0.1"], } ) - self.network.find_subnet = mock.Mock(return_value=_testsubnet) + self.network_client.find_subnet = mock.Mock(return_value=_testsubnet) arglist = [ '--host-route', 'destination=10.30.30.30/24,gateway=10.30.30.1', @@ -1250,7 +1270,7 @@ class TestSetSubnet(TestSubnet): 'allocation_pools': [{'start': '8.8.8.100', 'end': '8.8.8.150'}], 'dns_nameservers': ["10.1.10.1"], } - self.network.update_subnet.assert_called_once_with( + self.network_client.update_subnet.assert_called_once_with( _testsubnet, **attrs ) self.assertIsNone(result) @@ -1267,7 +1287,7 @@ class TestSetSubnet(TestSubnet): 'dns_nameservers': ['10.0.0.1'], } ) - self.network.find_subnet = mock.Mock(return_value=_testsubnet) + self.network_client.find_subnet = mock.Mock(return_value=_testsubnet) arglist = [ '--no-host-route', '--no-allocation-pool', @@ -1286,7 +1306,7 @@ class TestSetSubnet(TestSubnet): 'allocation_pools': [], 'dns_nameservers': [], } - self.network.update_subnet.assert_called_once_with( + self.network_client.update_subnet.assert_called_once_with( _testsubnet, **attrs ) self.assertIsNone(result) @@ -1306,8 +1326,8 @@ class TestSetSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_subnet.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_subnet.called) + self.network_client.set_tags.assert_called_once_with( self._subnet, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -1336,15 +1356,17 @@ class TestSetSubnet(TestSubnet): 'segment_id': None, } ) - self.network.find_subnet = mock.Mock(return_value=_subnet) - self.network.find_segment = mock.Mock(return_value=_segment) + self.network_client.find_subnet = mock.Mock(return_value=_subnet) + self.network_client.find_segment = mock.Mock(return_value=_segment) arglist = ['--network-segment', _segment.id, _subnet.name] verifylist = [('network_segment', _segment.id)] parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) attrs = {'segment_id': _segment.id} - self.network.update_subnet.assert_called_once_with(_subnet, **attrs) - self.network.update_subnet.assert_called_with(_subnet, **attrs) + self.network_client.update_subnet.assert_called_once_with( + _subnet, **attrs + ) + self.network_client.update_subnet.assert_called_with(_subnet, **attrs) self.assertIsNone(result) @@ -1400,7 +1422,7 @@ class TestShowSubnet(TestSubnet): # Get the command object to test self.cmd = subnet_v2.ShowSubnet(self.app, self.namespace) - self.network.find_subnet = mock.Mock(return_value=self._subnet) + self.network_client.find_subnet = mock.Mock(return_value=self._subnet) def test_show_no_options(self): arglist = [] @@ -1427,7 +1449,7 @@ class TestShowSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.find_subnet.assert_called_once_with( + self.network_client.find_subnet.assert_called_once_with( self._subnet.name, ignore_missing=False ) @@ -1457,9 +1479,11 @@ class TestUnsetSubnet(TestSubnet): 'tags': ['green', 'red'], } ) - self.network.find_subnet = mock.Mock(return_value=self._testsubnet) - self.network.update_subnet = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.find_subnet = mock.Mock( + return_value=self._testsubnet + ) + self.network_client.update_subnet = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = subnet_v2.UnsetSubnet(self.app, self.namespace) @@ -1499,7 +1523,7 @@ class TestUnsetSubnet(TestSubnet): 'service_types': ['network:floatingip_agent_gateway'], 'gateway_ip': None, } - self.network.update_subnet.assert_called_once_with( + self.network_client.update_subnet.assert_called_once_with( self._testsubnet, **attrs ) self.assertIsNone(result) @@ -1618,8 +1642,8 @@ class TestUnsetSubnet(TestSubnet): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_subnet.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_subnet.called) + self.network_client.set_tags.assert_called_once_with( self._testsubnet, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) diff --git a/openstackclient/tests/unit/network/v2/test_subnet_pool.py b/openstackclient/tests/unit/network/v2/test_subnet_pool.py index 9a4532f606..30d06b9c27 100644 --- a/openstackclient/tests/unit/network/v2/test_subnet_pool.py +++ b/openstackclient/tests/unit/network/v2/test_subnet_pool.py @@ -29,7 +29,7 @@ class TestSubnetPool(network_fakes.TestNetworkV2): super(TestSubnetPool, self).setUp() # Get a shortcut to the network client - self.network = self.app.client_manager.network + self.network_client = self.app.client_manager.network # Get a shortcut to the ProjectManager Mock self.projects_mock = self.app.client_manager.identity.projects # Get a shortcut to the DomainManager Mock @@ -80,15 +80,15 @@ class TestCreateSubnetPool(TestSubnetPool): def setUp(self): super(TestCreateSubnetPool, self).setUp() - self.network.create_subnet_pool = mock.Mock( + self.network_client.create_subnet_pool = mock.Mock( return_value=self._subnet_pool ) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = subnet_pool.CreateSubnetPool(self.app, self.namespace) - self.network.find_address_scope = mock.Mock( + self.network_client.find_address_scope = mock.Mock( return_value=self._address_scope ) @@ -106,7 +106,7 @@ class TestCreateSubnetPool(TestSubnetPool): arglist, verifylist, ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) def test_create_no_pool_prefix(self): """Make sure --pool-prefix is a required argument""" @@ -138,13 +138,13 @@ class TestCreateSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'prefixes': ['10.0.10.0/24'], 'name': self._subnet_pool.name, } ) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -174,7 +174,7 @@ class TestCreateSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'default_prefixlen': int(self._subnet_pool.default_prefixlen), 'max_prefixlen': int(self._subnet_pool.max_prefixlen), @@ -225,7 +225,7 @@ class TestCreateSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'prefixes': ['10.0.10.0/24'], 'project_id': self.project.id, @@ -252,7 +252,7 @@ class TestCreateSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'prefixes': ['10.0.10.0/24'], 'address_scope_id': self._address_scope.id, @@ -280,7 +280,7 @@ class TestCreateSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'is_default': True, 'name': self._subnet_pool.name, @@ -308,7 +308,7 @@ class TestCreateSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'name': self._subnet_pool.name, 'prefixes': ['10.0.10.0/24'], @@ -333,7 +333,7 @@ class TestCreateSubnetPool(TestSubnetPool): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( **{ 'name': self._subnet_pool.name, 'prefixes': ['10.0.10.0/24'], @@ -365,15 +365,15 @@ class TestCreateSubnetPool(TestSubnetPool): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.create_subnet_pool.assert_called_once_with( + self.network_client.create_subnet_pool.assert_called_once_with( prefixes=['10.0.10.0/24'], name=self._subnet_pool.name ) if add_tags: - self.network.set_tags.assert_called_once_with( + self.network_client.set_tags.assert_called_once_with( self._subnet_pool, tests_utils.CompareBySet(['red', 'blue']) ) else: - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.set_tags.called) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, data) @@ -391,9 +391,9 @@ class TestDeleteSubnetPool(TestSubnetPool): def setUp(self): super(TestDeleteSubnetPool, self).setUp() - self.network.delete_subnet_pool = mock.Mock(return_value=None) + self.network_client.delete_subnet_pool = mock.Mock(return_value=None) - self.network.find_subnet_pool = ( + self.network_client.find_subnet_pool = ( network_fakes.FakeSubnetPool.get_subnet_pools(self._subnet_pools) ) @@ -411,7 +411,7 @@ class TestDeleteSubnetPool(TestSubnetPool): result = self.cmd.take_action(parsed_args) - self.network.delete_subnet_pool.assert_called_once_with( + self.network_client.delete_subnet_pool.assert_called_once_with( self._subnet_pools[0] ) self.assertIsNone(result) @@ -432,7 +432,7 @@ class TestDeleteSubnetPool(TestSubnetPool): calls = [] for s in self._subnet_pools: calls.append(call(s)) - self.network.delete_subnet_pool.assert_has_calls(calls) + self.network_client.delete_subnet_pool.assert_has_calls(calls) self.assertIsNone(result) def test_multi_subnet_pools_delete_with_exception(self): @@ -449,7 +449,9 @@ class TestDeleteSubnetPool(TestSubnetPool): parsed_args = self.check_parser(self.cmd, arglist, verifylist) find_mock_result = [self._subnet_pools[0], exceptions.CommandError] - self.network.find_subnet_pool = mock.Mock(side_effect=find_mock_result) + self.network_client.find_subnet_pool = mock.Mock( + side_effect=find_mock_result + ) try: self.cmd.take_action(parsed_args) @@ -457,13 +459,13 @@ class TestDeleteSubnetPool(TestSubnetPool): except exceptions.CommandError as e: self.assertEqual('1 of 2 subnet pools failed to delete.', str(e)) - self.network.find_subnet_pool.assert_any_call( + self.network_client.find_subnet_pool.assert_any_call( self._subnet_pools[0].name, ignore_missing=False ) - self.network.find_subnet_pool.assert_any_call( + self.network_client.find_subnet_pool.assert_any_call( 'unexist_subnet_pool', ignore_missing=False ) - self.network.delete_subnet_pool.assert_called_once_with( + self.network_client.delete_subnet_pool.assert_called_once_with( self._subnet_pools[0] ) @@ -516,7 +518,9 @@ class TestListSubnetPool(TestSubnetPool): # Get the command object to test self.cmd = subnet_pool.ListSubnetPool(self.app, self.namespace) - self.network.subnet_pools = mock.Mock(return_value=self._subnet_pools) + self.network_client.subnet_pools = mock.Mock( + return_value=self._subnet_pools + ) def test_subnet_pool_list_no_option(self): arglist = [] @@ -527,7 +531,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.subnet_pools.assert_called_once_with() + self.network_client.subnet_pools.assert_called_once_with() self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -542,7 +546,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.subnet_pools.assert_called_once_with() + self.network_client.subnet_pools.assert_called_once_with() self.assertEqual(self.columns_long, columns) self.assertCountEqual(self.data_long, list(data)) @@ -558,7 +562,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'shared': False, 'is_shared': False} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -574,7 +578,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'shared': True, 'is_shared': True} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -590,7 +594,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'is_default': False} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -606,7 +610,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'is_default': True} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -625,7 +629,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -647,13 +651,13 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'project_id': project.id} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) def test_subnet_pool_list_name(self): subnet_pool = network_fakes.FakeSubnetPool.create_one_subnet_pool() - self.network.find_network = mock.Mock(return_value=subnet_pool) + self.network_client.find_network = mock.Mock(return_value=subnet_pool) arglist = [ '--name', subnet_pool.name, @@ -666,13 +670,15 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'name': subnet_pool.name} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) def test_subnet_pool_list_address_scope(self): addr_scope = network_fakes.create_one_address_scope() - self.network.find_address_scope = mock.Mock(return_value=addr_scope) + self.network_client.find_address_scope = mock.Mock( + return_value=addr_scope + ) arglist = [ '--address-scope', addr_scope.id, @@ -685,7 +691,7 @@ class TestListSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) filters = {'address_scope_id': addr_scope.id} - self.network.subnet_pools.assert_called_once_with(**filters) + self.network_client.subnet_pools.assert_called_once_with(**filters) self.assertEqual(self.columns, columns) self.assertCountEqual(self.data, list(data)) @@ -709,7 +715,7 @@ class TestListSubnetPool(TestSubnetPool): parsed_args = self.check_parser(self.cmd, arglist, verifylist) columns, data = self.cmd.take_action(parsed_args) - self.network.subnet_pools.assert_called_once_with( + self.network_client.subnet_pools.assert_called_once_with( **{ 'tags': 'red,blue', 'any_tags': 'red,green', @@ -732,14 +738,14 @@ class TestSetSubnetPool(TestSubnetPool): def setUp(self): super(TestSetSubnetPool, self).setUp() - self.network.update_subnet_pool = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.update_subnet_pool = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) - self.network.find_subnet_pool = mock.Mock( + self.network_client.find_subnet_pool = mock.Mock( return_value=self._subnet_pool ) - self.network.find_address_scope = mock.Mock( + self.network_client.find_address_scope = mock.Mock( return_value=self._address_scope ) @@ -771,7 +777,7 @@ class TestSetSubnetPool(TestSubnetPool): 'default_prefixlen': 8, 'min_prefixlen': 8, } - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -801,7 +807,7 @@ class TestSetSubnetPool(TestSubnetPool): 'prefixes': prefixes, 'max_prefixlen': 16, } - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -817,8 +823,8 @@ class TestSetSubnetPool(TestSubnetPool): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_subnet_pool.called) - self.assertFalse(self.network.set_tags.called) + self.assertFalse(self.network_client.update_subnet_pool.called) + self.assertFalse(self.network_client.set_tags.called) self.assertIsNone(result) def test_set_len_negative(self): @@ -857,7 +863,7 @@ class TestSetSubnetPool(TestSubnetPool): attrs = { 'address_scope_id': self._address_scope.id, } - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -878,7 +884,7 @@ class TestSetSubnetPool(TestSubnetPool): attrs = { 'address_scope_id': None, } - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -919,7 +925,7 @@ class TestSetSubnetPool(TestSubnetPool): result = self.cmd.take_action(parsed_args) attrs = {'is_default': True} - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -940,7 +946,7 @@ class TestSetSubnetPool(TestSubnetPool): attrs = { 'is_default': False, } - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -983,7 +989,7 @@ class TestSetSubnetPool(TestSubnetPool): attrs = { 'description': "new_description", } - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **attrs ) self.assertIsNone(result) @@ -1000,7 +1006,7 @@ class TestSetSubnetPool(TestSubnetPool): ] parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.network.update_subnet_pool.assert_called_once_with( + self.network_client.update_subnet_pool.assert_called_once_with( self._subnet_pool, **{ 'default_quota': 20, @@ -1023,8 +1029,8 @@ class TestSetSubnetPool(TestSubnetPool): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_subnet_pool.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_subnet_pool.called) + self.network_client.set_tags.assert_called_once_with( self._subnet_pool, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result) @@ -1077,7 +1083,7 @@ class TestShowSubnetPool(TestSubnetPool): def setUp(self): super(TestShowSubnetPool, self).setUp() - self.network.find_subnet_pool = mock.Mock( + self.network_client.find_subnet_pool = mock.Mock( return_value=self._subnet_pool ) @@ -1107,7 +1113,7 @@ class TestShowSubnetPool(TestSubnetPool): columns, data = self.cmd.take_action(parsed_args) - self.network.find_subnet_pool.assert_called_once_with( + self.network_client.find_subnet_pool.assert_called_once_with( self._subnet_pool.name, ignore_missing=False ) self.assertEqual(self.columns, columns) @@ -1120,11 +1126,11 @@ class TestUnsetSubnetPool(TestSubnetPool): self._subnetpool = network_fakes.FakeSubnetPool.create_one_subnet_pool( {'tags': ['green', 'red']} ) - self.network.find_subnet_pool = mock.Mock( + self.network_client.find_subnet_pool = mock.Mock( return_value=self._subnetpool ) - self.network.update_subnet_pool = mock.Mock(return_value=None) - self.network.set_tags = mock.Mock(return_value=None) + self.network_client.update_subnet_pool = mock.Mock(return_value=None) + self.network_client.set_tags = mock.Mock(return_value=None) # Get the command object to test self.cmd = subnet_pool.UnsetSubnetPool(self.app, self.namespace) @@ -1143,8 +1149,8 @@ class TestUnsetSubnetPool(TestSubnetPool): parsed_args = self.check_parser(self.cmd, arglist, verifylist) result = self.cmd.take_action(parsed_args) - self.assertFalse(self.network.update_subnet_pool.called) - self.network.set_tags.assert_called_once_with( + self.assertFalse(self.network_client.update_subnet_pool.called) + self.network_client.set_tags.assert_called_once_with( self._subnetpool, tests_utils.CompareBySet(expected_args) ) self.assertIsNone(result)