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         subnet = Subnet(
  29             **{'name': 'foo', 'id': 'bar', 'project_id': 'proj1',
  30                'network_id': 'foo-bar'})
  31         network = Network(
  32             **{'name': 'foo', 'id': 'bar', 'project_id': 'proj1',
  33                'provider:network_type': 'flat', 'admin_state_up': False,
  34                'shared': True, 'router:external': False, 'subnets': [subnet],
  35                'mtu': 999})
  36         self.assertEqual('foo', network.name)
  37         self.assertEqual('bar', network.id)
  38         self.assertEqual('proj1', network.project_id)
  39         self.assertEqual('flat', network.type)
  40         self.assertFalse(network.admin_state_up)
  41         self.assertFalse(network.external)
  42         self.assertTrue(network.shared)
  43         self.assertEqual([subnet], network.subnets)
  44         self.assertEqual(999, network.mtu)
  45
  46     def test_construction_kwargs_2(self):
  47         subnet = Subnet(
  48             **{'name': 'foo', 'id': 'bar', 'project_id': 'proj1',
  49                'network_id': 'foo-bar'})
  50         network = Network(
  51             **{'name': 'foo', 'id': 'bar', 'project_id': 'proj1',
  52                'type': 'flat', 'admin_state_up': False, 'shared': True,
  53                'external': False, 'subnets': [subnet]})
  54         self.assertEqual('foo', network.name)
  55         self.assertEqual('bar', network.id)
  56         self.assertEqual('proj1', network.project_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         self.assertEqual([subnet], network.subnets)
  62
  63     def test_construction_named(self):
  64         subnet = Subnet(
  65             **{'name': 'foo', 'id': 'bar', 'project_id': 'proj1',
  66                'network_id': 'foo-bar'})
  67         network = Network(
  68             name='foo', id='bar', project_id='proj1', type='flat',
  69             admin_state_up=False, shared=True, external=False,
  70             subnets=[subnet])
  71         self.assertEqual('foo', network.name)
  72         self.assertEqual('bar', network.id)
  73         self.assertEqual('proj1', network.project_id)
  74         self.assertEqual('flat', network.type)
  75         self.assertFalse(network.admin_state_up)
  76         self.assertFalse(network.external)
  77         self.assertTrue(network.shared)
  78         self.assertEqual([subnet], network.subnets)
  79
  80
  81 class SubnetObjectTests(unittest.TestCase):
  82     """
  83     Tests the construction of the snaps.domain.network.Subnet class
  84     """
  85
  86     def test_construction_kwargs(self):
  87         subnet = Subnet(
  88             **{'name': 'foo', 'id': 'bar', 'project_id': 'proj1',
  89                'cidr': '10.0.0.0/24', 'ip_version': 4,
  90                'gateway_ip': '10.0.0.1', 'enable_dhcp': True,
  91                'dns_nameservers': ['8.8.8.8'], 'host_routes': list(),
  92                'ipv6_ra_mode': 'hello', 'ipv6_address_mode': 'world'})
  93         self.assertEqual('foo', subnet.name)
  94         self.assertEqual('bar', subnet.id)
  95         self.assertEqual('proj1', subnet.project_id)
  96         self.assertEqual('10.0.0.0/24', subnet.cidr)
  97         self.assertEqual(4, subnet.ip_version)
  98         self.assertEqual('10.0.0.1', subnet.gateway_ip)
  99         self.assertTrue(subnet.enable_dhcp)
 100         self.assertEqual(1, len(subnet.dns_nameservers))
 101         self.assertEqual('8.8.8.8', subnet.dns_nameservers[0])
 102         self.assertEqual(list(), subnet.host_routes)
 103         self.assertEqual('hello', subnet.ipv6_ra_mode)
 104         self.assertEqual('world', subnet.ipv6_address_mode)
 105
 106     def test_construction_named(self):
 107         subnet = Subnet(
 108             name='foo', id='bar', project_id='proj1', cidr='10.0.0.0/24',
 109             ip_version=4, gateway_ip='10.0.0.1', enable_dhcp=True,
 110             dns_nameservers=['8.8.8.8'], host_routes=list(),
 111             ipv6_ra_mode='hello', ipv6_address_mode='world')
 112         self.assertEqual('foo', subnet.name)
 113         self.assertEqual('bar', subnet.id)
 114         self.assertEqual('proj1', subnet.project_id)
 115         self.assertEqual('10.0.0.0/24', subnet.cidr)
 116         self.assertEqual(4, subnet.ip_version)
 117         self.assertEqual('10.0.0.1', subnet.gateway_ip)
 118         self.assertTrue(subnet.enable_dhcp)
 119         self.assertEqual(1, len(subnet.dns_nameservers))
 120         self.assertEqual('8.8.8.8', subnet.dns_nameservers[0])
 121         self.assertEqual(list(), subnet.host_routes)
 122         self.assertEqual('hello', subnet.ipv6_ra_mode)
 123         self.assertEqual('world', subnet.ipv6_address_mode)
 124
 125
 126 class PortDomainObjectTests(unittest.TestCase):
 127     """
 128     Tests the construction of the snaps.domain.network.Port class
 129     """
 130
 131     def test_construction_ips_kwargs(self):
 132         ips = ['10', '11']
 133         port = Port(
 134             **{'name': 'foo', 'id': 'bar', 'description': 'test desc',
 135                'ips': ips, 'mac_address': 'abc123',
 136                'allowed_address_pairs': list(), 'admin_state_up': False,
 137                'device_id': 'def456', 'device_owner': 'joe',
 138                'network_id': 'ghi789', 'port_security_enabled': False,
 139                'security_groups': list(), 'tenant_id': 'jkl098'})
 140         self.assertEqual('foo', port.name)
 141         self.assertEqual('bar', port.id)
 142         self.assertEqual('test desc', port.description)
 143         self.assertEqual(ips, port.ips)
 144         self.assertEqual('abc123', port.mac_address)
 145         self.assertEqual(list(), port.allowed_address_pairs)
 146         self.assertFalse(port.admin_state_up)
 147         self.assertEqual('def456', port.device_id)
 148         self.assertEqual('joe', port.device_owner)
 149         self.assertEqual('ghi789', port.network_id)
 150         self.assertFalse(port.port_security_enabled)
 151         self.assertEqual(list(), port.security_groups)
 152         self.assertEqual(list(), port.security_groups)
 153
 154     def test_construction_fixed_ips_kwargs(self):
 155         ips = ['10', '11']
 156         port = Port(
 157             **{'name': 'foo', 'id': 'bar', 'description': 'test desc',
 158                'fixed_ips': ips, 'mac_address': 'abc123',
 159                'allowed_address_pairs': list(), 'admin_state_up': False,
 160                'device_id': 'def456', 'device_owner': 'joe',
 161                'network_id': 'ghi789', 'port_security_enabled': False,
 162                'security_groups': list(), 'tenant_id': 'jkl098'})
 163         self.assertEqual('foo', port.name)
 164         self.assertEqual('bar', port.id)
 165         self.assertEqual('test desc', port.description)
 166         self.assertEqual(ips, port.ips)
 167         self.assertEqual('abc123', port.mac_address)
 168         self.assertEqual(list(), port.allowed_address_pairs)
 169         self.assertFalse(port.admin_state_up)
 170         self.assertEqual('def456', port.device_id)
 171         self.assertEqual('joe', port.device_owner)
 172         self.assertEqual('ghi789', port.network_id)
 173         self.assertFalse(port.port_security_enabled)
 174         self.assertEqual(list(), port.security_groups)
 175         self.assertEqual(list(), port.security_groups)
 176
 177     def test_construction_named_ips(self):
 178         ips = ['10', '11']
 179         port = Port(
 180             mac_address='abc123', description='test desc', ips=ips, id='bar',
 181             name='foo', allowed_address_pairs=list(), admin_state_up=False,
 182             device_id='def456', device_owner='joe', network_id='ghi789',
 183             port_security_enabled=False, security_groups=list(),
 184             tenant_id='jkl098')
 185         self.assertEqual('foo', port.name)
 186         self.assertEqual('bar', port.id)
 187         self.assertEqual('test desc', port.description)
 188         self.assertEqual(ips, port.ips)
 189         self.assertEqual('abc123', port.mac_address)
 190         self.assertEqual(list(), port.allowed_address_pairs)
 191         self.assertFalse(port.admin_state_up)
 192         self.assertEqual('def456', port.device_id)
 193         self.assertEqual('joe', port.device_owner)
 194         self.assertEqual('ghi789', port.network_id)
 195         self.assertFalse(port.port_security_enabled)
 196         self.assertEqual(list(), port.security_groups)
 197         self.assertEqual(list(), port.security_groups)
 198
 199     def test_construction_named_fixed_ips(self):
 200         ips = ['10', '11']
 201         port = Port(
 202             mac_address='abc123', description='test desc', fixed_ips=ips,
 203             id='bar', name='foo', allowed_address_pairs=list(),
 204             admin_state_up=False, device_id='def456', device_owner='joe',
 205             network_id='ghi789', port_security_enabled=False,
 206             security_groups=list(), tenant_id='jkl098')
 207         self.assertEqual('foo', port.name)
 208         self.assertEqual('bar', port.id)
 209         self.assertEqual('test desc', port.description)
 210         self.assertEqual(ips, port.ips)
 211         self.assertEqual('abc123', port.mac_address)
 212         self.assertEqual(list(), port.allowed_address_pairs)
 213         self.assertFalse(port.admin_state_up)
 214         self.assertEqual('def456', port.device_id)
 215         self.assertEqual('joe', port.device_owner)
 216         self.assertEqual('ghi789', port.network_id)
 217         self.assertFalse(port.port_security_enabled)
 218         self.assertEqual(list(), port.security_groups)
 219         self.assertEqual(list(), port.security_groups)
 220
 221
 222 class RouterDomainObjectTests(unittest.TestCase):
 223     """
 224     Tests the construction of the snaps.domain.network.Router class
 225     """
 226
 227     def test_construction_kwargs(self):
 228         router = Router(
 229             **{'name': 'name', 'id': 'id', 'status': 'hello',
 230                'tenant_id': '1234', 'admin_state_up': 'yes',
 231                'external_gateway_info': 'no'})
 232         self.assertEqual('name', router.name)
 233         self.assertEqual('id', router.id)
 234         self.assertEqual('hello', router.status)
 235         self.assertEqual('1234', router.tenant_id)
 236         self.assertEqual('yes', router.admin_state_up)
 237         self.assertIsNone(router.external_fixed_ips)
 238         self.assertIsNone(router.external_network_id)
 239
 240     def test_construction_named(self):
 241         router = Router(
 242             external_gateway_info='no', admin_state_up='yes', tenant_id='1234',
 243             status='hello', id='id', name='name')
 244         self.assertEqual('name', router.name)
 245         self.assertEqual('id', router.id)
 246         self.assertEqual('hello', router.status)
 247         self.assertEqual('1234', router.tenant_id)
 248         self.assertEqual('yes', router.admin_state_up)
 249         self.assertIsNone(router.external_fixed_ips)
 250         self.assertIsNone(router.external_network_id)
 251
 252     def test_ext_gateway_named(self):
 253         router = Router(
 254             external_fixed_ips=['456', '789'], external_network_id='123',
 255             admin_state_up='yes', tenant_id='1234', status='hello', id='id',
 256             name='name')
 257         self.assertEqual('name', router.name)
 258         self.assertEqual('id', router.id)
 259         self.assertEqual('hello', router.status)
 260         self.assertEqual('1234', router.tenant_id)
 261         self.assertEqual('yes', router.admin_state_up)
 262         self.assertEqual(['456', '789'], router.external_fixed_ips)
 263         self.assertEqual('123', router.external_network_id)
 264
 265     def test_ext_net_ips_named(self):
 266         ext_gateway = {'network_id': '123',
 267                        'external_fixed_ips': ['456', '789']}
 268         router = Router(
 269             external_gateway_info=ext_gateway, admin_state_up='yes',
 270             tenant_id='1234', status='hello', id='id', name='name')
 271         self.assertEqual('name', router.name)
 272         self.assertEqual('id', router.id)
 273         self.assertEqual('hello', router.status)
 274         self.assertEqual('1234', router.tenant_id)
 275         self.assertEqual('yes', router.admin_state_up)
 276         self.assertEqual(['456', '789'], router.external_fixed_ips)
 277         self.assertEqual('123', router.external_network_id)
 278
 279
 280 class InterfaceRouterDomainObjectTests(unittest.TestCase):
 281     """
 282     Tests the construction of the snaps.domain.network.InterfaceRouter class
 283     """
 284
 285     def test_construction_kwargs(self):
 286         intf_router = InterfaceRouter(
 287             **{'id': 'id', 'subnet_id': 'foo', 'port_id': 'bar'})
 288         self.assertEqual('id', intf_router.id)
 289         self.assertEqual('foo', intf_router.subnet_id)
 290         self.assertEqual('bar', intf_router.port_id)
 291
 292     def test_construction_named(self):
 293         intf_router = InterfaceRouter(port_id='bar', subnet_id='foo', id='id')
 294         self.assertEqual('id', intf_router.id)
 295         self.assertEqual('foo', intf_router.subnet_id)
 296         self.assertEqual('bar', intf_router.port_id)
 297
 298
 299 class SecurityGroupDomainObjectTests(unittest.TestCase):
 300     """
 301     Tests the construction of the snaps.domain.network.SecurityGroup class
 302     """
 303
 304     def test_construction_proj_id_kwargs(self):
 305         sec_grp = SecurityGroup(
 306             **{'name': 'name', 'id': 'id', 'project_id': 'foo',
 307                'description': 'test desc',
 308                'rules': [
 309                     {'id': 'id', 'security_group_id': 'grp_id',
 310                      'description': 'desc', 'direction': 'dir',
 311                      'ethertype': 'eType', 'port_range_min': '10.0.0.100',
 312                      'port_range_max': '10.0.0.200', 'protocol': 'proto',
 313                      'remote_group_id': 'group_id',
 314                      'remote_ip_prefix': 'ip_prefix'}
 315                ]})
 316         self.assertEqual('name', sec_grp.name)
 317         self.assertEqual('id', sec_grp.id)
 318         self.assertEqual('test desc', sec_grp.description)
 319         self.assertEqual('foo', sec_grp.project_id)
 320
 321         self.assertEqual(1, len(sec_grp.rules))
 322         rule = sec_grp.rules[0]
 323         self.assertEqual('id', rule.id)
 324         self.assertEqual('grp_id', rule.security_group_id)
 325         self.assertEqual('desc', rule.description)
 326         self.assertEqual('dir', rule.direction)
 327         self.assertEqual('eType', rule.ethertype)
 328         self.assertEqual('10.0.0.100', rule.port_range_min)
 329         self.assertEqual('10.0.0.200', rule.port_range_max)
 330         self.assertEqual('proto', rule.protocol)
 331         self.assertEqual('group_id', rule.remote_group_id)
 332         self.assertEqual('ip_prefix', rule.remote_ip_prefix)
 333
 334     def test_construction_tenant_id_kwargs(self):
 335         sec_grp = SecurityGroup(
 336             **{'name': 'name', 'id': 'id',
 337                'tenant_id': 'foo'})
 338         self.assertEqual('name', sec_grp.name)
 339         self.assertEqual('id', sec_grp.id)
 340         self.assertEqual('foo', sec_grp.project_id)
 341         self.assertIsNone(sec_grp.description)
 342         self.assertEqual(0, len(sec_grp.rules))
 343
 344     def test_construction_named(self):
 345         rules = [SecurityGroupRule(
 346             **{'id': 'id', 'security_group_id': 'grp_id',
 347                'description': 'desc', 'direction': 'dir',
 348                'ethertype': 'eType', 'port_range_min': '10.0.0.100',
 349                'port_range_max': '10.0.0.200', 'protocol': 'proto',
 350                'remote_group_id': 'group_id',
 351                'remote_ip_prefix': 'ip_prefix'}
 352         )]
 353         sec_grp = SecurityGroup(description='test desc', tenant_id='foo',
 354                                 id='id', name='name', rules=rules)
 355         self.assertEqual('name', sec_grp.name)
 356         self.assertEqual('id', sec_grp.id)
 357         self.assertEqual('test desc', sec_grp.description)
 358         self.assertEqual('foo', sec_grp.project_id)
 359
 360         self.assertEqual(1, len(sec_grp.rules))
 361         rule = sec_grp.rules[0]
 362         self.assertEqual('id', rule.id)
 363         self.assertEqual('grp_id', rule.security_group_id)
 364         self.assertEqual('desc', rule.description)
 365         self.assertEqual('dir', rule.direction)
 366         self.assertEqual('eType', rule.ethertype)
 367         self.assertEqual('10.0.0.100', rule.port_range_min)
 368         self.assertEqual('10.0.0.200', rule.port_range_max)
 369         self.assertEqual('proto', rule.protocol)
 370         self.assertEqual('group_id', rule.remote_group_id)
 371         self.assertEqual('ip_prefix', rule.remote_ip_prefix)
 372
 373
 374 class SecurityGroupRuleDomainObjectTests(unittest.TestCase):
 375     """
 376     Tests the construction of the snaps.domain.network.SecurityGroupRule class
 377     """
 378
 379     def test_construction_kwargs(self):
 380         sec_grp_rule = SecurityGroupRule(
 381             **{'id': 'id', 'security_group_id': 'grp_id',
 382                'description': 'desc', 'direction': 'dir', 'ethertype': 'eType',
 383                'port_range_min': '10.0.0.100', 'port_range_max': '10.0.0.200',
 384                'protocol': 'proto', 'remote_group_id': 'group_id',
 385                'remote_ip_prefix': 'ip_prefix'})
 386         self.assertEqual('id', sec_grp_rule.id)
 387         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
 388         self.assertEqual('desc', sec_grp_rule.description)
 389         self.assertEqual('dir', sec_grp_rule.direction)
 390         self.assertEqual('eType', sec_grp_rule.ethertype)
 391         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
 392         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
 393         self.assertEqual('proto', sec_grp_rule.protocol)
 394         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
 395         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)
 396
 397     def test_construction_named(self):
 398         sec_grp_rule = SecurityGroupRule(
 399             remote_ip_prefix='ip_prefix', remote_group_id='group_id',
 400             protocol='proto', port_range_min='10.0.0.100',
 401             port_range_max='10.0.0.200', ethertype='eType',
 402             direction='dir', description='desc', security_group_id='grp_id',
 403             id='id')
 404         self.assertEqual('id', sec_grp_rule.id)
 405         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
 406         self.assertEqual('desc', sec_grp_rule.description)
 407         self.assertEqual('dir', sec_grp_rule.direction)
 408         self.assertEqual('eType', sec_grp_rule.ethertype)
 409         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
 410         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
 411         self.assertEqual('proto', sec_grp_rule.protocol)
 412         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
 413         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)