Support both admin role name
[snaps.git] / snaps / openstack / tests / create_user_tests.py
index 1f7a163..c15a71f 100644 (file)
@@ -1,4 +1,4 @@
-# Copyright (c) 2016 Cable Television Laboratories, Inc. ("CableLabs")
+# Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
 #                    and others.  All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # 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 uuid
 import unittest
+import uuid
+
+from snaps.config.user import UserConfig
 from snaps.openstack.create_user import OpenStackUser, UserSettings
 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
 from snaps.openstack.utils import keystone_utils
@@ -32,7 +34,7 @@ class UserSettingsUnitTests(unittest.TestCase):
 
     def test_empty_config(self):
         with self.assertRaises(Exception):
-            UserSettings(config=dict())
+            UserSettings(**dict())
 
     def test_name_only(self):
         with self.assertRaises(Exception):
@@ -40,7 +42,7 @@ class UserSettingsUnitTests(unittest.TestCase):
 
     def test_config_with_name_only(self):
         with self.assertRaises(Exception):
-            UserSettings(config={'name': 'foo'})
+            UserSettings(**{'name': 'foo'})
 
     def test_name_pass_enabled_str(self):
         with self.assertRaises(Exception):
@@ -48,39 +50,44 @@ class UserSettingsUnitTests(unittest.TestCase):
 
     def test_config_with_name_pass_enabled_str(self):
         with self.assertRaises(Exception):
-            UserSettings(config={'name': 'foo', 'password': 'bar', 'enabled': 'true'})
+            UserSettings(
+                **{'name': 'foo', 'password': 'bar', 'enabled': 'true'})
 
     def test_name_pass_only(self):
         settings = UserSettings(name='foo', password='bar')
-        self.assertEquals('foo', settings.name)
-        self.assertEquals('bar', settings.password)
+        self.assertEqual('foo', settings.name)
+        self.assertEqual('bar', settings.password)
         self.assertIsNone(settings.project_name)
         self.assertIsNone(settings.email)
         self.assertTrue(settings.enabled)
 
     def test_config_with_name_pass_only(self):
-        settings = UserSettings(config={'name': 'foo', 'password': 'bar'})
-        self.assertEquals('foo', settings.name)
-        self.assertEquals('bar', settings.password)
+        settings = UserSettings(**{'name': 'foo', 'password': 'bar'})
+        self.assertEqual('foo', settings.name)
+        self.assertEqual('bar', settings.password)
         self.assertIsNone(settings.project_name)
         self.assertIsNone(settings.email)
         self.assertTrue(settings.enabled)
 
     def test_all(self):
-        settings = UserSettings(name='foo', password='bar', project_name='proj-foo', email='foo@bar.com', enabled=False)
-        self.assertEquals('foo', settings.name)
-        self.assertEquals('bar', settings.password)
-        self.assertEquals('proj-foo', settings.project_name)
-        self.assertEquals('foo@bar.com', settings.email)
+        settings = UserSettings(name='foo', password='bar',
+                                project_name='proj-foo', email='foo@bar.com',
+                                enabled=False)
+        self.assertEqual('foo', settings.name)
+        self.assertEqual('bar', settings.password)
+        self.assertEqual('proj-foo', settings.project_name)
+        self.assertEqual('foo@bar.com', settings.email)
         self.assertFalse(settings.enabled)
 
     def test_config_all(self):
-        settings = UserSettings(config={'name': 'foo', 'password': 'bar', 'project_name': 'proj-foo',
-                                        'email': 'foo@bar.com', 'enabled': False})
-        self.assertEquals('foo', settings.name)
-        self.assertEquals('bar', settings.password)
-        self.assertEquals('proj-foo', settings.project_name)
-        self.assertEquals('foo@bar.com', settings.email)
+        settings = UserSettings(**{'name': 'foo', 'password': 'bar',
+                                   'project_name': 'proj-foo',
+                                   'email': 'foo@bar.com',
+                                   'enabled': False})
+        self.assertEqual('foo', settings.name)
+        self.assertEqual('bar', settings.password)
+        self.assertEqual('proj-foo', settings.project_name)
+        self.assertEqual('foo@bar.com', settings.email)
         self.assertFalse(settings.enabled)
 
 
