# See the License for the specific language governing permissions and
# limitations under the License.
import logging
+
from keystoneclient.exceptions import NotFound
from snaps.openstack.os_credentials import OSCreds
-
from snaps.openstack.utils import keystone_utils
__author__ = 'spisarski'
def create(self, cleanup=False):
"""
Creates the user in OpenStack if it does not already exist
- :param cleanup: Denotes whether or not this is being called for cleanup or not
+ :param cleanup: Denotes whether or not this is being called for cleanup
:return: The OpenStack user object
"""
self.__keystone = keystone_utils.keystone_client(self.__os_creds)
- self.__user = keystone_utils.get_user(self.__keystone, self.user_settings.name)
+ self.__user = keystone_utils.get_user(self.__keystone,
+ self.user_settings.name)
if self.__user:
logger.info('Found user with name - ' + self.user_settings.name)
elif not cleanup:
- self.__user = keystone_utils.create_user(self.__keystone, self.user_settings)
+ self.__user = keystone_utils.create_user(self.__keystone,
+ self.user_settings)
else:
logger.info('Did not create user due to cleanup mode')
def get_os_creds(self, project_name=None):
"""
Returns an OSCreds object based on this user account and a project
- :param project_name: the name of the project to leverage in the credentials
+ :param project_name: the name of the project to leverage in the
+ credentials
:return:
"""
- return OSCreds(username=self.user_settings.name,
- password=self.user_settings.password,
- auth_url=self.__os_creds.auth_url,
- project_name=project_name,
- identity_api_version=self.__os_creds.identity_api_version,
- user_domain_id=self.__os_creds.user_domain_id,
- project_domain_id=self.__os_creds.project_domain_id,
- proxy_settings=self.__os_creds.proxy_settings)
+ return OSCreds(
+ username=self.user_settings.name,
+ password=self.user_settings.password,
+ auth_url=self.__os_creds.auth_url,
+ project_name=project_name,
+ identity_api_version=self.__os_creds.identity_api_version,
+ user_domain_id=self.__os_creds.user_domain_id,
+ project_domain_id=self.__os_creds.project_domain_id,
+ proxy_settings=self.__os_creds.proxy_settings)
class UserSettings:
- def __init__(self, config=None, name=None, password=None, project_name=None, domain_name='default', email=None,
- enabled=True):
+ def __init__(self, **kwargs):
"""
Constructor
- :param config: dict() object containing the configuration settings using the attribute names below as each
- member's the key and overrides any of the other parameters.
:param name: the user's name (required)
:param password: the user's password (required)
:param project_name: the user's primary project name (optional)
- :param domain_name: the user's domain name (default='default'). For v3 APIs
+ :param domain_name: the user's domain name (default='default'). For v3
+ APIs
:param email: the user's email address (optional)
- :param enabled: denotes whether or not the user is enabled (default True)
+ :param enabled: denotes whether or not the user is enabled
+ (default True)
"""
- if config:
- self.name = config.get('name')
- self.password = config.get('password')
- self.project_name = config.get('project_name')
- self.email = config.get('email')
-
- if config.get('domain_name'):
- self.domain_name = config['domain_name']
- else:
- self.domain_name = domain_name
-
- if config.get('enabled') is not None:
- self.enabled = config['enabled']
- else:
- self.enabled = enabled
+ self.name = kwargs.get('name')
+ self.password = kwargs.get('password')
+ self.project_name = kwargs.get('project_name')
+ self.email = kwargs.get('email')
+
+ if kwargs.get('domain_name'):
+ self.domain_name = kwargs['domain_name']
+ else:
+ self.domain_name = 'default'
+
+ if kwargs.get('enabled') is not None:
+ self.enabled = kwargs['enabled']
else:
- self.name = name
- self.password = password
- self.project_name = project_name
- self.email = email
- self.enabled = enabled
- self.domain_name = domain_name
+ self.enabled = True
if not self.name or not self.password:
- raise Exception('The attributes name and password are required for UserSettings')
+ raise Exception(
+ 'The attributes name and password are required for '
+ 'UserSettings')
if not isinstance(self.enabled, bool):
raise Exception('The attribute enabled must be of type boolean')
# 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.openstack.create_user import OpenStackUser, UserSettings
from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
from snaps.openstack.utils import keystone_utils
def test_empty_config(self):
with self.assertRaises(Exception):
- UserSettings(config=dict())
+ UserSettings(**dict())
def test_name_only(self):
with self.assertRaises(Exception):
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):
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.assertTrue(settings.enabled)
def test_config_with_name_pass_only(self):
- settings = UserSettings(config={'name': 'foo', 'password': 'bar'})
+ settings = UserSettings(**{'name': 'foo', 'password': 'bar'})
self.assertEqual('foo', settings.name)
self.assertEqual('bar', settings.password)
self.assertIsNone(settings.project_name)
self.assertTrue(settings.enabled)
def test_all(self):
- settings = UserSettings(name='foo', password='bar', project_name='proj-foo', email='foo@bar.com', enabled=False)
+ 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.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})
+ 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)
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 = UserSettings(name=guid + '-name',
+ password=guid + '-password')
self.keystone = keystone_utils.keystone_client(self.os_creds)
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.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.assertEqual(created_user, retrieved_user)
Code Review / snaps.git / commitdiff