Replace device owners hard coded strings to neutron_lib constants

Change-Id: I821f0def82164ab1303188d3f1bdfd85473470cd
This commit is contained in:
Edan David 2016-07-06 08:40:04 -04:00
parent b6592c7372
commit 151d94521f
11 changed files with 30 additions and 23 deletions

View File

@ -15,6 +15,7 @@
import sys import sys
from neutron_lib import constants
from oslo_config import cfg from oslo_config import cfg
from oslo_log import log as logging from oslo_log import log as logging
@ -63,9 +64,9 @@ class FdbPopulationAgentExtension(agent_extension.AgentCoreResourceExtension):
# the message will be sent to the wire, causing the message # the message will be sent to the wire, causing the message
# to get lost in case the sender uses direct port and is # to get lost in case the sender uses direct port and is
# located on the same hypervisor as the network node. # located on the same hypervisor as the network node.
PERMITTED_DEVICE_OWNERS = {'compute', PERMITTED_DEVICE_OWNERS = {constants.DEVICE_OWNER_COMPUTE_PREFIX,
'network:router_interface', constants.DEVICE_OWNER_ROUTER_INTF,
'network:dhcp'} constants.DEVICE_OWNER_DHCP}
class FdbTableTracker(object): class FdbTableTracker(object):
"""FDB table tracker is a helper class """FDB table tracker is a helper class

View File

@ -27,7 +27,7 @@ from neutron.extensions import portbindings
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
DEVICE_OWNER_NETWORK_PROBE = 'network:probe' DEVICE_OWNER_NETWORK_PROBE = constants.DEVICE_OWNER_NETWORK_PREFIX + 'probe'
DEVICE_OWNER_COMPUTE_PROBE = constants.DEVICE_OWNER_COMPUTE_PREFIX + 'probe' DEVICE_OWNER_COMPUTE_PROBE = constants.DEVICE_OWNER_COMPUTE_PREFIX + 'probe'

View File

@ -212,7 +212,7 @@ class OVSAgentTestFramework(base.BaseOVSLinuxTestCase):
'physical_network': network.get('physical_network', 'physnet'), 'physical_network': network.get('physical_network', 'physnet'),
'segmentation_id': network.get('segmentation_id', 1), 'segmentation_id': network.get('segmentation_id', 1),
'fixed_ips': port['fixed_ips'], 'fixed_ips': port['fixed_ips'],
'device_owner': 'compute', 'device_owner': n_const.DEVICE_OWNER_COMPUTE_PREFIX,
'port_security_enabled': True, 'port_security_enabled': True,
'security_groups': ['default'], 'security_groups': ['default'],
'admin_state_up': True} 'admin_state_up': True}

View File

@ -129,7 +129,7 @@ class IpamTestCase(testlib_api.SqlTestCase):
'admin_state_up': True, 'admin_state_up': True,
'status': constants.PORT_STATUS_ACTIVE, 'status': constants.PORT_STATUS_ACTIVE,
'device_id': 'test_dev_id', 'device_id': 'test_dev_id',
'device_owner': 'compute', 'device_owner': constants.DEVICE_OWNER_COMPUTE_PREFIX,
'fixed_ips': port_fixed_ips} 'fixed_ips': port_fixed_ips}
self.plugin.create_port(self.cxt, {'port': port}) self.plugin.create_port(self.cxt, {'port': port})

View File

