1 # Copyright (c) 2016 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.
17 from snaps.openstack.create_user import OpenStackUser, UserSettings
18 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
19 from snaps.openstack.utils import keystone_utils
21 __author__ = 'spisarski'
24 class UserSettingsUnitTests(unittest.TestCase):
26 Tests the construction of the UserSettings class
29 def test_no_params(self):
30 with self.assertRaises(Exception):
33 def test_empty_config(self):
34 with self.assertRaises(Exception):
35 UserSettings(config=dict())
37 def test_name_only(self):
38 with self.assertRaises(Exception):
39 UserSettings(name='foo')
41 def test_config_with_name_only(self):
42 with self.assertRaises(Exception):
43 UserSettings(config={'name': 'foo'})
45 def test_name_pass_enabled_str(self):
46 with self.assertRaises(Exception):
47 UserSettings(name='foo', password='bar', enabled='true')
49 def test_config_with_name_pass_enabled_str(self):
50 with self.assertRaises(Exception):
51 UserSettings(config={'name': 'foo', 'password': 'bar', 'enabled': 'true'})
53 def test_name_pass_only(self):
54 settings = UserSettings(name='foo', password='bar')
55 self.assertEquals('foo', settings.name)
56 self.assertEquals('bar', settings.password)
57 self.assertIsNone(settings.project_name)
58 self.assertIsNone(settings.email)
59 self.assertTrue(settings.enabled)
61 def test_config_with_name_pass_only(self):
62 settings = UserSettings(config={'name': 'foo', 'password': 'bar'})
63 self.assertEquals('foo', settings.name)
64 self.assertEquals('bar', settings.password)
65 self.assertIsNone(settings.project_name)
66 self.assertIsNone(settings.email)
67 self.assertTrue(settings.enabled)
70 settings = UserSettings(name='foo', password='bar', project_name='proj-foo', email='foo@bar.com', enabled=False)
71 self.assertEquals('foo', settings.name)
72 self.assertEquals('bar', settings.password)
73 self.assertEquals('proj-foo', settings.project_name)
74 self.assertEquals('foo@bar.com', settings.email)
75 self.assertFalse(settings.enabled)
77 def test_config_all(self):
78 settings = UserSettings(config={'name': 'foo', 'password': 'bar', 'project_name': 'proj-foo',
79 'email': 'foo@bar.com', 'enabled': False})
80 self.assertEquals('foo', settings.name)
81 self.assertEquals('bar', settings.password)
82 self.assertEquals('proj-foo', settings.project_name)
83 self.assertEquals('foo@bar.com', settings.email)
84 self.assertFalse(settings.enabled)
87 class CreateUserSuccessTests(OSComponentTestCase):
89 Test for the CreateImage class defined in create_image.py
94 Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
97 guid = str(uuid.uuid4())[:-19]
98 guid = self.__class__.__name__ + '-' + guid
99 self.user_settings = UserSettings(name=guid + '-name', password=guid + '-password')
101 self.keystone = keystone_utils.keystone_client(self.os_creds)
103 # Initialize for cleanup
104 self.user_creator = None
108 Cleans the image and downloaded image file
110 if self.user_creator:
111 self.user_creator.clean()
113 def test_create_user(self):
115 Tests the creation of an OpenStack user.
117 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
118 created_user = self.user_creator.create()
119 self.assertIsNotNone(created_user)
121 retrieved_user = keystone_utils.get_user(self.keystone, self.user_settings.name)
122 self.assertIsNotNone(retrieved_user)
123 self.assertEquals(created_user, retrieved_user)
125 def test_create_user_2x(self):
127 Tests the creation of an OpenStack user twice to ensure it only creates one.
129 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
130 created_user = self.user_creator.create()
131 self.assertIsNotNone(created_user)
133 retrieved_user = keystone_utils.get_user(self.keystone, self.user_settings.name)
134 self.assertIsNotNone(retrieved_user)
135 self.assertEquals(created_user, retrieved_user)
137 # Create user for the second time to ensure it is the same
138 user2 = OpenStackUser(self.os_creds, self.user_settings).create()
139 self.assertEquals(retrieved_user, user2)
141 def test_create_delete_user(self):
143 Tests the creation of an OpenStack user then delete.
146 self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
147 created_user = self.user_creator.create()
148 self.assertIsNotNone(created_user)
150 keystone_utils.delete_user(self.keystone, created_user)
153 self.user_creator.clean()
154 self.assertIsNone(self.user_creator.get_user())