A common library that interfaces with VMware NSX.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

162 lines
7.3 KiB

# Copyright (c) 2017 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.
from vmware_nsxlib.tests.unit.v3 import nsxlib_testcase
from vmware_nsxlib.v3 import native_dhcp
class TestNativeDhcp(nsxlib_testcase.NsxLibTestCase):
"""Tests for vmware_nsxlib.v3.native_dhcp.NsxLibNativeDhcp."""
def setUp(self, *args, **kwargs):
super(TestNativeDhcp, self).setUp(with_mocks=False)
self.handler = native_dhcp.NsxLibNativeDhcp(
self.nsxlib.client,
nsxlib_testcase.get_default_nsxlib_config())
self.net_dns_domain = 'a.com'
self.subnet_dns_nameserver = '1.1.1.1'
self.default_dns_domain = 'b.com'
self.default_dns_nameserver = '2.2.2.2'
def _get_server_config(self, with_net_dns=True, with_default_dns=True,
tags=None, gateway_ip='2.2.2.2', cidr='5.5.0.0/24',
port_ip='5.5.0.1', net_name='dummy',
net_id='dummy_uuid'):
name = self.handler.build_server_name(net_name, net_id)
if not tags:
tags = []
dns_domain = None
dns_nameservers = None
if with_default_dns:
dns_domain = self.default_dns_domain
dns_nameservers = [self.default_dns_nameserver]
if with_net_dns:
dns_domain = self.net_dns_domain
dns_nameservers = [self.subnet_dns_nameserver]
return self.handler.build_server(name, port_ip, cidr, gateway_ip,
dns_domain, dns_nameservers,
tags=tags)
def test_build_server_config_dns_from_net_no_defaults(self):
# Verify that net/subnet dns params are used if exist
result = self._get_server_config(with_net_dns=True,
with_default_dns=False)
self.assertEqual(self.net_dns_domain, result['domain_name'])
self.assertEqual([self.subnet_dns_nameserver],
result['dns_nameservers'])
def test_build_server_config_dns_from_net_with_defaults(self):
# Verify that net/subnet dns params are used if exist, even if there
# are defaults
result = self._get_server_config(with_net_dns=True,
with_default_dns=True)
self.assertEqual(self.net_dns_domain, result['domain_name'])
self.assertEqual([self.subnet_dns_nameserver],
result['dns_nameservers'])
def test_build_server_config_dns_from_defaults(self):
# Verify that default dns params are used if net/subnet dns params
# are missing
result = self._get_server_config(with_net_dns=False,
with_default_dns=True)
self.assertEqual(self.default_dns_domain, result['domain_name'])
self.assertEqual([self.default_dns_nameserver],
result['dns_nameservers'])
def test_build_server_config_dns_from_config(self):
# Verify that config dns params are used if net/subnet and default
# dns params are missing
result = self._get_server_config(with_net_dns=False,
with_default_dns=False)
self.assertEqual(nsxlib_testcase.DNS_DOMAIN, result['domain_name'])
self.assertEqual(nsxlib_testcase.DNS_NAMESERVERS,
result['dns_nameservers'])
def test_build_server_config_with_tags(self):
tags = [{'scope': 'a', 'value': 'a'}]
result = self._get_server_config(tags=tags)
self.assertEqual(tags, result['tags'])
def test_build_server_config_with_gateway(self):
gw_ip = '10.10.10.10'
result = self._get_server_config(gateway_ip=gw_ip)
self.assertEqual(gw_ip, result['gateway_ip'])
def test_build_server_config_with_server_ip(self):
result = self._get_server_config(cidr='7.7.7.0/24', port_ip='7.7.7.14')
self.assertEqual('7.7.7.14/24', result['server_ip'])
def test_build_server_config_with_name(self):
net_name = 'net1'
net_id = 'uuid1uuid2'
result = self._get_server_config(net_name=net_name, net_id=net_id)
self.assertEqual('%s_%s...%s' % (net_name, net_id[:5], net_id[-5:]),
result['name'])
def test_build_server_config_no_name(self):
net_id = 'uuid1uuid2'
result = self._get_server_config(net_name=None, net_id=net_id)
self.assertEqual('dhcpserver_%s...%s' % (net_id[:5], net_id[-5:]),
result['name'])
def test_build_static_routes(self):
gateway_ip = '2.2.2.2'
cidr = '5.5.0.0/24'
host_routes = [{'nexthop': '81.0.200.254',
'destination': '91.255.255.0/24'}]
static_routes, gateway_ip = self.handler.build_static_routes(
gateway_ip, cidr, host_routes)
expected = [{'network': '5.5.0.0/24', 'next_hop': '0.0.0.0'},
{'network': '91.255.255.0/24', 'next_hop': '81.0.200.254'},
{'network': '0.0.0.0/0', 'next_hop': '2.2.2.2'}]
self.assertEqual(expected, static_routes)
self.assertEqual('2.2.2.2', gateway_ip)
def test_build_static_routes_gw_none(self):
gateway_ip = None
cidr = '5.5.0.0/24'
host_routes = [{'nexthop': '81.0.200.254',
'destination': '91.255.255.0/24'}]
static_routes, gateway_ip = self.handler.build_static_routes(
gateway_ip, cidr, host_routes)
expected = [{'network': '5.5.0.0/24', 'next_hop': '0.0.0.0'},
{'network': '91.255.255.0/24', 'next_hop': '81.0.200.254'}]
self.assertEqual(expected, static_routes)
self.assertIsNone(gateway_ip)
def test_build_static_routes_no_host_routes(self):
gateway_ip = '2.2.2.2'
cidr = '5.5.0.0/24'
host_routes = []
static_routes, gateway_ip = self.handler.build_static_routes(
gateway_ip, cidr, host_routes)
expected = [{'network': '5.5.0.0/24', 'next_hop': '0.0.0.0'},
{'network': '0.0.0.0/0', 'next_hop': '2.2.2.2'}]
self.assertEqual(expected, static_routes)
self.assertEqual('2.2.2.2', gateway_ip)
def test_build_static_routes_gw_none_host_route_any(self):
gateway_ip = None
cidr = '5.5.0.0/24'
host_routes = [{'nexthop': '81.0.200.254',
'destination': '0.0.0.0/0'}]
static_routes, gateway_ip = self.handler.build_static_routes(
gateway_ip, cidr, host_routes)
expected = [{'network': '5.5.0.0/24', 'next_hop': '0.0.0.0'},
{'network': '0.0.0.0/0', 'next_hop': '81.0.200.254'}]
self.assertEqual(expected, static_routes)
self.assertEqual('81.0.200.254', gateway_ip)