urgent bug fixes for danube (2)
[doctor.git] / 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
15
16 def get_identity_auth():
17     auth_url = os.environ['OS_AUTH_URL']
18     username = os.environ['OS_USERNAME']
19     password = os.environ['OS_PASSWORD']
20     user_domain_name = os.environ.get('OS_USER_DOMAIN_NAME')
21     project_name = os.environ.get('OS_PROJECT_NAME') or os.environ.get('OS_TENANT_NAME')
22     project_domain_name = os.environ.get('OS_PROJECT_DOMAIN_NAME')
23     if auth_url.endswith('v3'):
24         return v3.Password(auth_url=auth_url,
25                            username=username,
26                            password=password,
27                            user_domain_name=user_domain_name,
28                            project_name=project_name,
29                            project_domain_name=project_domain_name)
30     else:
31         return v2.Password(auth_url=auth_url,
32                            username=username,
33                            password=password,
34                            tenant_name=project_name)