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