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