299 lines
13 KiB
Python
299 lines
13 KiB
Python
# Copyright (c) 2015 OpenStack Foundation.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import mock
|
|
from oslo_config import cfg
|
|
import six
|
|
|
|
from neutron.api.v2 import attributes
|
|
from neutron.common import exceptions as n_exc
|
|
from neutron import context
|
|
from neutron.extensions import external_net
|
|
from neutron.extensions import extraroute
|
|
from neutron.extensions import l3
|
|
from neutron.extensions import l3_ext_gw_mode
|
|
from neutron.extensions import providernet as pnet
|
|
from neutron import manager
|
|
import neutron.tests.unit.db.test_db_base_plugin_v2 as test_plugin
|
|
from neutron.tests.unit.extensions import test_extra_dhcp_opt as test_dhcpopts
|
|
import neutron.tests.unit.extensions.test_extraroute as test_ext_route
|
|
import neutron.tests.unit.extensions.test_l3 as test_l3_plugin
|
|
import neutron.tests.unit.extensions.test_l3_ext_gw_mode as test_ext_gw_mode
|
|
import neutron.tests.unit.extensions.test_securitygroup as ext_sg
|
|
from neutron import version
|
|
from vmware_nsx.neutron.plugins.vmware.common import utils
|
|
from vmware_nsx.neutron.plugins.vmware.nsxlib import v3 as nsxlib
|
|
from vmware_nsx.neutron.plugins.vmware.nsxlib.v3 import dfw_api as firewall
|
|
from vmware_nsx.tests.unit import vmware
|
|
from vmware_nsx.tests.unit.vmware import nsx_v3_mocks
|
|
|
|
PLUGIN_NAME = ('vmware_nsx.neutron.plugins.vmware.'
|
|
'plugins.nsx_v3_plugin.NsxV3Plugin')
|
|
|
|
|
|
class NsxPluginV3TestCase(test_plugin.NeutronDbPluginV2TestCase):
|
|
|
|
def setUp(self,
|
|
plugin=PLUGIN_NAME,
|
|
ext_mgr=None,
|
|
service_plugins=None):
|
|
cfg.CONF.set_override('nsx_manager', '1.2.3.4', 'nsx_v3')
|
|
# Mock entire nsxlib methods as this is the best approach to perform
|
|
# white-box testing on the plugin class
|
|
# TODO(salv-orlando): supply unit tests for nsxlib.v3
|
|
nsxlib.create_logical_switch = nsx_v3_mocks.create_logical_switch
|
|
nsxlib.delete_logical_switch = mock.Mock()
|
|
nsxlib.get_logical_switch = nsx_v3_mocks.get_logical_switch
|
|
nsxlib.update_logical_switch = nsx_v3_mocks.update_logical_switch
|
|
nsxlib.create_logical_port = nsx_v3_mocks.create_logical_port
|
|
nsxlib.delete_logical_port = mock.Mock()
|
|
nsxlib.get_logical_port = nsx_v3_mocks.get_logical_port
|
|
nsxlib.update_logical_port = nsx_v3_mocks.update_logical_port
|
|
firewall.add_rules_in_section = nsx_v3_mocks.add_rules_in_section
|
|
firewall.nsclient.create_resource = nsx_v3_mocks.create_resource
|
|
firewall.nsclient.update_resource = nsx_v3_mocks.update_resource
|
|
firewall.nsclient.get_resource = nsx_v3_mocks.get_resource
|
|
firewall.nsclient.delete_resource = nsx_v3_mocks.delete_resource
|
|
|
|
super(NsxPluginV3TestCase, self).setUp(plugin=plugin,
|
|
ext_mgr=ext_mgr)
|
|
self.v3_mock = nsx_v3_mocks.NsxV3Mock()
|
|
nsxlib.get_edge_cluster = self.v3_mock.get_edge_cluster
|
|
nsxlib.get_logical_router = self.v3_mock.get_logical_router
|
|
|
|
def _create_network(self, fmt, name, admin_state_up,
|
|
arg_list=None, providernet_args=None, **kwargs):
|
|
data = {'network': {'name': name,
|
|
'admin_state_up': admin_state_up,
|
|
'tenant_id': self._tenant_id}}
|
|
# Fix to allow the router:external attribute and any other
|
|
# attributes containing a colon to be passed with
|
|
# a double underscore instead
|
|
kwargs = dict((k.replace('__', ':'), v) for k, v in kwargs.items())
|
|
if external_net.EXTERNAL in kwargs:
|
|
arg_list = (external_net.EXTERNAL, ) + (arg_list or ())
|
|
|
|
attrs = kwargs
|
|
if providernet_args:
|
|
attrs.update(providernet_args)
|
|
for arg in (('admin_state_up', 'tenant_id', 'shared') +
|
|
(arg_list or ())):
|
|
# Arg must be present and not empty
|
|
if arg in kwargs and kwargs[arg]:
|
|
data['network'][arg] = kwargs[arg]
|
|
network_req = self.new_create_request('networks', data, fmt)
|
|
if (kwargs.get('set_context') and 'tenant_id' in kwargs):
|
|
# create a specific auth context for this request
|
|
network_req.environ['neutron.context'] = context.Context(
|
|
'', kwargs['tenant_id'])
|
|
return network_req.get_response(self.api)
|
|
|
|
|
|
class TestNetworksV2(test_plugin.TestNetworksV2, NsxPluginV3TestCase):
|
|
pass
|
|
|
|
|
|
class TestPortsV2(test_plugin.TestPortsV2, NsxPluginV3TestCase):
|
|
pass
|
|
|
|
|
|
class SecurityGroupsTestCase(ext_sg.SecurityGroupDBTestCase):
|
|
|
|
def setUp(self,
|
|
plugin=PLUGIN_NAME,
|
|
ext_mgr=None):
|
|
nsxlib.create_logical_switch = nsx_v3_mocks.create_logical_switch
|
|
nsxlib.create_logical_port = nsx_v3_mocks.create_logical_port
|
|
nsxlib.update_logical_port = nsx_v3_mocks.update_logical_port
|
|
nsxlib.delete_logical_port = mock.Mock()
|
|
nsxlib.delete_logical_switch = mock.Mock()
|
|
nsxlib.get_logical_port = nsx_v3_mocks.get_logical_port
|
|
nsxlib.update_logical_port = nsx_v3_mocks.update_logical_port
|
|
firewall.add_rules_in_section = nsx_v3_mocks.add_rules_in_section
|
|
firewall.nsclient.create_resource = nsx_v3_mocks.create_resource
|
|
firewall.nsclient.update_resource = nsx_v3_mocks.update_resource
|
|
firewall.nsclient.get_resource = nsx_v3_mocks.get_resource
|
|
firewall.nsclient.delete_resource = nsx_v3_mocks.delete_resource
|
|
|
|
super(SecurityGroupsTestCase, self).setUp(plugin=PLUGIN_NAME,
|
|
ext_mgr=ext_mgr)
|
|
|
|
|
|
class TestSecurityGroups(ext_sg.TestSecurityGroups, SecurityGroupsTestCase):
|
|
pass
|
|
|
|
|
|
class DHCPOptsTestCase(test_dhcpopts.TestExtraDhcpOpt, NsxPluginV3TestCase):
|
|
|
|
def setUp(self, plugin=None):
|
|
super(test_dhcpopts.ExtraDhcpOptDBTestCase, self).setUp(
|
|
plugin=PLUGIN_NAME)
|
|
|
|
|
|
class TestL3ExtensionManager(object):
|
|
|
|
def get_resources(self):
|
|
# Simulate extension of L3 attribute map
|
|
# First apply attribute extensions
|
|
for key in l3.RESOURCE_ATTRIBUTE_MAP.keys():
|
|
l3.RESOURCE_ATTRIBUTE_MAP[key].update(
|
|
l3_ext_gw_mode.EXTENDED_ATTRIBUTES_2_0.get(key, {}))
|
|
l3.RESOURCE_ATTRIBUTE_MAP[key].update(
|
|
extraroute.EXTENDED_ATTRIBUTES_2_0.get(key, {}))
|
|
# Finally add l3 resources to the global attribute map
|
|
attributes.RESOURCE_ATTRIBUTE_MAP.update(
|
|
l3.RESOURCE_ATTRIBUTE_MAP)
|
|
return l3.L3.get_resources()
|
|
|
|
def get_actions(self):
|
|
return []
|
|
|
|
def get_request_extensions(self):
|
|
return []
|
|
|
|
|
|
def backup_l3_attribute_map():
|
|
"""Return a backup of the original l3 attribute map."""
|
|
return dict((res, attrs.copy()) for
|
|
(res, attrs) in six.iteritems(l3.RESOURCE_ATTRIBUTE_MAP))
|
|
|
|
|
|
def restore_l3_attribute_map(map_to_restore):
|
|
"""Ensure changes made by fake ext mgrs are reverted."""
|
|
l3.RESOURCE_ATTRIBUTE_MAP = map_to_restore
|
|
|
|
|
|
class L3NatTest(test_l3_plugin.L3BaseForIntTests, NsxPluginV3TestCase):
|
|
|
|
def _restore_l3_attribute_map(self):
|
|
l3.RESOURCE_ATTRIBUTE_MAP = self._l3_attribute_map_bk
|
|
|
|
def setUp(self, plugin=PLUGIN_NAME, ext_mgr=None,
|
|
service_plugins=None):
|
|
self._l3_attribute_map_bk = backup_l3_attribute_map()
|
|
cfg.CONF.set_override('api_extensions_path', vmware.NSXEXT_PATH)
|
|
cfg.CONF.set_default('max_routes', 3)
|
|
self.addCleanup(restore_l3_attribute_map, self._l3_attribute_map_bk)
|
|
ext_mgr = ext_mgr or TestL3ExtensionManager()
|
|
super(L3NatTest, self).setUp(
|
|
plugin=plugin, ext_mgr=ext_mgr, service_plugins=service_plugins)
|
|
self.plugin_instance = manager.NeutronManager.get_plugin()
|
|
self._plugin_name = "%s.%s" % (
|
|
self.plugin_instance.__module__,
|
|
self.plugin_instance.__class__.__name__)
|
|
self._plugin_class = self.plugin_instance.__class__
|
|
nsxlib.create_logical_port = self.v3_mock.create_logical_port
|
|
nsxlib.create_logical_router = self.v3_mock.create_logical_router
|
|
nsxlib.update_logical_router = self.v3_mock.update_logical_router
|
|
nsxlib.delete_logical_router = self.v3_mock.delete_logical_router
|
|
nsxlib.get_logical_router_port_by_ls_id = (
|
|
self.v3_mock.get_logical_router_port_by_ls_id)
|
|
nsxlib.create_logical_router_port = (
|
|
self.v3_mock.create_logical_router_port)
|
|
nsxlib.update_logical_router_port = (
|
|
self.v3_mock.update_logical_router_port)
|
|
nsxlib.delete_logical_router_port = (
|
|
self.v3_mock.delete_logical_router_port)
|
|
nsxlib.add_nat_rule = self.v3_mock.add_nat_rule
|
|
nsxlib.delete_nat_rule = self.v3_mock.delete_nat_rule
|
|
nsxlib.delete_nat_rule_by_values = (
|
|
self.v3_mock.delete_nat_rule_by_values)
|
|
nsxlib.get_logical_router_ports_by_router_id = (
|
|
self.v3_mock.get_logical_router_ports_by_router_id)
|
|
nsxlib.update_logical_router_advertisement = (
|
|
self.v3_mock.update_logical_router_advertisement)
|
|
|
|
def _create_l3_ext_network(
|
|
self, physical_network=nsx_v3_mocks.DEFAULT_TIER0_ROUTER_UUID):
|
|
name = 'l3_ext_net'
|
|
net_type = utils.NetworkTypes.L3_EXT
|
|
providernet_args = {pnet.NETWORK_TYPE: net_type,
|
|
pnet.PHYSICAL_NETWORK: physical_network}
|
|
return self.network(name=name,
|
|
router__external=True,
|
|
providernet_args=providernet_args,
|
|
arg_list=(pnet.NETWORK_TYPE,
|
|
pnet.PHYSICAL_NETWORK))
|
|
|
|
|
|
class TestL3NatTestCase(L3NatTest,
|
|
test_l3_plugin.L3NatDBIntTestCase,
|
|
NsxPluginV3TestCase,
|
|
test_ext_route.ExtraRouteDBTestCaseBase):
|
|
|
|
def _test_create_l3_ext_network(
|
|
self, physical_network=nsx_v3_mocks.DEFAULT_TIER0_ROUTER_UUID):
|
|
name = 'l3_ext_net'
|
|
net_type = utils.NetworkTypes.L3_EXT
|
|
expected = [('subnets', []), ('name', name), ('admin_state_up', True),
|
|
('status', 'ACTIVE'), ('shared', False),
|
|
(external_net.EXTERNAL, True),
|
|
(pnet.NETWORK_TYPE, net_type),
|
|
(pnet.PHYSICAL_NETWORK, physical_network)]
|
|
with self._create_l3_ext_network(physical_network) as net:
|
|
for k, v in expected:
|
|
self.assertEqual(net['network'][k], v)
|
|
|
|
def test_create_l3_ext_network_with_default_tier0(self):
|
|
self._test_create_l3_ext_network()
|
|
|
|
def test_floatingip_with_invalid_create_port(self):
|
|
self._test_floatingip_with_invalid_create_port(self._plugin_name)
|
|
|
|
def test_routes_update_for_multiple_routers(self):
|
|
self.skipTest('not supported')
|
|
|
|
def test_floatingip_multi_external_one_internal(self):
|
|
self.skipTest('not supported')
|
|
|
|
def test_multiple_subnets_on_different_routers(self):
|
|
with self.network() as network:
|
|
with self.subnet(network=network) as s1,\
|
|
self.subnet(network=network,
|
|
cidr='11.0.0.0/24') as s2,\
|
|
self.router() as r1,\
|
|
self.router() as r2:
|
|
self._router_interface_action('add', r1['router']['id'],
|
|
s1['subnet']['id'], None)
|
|
self.assertRaises(n_exc.InvalidInput,
|
|
self.plugin_instance.add_router_interface,
|
|
context.get_admin_context(),
|
|
r2['router']['id'],
|
|
{'subnet_id': s2['subnet']['id']})
|
|
self._router_interface_action('remove', r1['router']['id'],
|
|
s1['subnet']['id'], None)
|
|
self._router_interface_action('add', r2['router']['id'],
|
|
s2['subnet']['id'], None)
|
|
self._router_interface_action('remove', r2['router']['id'],
|
|
s2['subnet']['id'], None)
|
|
|
|
|
|
class ExtGwModeTestCase(L3NatTest,
|
|
test_ext_gw_mode.ExtGwModeIntTestCase):
|
|
pass
|
|
|
|
|
|
class TestNsxV3Utils(NsxPluginV3TestCase):
|
|
|
|
def test_build_v3_tags_payload(self):
|
|
result = utils.build_v3_tags_payload(
|
|
{'id': 'fake_id',
|
|
'tenant_id': 'fake_tenant_id'})
|
|
expected = [{'scope': 'neutron-id', 'tag': 'fake_id'},
|
|
{'scope': 'os-tid', 'tag': 'fake_tenant_id'},
|
|
{'scope': 'os-api-version',
|
|
'tag': version.version_info.release_string()}]
|
|
self.assertEqual(expected, result)
|