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.
18 from snaps.openstack.create_user import OpenStackUser, UserSettings
19 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
20 from snaps.openstack.utils import keystone_utils
22 __author__ = 'spisarski'
25 class UserSettingsUnitTests(unittest.TestCase):
27 Tests the construction of the UserSettings class
30 def test_no_params(self):
31 with self.assertRaises(Exception):
34 def test_empty_config(self):
35 with self.assertRaises(Exception):
36 UserSettings(**dict())
38 def test_name_only(self):
39 with self.assertRaises(Exception):
40 UserSettings(name='foo')
42 def test_config_with_name_only(self):
43 with self.assertRaises(Exception):
44 UserSettings(**{'name': 'foo'})
46 def test_name_pass_enabled_str(self):
47 with self.assertRaises(Exception):
48 UserSettings(name='foo', password='bar', enabled='true')
50 def test_config_with_name_pass_enabled_str(self):
51 with self.assertRaises(Exception):
53 **{'name': 'foo', 'password': 'bar', 'enabled': 'true'})
55 def test_name_pass_only(self):
56 settings = UserSettings(name='foo', password='bar')
57 self.assertEqual('foo', settings.name)
58 self.assertEqual('bar', settings.password)
59 self.assertIsNone(settings.project_name)
60 self.assertIsNone(settings.email)
61 self.assertTrue(settings.enabled)
63 def test_config_with_name_pass_only(self):
64 settings = UserSettings(**{'name': 'foo', 'password': 'bar'})
65 self.assertEqual('foo', settings.name)
66 self.assertEqual('bar', settings.password)
67 self.assertIsNone(settings.project_name)
68 self.assertIsNone(settings.email)
69 self.assertTrue(settings.enabled)
72 settings = UserSettings(name='foo', password='bar',
73 project_name='proj-foo', email='foo@bar.com',
75 self.assertEqual('foo', settings.name)
76 self.assertEqual('bar', settings.password)
77 self.assertEqual('proj-foo', settings.project_name)
78 self.assertEqual('foo@bar.com', settings.email)
79 self.assertFalse(settings.enabled)
81 def test_config_all(self):
82 settings = UserSettings(**{'name': 'foo', 'password': 'bar',
83 'project_name': 'proj-foo',
84 'email': 'foo@bar.com',
86 self.assertEqual('foo', settings.name)
87 self.assertEqual('bar', settings.password)
88 self.assertEqual('proj-foo', settings.project_name)
89 self.assertEqual('foo@bar.com', settings.email)
90 self.assertFalse(settings.enabled)
93 class CreateUserSuccessTests(OSComponentTestCase):
95 Test for the CreateImage class defined in create_image.py
100 Instantiates the CreateImage object that is responsible for downloading
101 and creating an OS image file within OpenStack
103 guid = str(uuid.uuid4())[:-19]
104 guid = self.__class__.__name__ + '-' + guid
105 self.user_settings = UserSettings(
107 password=guid + '-password',
108 roles={'admin': self.os_creds.project_name},
109 domain_name=self.os_creds.user_domain_name)
111 self.keystone = keystone_utils.keystone_client(self.os_creds)
113 # Initialize for cleanup
114 self.user_creator = None
118 Cleans the image and downloaded image file
120 if self.user_creator:
121 self.user_creator.clean()
123 def test_create_user(self):
125 Tests the creation of an OpenStack user.
127 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
128 created_user = self.user_creator.create()
129 self.assertIsNotNone(created_user)
131 retrieved_user = keystone_utils.get_user(self.keystone,
132 self.user_settings.name)
133 self.assertIsNotNone(retrieved_user)
134 self.assertEqual(created_user, retrieved_user)
136 def test_create_user_2x(self):
138 Tests the creation of an OpenStack user twice to ensure it only creates
141 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
142 created_user = self.user_creator.create()
143 self.assertIsNotNone(created_user)
145 retrieved_user = keystone_utils.get_user(self.keystone,
146 self.user_settings.name)
147 self.assertIsNotNone(retrieved_user)
148 self.assertEqual(created_user, retrieved_user)
150 # Create user for the second time to ensure it is the same
151 user2 = OpenStackUser(self.os_creds, self.user_settings).create()
152 self.assertEqual(retrieved_user, user2)
154 def test_create_delete_user(self):
156 Tests the creation of an OpenStack user then delete.
159 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
160 created_user = self.user_creator.create()
161 self.assertIsNotNone(created_user)
163 keystone_utils.delete_user(self.keystone, created_user)
166 self.user_creator.clean()
167 self.assertIsNone(self.user_creator.get_user())
169 def test_create_admin_user(self):
171 Tests the creation of an OpenStack user.
173 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
174 created_user = self.user_creator.create()
175 self.assertIsNotNone(created_user)
177 retrieved_user = keystone_utils.get_user(self.keystone,
178 self.user_settings.name)
179 self.assertIsNotNone(retrieved_user)
180 self.assertEqual(created_user, retrieved_user)
182 role = keystone_utils.get_role_by_name(self.keystone, 'admin')
183 self.assertIsNotNone(role)
185 os_proj = keystone_utils.get_project(
186 keystone=self.keystone, project_name=self.os_creds.project_name)
187 user_roles = keystone_utils.get_roles_by_user(
188 self.keystone, retrieved_user, os_proj)
189 self.assertIsNotNone(user_roles)
190 self.assertEqual(1, len(user_roles))
191 self.assertEqual(role.id, user_roles[0].id)