Remove unnecessary call to get_dhcp_port from DeviceManager

Change-Id: I35459c352e712ea009589f3050205a4c307b909a
Closes-Bug: #1263217

diff --git a/neutron/agent/linux/dhcp.py b/neutron/agent/linux/dhcp.py
index 1a508c15c78f75a7ca270f8f1d7d1dc3d0b71a78..5740aecaed9382bee22071d54048119bbfc4e3dc 100644 (file)
--- a/neutron/agent/linux/dhcp.py
+++ b/neutron/agent/linux/dhcp.py
@@ -624,26 +624,21 @@ class DeviceManager(object):
         host_uuid = uuid.uuid5(uuid.NAMESPACE_DNS, socket.gethostname())
         return 'dhcp%s-%s' % (host_uuid, network.id)
 
-    def _get_device(self, network):
+    def _get_device(self, network, port):
         """Return DHCP ip_lib device for this host on the network."""
-        device_id = self.get_device_id(network)
-        port = self.plugin.get_dhcp_port(network.id, device_id)
-        if port:
-            interface_name = self.get_interface_name(network, port)
-            return ip_lib.IPDevice(interface_name,
-                                   self.root_helper,
-                                   network.namespace)
-        else:
-            raise exceptions.NetworkNotFound(net_id=network.id)
+        interface_name = self.get_interface_name(network, port)
+        return ip_lib.IPDevice(interface_name,
+                               self.root_helper,
+                               network.namespace)
 
-    def _set_default_route(self, network):
+    def _set_default_route(self, network, port):
         """Sets the default gateway for this dhcp namespace.
 
         This method is idempotent and will only adjust the route if adjusting
         it would change it from what it already is.  This makes it safe to call
         and avoids unnecessary perturbation of the system.
         """
-        device = self._get_device(network)
+        device = self._get_device(network, port)
         gateway = device.route.get_gateway()
         if gateway:
             gateway = gateway['gateway']
@@ -778,14 +773,18 @@ class DeviceManager(object):
             device.route.pullup_route(interface_name)
 
         if self.conf.use_namespaces:
-            self._set_default_route(network)
+            self._set_default_route(network, port)
 
         return interface_name
 
     def update(self, network):
         """Update device settings for the network's DHCP on this host."""
         if self.conf.use_namespaces:
-            self._set_default_route(network)
+            device_id = self.get_device_id(network)
+            port = self.plugin.get_dhcp_port(network.id, device_id)
+            if not port:
+                raise exceptions.NetworkNotFound(net_id=network.id)
+            self._set_default_route(network, port)
 
     def destroy(self, network, device_name):
         """Destroy the device used for the network's DHCP on this host."""

diff --git a/neutron/tests/unit/test_dhcp_agent.py b/neutron/tests/unit/test_dhcp_agent.py
index f9f2002f3f33f876144845704ee61c3ea7e42ac7..090bf3a55b6ff734df0c4d095cc90c97961e2016 100644 (file)
--- a/neutron/tests/unit/test_dhcp_agent.py
+++ b/neutron/tests/unit/test_dhcp_agent.py
@@ -1159,7 +1159,7 @@ class TestDeviceManager(base.BaseTestCase):
                                            namespace=net.namespace))
         self.mock_driver.assert_has_calls(expected)
 
-        dh._set_default_route.assert_called_once_with(net)
+        dh._set_default_route.assert_called_once_with(net, port)
 
     def test_setup(self):
         cfg.CONF.set_override('enable_metadata_network', False)
@@ -1314,15 +1314,21 @@ class TestDeviceManager(base.BaseTestCase):
         dh = dhcp.DeviceManager(cfg.CONF, cfg.CONF.root_helper, None)
         dh._set_default_route = mock.Mock()
 
-        dh.update(True)
+        network = mock.Mock()
+        port = mock.Mock()
+        dh.plugin = mock.Mock()
+        dh.plugin.get_dhcp_port.return_value = port
+        dh.update(network)
 
