1 # Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
2 # and others. All rights reserved.
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:
8 # http://www.apache.org/licenses/LICENSE-2.0
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.
17 from snaps.domain.network import (
18 Port, SecurityGroup, SecurityGroupRule, Router, InterfaceRouter)
21 class PortDomainObjectTests(unittest.TestCase):
23 Tests the construction of the snaps.domain.network.Port class
26 def test_construction_kwargs(self):
29 **{'name': 'name', 'id': 'id', 'ips': ips})
30 self.assertEqual('name', port.name)
31 self.assertEqual('id', port.id)
32 self.assertEqual(ips, port.ips)
34 def test_construction_named(self):
36 port = Port(ips=ips, id='id', name='name')
37 self.assertEqual('name', port.name)
38 self.assertEqual('id', port.id)
39 self.assertEqual(ips, port.ips)
42 class RouterDomainObjectTests(unittest.TestCase):
44 Tests the construction of the snaps.domain.network.Router class
47 def test_construction_kwargs(self):
49 **{'name': 'name', 'id': 'id', 'status': 'hello',
50 'tenant_id': '1234', 'admin_state_up': 'yes',
51 'external_gateway_info': 'no'})
52 self.assertEqual('name', sec_grp.name)
53 self.assertEqual('id', sec_grp.id)
54 self.assertEqual('hello', sec_grp.status)
55 self.assertEqual('1234', sec_grp.tenant_id)
56 self.assertEqual('yes', sec_grp.admin_state_up)
57 self.assertEqual('no', sec_grp.external_gateway_info)
59 def test_construction_named(self):
61 external_gateway_info='no', admin_state_up='yes', tenant_id='1234',
62 status='hello', id='id', name='name')
63 self.assertEqual('name', sec_grp.name)
64 self.assertEqual('id', sec_grp.id)
65 self.assertEqual('hello', sec_grp.status)
66 self.assertEqual('1234', sec_grp.tenant_id)
67 self.assertEqual('yes', sec_grp.admin_state_up)
68 self.assertEqual('no', sec_grp.external_gateway_info)
71 class InterfaceRouterDomainObjectTests(unittest.TestCase):
73 Tests the construction of the snaps.domain.network.InterfaceRouter class
76 def test_construction_kwargs(self):
77 sec_grp = InterfaceRouter(
78 **{'id': 'id', 'subnet_id': 'foo', 'port_id': 'bar'})
79 self.assertEqual('id', sec_grp.id)
80 self.assertEqual('foo', sec_grp.subnet_id)
81 self.assertEqual('bar', sec_grp.port_id)
83 def test_construction_named(self):
84 sec_grp = InterfaceRouter(port_id='bar', subnet_id='foo', id='id')
85 self.assertEqual('id', sec_grp.id)
86 self.assertEqual('foo', sec_grp.subnet_id)
87 self.assertEqual('bar', sec_grp.port_id)
90 class SecurityGroupDomainObjectTests(unittest.TestCase):
92 Tests the construction of the snaps.domain.network.SecurityGroup class
95 def test_construction_proj_id_kwargs(self):
96 sec_grp = SecurityGroup(
97 **{'name': 'name', 'id': 'id',
99 self.assertEqual('name', sec_grp.name)
100 self.assertEqual('id', sec_grp.id)
101 self.assertEqual('foo', sec_grp.project_id)
103 def test_construction_tenant_id_kwargs(self):
104 sec_grp = SecurityGroup(
105 **{'name': 'name', 'id': 'id',
107 self.assertEqual('name', sec_grp.name)
108 self.assertEqual('id', sec_grp.id)
109 self.assertEqual('foo', sec_grp.project_id)
111 def test_construction_named(self):
112 sec_grp = SecurityGroup(tenant_id='foo', id='id', name='name')
113 self.assertEqual('name', sec_grp.name)
114 self.assertEqual('id', sec_grp.id)
115 self.assertEqual('foo', sec_grp.project_id)
118 class SecurityGroupRuleDomainObjectTests(unittest.TestCase):
120 Tests the construction of the snaps.domain.network.SecurityGroupRule class
123 def test_construction_kwargs(self):
124 sec_grp_rule = SecurityGroupRule(
125 **{'id': 'id', 'security_group_id': 'grp_id',
126 'description': 'desc', 'direction': 'dir', 'ethertype': 'eType',
127 'port_range_min': '10.0.0.100', 'port_range_max': '10.0.0.200',
128 'protocol': 'proto', 'remote_group_id': 'group_id',
129 'remote_ip_prefix': 'ip_prefix'})
130 self.assertEqual('id', sec_grp_rule.id)
131 self.assertEqual('grp_id', sec_grp_rule.security_group_id)
132 self.assertEqual('desc', sec_grp_rule.description)
133 self.assertEqual('dir', sec_grp_rule.direction)
134 self.assertEqual('eType', sec_grp_rule.ethertype)
135 self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
136 self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
137 self.assertEqual('proto', sec_grp_rule.protocol)
138 self.assertEqual('group_id', sec_grp_rule.remote_group_id)
139 self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)
141 def test_construction_named(self):
142 sec_grp_rule = SecurityGroupRule(
143 remote_ip_prefix='ip_prefix', remote_group_id='group_id',
144 protocol='proto', port_range_min='10.0.0.100',
145 port_range_max='10.0.0.200', ethertype='eType',
146 direction='dir', description='desc', security_group_id='grp_id',
148 self.assertEqual('id', sec_grp_rule.id)
149 self.assertEqual('grp_id', sec_grp_rule.security_group_id)
150 self.assertEqual('desc', sec_grp_rule.description)
151 self.assertEqual('dir', sec_grp_rule.direction)
152 self.assertEqual('eType', sec_grp_rule.ethertype)
153 self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
154 self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
155 self.assertEqual('proto', sec_grp_rule.protocol)
156 self.assertEqual('group_id', sec_grp_rule.remote_group_id)
157 self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)