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.config.cluster_template import ClusterTemplateConfig, \
18 ClusterTypeConfigError, ServerType, DockerStorageDriver, \
19 ContainerOrchestrationEngine
22 class ClusterTemplateConfigUnitTests(unittest.TestCase):
24 Tests the construction of the ClusterTemplateConfig class
27 def test_no_params(self):
28 with self.assertRaises(ClusterTypeConfigError):
29 ClusterTemplateConfig()
31 def test_empty_config(self):
32 with self.assertRaises(ClusterTypeConfigError):
33 ClusterTemplateConfig(config=dict())
35 def test_name_only(self):
36 with self.assertRaises(ClusterTypeConfigError):
37 ClusterTemplateConfig(name='foo')
39 def test_minimal_named(self):
40 config = ClusterTemplateConfig(
41 name='foo', image='bar', keypair='keys', external_net='external')
42 self.assertIsNotNone(config)
43 self.assertEqual('foo', config.name)
44 self.assertEqual('bar', config.image)
45 self.assertEqual('keys', config.keypair)
46 self.assertIsNone(config.network_driver)
47 self.assertEqual('external', config.external_net)
48 self.assertTrue(config.floating_ip_enabled)
49 self.assertEqual(3, config.docker_volume_size)
50 self.assertEqual(ServerType.vm, config.server_type)
51 self.assertIsNone(config.flavor)
52 self.assertIsNone(config.master_flavor)
53 self.assertEqual(ContainerOrchestrationEngine.kubernetes, config.coe)
54 self.assertIsNone(config.fixed_net)
55 self.assertIsNone(config.fixed_subnet)
56 self.assertTrue(config.registry_enabled)
57 self.assertIsNone(config.insecure_registry)
58 self.assertEqual(DockerStorageDriver.devicemapper,
59 config.docker_storage_driver)
60 self.assertIsNone(config.dns_nameserver)
61 self.assertFalse(config.public)
62 self.assertFalse(config.tls_disabled)
63 self.assertIsNone(config.http_proxy)
64 self.assertIsNone(config.https_proxy)
65 self.assertIsNone(config.no_proxy)
66 self.assertIsNone(config.volume_driver)
67 self.assertTrue(config.master_lb_enabled)
68 self.assertIsNone(config.labels)
70 def test_minimal_config(self):
71 config = ClusterTemplateConfig(
72 **{'name': 'foo', 'image': 'bar', 'keypair': 'keys',
73 'external_net': 'external'})
74 self.assertIsNotNone(config)
75 self.assertEqual('foo', config.name)
76 self.assertEqual('bar', config.image)
77 self.assertEqual('keys', config.keypair)
78 self.assertIsNone(config.network_driver)
79 self.assertEqual('external', config.external_net)
80 self.assertTrue(config.floating_ip_enabled)
81 self.assertEqual(3, config.docker_volume_size)
82 self.assertEqual(ServerType.vm, config.server_type)
83 self.assertIsNone(config.flavor)
84 self.assertIsNone(config.master_flavor)
85 self.assertEqual(ContainerOrchestrationEngine.kubernetes, config.coe)
86 self.assertIsNone(config.fixed_net)
87 self.assertIsNone(config.fixed_subnet)
88 self.assertTrue(config.registry_enabled)
89 self.assertIsNone(config.insecure_registry)
90 self.assertEqual(DockerStorageDriver.devicemapper,
91 config.docker_storage_driver)
92 self.assertIsNone(config.dns_nameserver)
93 self.assertFalse(config.public)
94 self.assertFalse(config.tls_disabled)
95 self.assertIsNone(config.http_proxy)
96 self.assertIsNone(config.https_proxy)
97 self.assertIsNone(config.no_proxy)
98 self.assertIsNone(config.volume_driver)
99 self.assertTrue(config.master_lb_enabled)
100 self.assertIsNone(config.labels)
102 def test_all_named(self):
103 labels = {'foo': 'bar'}
104 config = ClusterTemplateConfig(
105 name='foo', image='bar', keypair='keys', network_driver='driver',
106 external_net='external', docker_volume_size=99,
107 server_type=ServerType.baremetal, flavor='testFlavor',
108 master_flavor='masterFlavor',
109 coe=ContainerOrchestrationEngine.kubernetes, fixed_net='fixedNet',
110 fixed_subnet='fixedSubnet', registry_enabled=False,
111 docker_storage_driver=DockerStorageDriver.overlay,
112 dns_nameserver='8.8.4.4', public=True, tls=False,
113 http_proxy='http://foo:8080', https_proxy='https://foo:443',
114 no_proxy='foo,bar', volume_driver='volDriver',
115 master_lb_enabled=False, labels=labels)
116 self.assertIsNotNone(config)
117 self.assertEqual('foo', config.name)
118 self.assertEqual('bar', config.image)
119 self.assertEqual('keys', config.keypair)
120 self.assertEqual('driver', config.network_driver)
121 self.assertEqual('external', config.external_net)
122 self.assertEqual(99, config.docker_volume_size)
123 self.assertEqual(ServerType.baremetal, config.server_type)
124 self.assertEqual('testFlavor', config.flavor)
125 self.assertEqual('masterFlavor', config.master_flavor)
126 self.assertEqual(ContainerOrchestrationEngine.kubernetes, config.coe)
127 self.assertEqual('fixedNet', config.fixed_net)
128 self.assertEqual('fixedSubnet', config.fixed_subnet)
129 self.assertFalse(config.registry_enabled)
130 self.assertEqual(DockerStorageDriver.overlay,
131 config.docker_storage_driver)
132 self.assertEqual('8.8.4.4', config.dns_nameserver)
133 self.assertTrue(config.public)
134 self.assertFalse(config.tls_disabled)
135 self.assertEqual('http://foo:8080', config.http_proxy)
136 self.assertEqual('https://foo:443', config.https_proxy)
137 self.assertEqual('foo,bar', config.no_proxy)
138 self.assertEqual('volDriver', config.volume_driver)
139 self.assertFalse(config.master_lb_enabled)
140 self.assertEqual(labels, config.labels)
142 def test_all_config(self):
143 labels = {'foo': 'bar'}
144 config = ClusterTemplateConfig(**{
145 'name': 'foo', 'image': 'bar', 'keypair': 'keys',
146 'network_driver': 'driver', 'external_net': 'external',
147 'docker_volume_size': '99', 'server_type': 'baremetal',
148 'flavor': 'testFlavor', 'master_flavor': 'masterFlavor',
149 'coe': 'kubernetes', 'fixed_net': 'fixedNet',
150 'fixed_subnet': 'fixedSubnet', 'registry_enabled': 'false',
151 'docker_storage_driver': 'overlay', 'dns_nameserver': '8.8.4.4',
152 'public': 'true', 'tls': 'false', 'http_proxy': 'http://foo:8080',
153 'https_proxy': 'https://foo:443', 'no_proxy': 'foo,bar',
154 'volume_driver': 'volDriver', 'master_lb_enabled': 'false',
156 self.assertIsNotNone(config)
157 self.assertEqual('foo', config.name)
158 self.assertEqual('bar', config.image)
159 self.assertEqual('keys', config.keypair)
160 self.assertEqual('driver', config.network_driver)
161 self.assertEqual('external', config.external_net)
162 self.assertEqual(99, config.docker_volume_size)
163 self.assertEqual(ServerType.baremetal, config.server_type)
164 self.assertEqual('testFlavor', config.flavor)
165 self.assertEqual('masterFlavor', config.master_flavor)
166 self.assertEqual(ContainerOrchestrationEngine.kubernetes, config.coe)
167 self.assertEqual('fixedNet', config.fixed_net)
168 self.assertEqual('fixedSubnet', config.fixed_subnet)
169 self.assertFalse(config.registry_enabled)
170 self.assertEqual(DockerStorageDriver.overlay,
171 config.docker_storage_driver)
172 self.assertEqual('8.8.4.4', config.dns_nameserver)
173 self.assertTrue(config.public)
174 self.assertFalse(config.tls_disabled)
175 self.assertEqual('http://foo:8080', config.http_proxy)
176 self.assertEqual('https://foo:443', config.https_proxy)
177 self.assertEqual('foo,bar', config.no_proxy)
178 self.assertEqual('volDriver', config.volume_driver)
179 self.assertFalse(config.master_lb_enabled)
180 self.assertEqual(labels, config.labels)