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_kwargs(self):
 111         ips = ['10', '11']
 112         port = Port(
 113             **{'name': 'name', 'id': 'id', 'ips': ips})
 114         self.assertEqual('name', port.name)
 115         self.assertEqual('id', port.id)
 116         self.assertEqual(ips, port.ips)
 117
 118     def test_construction_named(self):
 119         ips = ['10', '11']
 120         port = Port(ips=ips, id='id', name='name')
 121         self.assertEqual('name', port.name)
 122         self.assertEqual('id', port.id)
 123         self.assertEqual(ips, port.ips)
 124
 125
 126 class RouterDomainObjectTests(unittest.TestCase):
 127     """
 128     Tests the construction of the snaps.domain.network.Router class
 129     """
 130
 131     def test_construction_kwargs(self):
 132         router = Router(
 133             **{'name': 'name', 'id': 'id', 'status': 'hello',
 134                'tenant_id': '1234', 'admin_state_up': 'yes',
 135                'external_gateway_info': 'no'})
 136         self.assertEqual('name', router.name)
 137         self.assertEqual('id', router.id)
 138         self.assertEqual('hello', router.status)
 139         self.assertEqual('1234', router.tenant_id)
 140         self.assertEqual('yes', router.admin_state_up)
 141         self.assertEqual('no', router.external_gateway_info)
 142
 143     def test_construction_named(self):
 144         router = Router(
 145             external_gateway_info='no', admin_state_up='yes', tenant_id='1234',
 146             status='hello', id='id', name='name')
 147         self.assertEqual('name', router.name)
 148         self.assertEqual('id', router.id)
 149         self.assertEqual('hello', router.status)
 150         self.assertEqual('1234', router.tenant_id)
 151         self.assertEqual('yes', router.admin_state_up)
 152         self.assertEqual('no', router.external_gateway_info)
 153
 154
 155 class InterfaceRouterDomainObjectTests(unittest.TestCase):
 156     """
 157     Tests the construction of the snaps.domain.network.InterfaceRouter class
 158     """
 159
 160     def test_construction_kwargs(self):
 161         intf_router = InterfaceRouter(
 162             **{'id': 'id', 'subnet_id': 'foo', 'port_id': 'bar'})
 163         self.assertEqual('id', intf_router.id)
 164         self.assertEqual('foo', intf_router.subnet_id)
 165         self.assertEqual('bar', intf_router.port_id)
 166
 167     def test_construction_named(self):
 168         intf_router = InterfaceRouter(port_id='bar', subnet_id='foo', id='id')
 169         self.assertEqual('id', intf_router.id)
 170         self.assertEqual('foo', intf_router.subnet_id)
 171         self.assertEqual('bar', intf_router.port_id)
 172
 173
 174 class SecurityGroupDomainObjectTests(unittest.TestCase):
 175     """
 176     Tests the construction of the snaps.domain.network.SecurityGroup class
 177     """
 178
 179     def test_construction_proj_id_kwargs(self):
 180         sec_grp = SecurityGroup(
 181             **{'name': 'name', 'id': 'id', 'project_id': 'foo',
 182                'description': 'test desc'})
 183         self.assertEqual('name', sec_grp.name)
 184         self.assertEqual('id', sec_grp.id)
 185         self.assertEqual('test desc', sec_grp.description)
 186         self.assertEqual('foo', sec_grp.project_id)
 187
 188     def test_construction_tenant_id_kwargs(self):
 189         sec_grp = SecurityGroup(
 190             **{'name': 'name', 'id': 'id',
 191                'tenant_id': 'foo'})
 192         self.assertEqual('name', sec_grp.name)
 193         self.assertEqual('id', sec_grp.id)
 194         self.assertEqual('foo', sec_grp.project_id)
 195         self.assertIsNone(sec_grp.description)
 196
 197     def test_construction_named(self):
 198         sec_grp = SecurityGroup(description='test desc', tenant_id='foo',
 199                                 id='id', name='name')
 200         self.assertEqual('name', sec_grp.name)
 201         self.assertEqual('id', sec_grp.id)
 202         self.assertEqual('test desc', sec_grp.description)
 203         self.assertEqual('foo', sec_grp.project_id)
 204
 205
 206 class SecurityGroupRuleDomainObjectTests(unittest.TestCase):
 207     """
 208     Tests the construction of the snaps.domain.network.SecurityGroupRule class
 209     """
 210
 211     def test_construction_kwargs(self):
 212         sec_grp_rule = SecurityGroupRule(
 213             **{'id': 'id', 'security_group_id': 'grp_id',
 214                'description': 'desc', 'direction': 'dir', 'ethertype': 'eType',
 215                'port_range_min': '10.0.0.100', 'port_range_max': '10.0.0.200',
 216                'protocol': 'proto', 'remote_group_id': 'group_id',
 217                'remote_ip_prefix': 'ip_prefix'})
 218         self.assertEqual('id', sec_grp_rule.id)
 219         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
 220         self.assertEqual('desc', sec_grp_rule.description)
 221         self.assertEqual('dir', sec_grp_rule.direction)
 222         self.assertEqual('eType', sec_grp_rule.ethertype)
 223         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
 224         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
 225         self.assertEqual('proto', sec_grp_rule.protocol)
 226         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
 227         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)
 228
 229     def test_construction_named(self):
 230         sec_grp_rule = SecurityGroupRule(
 231             remote_ip_prefix='ip_prefix', remote_group_id='group_id',
 232             protocol='proto', port_range_min='10.0.0.100',
 233             port_range_max='10.0.0.200', ethertype='eType',
 234             direction='dir', description='desc', security_group_id='grp_id',
 235             id='id')
 236         self.assertEqual('id', sec_grp_rule.id)
 237         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
 238         self.assertEqual('desc', sec_grp_rule.description)
 239         self.assertEqual('dir', sec_grp_rule.direction)
 240         self.assertEqual('eType', sec_grp_rule.ethertype)
 241         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
 242         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
 243         self.assertEqual('proto', sec_grp_rule.protocol)
 244         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
 245         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)