@ -16,6 +16,7 @@
import copy import copy
import mock import mock
from neutron_lib import constants
from oslo_config import cfg from oslo_config import cfg
import six import six
@ -31,7 +32,7 @@ from neutron.tests import base
class FdbPopulationExtensionTestCase(base.BaseTestCase): class FdbPopulationExtensionTestCase(base.BaseTestCase):
UPDATE_MSG = {u'device_owner': u'network:router_interface', UPDATE_MSG = {u'device_owner': constants.DEVICE_OWNER_ROUTER_INTF,
u'physical_network': u'physnet1', u'physical_network': u'physnet1',
u'mac_address': u'fa:16:3e:ba:bc:21', u'mac_address': u'fa:16:3e:ba:bc:21',
u'port_id': u'17ceda02-43e1-48d8-beb6-35885b20cae6'} u'port_id': u'17ceda02-43e1-48d8-beb6-35885b20cae6'}
@ -128,7 +129,7 @@ class FdbPopulationExtensionTestCase(base.BaseTestCase):
fdb_extension = self._get_fdb_extension(mock_execute, '') fdb_extension = self._get_fdb_extension(mock_execute, '')
mock_execute.reset_mock() mock_execute.reset_mock()
details = copy.deepcopy(self.UPDATE_MSG) details = copy.deepcopy(self.UPDATE_MSG)
details['device_owner'] = 'neutron:LOADBALANCER' details['device_owner'] = constants.DEVICE_OWNER_LOADBALANCER
fdb_extension.handle_port(context=None, details=details) fdb_extension.handle_port(context=None, details=details)
self.assertFalse(mock_execute.called) self.assertFalse(mock_execute.called)
updated_macs_for_device = ( updated_macs_for_device = (

View File

@ -67,7 +67,7 @@ class TestIpamNonPluggableBackend(base.BaseTestCase):
'network_id': 'fbb9b578-95eb-4b79-a116-78e5c4927176', 'network_id': 'fbb9b578-95eb-4b79-a116-78e5c4927176',
'fixed_ips': n_const.ATTR_NOT_SPECIFIED, 'fixed_ips': n_const.ATTR_NOT_SPECIFIED,
'mac_address': '12:34:56:78:44:ab', 'mac_address': '12:34:56:78:44:ab',
'device_owner': 'compute'}} 'device_owner': n_const.DEVICE_OWNER_COMPUTE_PREFIX}}
expected = [] expected = []
for subnet in subnets: for subnet in subnets:
addr = str(ipv6_utils.get_ipv6_addr_by_EUI64( addr = str(ipv6_utils.get_ipv6_addr_by_EUI64(
@ -100,7 +100,7 @@ class TestIpamNonPluggableBackend(base.BaseTestCase):
'network_id': 'fbb9b578-95eb-4b79-a116-78e5c4927176', 'network_id': 'fbb9b578-95eb-4b79-a116-78e5c4927176',
'fixed_ips': n_const.ATTR_NOT_SPECIFIED, 'fixed_ips': n_const.ATTR_NOT_SPECIFIED,
'mac_address': '12:34:56:78:44:ab', 'mac_address': '12:34:56:78:44:ab',
'device_owner': 'compute'}} 'device_owner': n_const.DEVICE_OWNER_COMPUTE_PREFIX}}
expected = [] expected = []
for subnet in subnets: for subnet in subnets:
addr = str(ipv6_utils.get_ipv6_addr_by_EUI64( addr = str(ipv6_utils.get_ipv6_addr_by_EUI64(

View File

@ -75,7 +75,9 @@ class TestDbBasePluginIpam(test_db_base.NeutronDbPluginV2TestCase):
'driver': mock.Mock(), 'driver': mock.Mock(),
'subnet': mock.Mock(), 'subnet': mock.Mock(),
'subnets': mock.Mock(), 'subnets': mock.Mock(),
'port': {'device_owner': 'compute:None'}, 'port': {
'device_owner': n_const.DEVICE_OWNER_COMPUTE_PREFIX + 'None'
},
'subnet_request': ipam_req.SpecificSubnetRequest( 'subnet_request': ipam_req.SpecificSubnetRequest(
self.tenant_id, self.tenant_id,
self.subnet_id, self.subnet_id,

View File

@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
import mock import mock
from neutron_lib import constants as n_const
from neutron_lib import exceptions as n_exc from neutron_lib import exceptions as n_exc
import testtools import testtools
@ -161,7 +162,7 @@ class TestL3_NAT_dbonly_mixin(base.BaseTestCase):
def test_prevent_l3_port_no_fixed_ips(self, gp): def test_prevent_l3_port_no_fixed_ips(self, gp):
# without fixed IPs is allowed # without fixed IPs is allowed
gp.return_value.get_port.return_value = { gp.return_value.get_port.return_value = {
'device_owner': 'network:router_interface', 'fixed_ips': [], 'device_owner': n_const.DEVICE_OWNER_ROUTER_INTF, 'fixed_ips': [],
'id': 'f' 'id': 'f'
} }
self.db.prevent_l3_port_deletion(None, None) self.db.prevent_l3_port_deletion(None, None)
@ -170,7 +171,7 @@ class TestL3_NAT_dbonly_mixin(base.BaseTestCase):
def test_prevent_l3_port_no_router(self, gp): def test_prevent_l3_port_no_router(self, gp):
# without router is allowed # without router is allowed
gp.return_value.get_port.return_value = { gp.return_value.get_port.return_value = {
'device_owner': 'network:router_interface', 'device_owner': n_const.DEVICE_OWNER_ROUTER_INTF,
'device_id': '44', 'id': 'f', 'device_id': '44', 'id': 'f',
'fixed_ips': [{'ip_address': '1.1.1.1', 'subnet_id': '4'}]} 'fixed_ips': [{'ip_address': '1.1.1.1', 'subnet_id': '4'}]}
self.db.get_router = mock.Mock() self.db.get_router = mock.Mock()
@ -180,7 +181,7 @@ class TestL3_NAT_dbonly_mixin(base.BaseTestCase):
@mock.patch.object(manager.NeutronManager, 'get_plugin') @mock.patch.object(manager.NeutronManager, 'get_plugin')
def test_prevent_l3_port_existing_router(self, gp): def test_prevent_l3_port_existing_router(self, gp):
gp.return_value.get_port.return_value = { gp.return_value.get_port.return_value = {
'device_owner': 'network:router_interface', 'device_owner': n_const.DEVICE_OWNER_ROUTER_INTF,
'device_id': 'some_router', 'id': 'f', 'device_id': 'some_router', 'id': 'f',
'fixed_ips': [{'ip_address': '1.1.1.1', 'subnet_id': '4'}]} 'fixed_ips': [{'ip_address': '1.1.1.1', 'subnet_id': '4'}]}
self.db.get_router = mock.Mock() self.db.get_router = mock.Mock()
@ -190,7 +191,7 @@ class TestL3_NAT_dbonly_mixin(base.BaseTestCase):
@mock.patch.object(manager.NeutronManager, 'get_plugin') @mock.patch.object(manager.NeutronManager, 'get_plugin')
def test_prevent_l3_port_existing_floating_ip(self, gp): def test_prevent_l3_port_existing_floating_ip(self, gp):
gp.return_value.get_port.return_value = { gp.return_value.get_port.return_value = {
'device_owner': 'network:floatingip', 'device_owner': n_const.DEVICE_OWNER_FLOATINGIP,
'device_id': 'some_flip', 'id': 'f', 'device_id': 'some_flip', 'id': 'f',
'fixed_ips': [{'ip_address': '1.1.1.1', 'subnet_id': '4'}]} 'fixed_ips': [{'ip_address': '1.1.1.1', 'subnet_id': '4'}]}
self.db.get_floatingip = mock.Mock() self.db.get_floatingip = mock.Mock()

View File

@ -288,7 +288,8 @@ class TestAddressRequestFactory(base.BaseTestCase):
def test_specific_address_request_is_loaded(self): def test_specific_address_request_is_loaded(self):
for address in ('10.12.0.15', 'fffe::1'): for address in ('10.12.0.15', 'fffe::1'):
ip = {'ip_address': address} ip = {'ip_address': address}
port = {'device_owner': 'compute:None'} port = {'device_owner':
constants.DEVICE_OWNER_COMPUTE_PREFIX + 'None'}
self.assertIsInstance( self.assertIsInstance(
ipam_req.AddressRequestFactory.get_request(None, port, ip), ipam_req.AddressRequestFactory.get_request(None, port, ip),
ipam_req.SpecificAddressRequest) ipam_req.SpecificAddressRequest)
@ -296,7 +297,8 @@ class TestAddressRequestFactory(base.BaseTestCase):
def test_any_address_request_is_loaded(self): def test_any_address_request_is_loaded(self):
for addr in [None, '']: for addr in [None, '']:
ip = {'ip_address': addr} ip = {'ip_address': addr}
port = {'device_owner': 'compute:None'} port = {'device_owner':
constants.DEVICE_OWNER_COMPUTE_PREFIX + 'None'}
self.assertIsInstance( self.assertIsInstance(
ipam_req.AddressRequestFactory.get_request(None, port, ip), ipam_req.AddressRequestFactory.get_request(None, port, ip),
ipam_req.AnyAddressRequest) ipam_req.AnyAddressRequest)
@ -305,14 +307,14 @@ class TestAddressRequestFactory(base.BaseTestCase):
ip = {'mac': '6c:62:6d:de:cf:49', ip = {'mac': '6c:62:6d:de:cf:49',
'subnet_cidr': '2001:470:abcd::/64', 'subnet_cidr': '2001:470:abcd::/64',
'eui64_address': True} 'eui64_address': True}
port = {'device_owner': 'compute:None'} port = {'device_owner': constants.DEVICE_OWNER_COMPUTE_PREFIX + 'None'}
self.assertIsInstance( self.assertIsInstance(
ipam_req.AddressRequestFactory.get_request(None, port, ip), ipam_req.AddressRequestFactory.get_request(None, port, ip),
ipam_req.AutomaticAddressRequest) ipam_req.AutomaticAddressRequest)
def test_prefernext_address_request_on_dhcp_port(self): def test_prefernext_address_request_on_dhcp_port(self):
ip = {} ip = {}
port = {'device_owner': 'network:dhcp'} port = {'device_owner': constants.DEVICE_OWNER_DHCP}
self.assertIsInstance( self.assertIsInstance(
ipam_req.AddressRequestFactory.get_request(None, port, ip), ipam_req.AddressRequestFactory.get_request(None, port, ip),
ipam_req.PreferNextAddressRequest) ipam_req.PreferNextAddressRequest)

View File

@ -53,7 +53,7 @@ class TestPortContext(base.BaseTestCase):
network = mock.MagicMock() network = mock.MagicMock()
binding = mock.Mock() binding = mock.Mock()
port = {'device_owner': 'compute', port = {'device_owner': constants.DEVICE_OWNER_COMPUTE_PREFIX,
portbindings.HOST_ID: 'host'} portbindings.HOST_ID: 'host'}
binding.host = 'foohost' binding.host = 'foohost'
@ -92,7 +92,7 @@ class TestPortContext(base.BaseTestCase):
network = mock.MagicMock() network = mock.MagicMock()
binding = mock.Mock() binding = mock.Mock()
port = {'device_owner': 'compute', port = {'device_owner': constants.DEVICE_OWNER_COMPUTE_PREFIX,
'status': 'status'} 'status': 'status'}
binding.status = 'foostatus' binding.status = 'foostatus'

View File

@ -1025,7 +1025,7 @@ class L3DvrSchedulerTestCase(testlib_api.SqlTestCase):
'context': self.adminContext, 'context': self.adminContext,
'original_port': None, 'original_port': None,
'port': { 'port': {
'device_owner': 'network:None', 'device_owner': constants.DEVICE_OWNER_NETWORK_PREFIX + 'None',
} }
} }
l3plugin = mock.Mock() l3plugin = mock.Mock()
@ -1374,7 +1374,7 @@ class L3DvrSchedulerTestCase(testlib_api.SqlTestCase):
dvr_port = { dvr_port = {
'id': 'dvr_port1', 'id': 'dvr_port1',
'device_id': 'r1', 'device_id': 'r1',
'device_owner': 'network:router_interface_distributed', 'device_owner': constants.DEVICE_OWNER_DVR_INTERFACE,
'fixed_ips': [] 'fixed_ips': []
} }
r1 = { r1 = {