Added support for Keystone v3
[promise.git] / promise / test / functest / run_promise_tests.py
1 #!/usr/bin/python
2 #
3 # Copyright (c) 2015 All rights reserved
4 # 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 #
8 # http://www.apache.org/licenses/LICENSE-2.0
9 #
10 import argparse
11 import json
12 import logging
13 import os
14 import re
15 import subprocess
16 import sys
17 import time
18
19 import functest.utils.openstack_utils as os_utils
20 from functest.utils.constants import CONST
21
22 parser = argparse.ArgumentParser()
23
24 parser.add_argument("-d", "--debug", help="Debug mode", action="store_true")
25 parser.add_argument("-r", "--report",
26                     help="Create json result file",
27                     action="store_true")
28 args = parser.parse_args()
29
30
31 PROMISE_REPO_DIR = '/src/promise'
32 RESULTS_DIR = CONST.dir_results
33
34 PROMISE_TENANT_NAME = CONST.promise_tenant_name
35 PROMISE_PROJECT_NAME = CONST.promise_tenant_name
36 PROMISE_PROJECT_DESCRIPTION = CONST.promise_tenant_description
37 PROMISE_USER_NAME = CONST.promise_user_name
38 PROMISE_USER_PWD = CONST.promise_user_pwd
39 PROMISE_IMAGE_NAME = CONST.promise_image_name
40 PROMISE_FLAVOR_NAME = CONST.promise_flavor_name
41 PROMISE_FLAVOR_VCPUS = CONST.promise_flavor_vcpus
42 PROMISE_FLAVOR_RAM = CONST.promise_flavor_ram
43 PROMISE_FLAVOR_DISK = CONST.promise_flavor_disk
44
45
46 GLANCE_IMAGE_FILENAME = CONST.openstack_image_file_name
47 GLANCE_IMAGE_FORMAT = CONST.openstack_image_disk_format
48 GLANCE_IMAGE_NAME = CONST.openstack_image_file_name
49 GLANCE_IMAGE_PATH = os.path.join(CONST.dir_functest_images,
50                                  GLANCE_IMAGE_FILENAME)
51
52 PROMISE_NET_NAME = CONST.promise_network_name
53 PROMISE_SUBNET_NAME = CONST.promise_subnet_name
54 PROMISE_SUBNET_CIDR = CONST.promise_subnet_cidr
55 PROMISE_ROUTER_NAME = CONST.promise_router_name
56
57
58 """ logging configuration """
59 logger = logging.getLogger('promise')
60
61
62 def main():
63     return_code = -1
64     os_auth = os.environ["OS_AUTH_URL"]
65
66     creds = os_utils.get_credentials()
67
68     try:
69         logger.info("Env variables")
70         logger.info("OS_AUTH_URL: %s" % os.environ["OS_AUTH_URL"])
71         logger.info("OS_IDENTITY_API_VERSION: %s " %
72                     os.environ["OS_IDENTITY_API_VERSION"])
73     except KeyError:
74         logger.error("Please set the OS environment variables")
75
76     keystone_client = os_utils.get_keystone_client()
77
78     user_id = os_utils.get_user_id(keystone_client, creds['username'])
79     if user_id == '':
80         logger.error("Error : Failed to get id of %s user" %
81                      creds['username'])
82         return return_code
83
84     logger.info("Creating project '%s'..." % PROMISE_PROJECT_NAME)
85     project_id = os_utils.create_tenant(
86         keystone_client, PROMISE_PROJECT_NAME, PROMISE_PROJECT_DESCRIPTION)
87     if not project_id:
88         logger.error("Error : Failed to create %s project"
89                      % PROMISE_PROJECT_NAME)
90         return return_code
91     logger.debug("Project '%s' created successfully." % PROMISE_PROJECT_NAME)
92
93     roles_name = ["admin", "Admin"]
94     role_id = ''
95     for role_name in roles_name:
96         if role_id == '':
97             role_id = os_utils.get_role_id(keystone_client, role_name)
98
99     if role_id == '':
100         logger.error("Error : Failed to get id for %s role" % role_name)
101         return return_code
102
103     logger.info("Adding role '%s' to project '%s'..."
104                 % (role_id, PROMISE_PROJECT_NAME))
105     if not os_utils.add_role_user(keystone_client, user_id,
106                                   role_id, project_id):
107         logger.error("Error : Failed to add %s on project %s" %
108                      (creds['username'], PROMISE_PROJECT_NAME))
109         return return_code
110     logger.debug("Role added successfully.")
111
112     logger.info("Creating user '%s'..." % PROMISE_USER_NAME)
113     user_id = os_utils.create_user(
114         keystone_client, PROMISE_USER_NAME, PROMISE_USER_PWD, None, project_id)
115
116     if not user_id:
117         logger.error("Error : Failed to create %s user" % PROMISE_USER_NAME)
118         return return_code
119     logger.debug("User '%s' created successfully." % PROMISE_USER_NAME)
120
121     nova_client = os_utils.get_nova_client()
122     glance_client = os_utils.get_glance_client()
123
124     logger.info("Creating image '%s' from '%s'..." % (PROMISE_IMAGE_NAME,
125                                                       GLANCE_IMAGE_PATH))
126
127     logger.info("Upload some OS images if it doesn't exist")
128
129     image_id = os_utils.get_image_id(glance_client, GLANCE_IMAGE_NAME)
130
131     if image_id == '':
132         logger.info("%s image doesn't exist on glance repo" % GLANCE_IMAGE_NAME)
133         logger.info("Try downloading this image and upload on glance !")
134         image_id = os_utils.create_glance_image(
135             glance_client, GLANCE_IMAGE_NAME, GLANCE_IMAGE_PATH)
136
137     if image_id == '':
138         logger.error("Failed to create the Glance image...")
139         return return_code
140
141     logger.debug("Image '%s' with ID '%s' created successfully."
142                  % (PROMISE_IMAGE_NAME, image_id))
143     flavor_id = os_utils.get_flavor_id(nova_client, PROMISE_FLAVOR_NAME)
144     if flavor_id == '':
145         logger.info("Creating flavor '%s'..." % PROMISE_FLAVOR_NAME)
146         flavor_id = os_utils.create_flavor(nova_client,
147                                            PROMISE_FLAVOR_NAME,
148                                            PROMISE_FLAVOR_RAM,
149                                            PROMISE_FLAVOR_DISK,
150                                            PROMISE_FLAVOR_VCPUS)
151         if not flavor_id:
152             logger.error("Failed to create the Flavor...")
153             return return_code
154         logger.debug("Flavor '%s' with ID '%s' created successfully." %
155                      (PROMISE_FLAVOR_NAME, flavor_id))
156     else:
157         logger.debug("Using existing flavor '%s' with ID '%s'..."
158                      % (PROMISE_FLAVOR_NAME, flavor_id))
159
160     network_dic = os_utils.create_shared_network_full(PROMISE_NET_NAME,
161                                                       PROMISE_SUBNET_NAME,
162                                                       PROMISE_ROUTER_NAME,
163                                                       PROMISE_SUBNET_CIDR)
164     if not network_dic:
165         logger.error("Failed to create the private network...")
166         return return_code
167
168     logger.info("Exporting environment variables...")
169     os.environ["NODE_ENV"] = "functest"
170     os.environ["OS_PASSWORD"] = PROMISE_USER_PWD
171     os.environ["OS_TEST_IMAGE"] = image_id
172     os.environ["OS_TEST_FLAVOR"] = flavor_id
173     os.environ["OS_TEST_NETWORK"] = network_dic["net_id"]
174     os.environ["OS_TENANT_NAME"] = PROMISE_TENANT_NAME
175     os.environ["OS_USERNAME"] = PROMISE_USER_NAME
176
177     os.chdir(PROMISE_REPO_DIR + '/source/')
178     results_file_name = os.path.join(RESULTS_DIR, 'promise-results.json')
179     results_file = open(results_file_name, 'w+')
180     cmd = 'npm run -s test -- --reporter json'
181
182     logger.info("Running command: %s" % cmd)
183     ret = subprocess.call(cmd, shell=True, stdout=results_file,
184                           stderr=subprocess.STDOUT)
185     results_file.close()
186
187     if ret == 0:
188         logger.info("The test succeeded.")
189         return_code = 0
190     else:
191         logger.info("The command '%s' failed." % cmd)
192
193     # Print output of file
194     with open(results_file_name, 'r') as results_file:
195         data = results_file.read()
196         logger.debug("\n%s" % data)
197         json_data = json.loads(data)
198
199         suites = json_data["stats"]["suites"]
200         tests = json_data["stats"]["tests"]
201         passes = json_data["stats"]["passes"]
202         pending = json_data["stats"]["pending"]
203         failures = json_data["stats"]["failures"]
204         start_time_json = json_data["stats"]["start"]
205         end_time = json_data["stats"]["end"]
206         duration = float(json_data["stats"]["duration"]) / float(1000)
207
208     logger.info("\n"
209                 "****************************************\n"
210                 "          Promise test report\n\n"
211                 "****************************************\n"
212                 " Suites:  \t%s\n"
213                 " Tests:   \t%s\n"
214                 " Passes:  \t%s\n"
215                 " Pending: \t%s\n"
216                 " Failures:\t%s\n"
217                 " Start:   \t%s\n"
218                 " End:     \t%s\n"
219                 " Duration:\t%s\n"
220                 "****************************************\n\n"
221                 % (suites, tests, passes, pending, failures,
222                    start_time_json, end_time, duration))
223     end_time = time.time()
224
225     return return_code
226
227
228 if __name__ == '__main__':
229     logging.basicConfig(level=logging.INFO)
230     sys.exit(main())