snaps/openstack/tests/create_user_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 uuid
  16 import unittest
  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
  20
  21 __author__ = 'spisarski'
  22
  23
  24 class UserSettingsUnitTests(unittest.TestCase):
  25     """
  26     Tests the construction of the UserSettings class
  27     """
  28
  29     def test_no_params(self):
  30         with self.assertRaises(Exception):
  31             UserSettings()
  32
  33     def test_empty_config(self):
  34         with self.assertRaises(Exception):
  35             UserSettings(config=dict())
  36
  37     def test_name_only(self):
  38         with self.assertRaises(Exception):
  39             UserSettings(name='foo')
  40
  41     def test_config_with_name_only(self):
  42         with self.assertRaises(Exception):
  43             UserSettings(config={'name': 'foo'})
  44
  45     def test_name_pass_enabled_str(self):
  46         with self.assertRaises(Exception):
  47             UserSettings(name='foo', password='bar', enabled='true')
  48
  49     def test_config_with_name_pass_enabled_str(self):
  50         with self.assertRaises(Exception):
  51             UserSettings(config={'name': 'foo', 'password': 'bar', 'enabled': 'true'})
  52
  53     def test_name_pass_only(self):
  54         settings = UserSettings(name='foo', password='bar')
  55         self.assertEqual('foo', settings.name)
  56         self.assertEqual('bar', settings.password)
  57         self.assertIsNone(settings.project_name)
  58         self.assertIsNone(settings.email)
  59         self.assertTrue(settings.enabled)
  60
  61     def test_config_with_name_pass_only(self):
  62         settings = UserSettings(config={'name': 'foo', 'password': 'bar'})
  63         self.assertEqual('foo', settings.name)
  64         self.assertEqual('bar', settings.password)
  65         self.assertIsNone(settings.project_name)
  66         self.assertIsNone(settings.email)
  67         self.assertTrue(settings.enabled)
  68
  69     def test_all(self):
  70         settings = UserSettings(name='foo', password='bar', project_name='proj-foo', email='foo@bar.com', enabled=False)
  71         self.assertEqual('foo', settings.name)
  72         self.assertEqual('bar', settings.password)
  73         self.assertEqual('proj-foo', settings.project_name)
  74         self.assertEqual('foo@bar.com', settings.email)
  75         self.assertFalse(settings.enabled)
  76
  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.assertEqual('foo', settings.name)
  81         self.assertEqual('bar', settings.password)
  82         self.assertEqual('proj-foo', settings.project_name)
  83         self.assertEqual('foo@bar.com', settings.email)
  84         self.assertFalse(settings.enabled)
  85
  86
  87 class CreateUserSuccessTests(OSComponentTestCase):
  88     """
  89     Test for the CreateImage class defined in create_image.py
  90     """
  91
  92     def setUp(self):
  93         """
  94         Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
  95         within OpenStack
  96         """
  97         guid = str(uuid.uuid4())[:-19]
  98         guid = self.__class__.__name__ + '-' + guid
  99         self.user_settings = UserSettings(name=guid + '-name', password=guid + '-password')
 100
 101         self.keystone = keystone_utils.keystone_client(self.os_creds)
 102
 103         # Initialize for cleanup
 104         self.user_creator = None
 105
 106     def tearDown(self):
 107         """
 108         Cleans the image and downloaded image file
 109         """
 110         if self.user_creator:
 111             self.user_creator.clean()
 112
 113     def test_create_user(self):
 114         """
 115         Tests the creation of an OpenStack user.
 116         """
 117         self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
 118         created_user = self.user_creator.create()
 119         self.assertIsNotNone(created_user)
 120
 121         retrieved_user = keystone_utils.get_user(self.keystone, self.user_settings.name)
 122         self.assertIsNotNone(retrieved_user)
 123         self.assertEqual(created_user, retrieved_user)
 124
 125     def test_create_user_2x(self):
 126         """
 127         Tests the creation of an OpenStack user twice to ensure it only creates one.
 128         """
 129         self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
 130         created_user = self.user_creator.create()
 131         self.assertIsNotNone(created_user)
 132
 133         retrieved_user = keystone_utils.get_user(self.keystone, self.user_settings.name)
 134         self.assertIsNotNone(retrieved_user)
 135         self.assertEqual(created_user, retrieved_user)
 136
 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.assertEqual(retrieved_user, user2)
 140
 141     def test_create_delete_user(self):
 142         """
 143         Tests the creation of an OpenStack user then delete.
 144         """
 145         # Create Image
 146         self.user_creator = OpenStackUser(self.os_creds, self.user_settings)
 147         created_user = self.user_creator.create()
 148         self.assertIsNotNone(created_user)
 149
 150         keystone_utils.delete_user(self.keystone, created_user)
 151
 152         # Delete user
 153         self.user_creator.clean()
 154         self.assertIsNone(self.user_creator.get_user())