diff --git a/neutron/tests/unit/agent/l3/test_agent.py b/neutron/tests/unit/agent/l3/test_agent.py index 6fda2ead05a..82f3e9da828 100644 --- a/neutron/tests/unit/agent/l3/test_agent.py +++ b/neutron/tests/unit/agent/l3/test_agent.py @@ -878,11 +878,10 @@ class TestBasicRouterOperations(BasicRouterOperationsFramework): self.assertEqual(distributed, ri.process_floating_ip_addresses.called) self.assertEqual(distributed, ri.process_floating_ip_nat_rules.called) - @mock.patch('neutron.agent.linux.ip_lib.IPDevice') - def _test_process_floating_ip_addresses_add(self, ri, agent, IPDevice): + def _test_process_floating_ip_addresses_add(self, ri, agent): floating_ips = ri.get_floating_ips() fip_id = floating_ips[0]['id'] - IPDevice.return_value = device = mock.Mock() + device = self.mock_ip_dev device.addr.list.return_value = [] ri.iptables_manager.ipv4['nat'] = mock.MagicMock() ex_gw_port = {'id': _uuid(), 'network_id': mock.sentinel.ext_net_id}