187 lines
8.6 KiB
Python
187 lines
8.6 KiB
Python
# Copyright 2018 VMware, Inc.
|
|
# All Rights Reserved
|
|
#
|
|
# 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 copy
|
|
|
|
import mock
|
|
|
|
from oslo_utils import uuidutils
|
|
|
|
from vmware_nsxlib.tests.unit.v3 import nsxlib_testcase
|
|
from vmware_nsxlib.tests.unit.v3 import test_constants
|
|
from vmware_nsxlib.v3 import exceptions as nsxlib_exc
|
|
from vmware_nsxlib.v3 import nsx_constants
|
|
|
|
|
|
class TestRouter(nsxlib_testcase.NsxClientTestCase):
|
|
|
|
def test_validate_tier0(self):
|
|
tier0_groups_dict = {}
|
|
tier0_uuid = uuidutils.generate_uuid()
|
|
rtr = {'edge_cluster_id': test_constants.FAKE_EDGE_CLUSTER_ID}
|
|
with mock.patch.object(self.nsxlib.router._router_client, 'get',
|
|
return_value=rtr),\
|
|
mock.patch.object(
|
|
self.nsxlib.edge_cluster, 'get',
|
|
return_value=test_constants.FAKE_EDGE_CLUSTER):
|
|
self.nsxlib.router.validate_tier0(tier0_groups_dict, tier0_uuid)
|
|
self.assertEqual(
|
|
tier0_groups_dict[tier0_uuid]['edge_cluster_uuid'],
|
|
test_constants.FAKE_EDGE_CLUSTER_ID)
|
|
self.assertEqual(
|
|
tier0_groups_dict[tier0_uuid]['member_index_list'], [0])
|
|
|
|
def test_validate_tier0_fail(self):
|
|
tier0_groups_dict = {}
|
|
tier0_uuid = uuidutils.generate_uuid()
|
|
edge_cluster = copy.copy(test_constants.FAKE_EDGE_CLUSTER)
|
|
edge_cluster['members'] = []
|
|
with mock.patch.object(self.nsxlib.router._router_client, 'get'),\
|
|
mock.patch.object(self.nsxlib.edge_cluster, 'get',
|
|
return_value=edge_cluster):
|
|
self.assertRaises(
|
|
nsxlib_exc.NsxLibInvalidInput,
|
|
self.nsxlib.router.validate_tier0,
|
|
tier0_groups_dict, tier0_uuid)
|
|
|
|
def test_add_router_link_port(self):
|
|
tags = [{'scope': 'a', 'tag': 'b'}]
|
|
tier0_uuid = uuidutils.generate_uuid()
|
|
tier1_uuid = uuidutils.generate_uuid()
|
|
with mock.patch.object(self.nsxlib.router._router_port_client,
|
|
'create') as port_create:
|
|
tier0_link_port = mock.MagicMock()
|
|
tier1_link_port = mock.MagicMock()
|
|
port_create.side_effect = [tier0_link_port, tier1_link_port]
|
|
self.assertEqual(
|
|
(tier0_link_port, tier1_link_port),
|
|
self.nsxlib.router.add_router_link_port(
|
|
tier1_uuid, tier0_uuid, tags))
|
|
self.assertEqual(port_create.call_count, 2)
|
|
|
|
def test_remove_router_link_port(self):
|
|
tier1_uuid = uuidutils.generate_uuid()
|
|
with mock.patch.object(
|
|
self.nsxlib.router._router_port_client, 'get_tier1_link_port',
|
|
return_value=test_constants.FAKE_ROUTER_LINKT1_PORT) as port_get,\
|
|
mock.patch.object(self.nsxlib.router._router_port_client,
|
|
'delete') as port_delete:
|
|
self.nsxlib.router.remove_router_link_port(tier1_uuid)
|
|
self.assertEqual(port_get.call_count, 1)
|
|
self.assertEqual(port_delete.call_count, 2)
|
|
|
|
def test_add_centralized_service_port(self):
|
|
logical_router_id = uuidutils.generate_uuid()
|
|
logical_port_id = uuidutils.generate_uuid()
|
|
display_name = mock.Mock()
|
|
tags = mock.Mock()
|
|
address_groups = mock.Mock()
|
|
port = mock.Mock()
|
|
with mock.patch.object(
|
|
self.nsxlib.router._router_port_client, 'create',
|
|
return_value=port) as create_port:
|
|
csp = self.nsxlib.router.add_centralized_service_port(
|
|
logical_router_id, display_name=display_name, tags=tags,
|
|
logical_port_id=logical_port_id, address_groups=address_groups)
|
|
create_port.assert_called_once_with(
|
|
logical_router_id, display_name=display_name, tags=tags,
|
|
logical_port_id=logical_port_id, address_groups=address_groups,
|
|
resource_type=nsx_constants.LROUTERPORT_CENTRALIZED)
|
|
self.assertEqual(csp, port)
|
|
|
|
def test_create_logical_router_intf_port_by_ls_id(self):
|
|
logical_router_id = uuidutils.generate_uuid()
|
|
display_name = 'dummy'
|
|
tags = []
|
|
ls_id = uuidutils.generate_uuid()
|
|
logical_switch_port_id = uuidutils.generate_uuid()
|
|
address_groups = []
|
|
with mock.patch.object(
|
|
self.nsxlib.router._router_port_client,
|
|
"get_by_lswitch_id",
|
|
side_effect=nsxlib_exc.ResourceNotFound()) as get_port,\
|
|
mock.patch.object(self.nsxlib.router._router_port_client,
|
|
"create") as create_port:
|
|
self.nsxlib.router.create_logical_router_intf_port_by_ls_id(
|
|
logical_router_id,
|
|
display_name,
|
|
tags,
|
|
ls_id,
|
|
logical_switch_port_id,
|
|
address_groups)
|
|
get_port.assert_called_once_with(ls_id)
|
|
create_port.assert_called_once_with(
|
|
logical_router_id, display_name, tags,
|
|
nsx_constants.LROUTERPORT_DOWNLINK,
|
|
logical_switch_port_id, address_groups, urpf_mode=None,
|
|
relay_service_uuid=None)
|
|
|
|
def test_add_fip_nat_rules(self):
|
|
with mock.patch.object(self.nsxlib.logical_router,
|
|
"add_nat_rule") as add_rule:
|
|
self.nsxlib.router.add_fip_nat_rules(
|
|
test_constants.FAKE_ROUTER_UUID,
|
|
'1.1.1.1', '2.2.2.2')
|
|
self.assertEqual(add_rule.call_count, 2)
|
|
|
|
def test_get_tier0_router_tz(self):
|
|
tier0_uuid = uuidutils.generate_uuid()
|
|
self.nsxlib.feature_supported = mock.MagicMock()
|
|
self.nsxlib.feature_supported.return_value = False
|
|
with mock.patch.object(self.nsxlib.router._router_client, 'get',
|
|
return_value=test_constants.FAKE_TIERO_ROUTER),\
|
|
mock.patch.object(self.nsxlib.edge_cluster, 'get',
|
|
return_value=test_constants.FAKE_EDGE_CLUSTER),\
|
|
mock.patch.object(self.nsxlib.transport_node, 'get',
|
|
return_value=test_constants.FAKE_TRANS_NODE):
|
|
tzs = self.nsxlib.router.get_tier0_router_tz(tier0_uuid)
|
|
self.assertEqual(tzs, [test_constants.FAKE_TZ_UUID])
|
|
|
|
def test_get_tier0_router_overlay_tz(self):
|
|
tier0_uuid = uuidutils.generate_uuid()
|
|
self.nsxlib.feature_supported = mock.MagicMock()
|
|
self.nsxlib.feature_supported.return_value = False
|
|
with mock.patch.object(self.nsxlib.router._router_client, 'get',
|
|
return_value=test_constants.FAKE_TIERO_ROUTER),\
|
|
mock.patch.object(self.nsxlib.edge_cluster, 'get',
|
|
return_value=test_constants.FAKE_EDGE_CLUSTER),\
|
|
mock.patch.object(self.nsxlib.transport_node, 'get',
|
|
return_value=test_constants.FAKE_TRANS_NODE),\
|
|
mock.patch.object(self.nsxlib.transport_zone, 'get_transport_type',
|
|
return_value="OVERLAY"):
|
|
tz = self.nsxlib.router.get_tier0_router_overlay_tz(tier0_uuid)
|
|
self.assertEqual(tz, test_constants.FAKE_TZ_UUID)
|
|
|
|
def test_get_tier0_router_overlay_tz_via_advanced_config(self):
|
|
tier0_uuid = uuidutils.generate_uuid()
|
|
with mock.patch.object(self.nsxlib.router._router_client, 'get',
|
|
return_value=test_constants.FAKE_TIERO_ROUTER):
|
|
tz = self.nsxlib.router.get_tier0_router_overlay_tz(tier0_uuid)
|
|
self.assertEqual(tz, test_constants.FAKE_TZ_UUID)
|
|
|
|
def test_get_connected_t0_transit_net(self):
|
|
t1_uuid = uuidutils.generate_uuid()
|
|
transit_net = '1.1.1.0'
|
|
link_port = {
|
|
'subnets': [{'ip_addresses': [transit_net],
|
|
'prefix_length': '31'}]}
|
|
with mock.patch.object(self.nsxlib.router._router_port_client,
|
|
'get_tier1_link_port',
|
|
return_value=link_port) as get_port:
|
|
net = self.nsxlib.router.get_connected_t0_transit_net(t1_uuid)
|
|
get_port.assert_called_with(t1_uuid)
|
|
self.assertEqual('%s' % (transit_net), net)
|