Initial patch with all code from CableLabs repository.

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

diff --git a/snaps/openstack/utils/tests/glance_utils_tests.py b/snaps/openstack/utils/tests/glance_utils_tests.py
new file mode 100644 (file)
index 0000000..d13908b
--- /dev/null
+++ b/snaps/openstack/utils/tests/glance_utils_tests.py
@@ -0,0 +1,115 @@
+# Copyright (c) 2016 Cable Television Laboratories, Inc. ("CableLabs")
+#                    and others.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import os
+import shutil
+import uuid
+
+from snaps import file_utils
+from snaps.openstack.tests import openstack_tests
+
+from snaps.openstack.utils import nova_utils
+from snaps.openstack.tests import validation_utils
+from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
+from snaps.openstack.utils import glance_utils
+
+__author__ = 'spisarski'
+
+
+class GlanceSmokeTests(OSComponentTestCase):
+    """
+    Tests to ensure that the neutron client can communicate with the cloud
+    """
+
+    def test_glance_connect_success(self):
+        """
+        Tests to ensure that the proper credentials can connect.
+        """
+        glance = glance_utils.glance_client(self.os_creds)
+
+        users = glance.images.list()
+        self.assertIsNotNone(users)
+
+    def test_glance_connect_fail(self):
+        """
+        Tests to ensure that the improper credentials cannot connect.
+        """
+        from snaps.openstack.os_credentials import OSCreds
+
+        with self.assertRaises(Exception):
+            neutron = glance_utils.glance_client(OSCreds('user', 'pass', 'url', 'project'))
+            neutron.list_networks()
+
+
+class GlanceUtilsTests(OSComponentTestCase):
+    """
+    Test for the CreateImage class defined in create_image.py
+    """
+
+    def setUp(self):
+        """
+        Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
+        within OpenStack
+        """
+        guid = uuid.uuid4()
+        self.image_name = self.__class__.__name__ + '-' + str(guid)
+        self.image = None
+        self.nova = nova_utils.nova_client(self.os_creds)
+        self.glance = glance_utils.glance_client(self.os_creds)
+
+        self.tmp_dir = 'tmp/' + str(guid)
+        if not os.path.exists(self.tmp_dir):
+            os.makedirs(self.tmp_dir)
+
+    def tearDown(self):
+        """
+        Cleans the remote OpenStack objects
+        """
+        if self.image:
+            glance_utils.delete_image(self.glance, self.image)
+
+        if os.path.exists(self.tmp_dir) and os.path.isdir(self.tmp_dir):
+            shutil.rmtree(self.tmp_dir)
+
+    def test_create_image_minimal_url(self):
+        """
+        Tests the glance_utils.create_image() function with a URL
+        """
+        os_image_settings = openstack_tests.cirros_url_image(name=self.image_name)
+
+        self.image = glance_utils.create_image(self.glance, os_image_settings)
+        self.assertIsNotNone(self.image)
+
+        self.assertEqual(self.image_name, self.image.name)
+
+        image = glance_utils.get_image(self.nova, self.glance, os_image_settings.name)
+        self.assertIsNotNone(image)
+
+        validation_utils.objects_equivalent(self.image, image)
+
+    def test_create_image_minimal_file(self):
+        """
+        Tests the glance_utils.create_image() function with a file
+        """
+        url_image_settings = openstack_tests.cirros_url_image('foo')
+        image_file = file_utils.download(url_image_settings.url, self.tmp_dir)
+        file_image_settings = openstack_tests.file_image_test_settings(name=self.image_name, file_path=image_file.name)
+
+        self.image = glance_utils.create_image(self.glance, file_image_settings)
+        self.assertIsNotNone(self.image)
+        self.assertEqual(self.image_name, self.image.name)
+
+        image = glance_utils.get_image(self.nova, self.glance, file_image_settings.name)
+        self.assertIsNotNone(image)
+        validation_utils.objects_equivalent(self.image, image)