tests/identity_auth.py

   1 ##############################################################################
   2 # Copyright (c) 2017 NEC Corporation and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import os
  11
  12 from keystoneauth1.identity import v2
  13 from keystoneauth1.identity import v3
  14 from keystoneauth1 import session
  15
  16
  17 def get_identity_auth():
  18     auth_url = os.environ['OS_AUTH_URL']
  19     username = os.environ['OS_USERNAME']
  20     password = os.environ['OS_PASSWORD']
  21     user_domain_name = os.environ.get('OS_USER_DOMAIN_NAME')
  22     project_name = os.environ.get('OS_PROJECT_NAME') or os.environ.get('OS_TENANT_NAME')
  23     project_domain_name = os.environ.get('OS_PROJECT_DOMAIN_NAME')
  24     if auth_url.endswith('v3'):
  25         return v3.Password(auth_url=auth_url,
  26                            username=username,
  27                            password=password,
  28                            user_domain_name=user_domain_name,
  29                            project_name=project_name,
  30                            project_domain_name=project_domain_name)
  31     else:
  32         return v2.Password(auth_url=auth_url,
  33                            username=username,
  34                            password=password,
  35                            tenant_name=project_name)
  36
  37
  38 def get_session(auth=None):
  39     """Get a user credentials auth session."""
  40     if auth is None:
  41         auth = get_identity_auth()
  42     return session.Session(auth=auth)