@@ -91,14 +98,19 @@ class CreateUserSuccessTests(OSComponentTestCase):
 
     def setUp(self):
         """
-        Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
-        within OpenStack
+        Instantiates the CreateImage object that is responsible for downloading
+        and creating an OS image file within OpenStack
         """
         guid = str(uuid.uuid4())[:-19]
         guid = self.__class__.__name__ + '-' + guid
-        self.user_settings = UserSettings(name=guid + '-name', password=guid + '-password')
+        self.user_settings = UserConfig(
+            name=guid + '-name',
+            password=guid + '-password',
+            roles={'admin': self.os_creds.project_name,
+                   'Admin': self.os_creds.project_name},
+            domain_name=self.os_creds.user_domain_name)
 
-        self.keystone = keystone_utils.keystone_client(self.os_creds)
+        self.keystone = keystone_utils.keystone_client(self.os_creds, self.os_session)
 
         # Initialize for cleanup
         self.user_creator = None
@@ -110,6 +122,8 @@ class CreateUserSuccessTests(OSComponentTestCase):
         if self.user_creator:
             self.user_creator.clean()
 
+        super(self.__class__, self).__clean__()
+
     def test_create_user(self):
         """
         Tests the creation of an OpenStack user.
@@ -118,25 +132,28 @@ class CreateUserSuccessTests(OSComponentTestCase):
         created_user = self.user_creator.create()
         self.assertIsNotNone(created_user)
 
-        retrieved_user = keystone_utils.get_user(self.keystone, self.user_settings.name)
+        retrieved_user = keystone_utils.get_user(self.keystone,
+                                                 self.user_settings.name)
         self.assertIsNotNone(retrieved_user)
-        self.assertEquals(created_user, retrieved_user)
+        self.assertEqual(created_user, retrieved_user)
 
     def test_create_user_2x(self):
         """
-        Tests the creation of an OpenStack user twice to ensure it only creates one.
+        Tests the creation of an OpenStack user twice to ensure it only creates
+        one.
         """
         self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
         created_user = self.user_creator.create()
         self.assertIsNotNone(created_user)
 
-        retrieved_user = keystone_utils.get_user(self.keystone, self.user_settings.name)
+        retrieved_user = keystone_utils.get_user(self.keystone,
+                                                 self.user_settings.name)
         self.assertIsNotNone(retrieved_user)
-        self.assertEquals(created_user, retrieved_user)
+        self.assertEqual(created_user, retrieved_user)
 
         # Create user for the second time to ensure it is the same
         user2 = OpenStackUser(self.os_creds, self.user_settings).create()
-        self.assertEquals(retrieved_user, user2)
+        self.assertEqual(retrieved_user, user2)
 
     def test_create_delete_user(self):
         """
@@ -153,3 +170,28 @@ class CreateUserSuccessTests(OSComponentTestCase):
         self.user_creator.clean()
         self.assertIsNone(self.user_creator.get_user())
 
+    def test_create_admin_user(self):
+        """
+        Tests the creation of an OpenStack user.
+        """
+        self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
+        created_user = self.user_creator.create()
+        self.assertIsNotNone(created_user)
+
+        retrieved_user = keystone_utils.get_user(self.keystone,
+                                                 self.user_settings.name)
+        self.assertIsNotNone(retrieved_user)
+        self.assertEqual(created_user, retrieved_user)
+
+        role = keystone_utils.get_role_by_name(self.keystone, 'admin')
+        if not role:
+            role = keystone_utils.get_role_by_name(self.keystone, 'Admin')
+        self.assertIsNotNone(role)
+
+        os_proj = keystone_utils.get_project(
+            keystone=self.keystone, project_name=self.os_creds.project_name)
+        user_roles = keystone_utils.get_roles_by_user(
+            self.keystone, retrieved_user, os_proj)
+        self.assertIsNotNone(user_roles)
+        self.assertEqual(1, len(user_roles))
+        self.assertEqual(role.id, user_roles[0].id)