-        dh._set_default_route.assert_called_once_with(True)
+        dh._set_default_route.assert_called_once_with(network, port)
 
         # No namespaces, shouldn't set default route.
         cfg.CONF.set_override('use_namespaces', False)
         cfg.CONF.set_override('enable_metadata_network', False)
         dh = dhcp.DeviceManager(cfg.CONF, cfg.CONF.root_helper, None)
         dh._set_default_route = mock.Mock()
+        dh.plugin = mock.Mock()
+        dh.plugin.get_dhcp_port.return_value = port
 
         dh.update(FakeV4Network())
 
@@ -1333,6 +1339,8 @@ class TestDeviceManager(base.BaseTestCase):
         cfg.CONF.set_override('enable_metadata_network', True)
         dh = dhcp.DeviceManager(cfg.CONF, cfg.CONF.root_helper, None)
         dh._set_default_route = mock.Mock()
+        dh.plugin = mock.Mock()
+        dh.plugin.get_dhcp_port.return_value = port
 
         dh.update(FakeV4Network())
 
@@ -1356,7 +1364,7 @@ class TestDeviceManager(base.BaseTestCase):
         dh = self._get_device_manager_with_mock_device(cfg.CONF, device)
         network = FakeV4Network()
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         self.assertFalse(device.route.delete_gateway.called)
@@ -1370,7 +1378,7 @@ class TestDeviceManager(base.BaseTestCase):
         dh = self._get_device_manager_with_mock_device(cfg.CONF, device)
         network = FakeV4NetworkNoSubnet()
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         self.assertFalse(device.route.delete_gateway.called)
@@ -1384,7 +1392,7 @@ class TestDeviceManager(base.BaseTestCase):
         dh = self._get_device_manager_with_mock_device(cfg.CONF, device)
         network = FakeV4NetworkNoSubnet()
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         device.route.delete_gateway.assert_called_once_with('192.168.0.1')
@@ -1398,7 +1406,7 @@ class TestDeviceManager(base.BaseTestCase):
         dh = self._get_device_manager_with_mock_device(cfg.CONF, device)
         network = FakeV4NetworkNoGateway()
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         device.route.delete_gateway.assert_called_once_with('192.168.0.1')
@@ -1412,7 +1420,7 @@ class TestDeviceManager(base.BaseTestCase):
         dh = self._get_device_manager_with_mock_device(cfg.CONF, device)
         network = FakeV4Network()
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         self.assertFalse(device.route.delete_gateway.called)
@@ -1426,7 +1434,7 @@ class TestDeviceManager(base.BaseTestCase):
         dh = self._get_device_manager_with_mock_device(cfg.CONF, device)
         network = FakeV4Network()
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         self.assertFalse(device.route.delete_gateway.called)
@@ -1443,12 +1451,24 @@ class TestDeviceManager(base.BaseTestCase):
         subnet2.gateway_ip = '192.168.1.1'
         network.subnets = [subnet2, FakeV4Subnet()]
 
-        dh._set_default_route(network)
+        dh._set_default_route(network, mock.Mock())
 
         device.route.get_gateway.assert_called_once()
         self.assertFalse(device.route.delete_gateway.called)
         device.route.add_gateway.assert_called_once_with('192.168.1.1')
 
+    def test_get_device_dont_call_get_dhcp_port(self):
+        dh = dhcp.DeviceManager(cfg.CONF, cfg.CONF.root_helper, None)
+        dh.get_interface_name = mock.Mock()
+        dh.plugin = mock.Mock()
+        network = mock.Mock()
+        port = mock.Mock()
+
+        dh._get_device(network, port)
+
+        self.assertFalse(dh.plugin.get_dhcp_port.called)
+        dh.get_interface_name.assert_called_once_with(network, port)
+
 
 class TestDictModel(base.BaseTestCase):
     def test_basic_dict(self):