snaps/domain/test/network_tests.py

   1 # Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
   2 #                    and others.  All rights reserved.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at:
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 import unittest
  17 from snaps.domain.network import (
  18     Port, SecurityGroup, SecurityGroupRule, Router, InterfaceRouter, Network,
  19     Subnet)
  20
  21
  22 class NetworkObjectTests(unittest.TestCase):
  23     """
  24     Tests the construction of the snaps.domain.network.Network class
  25     """
  26
  27     def test_construction_kwargs_1(self):
  28         network = Network(
  29             **{'name': 'foo', 'id': 'bar', 'provider:network_type': 'flat',
  30                'admin_state_up': False, 'shared': True,
  31                'router:external': False})
  32         self.assertEqual('foo', network.name)
  33         self.assertEqual('bar', network.id)
  34         self.assertEqual('flat', network.type)
  35         self.assertFalse(network.admin_state_up)
  36         self.assertFalse(network.external)
  37         self.assertTrue(network.shared)
  38
  39     def test_construction_kwargs_2(self):
  40         network = Network(
  41             **{'name': 'foo', 'id': 'bar', 'type': 'flat',
  42                'admin_state_up': False, 'shared': True,
  43                'external': False})
  44         self.assertEqual('foo', network.name)
  45         self.assertEqual('bar', network.id)
  46         self.assertEqual('flat', network.type)
  47         self.assertFalse(network.admin_state_up)
  48         self.assertFalse(network.external)
  49         self.assertTrue(network.shared)
  50
  51     def test_construction_named(self):
  52         network = Network(
  53             name='foo', id='bar', type='flat', admin_state_up=False,
  54             shared=True, external=False)
  55         self.assertEqual('foo', network.name)
  56         self.assertEqual('bar', network.id)
  57         self.assertEqual('flat', network.type)
  58         self.assertFalse(network.admin_state_up)
  59         self.assertFalse(network.external)
  60         self.assertTrue(network.shared)
  61
  62
  63 class SubnetObjectTests(unittest.TestCase):
  64     """
  65     Tests the construction of the snaps.domain.network.Subnet class
  66     """
  67
  68     def test_construction_kwargs(self):
  69         subnet = Subnet(
  70             **{'name': 'foo', 'id': 'bar', 'cidr': '10.0.0.0/24',
  71                'ip_version': 4, 'gateway_ip': '10.0.0.1', 'enable_dhcp': True,
  72                'dns_nameservers': ['8.8.8.8'], 'host_routes': list(),
  73                'ipv6_ra_mode': 'hello', 'ipv6_address_mode': 'world'})
  74         self.assertEqual('foo', subnet.name)
  75         self.assertEqual('bar', subnet.id)
  76         self.assertEqual('10.0.0.0/24', subnet.cidr)
  77         self.assertEqual(4, subnet.ip_version)
  78         self.assertEqual('10.0.0.1', subnet.gateway_ip)
  79         self.assertTrue(subnet.enable_dhcp)
  80         self.assertEqual(1, len(subnet.dns_nameservers))
  81         self.assertEqual('8.8.8.8', subnet.dns_nameservers[0])
  82         self.assertEqual(list(), subnet.host_routes)
  83         self.assertEqual('hello', subnet.ipv6_ra_mode)
  84         self.assertEqual('world', subnet.ipv6_address_mode)
  85
  86     def test_construction_named(self):
  87         subnet = Subnet(
  88             name='foo', id='bar', cidr='10.0.0.0/24',
  89             ip_version=4, gateway_ip='10.0.0.1', enable_dhcp=True,
  90             dns_nameservers=['8.8.8.8'], host_routes=list(),
  91             ipv6_ra_mode='hello', ipv6_address_mode='world')
  92         self.assertEqual('foo', subnet.name)
  93         self.assertEqual('bar', subnet.id)
  94         self.assertEqual('10.0.0.0/24', subnet.cidr)
  95         self.assertEqual(4, subnet.ip_version)
  96         self.assertEqual('10.0.0.1', subnet.gateway_ip)
  97         self.assertTrue(subnet.enable_dhcp)
  98         self.assertEqual(1, len(subnet.dns_nameservers))
  99         self.assertEqual('8.8.8.8', subnet.dns_nameservers[0])
 100         self.assertEqual(list(), subnet.host_routes)
 101         self.assertEqual('hello', subnet.ipv6_ra_mode)
 102         self.assertEqual('world', subnet.ipv6_address_mode)
 103
 104
 105 class PortDomainObjectTests(unittest.TestCase):
 106     """
 107     Tests the construction of the snaps.domain.network.Port class
 108     """
 109
 110     def test_construction_ips_kwargs(self):
 111         ips = ['10', '11']
 112         port = Port(
 113             **{'name': 'foo', 'id': 'bar', 'description': 'test desc',
 114                'ips': ips, 'mac_address': 'abc123',
 115                'allowed_address_pairs': list(), 'admin_state_up': False,
 116                'device_id': 'def456', 'device_owner': 'joe',
 117                'network_id': 'ghi789', 'port_security_enabled': False,
 118                'security_groups': list(), 'tenant_id': 'jkl098'})
 119         self.assertEqual('foo', port.name)
 120         self.assertEqual('bar', port.id)
 121         self.assertEqual('test desc', port.description)
 122         self.assertEqual(ips, port.ips)
 123         self.assertEqual('abc123', port.mac_address)
 124         self.assertEqual(list(), port.allowed_address_pairs)
 125         self.assertFalse(port.admin_state_up)
 126         self.assertEqual('def456', port.device_id)
 127         self.assertEqual('joe', port.device_owner)
 128         self.assertEqual('ghi789', port.network_id)
 129         self.assertFalse(port.port_security_enabled)
 130         self.assertEqual(list(), port.security_groups)
 131         self.assertEqual(list(), port.security_groups)
 132
 133     def test_construction_fixed_ips_kwargs(self):
 134         ips = ['10', '11']
 135         port = Port(
 136             **{'name': 'foo', 'id': 'bar', 'description': 'test desc',
 137                'fixed_ips': ips, 'mac_address': 'abc123',
 138                'allowed_address_pairs': list(), 'admin_state_up': False,
 139                'device_id': 'def456', 'device_owner': 'joe',
 140                'network_id': 'ghi789', 'port_security_enabled': False,
 141                'security_groups': list(), 'tenant_id': 'jkl098'})
 142         self.assertEqual('foo', port.name)
 143         self.assertEqual('bar', port.id)
 144         self.assertEqual('test desc', port.description)
 145         self.assertEqual(ips, port.ips)
 146         self.assertEqual('abc123', port.mac_address)
 147         self.assertEqual(list(), port.allowed_address_pairs)
 148         self.assertFalse(port.admin_state_up)
 149         self.assertEqual('def456', port.device_id)
 150         self.assertEqual('joe', port.device_owner)
 151         self.assertEqual('ghi789', port.network_id)
 152         self.assertFalse(port.port_security_enabled)
 153         self.assertEqual(list(), port.security_groups)
 154         self.assertEqual(list(), port.security_groups)
 155
 156     def test_construction_named_ips(self):
 157         ips = ['10', '11']
 158         port = Port(
 159             mac_address='abc123', description='test desc', ips=ips, id='bar',
 160             name='foo', allowed_address_pairs=list(), admin_state_up=False,
 161             device_id='def456', device_owner='joe', network_id='ghi789',
 162             port_security_enabled=False, security_groups=list(),
 163             tenant_id='jkl098')
 164         self.assertEqual('foo', port.name)
 165         self.assertEqual('bar', port.id)
 166         self.assertEqual('test desc', port.description)
 167         self.assertEqual(ips, port.ips)
 168         self.assertEqual('abc123', port.mac_address)
 169         self.assertEqual(list(), port.allowed_address_pairs)
 170         self.assertFalse(port.admin_state_up)
 171         self.assertEqual('def456', port.device_id)
 172         self.assertEqual('joe', port.device_owner)
 173         self.assertEqual('ghi789', port.network_id)
 174         self.assertFalse(port.port_security_enabled)
 175         self.assertEqual(list(), port.security_groups)
 176         self.assertEqual(list(), port.security_groups)
 177
 178     def test_construction_named_fixed_ips(self):
 179         ips = ['10', '11']
 180         port = Port(
 181             mac_address='abc123', description='test desc', fixed_ips=ips,
 182             id='bar', name='foo', allowed_address_pairs=list(),
 183             admin_state_up=False, device_id='def456', device_owner='joe',
 184             network_id='ghi789', port_security_enabled=False,
 185             security_groups=list(), tenant_id='jkl098')
 186         self.assertEqual('foo', port.name)
 187         self.assertEqual('bar', port.id)
 188         self.assertEqual('test desc', port.description)
 189         self.assertEqual(ips, port.ips)
 190         self.assertEqual('abc123', port.mac_address)
 191         self.assertEqual(list(), port.allowed_address_pairs)
 192         self.assertFalse(port.admin_state_up)
 193         self.assertEqual('def456', port.device_id)
 194         self.assertEqual('joe', port.device_owner)
 195         self.assertEqual('ghi789', port.network_id)
 196         self.assertFalse(port.port_security_enabled)
 197         self.assertEqual(list(), port.security_groups)
 198         self.assertEqual(list(), port.security_groups)
 199
 200
 201 class RouterDomainObjectTests(unittest.TestCase):
 202     """
 203     Tests the construction of the snaps.domain.network.Router class
 204     """
 205
 206     def test_construction_kwargs(self):
 207         router = Router(
 208             **{'name': 'name', 'id': 'id', 'status': 'hello',
 209                'tenant_id': '1234', 'admin_state_up': 'yes',
 210                'external_gateway_info': 'no'})
 211         self.assertEqual('name', router.name)
 212         self.assertEqual('id', router.id)
 213         self.assertEqual('hello', router.status)
 214         self.assertEqual('1234', router.tenant_id)
 215         self.assertEqual('yes', router.admin_state_up)
 216         self.assertIsNone(router.external_fixed_ips)
 217         self.assertIsNone(router.external_network_id)
 218
 219     def test_construction_named(self):
 220         router = Router(
 221             external_gateway_info='no', admin_state_up='yes', tenant_id='1234',
 222             status='hello', id='id', name='name')
 223         self.assertEqual('name', router.name)
 224         self.assertEqual('id', router.id)
 225         self.assertEqual('hello', router.status)
 226         self.assertEqual('1234', router.tenant_id)
 227         self.assertEqual('yes', router.admin_state_up)
 228         self.assertIsNone(router.external_fixed_ips)
 229         self.assertIsNone(router.external_network_id)
 230
 231     def test_ext_gateway_named(self):
 232         ext_gateway = {'network_id': '123',
 233                        'external_fixed_ips': ['456', '789']}
 234         router = Router(
 235             external_fixed_ips=['456', '789'], external_network_id='123',
 236             admin_state_up='yes', tenant_id='1234', status='hello', id='id',
 237             name='name')
 238         self.assertEqual('name', router.name)
 239         self.assertEqual('id', router.id)
 240         self.assertEqual('hello', router.status)
 241         self.assertEqual('1234', router.tenant_id)
 242         self.assertEqual('yes', router.admin_state_up)
 243         self.assertEqual(['456', '789'], router.external_fixed_ips)
 244         self.assertEqual('123', router.external_network_id)
 245
 246
 247     def test_ext_net_ips_named(self):
 248         ext_gateway = {'network_id': '123',
 249                        'external_fixed_ips': ['456', '789']}
 250         router = Router(
 251             external_gateway_info=ext_gateway, admin_state_up='yes',
 252             tenant_id='1234', status='hello', id='id', name='name')
 253         self.assertEqual('name', router.name)
 254         self.assertEqual('id', router.id)
 255         self.assertEqual('hello', router.status)
 256         self.assertEqual('1234', router.tenant_id)
 257         self.assertEqual('yes', router.admin_state_up)
 258         self.assertEqual(['456', '789'], router.external_fixed_ips)
 259         self.assertEqual('123', router.external_network_id)
 260
 261
 262 class InterfaceRouterDomainObjectTests(unittest.TestCase):
 263     """
 264     Tests the construction of the snaps.domain.network.InterfaceRouter class
 265     """
 266
 267     def test_construction_kwargs(self):
 268         intf_router = InterfaceRouter(
 269             **{'id': 'id', 'subnet_id': 'foo', 'port_id': 'bar'})
 270         self.assertEqual('id', intf_router.id)
 271         self.assertEqual('foo', intf_router.subnet_id)
 272         self.assertEqual('bar', intf_router.port_id)
 273
 274     def test_construction_named(self):
 275         intf_router = InterfaceRouter(port_id='bar', subnet_id='foo', id='id')
 276         self.assertEqual('id', intf_router.id)
 277         self.assertEqual('foo', intf_router.subnet_id)
 278         self.assertEqual('bar', intf_router.port_id)
 279
 280
 281 class SecurityGroupDomainObjectTests(unittest.TestCase):
 282     """
 283     Tests the construction of the snaps.domain.network.SecurityGroup class
 284     """
 285
 286     def test_construction_proj_id_kwargs(self):
 287         sec_grp = SecurityGroup(
 288             **{'name': 'name', 'id': 'id', 'project_id': 'foo',
 289                'description': 'test desc'})
 290         self.assertEqual('name', sec_grp.name)
 291         self.assertEqual('id', sec_grp.id)
 292         self.assertEqual('test desc', sec_grp.description)
 293         self.assertEqual('foo', sec_grp.project_id)
 294
 295     def test_construction_tenant_id_kwargs(self):
 296         sec_grp = SecurityGroup(
 297             **{'name': 'name', 'id': 'id',
 298                'tenant_id': 'foo'})
 299         self.assertEqual('name', sec_grp.name)
 300         self.assertEqual('id', sec_grp.id)
 301         self.assertEqual('foo', sec_grp.project_id)
 302         self.assertIsNone(sec_grp.description)
 303
 304     def test_construction_named(self):
 305         sec_grp = SecurityGroup(description='test desc', tenant_id='foo',
 306                                 id='id', name='name')
 307         self.assertEqual('name', sec_grp.name)
 308         self.assertEqual('id', sec_grp.id)
 309         self.assertEqual('test desc', sec_grp.description)
 310         self.assertEqual('foo', sec_grp.project_id)
 311
 312
 313 class SecurityGroupRuleDomainObjectTests(unittest.TestCase):
 314     """
 315     Tests the construction of the snaps.domain.network.SecurityGroupRule class
 316     """
 317
 318     def test_construction_kwargs(self):
 319         sec_grp_rule = SecurityGroupRule(
 320             **{'id': 'id', 'security_group_id': 'grp_id',
 321                'description': 'desc', 'direction': 'dir', 'ethertype': 'eType',
 322                'port_range_min': '10.0.0.100', 'port_range_max': '10.0.0.200',
 323                'protocol': 'proto', 'remote_group_id': 'group_id',
 324                'remote_ip_prefix': 'ip_prefix'})
 325         self.assertEqual('id', sec_grp_rule.id)
 326         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
 327         self.assertEqual('desc', sec_grp_rule.description)
 328         self.assertEqual('dir', sec_grp_rule.direction)
 329         self.assertEqual('eType', sec_grp_rule.ethertype)
 330         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
 331         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
 332         self.assertEqual('proto', sec_grp_rule.protocol)
 333         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
 334         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)
 335
 336     def test_construction_named(self):
 337         sec_grp_rule = SecurityGroupRule(
 338             remote_ip_prefix='ip_prefix', remote_group_id='group_id',
 339             protocol='proto', port_range_min='10.0.0.100',
 340             port_range_max='10.0.0.200', ethertype='eType',
 341             direction='dir', description='desc', security_group_id='grp_id',
 342             id='id')
 343         self.assertEqual('id', sec_grp_rule.id)
 344         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
 345         self.assertEqual('desc', sec_grp_rule.description)
 346         self.assertEqual('dir', sec_grp_rule.direction)
 347         self.assertEqual('eType', sec_grp_rule.ethertype)
 348         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
 349         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
 350         self.assertEqual('proto', sec_grp_rule.protocol)
 351         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
 352         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)