snaps/domain/test/cluster_template_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 import unittest
  16
  17 from snaps.config.cluster_template import (
  18     ContainerOrchestrationEngine, ServerType, DockerStorageDriver)
  19 from snaps.domain.cluster_template import ClusterTemplate
  20
  21
  22 class ClusterTemplateUnitTests(unittest.TestCase):
  23     """
  24     Tests the construction of the ClusterTypeConfig class
  25     """
  26     def test_all_named(self):
  27         labels = {'foo': 'bar'}
  28         config = ClusterTemplate(
  29             id='tmplt-id', name='foo', image='bar', keypair='keys',
  30             network_driver='driver', external_net='external',
  31             docker_volume_size=99, server_type=ServerType.baremetal.value,
  32             flavor='testFlavor', master_flavor='masterFlavor',
  33             coe=ContainerOrchestrationEngine.kubernetes.value,
  34             fixed_net='fixedNet', fixed_subnet='fixedSubnet',
  35             registry_enabled=False,
  36             docker_storage_driver=DockerStorageDriver.overlay.value,
  37             dns_nameserver='8.8.4.4', public=True, tls=False,
  38             http_proxy='http://foo:8080', https_proxy='https://foo:443',
  39             no_proxy='foo,bar', volume_driver='volDriver',
  40             master_lb_enabled=False, labels=labels)
  41         self.assertIsNotNone(config)
  42         self.assertEqual('tmplt-id', config.id)
  43         self.assertEqual('foo', config.name)
  44         self.assertEqual('bar', config.image)
  45         self.assertEqual('keys', config.keypair)
  46         self.assertEqual('driver', config.network_driver)
  47         self.assertEqual('external', config.external_net)
  48         self.assertEqual(99, config.docker_volume_size)
  49         self.assertEqual(ServerType.baremetal.value, config.server_type)
  50         self.assertEqual('testFlavor', config.flavor)
  51         self.assertEqual('masterFlavor', config.master_flavor)
  52         self.assertEqual(ContainerOrchestrationEngine.kubernetes.value,
  53                          config.coe)
  54         self.assertEqual('fixedNet', config.fixed_net)
  55         self.assertEqual('fixedSubnet', config.fixed_subnet)
  56         self.assertFalse(config.registry_enabled)
  57         self.assertEqual(DockerStorageDriver.overlay.value,
  58                          config.docker_storage_driver)
  59         self.assertEqual('8.8.4.4', config.dns_nameserver)
  60         self.assertTrue(config.public)
  61         self.assertFalse(config.tls_disabled)
  62         self.assertEqual('http://foo:8080', config.http_proxy)
  63         self.assertEqual('https://foo:443', config.https_proxy)
  64         self.assertEqual('foo,bar', config.no_proxy)
  65         self.assertEqual('volDriver', config.volume_driver)
  66         self.assertFalse(config.master_lb_enabled)
  67         self.assertEqual(labels, config.labels)
  68
  69     def test_all_config(self):
  70         labels = {'foo': 'bar'}
  71         config = ClusterTemplate(**{
  72             'id': 'tmplt-id', 'name': 'foo', 'image': 'bar', 'keypair': 'keys',
  73             'network_driver': 'driver', 'external_net': 'external',
  74             'docker_volume_size': '99', 'server_type': 'baremetal',
  75             'flavor': 'testFlavor', 'master_flavor': 'masterFlavor',
  76             'coe': 'kubernetes', 'fixed_net': 'fixedNet',
  77             'fixed_subnet': 'fixedSubnet', 'registry_enabled': False,
  78             'docker_storage_driver': 'overlay', 'dns_nameserver': '8.8.4.4',
  79             'public': 'true', 'tls': 'false', 'http_proxy': 'http://foo:8080',
  80             'https_proxy': 'https://foo:443', 'no_proxy': 'foo,bar',
  81             'volume_driver': 'volDriver', 'master_lb_enabled': False,
  82             'labels': labels})
  83         self.assertIsNotNone(config)
  84         self.assertEqual('tmplt-id', config.id)
  85         self.assertEqual('foo', config.name)
  86         self.assertEqual('bar', config.image)
  87         self.assertEqual('keys', config.keypair)
  88         self.assertEqual('driver', config.network_driver)
  89         self.assertEqual('external', config.external_net)
  90         self.assertEqual(99, config.docker_volume_size)
  91         self.assertEqual(ServerType.baremetal.value, config.server_type)
  92         self.assertEqual('testFlavor', config.flavor)
  93         self.assertEqual('masterFlavor', config.master_flavor)
  94         self.assertEqual(ContainerOrchestrationEngine.kubernetes.value,
  95                          config.coe)
  96         self.assertEqual('fixedNet', config.fixed_net)
  97         self.assertEqual('fixedSubnet', config.fixed_subnet)
  98         self.assertFalse(config.registry_enabled)
  99         self.assertEqual(DockerStorageDriver.overlay.value,
 100                          config.docker_storage_driver)
 101         self.assertEqual('8.8.4.4', config.dns_nameserver)
 102         self.assertTrue(config.public)
 103         self.assertFalse(config.tls_disabled)
 104         self.assertEqual('http://foo:8080', config.http_proxy)
 105         self.assertEqual('https://foo:443', config.https_proxy)
 106         self.assertEqual('foo,bar', config.no_proxy)
 107         self.assertEqual('volDriver', config.volume_driver)
 108         self.assertFalse(config.master_lb_enabled)
 109         self.assertEqual(labels, config.labels)