Implemented the ability to create Magnum Cluster Type objects.

[snaps.git] / snaps / openstack / utils / tests / magnum_utils_tests.py

diff --git a/snaps/openstack/utils/tests/magnum_utils_tests.py b/snaps/openstack/utils/tests/magnum_utils_tests.py
index 48ead80..9e47900 100644 (file)
--- a/snaps/openstack/utils/tests/magnum_utils_tests.py
+++ b/snaps/openstack/utils/tests/magnum_utils_tests.py
@@ -13,8 +13,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging
+import uuid
 
+from snaps.config.cluster_template import ClusterTemplateConfig
+from snaps.config.keypair import KeypairConfig
+from snaps.openstack.create_image import OpenStackImage
+from snaps.openstack.create_keypairs import OpenStackKeypair
 from snaps.openstack.os_credentials import OSCreds
+from snaps.openstack.tests import openstack_tests
 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
 from snaps.openstack.utils import magnum_utils
 
@@ -35,7 +41,7 @@ class MagnumSmokeTests(OSComponentTestCase):
         magnum = magnum_utils.magnum_client(self.os_creds)
 
         # This should not throw an exception
-        magnum.clusters.list()
+        self.assertIsNotNone(magnum.clusters.list())
 
     def test_nova_connect_fail(self):
         """
@@ -48,3 +54,121 @@ class MagnumSmokeTests(OSComponentTestCase):
                         auth_url=self.os_creds.auth_url,
                         project_name=self.os_creds.project_name,
                         proxy_settings=self.os_creds.proxy_settings))
+
+
+class MagnumUtilsTests(OSComponentTestCase):
+    """
+    Tests individual functions within magnum_utils.py
+    """
+
+    def setUp(self):
+        self.guid = self.__class__.__name__ + '-' + str(uuid.uuid4())
+        self.cluster_type_name = self.guid + '-cluster-type'
+        self.magnum = magnum_utils.magnum_client(self.os_creds)
+
+        metadata = self.image_metadata
+        if not metadata:
+            metadata = dict()
+        if 'extra_properties' not in metadata:
+            metadata['extra_properties'] = dict()
+        metadata['extra_properties']['os_distro'] = 'cirros'
+
+        os_image_settings = openstack_tests.cirros_image_settings(
+            name=self.guid + '-image', image_metadata=metadata)
+
+        self.image_creator = OpenStackImage(self.os_creds, os_image_settings)
+
+        keypair_priv_filepath = 'tmp/' + self.guid
+        keypair_pub_filepath = keypair_priv_filepath + '.pub'
+
+        self.keypair_creator = OpenStackKeypair(
+            self.os_creds, KeypairConfig(
+                name=self.guid + '-keypair',
+                public_filepath=keypair_pub_filepath,
+                private_filepath=keypair_priv_filepath))
+
+        self.cluster_template = None
+
+        try:
+            self.image_creator.create()
+            self.keypair_creator.create()
+        except:
+            self.tearDown()
+            raise
+
+    def tearDown(self):
+        if self.cluster_template:
+            try:
+                magnum_utils.delete_cluster_template(
+                    self.magnum, self.cluster_template.id)
+            except:
+                pass
+        if self.keypair_creator:
+            try:
+                self.keypair_creator.clean()
+            except:
+                pass
+        if self.image_creator:
+            try:
+                self.image_creator.clean()
+            except:
+                pass
+
+    def test_create_cluster_template_simple(self):
+        config = ClusterTemplateConfig(
+            name=self.cluster_type_name,
+            image=self.image_creator.image_settings.name,
+            keypair=self.keypair_creator.keypair_settings.name,
+            external_net=self.ext_net_name)
+
+        self.cluster_template = magnum_utils.create_cluster_template(
+            self.magnum, config)
+        self.assertIsNotNone(self.cluster_template)
+        self.assertTrue(
+            validate_cluster_template(config, self.cluster_template))
+
+
+def validate_cluster_template(tmplt_config, tmplt_obj):
+    """
+    Returns true if the configuration matches the ClusterTemplate object
+    :param tmplt_config: the ClusterTemplateConfig object
+    :param tmplt_obj: the ClusterTemplate domain object
+    :return: T/F
+    """
+    if not tmplt_config.network_driver:
+        network_driver = 'flannel'
+    else:
+        network_driver = tmplt_config.network_driver
+
+    return (
+        tmplt_config.coe.value == tmplt_obj.coe and
+        tmplt_config.dns_nameserver == tmplt_obj.dns_nameserver and
+        tmplt_config.docker_storage_driver.value
+            == tmplt_obj.docker_storage_driver and
+        tmplt_config.docker_volume_size == tmplt_obj.docker_volume_size and
+        tmplt_config.external_net == tmplt_obj.external_net and
+        tmplt_config.fixed_net == tmplt_obj.fixed_net and
+        tmplt_config.fixed_subnet == tmplt_obj.fixed_subnet and
+        tmplt_config.flavor == tmplt_obj.flavor and
+        tmplt_config.floating_ip_enabled == tmplt_obj.floating_ip_enabled and
+        tmplt_config.http_proxy == tmplt_obj.http_proxy and
+        tmplt_config.https_proxy == tmplt_obj.https_proxy and
+        tmplt_config.no_proxy == tmplt_obj.no_proxy and
+        tmplt_config.image == tmplt_obj.image and
+        tmplt_config.insecure_registry == tmplt_obj.insecure_registry and
+        tmplt_config.keypair == tmplt_obj.keypair and
+        tmplt_config.labels == tmplt_obj.labels and
+        tmplt_config.master_flavor == tmplt_obj.master_flavor and
+        tmplt_config.master_lb_enabled == tmplt_obj.master_lb_enabled and
+        tmplt_config.name == tmplt_obj.name and
+        network_driver == tmplt_obj.network_driver and
+        tmplt_config.no_proxy == tmplt_obj.no_proxy and
+        tmplt_config.public == tmplt_obj.public and
+        tmplt_config.registry_enabled == tmplt_obj.registry_enabled and
+        tmplt_config.server_type.value == tmplt_obj.server_type and
+        tmplt_config.tls_disabled == tmplt_obj.tls_disabled and
+        tmplt_config.volume_driver == tmplt_obj.volume_driver
+    )
+    # def test_create_cluster_simple(self):
+    #     cluster = magnum_utils.create_cluster(self.magnum, 'foo')
+    #     self.assertIsNotNone(cluster)