Added support for Glance v2
[snaps.git] / snaps / openstack / utils / tests / glance_utils_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 os
16 import shutil
17 import uuid
18
19 from snaps import file_utils
20 from snaps.openstack.tests import openstack_tests
21
22 from snaps.openstack.tests import validation_utils
23 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
24 from snaps.openstack.utils import glance_utils
25
26 __author__ = 'spisarski'
27
28
29 class GlanceSmokeTests(OSComponentTestCase):
30     """
31     Tests to ensure that the neutron client can communicate with the cloud
32     """
33
34     def test_glance_connect_success(self):
35         """
36         Tests to ensure that the proper credentials can connect.
37         """
38         glance = glance_utils.glance_client(self.os_creds)
39
40         users = glance.images.list()
41         self.assertIsNotNone(users)
42
43     def test_glance_connect_fail(self):
44         """
45         Tests to ensure that the improper credentials cannot connect.
46         """
47         from snaps.openstack.os_credentials import OSCreds
48
49         with self.assertRaises(Exception):
50             neutron = glance_utils.glance_client(OSCreds('user', 'pass', 'url', 'project'))
51             neutron.list_networks()
52
53
54 class GlanceUtilsTests(OSComponentTestCase):
55     """
56     Test for the CreateImage class defined in create_image.py
57     """
58
59     def setUp(self):
60         """
61         Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
62         within OpenStack
63         """
64         guid = uuid.uuid4()
65         self.image_name = self.__class__.__name__ + '-' + str(guid)
66         self.image = None
67         self.glance = glance_utils.glance_client(self.os_creds)
68
69         self.tmp_dir = 'tmp/' + str(guid)
70         if not os.path.exists(self.tmp_dir):
71             os.makedirs(self.tmp_dir)
72
73     def tearDown(self):
74         """
75         Cleans the remote OpenStack objects
76         """
77         if self.image:
78             glance_utils.delete_image(self.glance, self.image)
79
80         if os.path.exists(self.tmp_dir) and os.path.isdir(self.tmp_dir):
81             shutil.rmtree(self.tmp_dir)
82
83     def test_create_image_minimal_url(self):
84         """
85         Tests the glance_utils.create_image() function with a URL
86         """
87         os_image_settings = openstack_tests.cirros_url_image(name=self.image_name)
88
89         self.image = glance_utils.create_image(self.glance, os_image_settings)
90         self.assertIsNotNone(self.image)
91
92         self.assertEqual(self.image_name, self.image.name)
93
94         image = glance_utils.get_image(self.glance, os_image_settings.name)
95         self.assertIsNotNone(image)
96
97         validation_utils.objects_equivalent(self.image, image)
98
99     def test_create_image_minimal_file(self):
100         """
101         Tests the glance_utils.create_image() function with a file
102         """
103         url_image_settings = openstack_tests.cirros_url_image('foo')
104         image_file = file_utils.download(url_image_settings.url, self.tmp_dir)
105         file_image_settings = openstack_tests.file_image_test_settings(name=self.image_name, file_path=image_file.name)
106
107         self.image = glance_utils.create_image(self.glance, file_image_settings)
108         self.assertIsNotNone(self.image)
109         self.assertEqual(self.image_name, self.image.name)
110
111         image = glance_utils.get_image(self.glance, file_image_settings.name)
112         self.assertIsNotNone(image)
113         validation_utils.objects_equivalent(self.image, image)