internal_ports = qclient.list_ports(
device_id=router_id,
- device_owner=n_const.DEVICE_OWNER_ROUTER_INTF)['ports']
+ device_owner=[n_const.DEVICE_OWNER_ROUTER_INTF,
+ n_const.DEVICE_OWNER_DVR_INTERFACE])['ports']
return tuple(p['network_id'] for p in internal_ports)
@utils.cache_method_results
from neutron.tests import base
+EXPECTED_OWNER_ROUTERS = [
+ constants.DEVICE_OWNER_ROUTER_INTF,
+ constants.DEVICE_OWNER_DVR_INTERFACE
+]
+
+
class FakeConf(object):
admin_user = 'neutron'
admin_password = 'password'
networks = self.handler._get_router_networks(router_id)
mock_list_ports.assert_called_once_with(
device_id=router_id,
- device_owner=constants.DEVICE_OWNER_ROUTER_INTF)
+ device_owner=EXPECTED_OWNER_ROUTERS)
self.assertEqual(expected, networks)
def _test_get_router_networks_twice_helper(self):
networks = self.handler._get_router_networks(router_id)
mock_list_ports.assert_called_once_with(
device_id=router_id,
- device_owner=constants.DEVICE_OWNER_ROUTER_INTF)
+ device_owner=EXPECTED_OWNER_ROUTERS)
self.assertEqual(expected_networks, networks)
networks = self.handler._get_router_networks(router_id)
new_qclient_call,
mock.call().list_ports(
device_id=router_id,
- device_owner=constants.DEVICE_OWNER_ROUTER_INTF
+ device_owner=EXPECTED_OWNER_ROUTERS
)
])