shade/shade/tests/unit/test_subnet.py

375 lines
15 KiB
Python

# Copyright 2017 OVH SAS
# 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 testtools
from shade import exc
from shade.tests.unit import base
class TestSubnet(base.RequestsMockTestCase):
network_name = 'network_name'
subnet_name = 'subnet_name'
subnet_id = '1f1696eb-7f47-47f6-835c-4889bff88604'
subnet_cidr = '192.168.199.0/24'
mock_network_rep = {
'id': '881d1bb7-a663-44c0-8f9f-ee2765b74486',
'name': network_name,
}
mock_subnet_rep = {
'allocation_pools': [{
'start': u'192.168.199.2',
'end': u'192.168.199.254'
}],
'cidr': subnet_cidr,
'created_at': '2017-04-24T20:22:23Z',
'description': '',
'dns_nameservers': [],
'enable_dhcp': False,
'gateway_ip': '192.168.199.1',
'host_routes': [],
'id': subnet_id,
'ip_version': 4,
'ipv6_address_mode': None,
'ipv6_ra_mode': None,
'name': subnet_name,
'network_id': mock_network_rep['id'],
'project_id': '861808a93da0484ea1767967c4df8a23',
'revision_number': 2,
'service_types': [],
'subnetpool_id': None,
'tags': []
}
def test_get_subnet(self):
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [self.mock_subnet_rep]})
])
r = self.cloud.get_subnet(self.subnet_name)
self.assertIsNotNone(r)
self.assertDictEqual(self.mock_subnet_rep, r)
self.assert_calls()
def test_create_subnet(self):
pool = [{'start': '192.168.199.2', 'end': '192.168.199.254'}]
dns = ['8.8.8.8']
routes = [{"destination": "0.0.0.0/0", "nexthop": "123.456.78.9"}]
mock_subnet_rep = copy.copy(self.mock_subnet_rep)
mock_subnet_rep['allocation_pools'] = pool
mock_subnet_rep['dns_nameservers'] = dns
mock_subnet_rep['host_routes'] = routes
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]}),
dict(method='POST',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnet': mock_subnet_rep},
validate=dict(
json={'subnet': {
'cidr': self.subnet_cidr,
'enable_dhcp': False,
'ip_version': 4,
'network_id': self.mock_network_rep['id'],
'allocation_pools': pool,
'dns_nameservers': dns,
'host_routes': routes}}))
])
subnet = self.cloud.create_subnet(self.network_name, self.subnet_cidr,
allocation_pools=pool,
dns_nameservers=dns,
host_routes=routes)
self.assertDictEqual(mock_subnet_rep, subnet)
self.assert_calls()
def test_create_subnet_string_ip_version(self):
'''Allow ip_version as a string'''
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]}),
dict(method='POST',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnet': self.mock_subnet_rep},
validate=dict(
json={'subnet': {
'cidr': self.subnet_cidr,
'enable_dhcp': False,
'ip_version': 4,
'network_id': self.mock_network_rep['id']}}))
])
subnet = self.cloud.create_subnet(
self.network_name, self.subnet_cidr, ip_version='4')
self.assertDictEqual(self.mock_subnet_rep, subnet)
self.assert_calls()
def test_create_subnet_bad_ip_version(self):
'''String ip_versions must be convertable to int'''
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]})
])
with testtools.ExpectedException(
exc.OpenStackCloudException,
"ip_version must be an integer"
):
self.cloud.create_subnet(
self.network_name, self.subnet_cidr, ip_version='4x')
self.assert_calls()
def test_create_subnet_without_gateway_ip(self):
pool = [{'start': '192.168.199.2', 'end': '192.168.199.254'}]
dns = ['8.8.8.8']
mock_subnet_rep = copy.copy(self.mock_subnet_rep)
mock_subnet_rep['allocation_pools'] = pool
mock_subnet_rep['dns_nameservers'] = dns
mock_subnet_rep['gateway_ip'] = None
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]}),
dict(method='POST',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnet': mock_subnet_rep},
validate=dict(
json={'subnet': {
'cidr': self.subnet_cidr,
'enable_dhcp': False,
'ip_version': 4,
'network_id': self.mock_network_rep['id'],
'allocation_pools': pool,
'gateway_ip': None,
'dns_nameservers': dns}}))
])
subnet = self.cloud.create_subnet(self.network_name, self.subnet_cidr,
allocation_pools=pool,
dns_nameservers=dns,
disable_gateway_ip=True)
self.assertDictEqual(mock_subnet_rep, subnet)
self.assert_calls()
def test_create_subnet_with_gateway_ip(self):
pool = [{'start': '192.168.199.8', 'end': '192.168.199.254'}]
gateway = '192.168.199.2'
dns = ['8.8.8.8']
mock_subnet_rep = copy.copy(self.mock_subnet_rep)
mock_subnet_rep['allocation_pools'] = pool
mock_subnet_rep['dns_nameservers'] = dns
mock_subnet_rep['gateway_ip'] = gateway
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]}),
dict(method='POST',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnet': mock_subnet_rep},
validate=dict(
json={'subnet': {
'cidr': self.subnet_cidr,
'enable_dhcp': False,
'ip_version': 4,
'network_id': self.mock_network_rep['id'],
'allocation_pools': pool,
'gateway_ip': gateway,
'dns_nameservers': dns}}))
])
subnet = self.cloud.create_subnet(self.network_name, self.subnet_cidr,
allocation_pools=pool,
dns_nameservers=dns,
gateway_ip=gateway)
self.assertDictEqual(mock_subnet_rep, subnet)
self.assert_calls()
def test_create_subnet_conflict_gw_ops(self):
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]})
])
gateway = '192.168.200.3'
self.assertRaises(exc.OpenStackCloudException,
self.cloud.create_subnet, 'kooky',
self.subnet_cidr, gateway_ip=gateway,
disable_gateway_ip=True)
self.assert_calls()
def test_create_subnet_bad_network(self):
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [self.mock_network_rep]})
])
self.assertRaises(exc.OpenStackCloudException,
self.cloud.create_subnet,
'duck', self.subnet_cidr)
self.assert_calls()
def test_create_subnet_non_unique_network(self):
net1 = dict(id='123', name=self.network_name)
net2 = dict(id='456', name=self.network_name)
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'networks.json']),
json={'networks': [net1, net2]})
])
self.assertRaises(exc.OpenStackCloudException,
self.cloud.create_subnet,
self.network_name, self.subnet_cidr)
self.assert_calls()
def test_delete_subnet(self):
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [self.mock_subnet_rep]}),
dict(method='DELETE',
uri=self.get_mock_url(
'network', 'public',
append=['v2.0', 'subnets', '%s.json' % self.subnet_id]),
json={})
])
self.assertTrue(self.cloud.delete_subnet(self.subnet_name))
self.assert_calls()
def test_delete_subnet_not_found(self):
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': []})
])
self.assertFalse(self.cloud.delete_subnet('goofy'))
self.assert_calls()
def test_delete_subnet_multiple_found(self):
subnet1 = dict(id='123', name=self.subnet_name)
subnet2 = dict(id='456', name=self.subnet_name)
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [subnet1, subnet2]})
])
self.assertRaises(exc.OpenStackCloudException,
self.cloud.delete_subnet,
self.subnet_name)
self.assert_calls()
def test_delete_subnet_multiple_using_id(self):
subnet1 = dict(id='123', name=self.subnet_name)
subnet2 = dict(id='456', name=self.subnet_name)
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [subnet1, subnet2]}),
dict(method='DELETE',
uri=self.get_mock_url(
'network', 'public',
append=['v2.0', 'subnets', '%s.json' % subnet1['id']]),
json={})
])
self.assertTrue(self.cloud.delete_subnet(subnet1['id']))
self.assert_calls()
def test_update_subnet(self):
expected_subnet = copy.copy(self.mock_subnet_rep)
expected_subnet['name'] = 'goofy'
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [self.mock_subnet_rep]}),
dict(method='PUT',
uri=self.get_mock_url(
'network', 'public',
append=['v2.0', 'subnets', '%s.json' % self.subnet_id]),
json={'subnet': expected_subnet},
validate=dict(
json={'subnet': {'name': 'goofy'}}))
])
subnet = self.cloud.update_subnet(self.subnet_id, subnet_name='goofy')
self.assertDictEqual(expected_subnet, subnet)
self.assert_calls()
def test_update_subnet_gateway_ip(self):
expected_subnet = copy.copy(self.mock_subnet_rep)
gateway = '192.168.199.3'
expected_subnet['gateway_ip'] = gateway
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [self.mock_subnet_rep]}),
dict(method='PUT',
uri=self.get_mock_url(
'network', 'public',
append=['v2.0', 'subnets', '%s.json' % self.subnet_id]),
json={'subnet': expected_subnet},
validate=dict(
json={'subnet': {'gateway_ip': gateway}}))
])
subnet = self.cloud.update_subnet(self.subnet_id, gateway_ip=gateway)
self.assertDictEqual(expected_subnet, subnet)
self.assert_calls()
def test_update_subnet_disable_gateway_ip(self):
expected_subnet = copy.copy(self.mock_subnet_rep)
expected_subnet['gateway_ip'] = None
self.register_uris([
dict(method='GET',
uri=self.get_mock_url(
'network', 'public', append=['v2.0', 'subnets.json']),
json={'subnets': [self.mock_subnet_rep]}),
dict(method='PUT',
uri=self.get_mock_url(
'network', 'public',
append=['v2.0', 'subnets', '%s.json' % self.subnet_id]),
json={'subnet': expected_subnet},
validate=dict(
json={'subnet': {'gateway_ip': None}}))
])
subnet = self.cloud.update_subnet(self.subnet_id,
disable_gateway_ip=True)
self.assertDictEqual(expected_subnet, subnet)
self.assert_calls()
def test_update_subnet_conflict_gw_ops(self):
self.assertRaises(exc.OpenStackCloudException,
self.cloud.update_subnet,
self.subnet_id, gateway_ip="192.168.199.3",
disable_gateway_ip=True)