blob: a0d84ea9f07c517cbc8c6c96737ff55e75c3bdf1 [file] [log] [blame]
Tsuyoshi Ozawa4e0238d2016-09-20 05:56:10 +09001#!/usr/bin/env python
Jan Tattermusch7897ae92017-06-07 22:57:36 +02002# Copyright 2015 gRPC authors.
Craig Tillerc2c79212015-02-16 12:00:01 -08003#
Jan Tattermusch7897ae92017-06-07 22:57:36 +02004# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
Craig Tillerc2c79212015-02-16 12:00:01 -08007#
Jan Tattermusch7897ae92017-06-07 22:57:36 +02008# http://www.apache.org/licenses/LICENSE-2.0
Craig Tillerc2c79212015-02-16 12:00:01 -08009#
Jan Tattermusch7897ae92017-06-07 22:57:36 +020010# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
Craig Tillerc2c79212015-02-16 12:00:01 -080015
Nicolas Nobleddef2462015-01-06 18:08:25 -080016"""Run tests in parallel."""
17
siddharthshukla0589e532016-07-07 16:08:01 +020018from __future__ import print_function
19
Nicolas Nobleddef2462015-01-06 18:08:25 -080020import argparse
Craig Tiller9279ac22016-01-20 17:05:23 -080021import ast
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -070022import collections
Nicolas Nobleddef2462015-01-06 18:08:25 -080023import glob
24import itertools
Craig Tiller261dd982015-01-16 16:41:45 -080025import json
David Garcia Quintas0727c102017-02-21 10:48:35 -080026import logging
Nicolas Nobleddef2462015-01-06 18:08:25 -080027import multiprocessing
Craig Tiller1cc11db2015-01-15 22:50:50 -080028import os
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -070029import os.path
Craig Tiller38fb8de2016-07-13 08:23:32 -070030import pipes
David Garcia Quintas79e389f2015-06-02 17:49:42 -070031import platform
32import random
Craig Tillerfe406ec2015-02-24 13:55:12 -080033import re
Craig Tiller82875232015-09-25 13:57:34 -070034import socket
David Garcia Quintas79e389f2015-06-02 17:49:42 -070035import subprocess
Nicolas Nobleddef2462015-01-06 18:08:25 -080036import sys
Craig Tillerf0a293e2015-10-12 10:05:50 -070037import tempfile
38import traceback
ctiller3040cb72015-01-07 12:13:17 -080039import time
siddharthshukla0589e532016-07-07 16:08:01 +020040from six.moves import urllib
Jan Tattermusch03c01062015-12-11 14:28:56 -080041import uuid
Siddharth Shuklad194f592017-03-11 19:12:43 +010042import six
Nicolas Nobleddef2462015-01-06 18:08:25 -080043
Jan Tattermusch5c79a312016-12-20 11:02:50 +010044import python_utils.jobset as jobset
45import python_utils.report_utils as report_utils
46import python_utils.watch_dirs as watch_dirs
Craig Tiller7dc4ea62017-02-02 16:08:05 -080047import python_utils.start_port_server as start_port_server
Matt Kwong52ff9862017-04-17 13:56:51 -070048try:
49 from python_utils.upload_test_results import upload_results_to_bq
50except (ImportError):
51 pass # It's ok to not import because this is only necessary to upload results to BQ.
Craig Tillerb361b4e2016-01-06 11:44:17 -080052
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070053gcp_utils_dir = os.path.abspath(os.path.join(
54 os.path.dirname(__file__), '../gcp/utils'))
55sys.path.append(gcp_utils_dir)
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070056
Jan Tattermusch3b5121b2016-02-22 17:41:05 -080057_ROOT = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../..'))
58os.chdir(_ROOT)
Craig Tiller2cc2b842015-02-27 11:38:31 -080059
60
Craig Tiller8f18ee62016-07-18 08:00:33 -070061_FORCE_ENVIRON_FOR_WRAPPERS = {
62 'GRPC_VERBOSITY': 'DEBUG',
63}
Craig Tiller06805272015-06-11 14:46:47 -070064
Craig Tiller123f1372016-06-15 15:06:14 -070065_POLLING_STRATEGIES = {
Vijay Paiadb76ee2017-05-16 08:50:55 -070066 'linux': ['epollsig', 'poll', 'poll-cv'],
Craig Tiller18545a32017-05-05 17:33:44 +000067# TODO(ctiller, sreecha): enable epoll1, epollex, epoll-thread-pool
Vijay Paiadb76ee2017-05-16 08:50:55 -070068 'mac': ['poll'],
Craig Tiller123f1372016-06-15 15:06:14 -070069}
70
71
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070072def get_flaky_tests(limit=None):
Craig Tillerd16abf82017-06-07 08:58:55 -070073 import big_query_utils
74
75 bq = big_query_utils.create_big_query()
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070076 query = """
77 SELECT
78 test_name,
79 SUM(result != 'PASSED'
80 AND result != 'SKIPPED') AS count_failed,
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070081 FROM
82 [grpc-testing:jenkins_test_results.aggregate_results]
83 WHERE
84 timestamp >= DATE_ADD(DATE(CURRENT_TIMESTAMP()), -1, "WEEK")
85 AND NOT REGEXP_MATCH(job_name, '.*portability.*')
86 AND REGEXP_MATCH(job_name, '.*master.*')
87 GROUP BY
88 test_name
89 HAVING
Craig Tillerc2587132017-06-06 15:26:26 -070090 count_failed > 0"""
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070091 if limit:
92 query += " limit {}".format(limit)
Craig Tillerd16abf82017-06-07 08:58:55 -070093 query_job = big_query_utils.sync_query_job(bq, 'grpc-testing', query)
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -070094 page = bq.jobs().getQueryResults(
95 pageToken=None,
96 **query_job['jobReference']).execute(num_retries=3)
97 flake_names = [row['f'][0]['v'] for row in page['rows']]
98 return flake_names
99
100
Craig Tillerd50993d2015-08-05 08:04:36 -0700101def platform_string():
Nicolas "Pixel" Noblef72d7b52015-12-03 03:07:43 +0100102 return jobset.platform_string()
Craig Tillerd50993d2015-08-05 08:04:36 -0700103
104
Craig Tiller38fb8de2016-07-13 08:23:32 -0700105_DEFAULT_TIMEOUT_SECONDS = 5 * 60
106
David Garcia Quintas03920252017-02-15 12:51:21 -0800107def run_shell_command(cmd, env=None, cwd=None):
108 try:
109 subprocess.check_output(cmd, shell=True, env=env, cwd=cwd)
110 except subprocess.CalledProcessError as e:
David Garcia Quintas0727c102017-02-21 10:48:35 -0800111 logging.exception("Error while running command '%s'. Exit status %d. Output:\n%s",
112 e.cmd, e.returncode, e.output)
David Garcia Quintas03920252017-02-15 12:51:21 -0800113 raise
Craig Tiller38fb8de2016-07-13 08:23:32 -0700114
Craig Tiller738c3342015-01-12 14:28:33 -0800115# SimpleConfig: just compile with CONFIG=config, and run the binary to test
Craig Tillera0f85172016-01-20 15:56:06 -0800116class Config(object):
Craig Tillerb50d1662015-01-15 17:28:21 -0800117
murgatroid99c36f6ea2016-10-03 09:24:09 -0700118 def __init__(self, config, environ=None, timeout_multiplier=1, tool_prefix=[], iomgr_platform='native'):
murgatroid99132ce6a2015-03-04 17:29:14 -0800119 if environ is None:
120 environ = {}
Craig Tiller738c3342015-01-12 14:28:33 -0800121 self.build_config = config
Craig Tiller547db2b2015-01-30 14:08:39 -0800122 self.environ = environ
murgatroid99132ce6a2015-03-04 17:29:14 -0800123 self.environ['CONFIG'] = config
Craig Tillera0f85172016-01-20 15:56:06 -0800124 self.tool_prefix = tool_prefix
Masood Malekghassemi26ea9e22015-10-09 15:19:17 -0700125 self.timeout_multiplier = timeout_multiplier
murgatroid99c36f6ea2016-10-03 09:24:09 -0700126 self.iomgr_platform = iomgr_platform
Craig Tiller738c3342015-01-12 14:28:33 -0800127
Craig Tiller38fb8de2016-07-13 08:23:32 -0700128 def job_spec(self, cmdline, timeout_seconds=_DEFAULT_TIMEOUT_SECONDS,
Craig Tillerde7edf82016-03-20 09:12:16 -0700129 shortname=None, environ={}, cpu_cost=1.0, flaky=False):
Craig Tiller49f61322015-03-03 13:02:11 -0800130 """Construct a jobset.JobSpec for a test under this config
131
132 Args:
133 cmdline: a list of strings specifying the command line the test
134 would like to run
Craig Tiller49f61322015-03-03 13:02:11 -0800135 """
Craig Tiller4fc90032015-05-21 10:39:52 -0700136 actual_environ = self.environ.copy()
siddharthshukla0589e532016-07-07 16:08:01 +0200137 for k, v in environ.items():
Craig Tiller4fc90032015-05-21 10:39:52 -0700138 actual_environ[k] = v
Craig Tiller336c21e2017-06-06 15:23:10 -0700139 if not flaky and shortname and shortname in flaky_tests:
140 print('Setting %s to flaky' % shortname)
141 flaky = True
Craig Tillera0f85172016-01-20 15:56:06 -0800142 return jobset.JobSpec(cmdline=self.tool_prefix + cmdline,
Jan Tattermusch9a7d30c2015-04-23 16:12:55 -0700143 shortname=shortname,
Craig Tiller4fc90032015-05-21 10:39:52 -0700144 environ=actual_environ,
Craig Tiller56c6b6a2016-01-20 08:27:37 -0800145 cpu_cost=cpu_cost,
Craig Tiller94d04a52016-01-20 10:58:23 -0800146 timeout_seconds=(self.timeout_multiplier * timeout_seconds if timeout_seconds else None),
Craig Tillerde7edf82016-03-20 09:12:16 -0700147 flake_retries=5 if flaky or args.allow_flakes else 0,
Craig Tiller35505de2015-10-08 13:31:33 -0700148 timeout_retries=3 if args.allow_flakes else 0)
Craig Tiller738c3342015-01-12 14:28:33 -0800149
150
murgatroid99cf08daf2015-09-21 15:33:16 -0700151def get_c_tests(travis, test_lang) :
152 out = []
153 platforms_str = 'ci_platforms' if travis else 'platforms'
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100154 with open('tools/run_tests/generated/tests.json') as f:
murgatroid9989899b12015-09-22 09:14:48 -0700155 js = json.load(f)
murgatroid99a3e244f2015-09-22 11:25:53 -0700156 return [tgt
157 for tgt in js
158 if tgt['language'] == test_lang and
159 platform_string() in tgt[platforms_str] and
160 not (travis and tgt['flaky'])]
murgatroid99cf08daf2015-09-21 15:33:16 -0700161
murgatroid99fafeeb32015-09-22 09:13:03 -0700162
Jan Tattermusch77db4322016-02-20 20:19:35 -0800163def _check_compiler(compiler, supported_compilers):
164 if compiler not in supported_compilers:
Jan Tattermuschb2531e22016-03-25 16:14:41 -0700165 raise Exception('Compiler %s not supported (on this platform).' % compiler)
166
167
168def _check_arch(arch, supported_archs):
169 if arch not in supported_archs:
170 raise Exception('Architecture %s not supported.' % arch)
Jan Tattermusch77db4322016-02-20 20:19:35 -0800171
172
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800173def _is_use_docker_child():
174 """Returns True if running running as a --use_docker child."""
175 return True if os.getenv('RUN_TESTS_COMMAND') else False
176
177
siddharthshukla2135a1b2016-08-04 02:11:53 +0200178_PythonConfigVars = collections.namedtuple(
179 '_ConfigVars', ['shell', 'builder', 'builder_prefix_arguments',
180 'venv_relative_python', 'toolchain', 'runner'])
181
182
183def _python_config_generator(name, major, minor, bits, config_vars):
184 return PythonConfig(
185 name,
186 config_vars.shell + config_vars.builder + config_vars.builder_prefix_arguments + [
187 _python_pattern_function(major=major, minor=minor, bits=bits)] + [
188 name] + config_vars.venv_relative_python + config_vars.toolchain,
189 config_vars.shell + config_vars.runner + [
190 os.path.join(name, config_vars.venv_relative_python[0])])
191
192
193def _pypy_config_generator(name, major, config_vars):
194 return PythonConfig(
195 name,
196 config_vars.shell + config_vars.builder + config_vars.builder_prefix_arguments + [
197 _pypy_pattern_function(major=major)] + [
198 name] + config_vars.venv_relative_python + config_vars.toolchain,
199 config_vars.shell + config_vars.runner + [
200 os.path.join(name, config_vars.venv_relative_python[0])])
201
202
203def _python_pattern_function(major, minor, bits):
204 # Bit-ness is handled by the test machine's environment
205 if os.name == "nt":
206 if bits == "64":
207 return '/c/Python{major}{minor}/python.exe'.format(
208 major=major, minor=minor, bits=bits)
209 else:
210 return '/c/Python{major}{minor}_{bits}bits/python.exe'.format(
211 major=major, minor=minor, bits=bits)
212 else:
213 return 'python{major}.{minor}'.format(major=major, minor=minor)
214
215
216def _pypy_pattern_function(major):
217 if major == '2':
218 return 'pypy'
219 elif major == '3':
220 return 'pypy3'
221 else:
222 raise ValueError("Unknown PyPy major version")
223
224
Craig Tillerc7449162015-01-16 14:42:10 -0800225class CLanguage(object):
226
Craig Tillere9c959d2015-01-18 10:23:26 -0800227 def __init__(self, make_target, test_lang):
Craig Tillerc7449162015-01-16 14:42:10 -0800228 self.make_target = make_target
Craig Tillerd50993d2015-08-05 08:04:36 -0700229 self.platform = platform_string()
Craig Tiller711bbe62015-08-19 12:35:16 -0700230 self.test_lang = test_lang
Craig Tillerc7449162015-01-16 14:42:10 -0800231
Jan Tattermusch77db4322016-02-20 20:19:35 -0800232 def configure(self, config, args):
233 self.config = config
234 self.args = args
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100235 if self.args.compiler == 'cmake':
236 _check_arch(self.args.arch, ['default'])
237 self._use_cmake = True
238 self._docker_distro = 'jessie'
239 self._make_options = []
240 elif self.platform == 'windows':
241 self._use_cmake = False
Jan Tattermuschc96caf82016-02-22 17:31:02 -0800242 self._make_options = [_windows_toolset_option(self.args.compiler),
243 _windows_arch_option(self.args.arch)]
Jan Tattermusch77db4322016-02-20 20:19:35 -0800244 else:
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100245 self._use_cmake = False
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800246 self._docker_distro, self._make_options = self._compiler_options(self.args.use_docker,
247 self.args.compiler)
murgatroid99c36f6ea2016-10-03 09:24:09 -0700248 if args.iomgr_platform == "uv":
249 cflags = '-DGRPC_UV '
250 try:
251 cflags += subprocess.check_output(['pkg-config', '--cflags', 'libuv']).strip() + ' '
murgatroid991687cab2016-10-11 11:42:01 -0700252 except (subprocess.CalledProcessError, OSError):
murgatroid99c36f6ea2016-10-03 09:24:09 -0700253 pass
254 try:
255 ldflags = subprocess.check_output(['pkg-config', '--libs', 'libuv']).strip() + ' '
murgatroid991687cab2016-10-11 11:42:01 -0700256 except (subprocess.CalledProcessError, OSError):
murgatroid99c36f6ea2016-10-03 09:24:09 -0700257 ldflags = '-luv '
258 self._make_options += ['EXTRA_CPPFLAGS={}'.format(cflags),
259 'EXTRA_LDLIBS={}'.format(ldflags)]
Jan Tattermusch77db4322016-02-20 20:19:35 -0800260
261 def test_specs(self):
Craig Tiller547db2b2015-01-30 14:08:39 -0800262 out = []
Jan Tattermusch77db4322016-02-20 20:19:35 -0800263 binaries = get_c_tests(self.args.travis, self.test_lang)
Craig Tiller946ce7a2016-04-06 10:35:58 -0700264 for target in binaries:
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100265 if self._use_cmake and target.get('boringssl', False):
266 # cmake doesn't build boringssl tests
267 continue
Craig Tiller123f1372016-06-15 15:06:14 -0700268 polling_strategies = (_POLLING_STRATEGIES.get(self.platform, ['all'])
Craig Tiller946ce7a2016-04-06 10:35:58 -0700269 if target.get('uses_polling', True)
270 else ['all'])
murgatroid992c287ca2016-10-07 09:55:35 -0700271 if self.args.iomgr_platform == 'uv':
272 polling_strategies = ['all']
Craig Tiller946ce7a2016-04-06 10:35:58 -0700273 for polling_strategy in polling_strategies:
274 env={'GRPC_DEFAULT_SSL_ROOTS_FILE_PATH':
Craig Tillerb29f1fe2017-03-28 15:49:23 -0700275 _ROOT + '/src/core/tsi/test_creds/ca.pem',
Craig Tiller8f18ee62016-07-18 08:00:33 -0700276 'GRPC_POLL_STRATEGY': polling_strategy,
277 'GRPC_VERBOSITY': 'DEBUG'}
Yuchen Zengfdae4bd2016-11-07 17:46:16 -0800278 resolver = os.environ.get('GRPC_DNS_RESOLVER', None);
279 if resolver:
280 env['GRPC_DNS_RESOLVER'] = resolver
Craig Tiller38fb8de2016-07-13 08:23:32 -0700281 shortname_ext = '' if polling_strategy=='all' else ' GRPC_POLL_STRATEGY=%s' % polling_strategy
Craig Tillerbc28cd62016-12-02 13:52:21 -0800282 timeout_scaling = 1
283 if polling_strategy == 'poll-cv':
284 timeout_scaling *= 5
Robbie Shadeca7effc2017-01-17 09:14:29 -0500285
Sree Kuchibhotla70d9ca42017-01-27 10:54:05 -0800286 if polling_strategy in target.get('excluded_poll_engines', []):
Sree Kuchibhotlab5517dd2017-01-27 14:18:18 -0800287 continue
Sree Kuchibhotla03370d32017-02-01 08:33:33 -0800288
Robbie Shadeca7effc2017-01-17 09:14:29 -0500289 # Scale overall test timeout if running under various sanitizers.
290 config = self.args.config
291 if ('asan' in config
292 or config == 'msan'
293 or config == 'tsan'
294 or config == 'ubsan'
295 or config == 'helgrind'
296 or config == 'memcheck'):
297 timeout_scaling *= 20
298
Craig Tillerb38197e2016-02-26 10:14:54 -0800299 if self.config.build_config in target['exclude_configs']:
300 continue
murgatroid99c36f6ea2016-10-03 09:24:09 -0700301 if self.args.iomgr_platform in target.get('exclude_iomgrs', []):
302 continue
Craig Tillerb38197e2016-02-26 10:14:54 -0800303 if self.platform == 'windows':
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100304 if self._use_cmake:
305 binary = 'cmake/build/%s/%s.exe' % (_MSBUILD_CONFIG[self.config.build_config], target['name'])
306 else:
307 binary = 'vsprojects/%s%s/%s.exe' % (
308 'x64/' if self.args.arch == 'x64' else '',
309 _MSBUILD_CONFIG[self.config.build_config],
310 target['name'])
Craig Tillerca62ff02016-02-24 22:22:57 -0800311 else:
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100312 if self._use_cmake:
313 binary = 'cmake/build/%s' % target['name']
314 else:
315 binary = 'bins/%s/%s' % (self.config.build_config, target['name'])
Craig Tillerab34b122016-11-28 13:19:12 -0800316 cpu_cost = target['cpu_cost']
317 if cpu_cost == 'capacity':
318 cpu_cost = multiprocessing.cpu_count()
Craig Tillerb38197e2016-02-26 10:14:54 -0800319 if os.path.isfile(binary):
320 if 'gtest' in target and target['gtest']:
321 # here we parse the output of --gtest_list_tests to build up a
322 # complete list of the tests contained in a binary
323 # for each test, we then add a job to run, filtering for just that
324 # test
325 with open(os.devnull, 'w') as fnull:
326 tests = subprocess.check_output([binary, '--gtest_list_tests'],
327 stderr=fnull)
328 base = None
329 for line in tests.split('\n'):
330 i = line.find('#')
331 if i >= 0: line = line[:i]
332 if not line: continue
333 if line[0] != ' ':
334 base = line.strip()
335 else:
336 assert base is not None
337 assert line[1] == ' '
338 test = base + line.strip()
David Garcia Quintas947e5302017-02-17 16:44:45 -0800339 cmdline = [binary, '--gtest_filter=%s' % test] + target['args']
Craig Tiller38fb8de2016-07-13 08:23:32 -0700340 out.append(self.config.job_spec(cmdline,
David Garcia Quintas947e5302017-02-17 16:44:45 -0800341 shortname='%s %s' % (' '.join(cmdline), shortname_ext),
Craig Tillerab34b122016-11-28 13:19:12 -0800342 cpu_cost=cpu_cost,
Craig Tillerbc28cd62016-12-02 13:52:21 -0800343 timeout_seconds=_DEFAULT_TIMEOUT_SECONDS * timeout_scaling,
Craig Tillerb38197e2016-02-26 10:14:54 -0800344 environ=env))
345 else:
346 cmdline = [binary] + target['args']
Craig Tiller38fb8de2016-07-13 08:23:32 -0700347 out.append(self.config.job_spec(cmdline,
348 shortname=' '.join(
349 pipes.quote(arg)
350 for arg in cmdline) +
351 shortname_ext,
Craig Tillerab34b122016-11-28 13:19:12 -0800352 cpu_cost=cpu_cost,
Craig Tillerc2278152016-03-21 08:59:54 -0700353 flaky=target.get('flaky', False),
Craig Tillerbc28cd62016-12-02 13:52:21 -0800354 timeout_seconds=target.get('timeout_seconds', _DEFAULT_TIMEOUT_SECONDS) * timeout_scaling,
Craig Tillerb38197e2016-02-26 10:14:54 -0800355 environ=env))
356 elif self.args.regex == '.*' or self.platform == 'windows':
siddharthshukla0589e532016-07-07 16:08:01 +0200357 print('\nWARNING: binary not found, skipping', binary)
Nicolas Noblee1445362015-05-11 17:40:26 -0700358 return sorted(out)
Craig Tillerc7449162015-01-16 14:42:10 -0800359
Jan Tattermusch77db4322016-02-20 20:19:35 -0800360 def make_targets(self):
Jan Tattermusch77db4322016-02-20 20:19:35 -0800361 if self.platform == 'windows':
Craig Tiller7bb3efd2015-09-01 08:04:03 -0700362 # don't build tools on windows just yet
363 return ['buildtests_%s' % self.make_target]
David Garcia Quintas8ff30322017-05-11 16:13:37 -0700364 return ['buildtests_%s' % self.make_target, 'tools_%s' % self.make_target,
365 'check_epollexclusive']
Craig Tillerc7449162015-01-16 14:42:10 -0800366
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800367 def make_options(self):
Jan Tattermuschc96caf82016-02-22 17:31:02 -0800368 return self._make_options;
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800369
murgatroid99256d3df2015-09-21 16:58:02 -0700370 def pre_build_steps(self):
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100371 if self._use_cmake:
372 if self.platform == 'windows':
373 return [['tools\\run_tests\\helper_scripts\\pre_build_cmake.bat']]
374 else:
375 return [['tools/run_tests/helper_scripts/pre_build_cmake.sh']]
Jan Tattermusch874aec02015-10-07 19:26:19 -0700376 else:
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100377 if self.platform == 'windows':
378 return [['tools\\run_tests\\helper_scripts\\pre_build_c.bat']]
379 else:
380 return []
murgatroid99256d3df2015-09-21 16:58:02 -0700381
Craig Tillerc7449162015-01-16 14:42:10 -0800382 def build_steps(self):
383 return []
384
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200385 def post_tests_steps(self):
386 if self.platform == 'windows':
387 return []
388 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100389 return [['tools/run_tests/helper_scripts/post_tests_c.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200390
murgatroid99a3e244f2015-09-22 11:25:53 -0700391 def makefile_name(self):
Jan Tattermuschc98bde62017-01-25 19:12:11 +0100392 if self._use_cmake:
393 return 'cmake/build/Makefile'
394 else:
395 return 'Makefile'
murgatroid99a3e244f2015-09-22 11:25:53 -0700396
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700397 def _clang_make_options(self, version_suffix=''):
398 return ['CC=clang%s' % version_suffix,
399 'CXX=clang++%s' % version_suffix,
400 'LD=clang%s' % version_suffix,
401 'LDXX=clang++%s' % version_suffix]
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800402
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700403 def _gcc_make_options(self, version_suffix):
404 return ['CC=gcc%s' % version_suffix,
405 'CXX=g++%s' % version_suffix,
406 'LD=gcc%s' % version_suffix,
407 'LDXX=g++%s' % version_suffix]
Jan Tattermusch9bb70622016-03-18 10:28:54 -0700408
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800409 def _compiler_options(self, use_docker, compiler):
410 """Returns docker distro and make options to use for given compiler."""
Jan Tattermuschfd3857b2016-06-03 12:24:03 -0700411 if not use_docker and not _is_use_docker_child():
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800412 _check_compiler(compiler, ['default'])
413
414 if compiler == 'gcc4.9' or compiler == 'default':
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800415 return ('jessie', [])
Matt Kwong029ed102016-11-01 18:04:47 -0700416 elif compiler == 'gcc4.8':
Matt Kwong1347e402016-11-02 18:07:40 -0700417 return ('jessie', self._gcc_make_options(version_suffix='-4.8'))
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800418 elif compiler == 'gcc5.3':
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800419 return ('ubuntu1604', [])
Ken Payson975b5102017-03-30 17:38:40 -0700420 elif compiler == 'gcc_musl':
421 return ('alpine', [])
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800422 elif compiler == 'clang3.4':
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700423 # on ubuntu1404, clang-3.4 alias doesn't exist, just use 'clang'
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800424 return ('ubuntu1404', self._clang_make_options())
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700425 elif compiler == 'clang3.5':
426 return ('jessie', self._clang_make_options(version_suffix='-3.5'))
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800427 elif compiler == 'clang3.6':
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700428 return ('ubuntu1604', self._clang_make_options(version_suffix='-3.6'))
429 elif compiler == 'clang3.7':
430 return ('ubuntu1604', self._clang_make_options(version_suffix='-3.7'))
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800431 else:
432 raise Exception('Compiler %s not supported.' % compiler)
433
Jan Tattermusch77db4322016-02-20 20:19:35 -0800434 def dockerfile_dir(self):
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800435 return 'tools/dockerfile/test/cxx_%s_%s' % (self._docker_distro,
436 _docker_arch_suffix(self.args.arch))
Jan Tattermusch788ee232016-01-26 12:19:44 -0800437
murgatroid99132ce6a2015-03-04 17:29:14 -0800438 def __str__(self):
439 return self.make_target
440
Craig Tillercc0535d2015-12-08 15:14:47 -0800441
murgatroid992c8d5162015-01-26 10:41:21 -0800442class NodeLanguage(object):
443
Jan Tattermusche477b842016-02-06 22:19:01 -0800444 def __init__(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800445 self.platform = platform_string()
Jan Tattermusche477b842016-02-06 22:19:01 -0800446
Jan Tattermusch77db4322016-02-20 20:19:35 -0800447 def configure(self, config, args):
448 self.config = config
449 self.args = args
murgatroid99eaf79642016-11-01 11:05:02 -0700450 # Note: electron ABI only depends on major and minor version, so that's all
451 # we should specify in the compiler argument
murgatroid999fab4382016-04-29 15:05:00 -0700452 _check_compiler(self.args.compiler, ['default', 'node0.12',
murgatroid99eaf79642016-11-01 11:05:02 -0700453 'node4', 'node5', 'node6',
murgatroid99f25f5052017-04-13 17:22:32 -0700454 'node7', 'electron1.3', 'electron1.6'])
murgatroid999fab4382016-04-29 15:05:00 -0700455 if self.args.compiler == 'default':
murgatroid99c5181982017-01-12 12:40:11 -0800456 self.runtime = 'node'
murgatroid991191b722017-02-08 11:56:52 -0800457 self.node_version = '7'
murgatroid999fab4382016-04-29 15:05:00 -0700458 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700459 if self.args.compiler.startswith('electron'):
460 self.runtime = 'electron'
461 self.node_version = self.args.compiler[8:]
462 else:
463 self.runtime = 'node'
464 # Take off the word "node"
465 self.node_version = self.args.compiler[4:]
Jan Tattermusch77db4322016-02-20 20:19:35 -0800466
467 def test_specs(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800468 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100469 return [self.config.job_spec(['tools\\run_tests\\helper_scripts\\run_node.bat'])]
Michael Lumishaaa876a2016-02-10 15:27:58 -0800470 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700471 run_script = 'run_node'
472 if self.runtime == 'electron':
473 run_script += '_electron'
murgatroid99c34cac22017-01-04 15:43:02 -0800474 return [self.config.job_spec(['tools/run_tests/helper_scripts/{}.sh'.format(run_script),
murgatroid99eaf79642016-11-01 11:05:02 -0700475 self.node_version],
Jan Tattermusch77db4322016-02-20 20:19:35 -0800476 None,
477 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
murgatroid992c8d5162015-01-26 10:41:21 -0800478
murgatroid99256d3df2015-09-21 16:58:02 -0700479 def pre_build_steps(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800480 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100481 return [['tools\\run_tests\\helper_scripts\\pre_build_node.bat']]
Michael Lumishaaa876a2016-02-10 15:27:58 -0800482 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700483 build_script = 'pre_build_node'
484 if self.runtime == 'electron':
485 build_script += '_electron'
murgatroid991191b722017-02-08 11:56:52 -0800486 return [['tools/run_tests/helper_scripts/{}.sh'.format(build_script),
487 self.node_version]]
murgatroid99256d3df2015-09-21 16:58:02 -0700488
Jan Tattermusch77db4322016-02-20 20:19:35 -0800489 def make_targets(self):
murgatroid99db5b1602015-10-01 13:20:11 -0700490 return []
murgatroid992c8d5162015-01-26 10:41:21 -0800491
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800492 def make_options(self):
493 return []
494
murgatroid992c8d5162015-01-26 10:41:21 -0800495 def build_steps(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800496 if self.platform == 'windows':
murgatroid9988113f72017-02-21 10:04:29 -0800497 if self.config == 'dbg':
murgatroid991191b722017-02-08 11:56:52 -0800498 config_flag = '--debug'
499 else:
500 config_flag = '--release'
501 return [['tools\\run_tests\\helper_scripts\\build_node.bat',
murgatroid991191b722017-02-08 11:56:52 -0800502 config_flag]]
Michael Lumishaaa876a2016-02-10 15:27:58 -0800503 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700504 build_script = 'build_node'
505 if self.runtime == 'electron':
506 build_script += '_electron'
507 # building for electron requires a patch version
508 self.node_version += '.0'
murgatroid991191b722017-02-08 11:56:52 -0800509 return [['tools/run_tests/helper_scripts/{}.sh'.format(build_script),
murgatroid99f94f64f2017-04-24 13:35:21 -0700510 self.node_version]]
Craig Tillerc7449162015-01-16 14:42:10 -0800511
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200512 def post_tests_steps(self):
513 return []
514
murgatroid99a3e244f2015-09-22 11:25:53 -0700515 def makefile_name(self):
516 return 'Makefile'
517
Jan Tattermusch77db4322016-02-20 20:19:35 -0800518 def dockerfile_dir(self):
519 return 'tools/dockerfile/test/node_jessie_%s' % _docker_arch_suffix(self.args.arch)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800520
murgatroid99132ce6a2015-03-04 17:29:14 -0800521 def __str__(self):
522 return 'node'
523
Craig Tiller99775822015-01-30 13:07:16 -0800524
Craig Tillerc7449162015-01-16 14:42:10 -0800525class PhpLanguage(object):
526
Jan Tattermusch77db4322016-02-20 20:19:35 -0800527 def configure(self, config, args):
528 self.config = config
529 self.args = args
530 _check_compiler(self.args.compiler, ['default'])
531
532 def test_specs(self):
Jan Tattermuschfffb2672016-12-19 15:24:45 +0100533 return [self.config.job_spec(['src/php/bin/run_tests.sh'],
Jan Tattermusch77db4322016-02-20 20:19:35 -0800534 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
Craig Tillerc7449162015-01-16 14:42:10 -0800535
murgatroid99256d3df2015-09-21 16:58:02 -0700536 def pre_build_steps(self):
537 return []
538
Jan Tattermusch77db4322016-02-20 20:19:35 -0800539 def make_targets(self):
Craig Tilleraf7cf542015-05-22 10:07:34 -0700540 return ['static_c', 'shared_c']
Craig Tillerc7449162015-01-16 14:42:10 -0800541
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800542 def make_options(self):
543 return []
544
Craig Tillerc7449162015-01-16 14:42:10 -0800545 def build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100546 return [['tools/run_tests/helper_scripts/build_php.sh']]
Craig Tillerc7449162015-01-16 14:42:10 -0800547
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200548 def post_tests_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100549 return [['tools/run_tests/helper_scripts/post_tests_php.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200550
murgatroid99a3e244f2015-09-22 11:25:53 -0700551 def makefile_name(self):
552 return 'Makefile'
553
Jan Tattermusch77db4322016-02-20 20:19:35 -0800554 def dockerfile_dir(self):
555 return 'tools/dockerfile/test/php_jessie_%s' % _docker_arch_suffix(self.args.arch)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800556
murgatroid99132ce6a2015-03-04 17:29:14 -0800557 def __str__(self):
558 return 'php'
559
Craig Tillerc7449162015-01-16 14:42:10 -0800560
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700561class Php7Language(object):
562
563 def configure(self, config, args):
564 self.config = config
565 self.args = args
566 _check_compiler(self.args.compiler, ['default'])
567
568 def test_specs(self):
Jan Tattermuschfffb2672016-12-19 15:24:45 +0100569 return [self.config.job_spec(['src/php/bin/run_tests.sh'],
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700570 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
571
572 def pre_build_steps(self):
573 return []
574
575 def make_targets(self):
576 return ['static_c', 'shared_c']
577
578 def make_options(self):
579 return []
580
581 def build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100582 return [['tools/run_tests/helper_scripts/build_php.sh']]
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700583
584 def post_tests_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100585 return [['tools/run_tests/helper_scripts/post_tests_php.sh']]
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700586
587 def makefile_name(self):
588 return 'Makefile'
589
590 def dockerfile_dir(self):
591 return 'tools/dockerfile/test/php7_jessie_%s' % _docker_arch_suffix(self.args.arch)
592
593 def __str__(self):
594 return 'php7'
595
596
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700597class PythonConfig(collections.namedtuple('PythonConfig', [
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700598 'name', 'build', 'run'])):
599 """Tuple of commands (named s.t. 'what it says on the tin' applies)"""
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700600
Nathaniel Manista840615e2015-01-22 20:31:47 +0000601class PythonLanguage(object):
602
Jan Tattermusch77db4322016-02-20 20:19:35 -0800603 def configure(self, config, args):
604 self.config = config
605 self.args = args
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700606 self.pythons = self._get_pythons(self.args)
Jan Tattermusch77db4322016-02-20 20:19:35 -0800607
608 def test_specs(self):
Jan Tattermusch072ebaa2016-03-01 18:33:12 -0800609 # load list of known test suites
Masood Malekghassemi1ff429d2016-06-02 16:39:20 -0700610 with open('src/python/grpcio_tests/tests/tests.json') as tests_json_file:
Jan Tattermusch072ebaa2016-03-01 18:33:12 -0800611 tests_json = json.load(tests_json_file)
Masood Malekghassemi2b841622015-07-28 17:39:02 -0700612 environment = dict(_FORCE_ENVIRON_FOR_WRAPPERS)
Masood Malekghassemi1c062bd2016-06-13 18:41:36 -0700613 return [self.config.job_spec(
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700614 config.run,
Masood Malekghassemie6a23e22016-06-28 13:58:42 -0700615 timeout_seconds=5*60,
siddharthshukla0589e532016-07-07 16:08:01 +0200616 environ=dict(list(environment.items()) +
Masood Malekghassemic4f5a2e2016-12-05 15:54:56 -0800617 [('GRPC_PYTHON_TESTRUNNER_FILTER', str(suite_name))]),
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700618 shortname='%s.test.%s' % (config.name, suite_name),)
Masood Malekghassemi1c062bd2016-06-13 18:41:36 -0700619 for suite_name in tests_json
620 for config in self.pythons]
Nathaniel Manista840615e2015-01-22 20:31:47 +0000621
murgatroid99256d3df2015-09-21 16:58:02 -0700622 def pre_build_steps(self):
623 return []
624
Jan Tattermusch77db4322016-02-20 20:19:35 -0800625 def make_targets(self):
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700626 return []
Nathaniel Manista840615e2015-01-22 20:31:47 +0000627
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800628 def make_options(self):
629 return []
630
Nathaniel Manista840615e2015-01-22 20:31:47 +0000631 def build_steps(self):
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700632 return [config.build for config in self.pythons]
Nathaniel Manista840615e2015-01-22 20:31:47 +0000633
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200634 def post_tests_steps(self):
Matt Kwong0ff19572017-03-30 18:22:21 -0700635 if self.config != 'gcov':
636 return []
637 else:
638 return [['tools/run_tests/helper_scripts/post_tests_python.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200639
murgatroid99a3e244f2015-09-22 11:25:53 -0700640 def makefile_name(self):
641 return 'Makefile'
642
Jan Tattermusch77db4322016-02-20 20:19:35 -0800643 def dockerfile_dir(self):
siddharthshuklac4782142016-06-28 18:48:47 +0200644 return 'tools/dockerfile/test/python_%s_%s' % (self.python_manager_name(), _docker_arch_suffix(self.args.arch))
645
646 def python_manager_name(self):
Ken Payson975b5102017-03-30 17:38:40 -0700647 if self.args.compiler in ['python3.5', 'python3.6']:
648 return 'pyenv'
649 elif self.args.compiler == 'python_alpine':
650 return 'alpine'
651 else:
652 return 'jessie'
Jan Tattermusch788ee232016-01-26 12:19:44 -0800653
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700654 def _get_pythons(self, args):
655 if args.arch == 'x86':
656 bits = '32'
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700657 else:
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700658 bits = '64'
siddharthshukla2135a1b2016-08-04 02:11:53 +0200659
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700660 if os.name == 'nt':
661 shell = ['bash']
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100662 builder = [os.path.abspath('tools/run_tests/helper_scripts/build_python_msys2.sh')]
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700663 builder_prefix_arguments = ['MINGW{}'.format(bits)]
664 venv_relative_python = ['Scripts/python.exe']
665 toolchain = ['mingw32']
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700666 else:
667 shell = []
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100668 builder = [os.path.abspath('tools/run_tests/helper_scripts/build_python.sh')]
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700669 builder_prefix_arguments = []
670 venv_relative_python = ['bin/python']
671 toolchain = ['unix']
siddharthshukla2135a1b2016-08-04 02:11:53 +0200672
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100673 runner = [os.path.abspath('tools/run_tests/helper_scripts/run_python.sh')]
siddharthshukla2135a1b2016-08-04 02:11:53 +0200674 config_vars = _PythonConfigVars(shell, builder, builder_prefix_arguments,
675 venv_relative_python, toolchain, runner)
676 python27_config = _python_config_generator(name='py27', major='2',
677 minor='7', bits=bits,
678 config_vars=config_vars)
679 python34_config = _python_config_generator(name='py34', major='3',
680 minor='4', bits=bits,
681 config_vars=config_vars)
682 python35_config = _python_config_generator(name='py35', major='3',
683 minor='5', bits=bits,
684 config_vars=config_vars)
685 python36_config = _python_config_generator(name='py36', major='3',
686 minor='6', bits=bits,
687 config_vars=config_vars)
688 pypy27_config = _pypy_config_generator(name='pypy', major='2',
689 config_vars=config_vars)
690 pypy32_config = _pypy_config_generator(name='pypy3', major='3',
691 config_vars=config_vars)
692
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700693 if args.compiler == 'default':
694 if os.name == 'nt':
Ken Payson631862f2017-06-01 14:27:52 -0700695 return (python35_config,)
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700696 else:
697 return (python27_config, python34_config,)
698 elif args.compiler == 'python2.7':
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700699 return (python27_config,)
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700700 elif args.compiler == 'python3.4':
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700701 return (python34_config,)
siddharthshuklac4782142016-06-28 18:48:47 +0200702 elif args.compiler == 'python3.5':
703 return (python35_config,)
704 elif args.compiler == 'python3.6':
705 return (python36_config,)
siddharthshukla2135a1b2016-08-04 02:11:53 +0200706 elif args.compiler == 'pypy':
707 return (pypy27_config,)
708 elif args.compiler == 'pypy3':
709 return (pypy32_config,)
Ken Payson975b5102017-03-30 17:38:40 -0700710 elif args.compiler == 'python_alpine':
711 return (python27_config,)
Jan Tattermusch825471c2016-04-25 16:52:25 -0700712 else:
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700713 raise Exception('Compiler %s not supported.' % args.compiler)
Jan Tattermusch825471c2016-04-25 16:52:25 -0700714
murgatroid99132ce6a2015-03-04 17:29:14 -0800715 def __str__(self):
716 return 'python'
717
Craig Tillerd625d812015-04-08 15:52:35 -0700718
murgatroid996a4c4fa2015-02-27 12:08:57 -0800719class RubyLanguage(object):
720
Jan Tattermusch77db4322016-02-20 20:19:35 -0800721 def configure(self, config, args):
722 self.config = config
723 self.args = args
724 _check_compiler(self.args.compiler, ['default'])
725
726 def test_specs(self):
Alexander Polcync44c16e2017-03-14 17:44:21 -0700727 tests = [self.config.job_spec(['tools/run_tests/helper_scripts/run_ruby.sh'],
728 timeout_seconds=10*60,
729 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
Alexander Polcync44c16e2017-03-14 17:44:21 -0700730 tests.append(self.config.job_spec(['tools/run_tests/helper_scripts/run_ruby_end2end_tests.sh'],
731 timeout_seconds=10*60,
732 environ=_FORCE_ENVIRON_FOR_WRAPPERS))
733 return tests
murgatroid996a4c4fa2015-02-27 12:08:57 -0800734
murgatroid99256d3df2015-09-21 16:58:02 -0700735 def pre_build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100736 return [['tools/run_tests/helper_scripts/pre_build_ruby.sh']]
murgatroid99256d3df2015-09-21 16:58:02 -0700737
Jan Tattermusch4651bef2016-02-23 08:31:25 -0800738 def make_targets(self):
murgatroid997d243df2016-02-18 09:58:05 -0800739 return []
murgatroid996a4c4fa2015-02-27 12:08:57 -0800740
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800741 def make_options(self):
742 return []
743
murgatroid996a4c4fa2015-02-27 12:08:57 -0800744 def build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100745 return [['tools/run_tests/helper_scripts/build_ruby.sh']]
murgatroid996a4c4fa2015-02-27 12:08:57 -0800746
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200747 def post_tests_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100748 return [['tools/run_tests/helper_scripts/post_tests_ruby.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200749
murgatroid99a3e244f2015-09-22 11:25:53 -0700750 def makefile_name(self):
751 return 'Makefile'
752
Jan Tattermusch77db4322016-02-20 20:19:35 -0800753 def dockerfile_dir(self):
754 return 'tools/dockerfile/test/ruby_jessie_%s' % _docker_arch_suffix(self.args.arch)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800755
murgatroid99132ce6a2015-03-04 17:29:14 -0800756 def __str__(self):
757 return 'ruby'
758
Craig Tillerd625d812015-04-08 15:52:35 -0700759
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800760class CSharpLanguage(object):
Jan Tattermusch77db4322016-02-20 20:19:35 -0800761
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700762 def __init__(self):
Craig Tillerd50993d2015-08-05 08:04:36 -0700763 self.platform = platform_string()
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700764
Jan Tattermusch77db4322016-02-20 20:19:35 -0800765 def configure(self, config, args):
766 self.config = config
767 self.args = args
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700768 if self.platform == 'windows':
Jan Tattermusche7f0b852017-02-08 19:06:10 -0800769 _check_compiler(self.args.compiler, ['coreclr', 'default'])
Jan Tattermuschb2531e22016-03-25 16:14:41 -0700770 _check_arch(self.args.arch, ['default'])
Jan Tattermusch3c344d22017-04-04 16:49:06 +0200771 self._cmake_arch_option = 'x64'
Jan Tattermusche7f0b852017-02-08 19:06:10 -0800772 self._make_options = []
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700773 else:
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700774 _check_compiler(self.args.compiler, ['default', 'coreclr'])
Jan Tattermusch3c344d22017-04-04 16:49:06 +0200775 self._docker_distro = 'jessie'
Jan Tattermusch76511a52016-06-17 14:00:57 -0700776
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700777 if self.platform == 'mac':
Jan Tattermusch2a322c22016-03-30 13:55:07 -0700778 # TODO(jtattermusch): EMBED_ZLIB=true currently breaks the mac build
Jan Tattermusch6d082202016-06-21 10:03:38 -0700779 self._make_options = ['EMBED_OPENSSL=true']
780 if self.args.compiler != 'coreclr':
781 # On Mac, official distribution of mono is 32bit.
Craig Tiller90ce7232017-04-12 06:53:09 -0700782 self._make_options += ['ARCH_FLAGS=-m32', 'LDFLAGS=-m32']
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700783 else:
784 self._make_options = ['EMBED_OPENSSL=true', 'EMBED_ZLIB=true']
Jan Tattermusch77db4322016-02-20 20:19:35 -0800785
786 def test_specs(self):
Jan Tattermusch03c01062015-12-11 14:28:56 -0800787 with open('src/csharp/tests.json') as f:
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700788 tests_by_assembly = json.load(f)
Jan Tattermusch03c01062015-12-11 14:28:56 -0800789
Jan Tattermuscha2d964c2016-02-22 17:33:09 -0800790 msbuild_config = _MSBUILD_CONFIG[self.config.build_config]
Jan Tattermusch78cb9312017-04-11 11:04:42 +0200791 nunit_args = ['--labels=All', '--noresult', '--workers=1']
Jan Tattermusch76511a52016-06-17 14:00:57 -0700792 assembly_subdir = 'bin/%s' % msbuild_config
793 assembly_extension = '.exe'
794
795 if self.args.compiler == 'coreclr':
Jan Tattermusch1f7ce192016-09-08 16:21:16 +0200796 assembly_subdir += '/netcoreapp1.0'
797 runtime_cmd = ['dotnet', 'exec']
798 assembly_extension = '.dll'
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700799 else:
Jan Tattermusch3c344d22017-04-04 16:49:06 +0200800 assembly_subdir += '/net45'
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700801 if self.platform == 'windows':
802 runtime_cmd = []
803 else:
804 runtime_cmd = ['mono']
Jan Tattermuschbf3b1532015-10-26 10:24:42 -0700805
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700806 specs = []
Siddharth Shuklad194f592017-03-11 19:12:43 +0100807 for assembly in six.iterkeys(tests_by_assembly):
Jan Tattermusch76511a52016-06-17 14:00:57 -0700808 assembly_file = 'src/csharp/%s/%s/%s%s' % (assembly,
809 assembly_subdir,
810 assembly,
811 assembly_extension)
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700812 if self.config.build_config != 'gcov' or self.platform != 'windows':
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700813 # normally, run each test as a separate process
814 for test in tests_by_assembly[assembly]:
815 cmdline = runtime_cmd + [assembly_file, '--test=%s' % test] + nunit_args
816 specs.append(self.config.job_spec(cmdline,
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700817 shortname='csharp.%s' % test,
818 environ=_FORCE_ENVIRON_FOR_WRAPPERS))
819 else:
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700820 # For C# test coverage, run all tests from the same assembly at once
821 # using OpenCover.Console (only works on Windows).
822 cmdline = ['src\\csharp\\packages\\OpenCover.4.6.519\\tools\\OpenCover.Console.exe',
823 '-target:%s' % assembly_file,
824 '-targetdir:src\\csharp',
825 '-targetargs:%s' % ' '.join(nunit_args),
826 '-filter:+[Grpc.Core]*',
827 '-register:user',
828 '-output:src\\csharp\\coverage_csharp_%s.xml' % assembly]
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700829
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700830 # set really high cpu_cost to make sure instances of OpenCover.Console run exclusively
831 # to prevent problems with registering the profiler.
832 run_exclusive = 1000000
Jan Tattermusch35e608f2016-04-09 16:35:06 -0700833 specs.append(self.config.job_spec(cmdline,
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700834 shortname='csharp.coverage.%s' % assembly,
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700835 cpu_cost=run_exclusive,
Jan Tattermusch77db4322016-02-20 20:19:35 -0800836 environ=_FORCE_ENVIRON_FOR_WRAPPERS))
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700837 return specs
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800838
murgatroid99256d3df2015-09-21 16:58:02 -0700839 def pre_build_steps(self):
Jan Tattermusch48423fc2015-10-07 18:59:16 -0700840 if self.platform == 'windows':
Jan Tattermusche7f0b852017-02-08 19:06:10 -0800841 return [['tools\\run_tests\\helper_scripts\\pre_build_csharp.bat', self._cmake_arch_option]]
Jan Tattermusch48423fc2015-10-07 18:59:16 -0700842 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100843 return [['tools/run_tests/helper_scripts/pre_build_csharp.sh']]
murgatroid99256d3df2015-09-21 16:58:02 -0700844
Jan Tattermusch77db4322016-02-20 20:19:35 -0800845 def make_targets(self):
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700846 return ['grpc_csharp_ext']
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800847
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800848 def make_options(self):
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700849 return self._make_options;
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800850
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800851 def build_steps(self):
Jan Tattermusch3c344d22017-04-04 16:49:06 +0200852 if self.platform == 'windows':
853 return [['tools\\run_tests\\helper_scripts\\build_csharp.bat']]
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700854 else:
Jan Tattermusch3c344d22017-04-04 16:49:06 +0200855 return [['tools/run_tests/helper_scripts/build_csharp.sh']]
Nathaniel Manista840615e2015-01-22 20:31:47 +0000856
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200857 def post_tests_steps(self):
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700858 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100859 return [['tools\\run_tests\\helper_scripts\\post_tests_csharp.bat']]
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700860 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100861 return [['tools/run_tests/helper_scripts/post_tests_csharp.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200862
murgatroid99a3e244f2015-09-22 11:25:53 -0700863 def makefile_name(self):
Jan Tattermusche7f0b852017-02-08 19:06:10 -0800864 if self.platform == 'windows':
865 return 'cmake/build/%s/Makefile' % self._cmake_arch_option
866 else:
867 return 'Makefile'
murgatroid99a3e244f2015-09-22 11:25:53 -0700868
Jan Tattermusch77db4322016-02-20 20:19:35 -0800869 def dockerfile_dir(self):
Jan Tattermusch76511a52016-06-17 14:00:57 -0700870 return 'tools/dockerfile/test/csharp_%s_%s' % (self._docker_distro,
871 _docker_arch_suffix(self.args.arch))
Jan Tattermusch788ee232016-01-26 12:19:44 -0800872
murgatroid99132ce6a2015-03-04 17:29:14 -0800873 def __str__(self):
874 return 'csharp'
875
Craig Tillerd625d812015-04-08 15:52:35 -0700876
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700877class ObjCLanguage(object):
878
Jan Tattermusch77db4322016-02-20 20:19:35 -0800879 def configure(self, config, args):
880 self.config = config
881 self.args = args
882 _check_compiler(self.args.compiler, ['default'])
883
884 def test_specs(self):
Jorge Canizales8a9fe2a2016-07-29 15:44:47 -0700885 return [
886 self.config.job_spec(['src/objective-c/tests/run_tests.sh'],
Jan Tattermuschfffb2672016-12-19 15:24:45 +0100887 timeout_seconds=60*60,
Jorge Canizales8a9fe2a2016-07-29 15:44:47 -0700888 shortname='objc-tests',
889 environ=_FORCE_ENVIRON_FOR_WRAPPERS),
890 self.config.job_spec(['src/objective-c/tests/build_example_test.sh'],
Nicolas "Pixel" Noble7f074e02016-08-11 21:00:08 +0200891 timeout_seconds=30*60,
Jorge Canizales8a9fe2a2016-07-29 15:44:47 -0700892 shortname='objc-examples-build',
893 environ=_FORCE_ENVIRON_FOR_WRAPPERS),
894 ]
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700895
murgatroid99256d3df2015-09-21 16:58:02 -0700896 def pre_build_steps(self):
897 return []
898
Jan Tattermusch77db4322016-02-20 20:19:35 -0800899 def make_targets(self):
Jorge Canizales6eade6d2016-07-11 00:34:14 -0700900 return ['interop_server']
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700901
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800902 def make_options(self):
903 return []
904
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700905 def build_steps(self):
Jorge Canizalesd0b32e92015-07-30 23:08:43 -0700906 return [['src/objective-c/tests/build_tests.sh']]
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700907
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200908 def post_tests_steps(self):
909 return []
910
murgatroid99a3e244f2015-09-22 11:25:53 -0700911 def makefile_name(self):
912 return 'Makefile'
913
Jan Tattermusch77db4322016-02-20 20:19:35 -0800914 def dockerfile_dir(self):
Jan Tattermusch788ee232016-01-26 12:19:44 -0800915 return None
916
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700917 def __str__(self):
918 return 'objc'
919
920
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100921class Sanity(object):
922
Jan Tattermusch77db4322016-02-20 20:19:35 -0800923 def configure(self, config, args):
924 self.config = config
925 self.args = args
926 _check_compiler(self.args.compiler, ['default'])
927
928 def test_specs(self):
Craig Tiller94d04a52016-01-20 10:58:23 -0800929 import yaml
Jan Tattermusch788ee232016-01-26 12:19:44 -0800930 with open('tools/run_tests/sanity/sanity_tests.yaml', 'r') as f:
Jan Tattermusch7dd2cc62016-12-20 17:15:39 +0100931 environ={'TEST': 'true'}
932 if _is_use_docker_child():
933 environ['CLANG_FORMAT_SKIP_DOCKER'] = 'true'
Craig Tiller34226af2016-06-24 16:46:25 -0700934 return [self.config.job_spec(cmd['script'].split(),
Jan Tattermusch7dd2cc62016-12-20 17:15:39 +0100935 timeout_seconds=30*60,
936 environ=environ,
Jan Tattermusch77db4322016-02-20 20:19:35 -0800937 cpu_cost=cmd.get('cpu_cost', 1))
Craig Tiller94d04a52016-01-20 10:58:23 -0800938 for cmd in yaml.load(f)]
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100939
murgatroid99256d3df2015-09-21 16:58:02 -0700940 def pre_build_steps(self):
941 return []
942
Jan Tattermusch77db4322016-02-20 20:19:35 -0800943 def make_targets(self):
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100944 return ['run_dep_checks']
945
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800946 def make_options(self):
947 return []
948
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100949 def build_steps(self):
950 return []
951
Nicolas "Pixel" Noble87879b32015-10-12 23:28:53 +0200952 def post_tests_steps(self):
953 return []
954
murgatroid99a3e244f2015-09-22 11:25:53 -0700955 def makefile_name(self):
956 return 'Makefile'
957
Jan Tattermusch77db4322016-02-20 20:19:35 -0800958 def dockerfile_dir(self):
Jan Tattermusche70b3c52016-02-07 20:21:02 -0800959 return 'tools/dockerfile/test/sanity'
Jan Tattermusch788ee232016-01-26 12:19:44 -0800960
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100961 def __str__(self):
962 return 'sanity'
963
murgatroid99b53e5d12016-10-18 09:55:28 -0700964class NodeExpressLanguage(object):
965 """Dummy Node express test target to enable running express performance
966 benchmarks"""
967
968 def __init__(self):
969 self.platform = platform_string()
970
971 def configure(self, config, args):
972 self.config = config
973 self.args = args
974 _check_compiler(self.args.compiler, ['default', 'node0.12',
975 'node4', 'node5', 'node6'])
976 if self.args.compiler == 'default':
977 self.node_version = '4'
978 else:
979 # Take off the word "node"
980 self.node_version = self.args.compiler[4:]
981
982 def test_specs(self):
983 return []
984
985 def pre_build_steps(self):
986 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100987 return [['tools\\run_tests\\helper_scripts\\pre_build_node.bat']]
murgatroid99b53e5d12016-10-18 09:55:28 -0700988 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100989 return [['tools/run_tests/helper_scripts/pre_build_node.sh', self.node_version]]
murgatroid99b53e5d12016-10-18 09:55:28 -0700990
991 def make_targets(self):
992 return []
993
994 def make_options(self):
995 return []
996
997 def build_steps(self):
998 return []
999
1000 def post_tests_steps(self):
1001 return []
1002
1003 def makefile_name(self):
1004 return 'Makefile'
1005
1006 def dockerfile_dir(self):
1007 return 'tools/dockerfile/test/node_jessie_%s' % _docker_arch_suffix(self.args.arch)
1008
1009 def __str__(self):
1010 return 'node_express'
Nicolas "Pixel" Noblee55cd7f2015-04-14 17:59:13 +02001011
Craig Tiller738c3342015-01-12 14:28:33 -08001012# different configurations we can run under
Jan Tattermusch5c79a312016-12-20 11:02:50 +01001013with open('tools/run_tests/generated/configs.json') as f:
Craig Tiller1dce9062016-01-20 17:01:56 -08001014 _CONFIGS = dict((cfg['config'], Config(**cfg)) for cfg in ast.literal_eval(f.read()))
Craig Tiller738c3342015-01-12 14:28:33 -08001015
1016
Craig Tillerc7449162015-01-16 14:42:10 -08001017_LANGUAGES = {
Craig Tillere9c959d2015-01-18 10:23:26 -08001018 'c++': CLanguage('cxx', 'c++'),
1019 'c': CLanguage('c', 'c'),
murgatroid992c8d5162015-01-26 10:41:21 -08001020 'node': NodeLanguage(),
murgatroid99b53e5d12016-10-18 09:55:28 -07001021 'node_express': NodeExpressLanguage(),
Nathaniel Manista840615e2015-01-22 20:31:47 +00001022 'php': PhpLanguage(),
Stanley Cheung2e2cdff2016-07-23 19:07:36 -07001023 'php7': Php7Language(),
Nathaniel Manista840615e2015-01-22 20:31:47 +00001024 'python': PythonLanguage(),
Jan Tattermusch1970a5b2015-03-03 15:17:25 -08001025 'ruby': RubyLanguage(),
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +01001026 'csharp': CSharpLanguage(),
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -07001027 'objc' : ObjCLanguage(),
Jan Tattermusch70a57e42016-02-20 18:50:27 -08001028 'sanity': Sanity()
Craig Tillereb272bc2015-01-30 13:13:14 -08001029 }
Nicolas Nobleddef2462015-01-06 18:08:25 -08001030
Jan Tattermusch77db4322016-02-20 20:19:35 -08001031
Jan Tattermuscha2d964c2016-02-22 17:33:09 -08001032_MSBUILD_CONFIG = {
Craig Tiller7bb3efd2015-09-01 08:04:03 -07001033 'dbg': 'Debug',
1034 'opt': 'Release',
Jan Tattermusche4a69182015-12-15 09:53:01 -08001035 'gcov': 'Debug',
Craig Tiller7bb3efd2015-09-01 08:04:03 -07001036 }
1037
David Garcia Quintase90cd372015-05-31 18:15:26 -07001038
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001039def _windows_arch_option(arch):
1040 """Returns msbuild cmdline option for selected architecture."""
Jan Tattermusch9be594f2016-01-25 18:08:47 -08001041 if arch == 'default' or arch == 'x86':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001042 return '/p:Platform=Win32'
Jan Tattermusch9be594f2016-01-25 18:08:47 -08001043 elif arch == 'x64':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001044 return '/p:Platform=x64'
1045 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001046 print('Architecture %s not supported.' % arch)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001047 sys.exit(1)
Jan Tattermusch788ee232016-01-26 12:19:44 -08001048
Jan Tattermuschf08018a2016-01-26 08:22:09 -08001049
1050def _check_arch_option(arch):
1051 """Checks that architecture option is valid."""
1052 if platform_string() == 'windows':
1053 _windows_arch_option(arch)
1054 elif platform_string() == 'linux':
1055 # On linux, we need to be running under docker with the right architecture.
Jan Tattermusch07fb0422016-01-26 10:46:56 -08001056 runtime_arch = platform.architecture()[0]
Jan Tattermuschf08018a2016-01-26 08:22:09 -08001057 if arch == 'default':
1058 return
1059 elif runtime_arch == '64bit' and arch == 'x64':
1060 return
1061 elif runtime_arch == '32bit' and arch == 'x86':
1062 return
1063 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001064 print('Architecture %s does not match current runtime architecture.' % arch)
Jan Tattermuschf08018a2016-01-26 08:22:09 -08001065 sys.exit(1)
1066 else:
1067 if args.arch != 'default':
siddharthshukla0589e532016-07-07 16:08:01 +02001068 print('Architecture %s not supported on current platform.' % args.arch)
Jan Tattermuschf08018a2016-01-26 08:22:09 -08001069 sys.exit(1)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001070
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001071
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001072def _windows_build_bat(compiler):
1073 """Returns name of build.bat for selected compiler."""
Jan Tattermuschbc98af12016-06-17 18:38:27 -07001074 # For CoreCLR, fall back to the default compiler for C core
Jan Tattermuschdd900d52017-02-09 16:26:41 -08001075 if compiler == 'default' or compiler == 'vs2013':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001076 return 'vsprojects\\build_vs2013.bat'
1077 elif compiler == 'vs2015':
1078 return 'vsprojects\\build_vs2015.bat'
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001079 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001080 print('Compiler %s not supported.' % compiler)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001081 sys.exit(1)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001082
1083
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001084def _windows_toolset_option(compiler):
1085 """Returns msbuild PlatformToolset for selected compiler."""
Jan Tattermuschbc98af12016-06-17 18:38:27 -07001086 # For CoreCLR, fall back to the default compiler for C core
1087 if compiler == 'default' or compiler == 'vs2013' or compiler == 'coreclr':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001088 return '/p:PlatformToolset=v120'
1089 elif compiler == 'vs2015':
1090 return '/p:PlatformToolset=v140'
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001091 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001092 print('Compiler %s not supported.' % compiler)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001093 sys.exit(1)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001094
1095
Jan Tattermusche70b3c52016-02-07 20:21:02 -08001096def _docker_arch_suffix(arch):
1097 """Returns suffix to dockerfile dir to use."""
1098 if arch == 'default' or arch == 'x64':
1099 return 'x64'
1100 elif arch == 'x86':
1101 return 'x86'
1102 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001103 print('Architecture %s not supported with current settings.' % arch)
Jan Tattermusche70b3c52016-02-07 20:21:02 -08001104 sys.exit(1)
1105
1106
David Garcia Quintase90cd372015-05-31 18:15:26 -07001107def runs_per_test_type(arg_str):
1108 """Auxilary function to parse the "runs_per_test" flag.
1109
1110 Returns:
1111 A positive integer or 0, the latter indicating an infinite number of
1112 runs.
1113
1114 Raises:
1115 argparse.ArgumentTypeError: Upon invalid input.
1116 """
1117 if arg_str == 'inf':
1118 return 0
1119 try:
1120 n = int(arg_str)
1121 if n <= 0: raise ValueError
Craig Tiller50e53e22015-06-01 20:18:21 -07001122 return n
David Garcia Quintase90cd372015-05-31 18:15:26 -07001123 except:
Adele Zhoue4c35612015-10-16 15:34:23 -07001124 msg = '\'{}\' is not a positive integer or \'inf\''.format(arg_str)
David Garcia Quintase90cd372015-05-31 18:15:26 -07001125 raise argparse.ArgumentTypeError(msg)
Jan Tattermuschc95eead2015-09-18 13:03:50 -07001126
siddharthshukla2135a1b2016-08-04 02:11:53 +02001127
David Garcia Quintas95b37b72017-02-15 16:49:49 -08001128def percent_type(arg_str):
1129 pct = float(arg_str)
1130 if pct > 100 or pct < 0:
1131 raise argparse.ArgumentTypeError(
1132 "'%f' is not a valid percentage in the [0, 100] range" % pct)
1133 return pct
1134
1135# This is math.isclose in python >= 3.5
1136def isclose(a, b, rel_tol=1e-09, abs_tol=0.0):
1137 return abs(a-b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol)
1138
1139
Jan Tattermuschc95eead2015-09-18 13:03:50 -07001140# parse command line
1141argp = argparse.ArgumentParser(description='Run grpc tests.')
1142argp.add_argument('-c', '--config',
Jan Tattermusch77db4322016-02-20 20:19:35 -08001143 choices=sorted(_CONFIGS.keys()),
1144 default='opt')
David Garcia Quintase90cd372015-05-31 18:15:26 -07001145argp.add_argument('-n', '--runs_per_test', default=1, type=runs_per_test_type,
1146 help='A positive integer or "inf". If "inf", all tests will run in an '
1147 'infinite loop. Especially useful in combination with "-f"')
Craig Tillerfe406ec2015-02-24 13:55:12 -08001148argp.add_argument('-r', '--regex', default='.*', type=str)
Vijay Pai488fd0e2016-06-13 12:37:12 -07001149argp.add_argument('--regex_exclude', default='', type=str)
Craig Tiller5f735a62016-01-20 09:31:15 -08001150argp.add_argument('-j', '--jobs', default=multiprocessing.cpu_count(), type=int)
Craig Tiller8451e872015-02-27 09:25:51 -08001151argp.add_argument('-s', '--slowdown', default=1.0, type=float)
David Garcia Quintas95b37b72017-02-15 16:49:49 -08001152argp.add_argument('-p', '--sample_percent', default=100.0, type=percent_type,
1153 help='Run a random sample with that percentage of tests')
ctiller3040cb72015-01-07 12:13:17 -08001154argp.add_argument('-f', '--forever',
1155 default=False,
1156 action='store_const',
1157 const=True)
Nicolas "Pixel" Noblea7df3f92015-02-26 22:07:04 +01001158argp.add_argument('-t', '--travis',
1159 default=False,
1160 action='store_const',
1161 const=True)
Nicolas Noble044db742015-01-14 16:57:24 -08001162argp.add_argument('--newline_on_success',
1163 default=False,
1164 action='store_const',
1165 const=True)
Craig Tiller686fb262015-01-15 07:39:09 -08001166argp.add_argument('-l', '--language',
Craig Tiller60f15e62015-05-13 09:05:17 -07001167 choices=['all'] + sorted(_LANGUAGES.keys()),
Craig Tiller686fb262015-01-15 07:39:09 -08001168 nargs='+',
Craig Tiller60f15e62015-05-13 09:05:17 -07001169 default=['all'])
Craig Tillercd43da82015-05-29 08:41:29 -07001170argp.add_argument('-S', '--stop_on_failure',
1171 default=False,
1172 action='store_const',
1173 const=True)
Jan Tattermuschc95eead2015-09-18 13:03:50 -07001174argp.add_argument('--use_docker',
1175 default=False,
1176 action='store_const',
1177 const=True,
Adele Zhoue4c35612015-10-16 15:34:23 -07001178 help='Run all the tests under docker. That provides ' +
1179 'additional isolation and prevents the need to install ' +
1180 'language specific prerequisites. Only available on Linux.')
Craig Tillerd4509a12015-09-28 09:18:40 -07001181argp.add_argument('--allow_flakes',
1182 default=False,
1183 action='store_const',
1184 const=True,
Adele Zhoue4c35612015-10-16 15:34:23 -07001185 help='Allow flaky tests to show as passing (re-runs failed tests up to five times)')
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001186argp.add_argument('--arch',
Jan Tattermusch9be594f2016-01-25 18:08:47 -08001187 choices=['default', 'x86', 'x64'],
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001188 default='default',
1189 help='Selects architecture to target. For some platforms "default" is the only supported choice.')
1190argp.add_argument('--compiler',
Jan Tattermuschc4cbe392016-02-22 19:29:38 -08001191 choices=['default',
Ken Payson975b5102017-03-30 17:38:40 -07001192 'gcc4.4', 'gcc4.6', 'gcc4.8', 'gcc4.9', 'gcc5.3', 'gcc_musl',
Jan Tattermusch6d258c52016-06-10 09:36:51 -07001193 'clang3.4', 'clang3.5', 'clang3.6', 'clang3.7',
Jan Tattermuschdc5e5092017-02-09 16:15:12 -08001194 'vs2013', 'vs2015',
Ken Payson975b5102017-03-30 17:38:40 -07001195 'python2.7', 'python3.4', 'python3.5', 'python3.6', 'pypy', 'pypy3', 'python_alpine',
murgatroid995a3f8622016-10-26 13:45:04 -07001196 'node0.12', 'node4', 'node5', 'node6', 'node7',
murgatroid99f25f5052017-04-13 17:22:32 -07001197 'electron1.3', 'electron1.6',
Jan Tattermuschc98bde62017-01-25 19:12:11 +01001198 'coreclr',
1199 'cmake'],
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001200 default='default',
Jan Tattermusch77db4322016-02-20 20:19:35 -08001201 help='Selects compiler to use. Allowed values depend on the platform and language.')
murgatroid99c36f6ea2016-10-03 09:24:09 -07001202argp.add_argument('--iomgr_platform',
1203 choices=['native', 'uv'],
1204 default='native',
1205 help='Selects iomgr platform to build on')
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001206argp.add_argument('--build_only',
1207 default=False,
1208 action='store_const',
1209 const=True,
Matt Kwong52ff9862017-04-17 13:56:51 -07001210 help='Perform all the build steps but don\'t run any tests.')
Craig Tiller5f735a62016-01-20 09:31:15 -08001211argp.add_argument('--measure_cpu_costs', default=False, action='store_const', const=True,
1212 help='Measure the cpu costs of tests')
Craig Tiller1676f912016-01-05 10:49:44 -08001213argp.add_argument('--update_submodules', default=[], nargs='*',
1214 help='Update some submodules before building. If any are updated, also run generate_projects. ' +
1215 'Submodules are specified as SUBMODULE_NAME:BRANCH; if BRANCH is omitted, master is assumed.')
Craig Tiller234b6e72015-05-23 10:12:40 -07001216argp.add_argument('-a', '--antagonists', default=0, type=int)
Nicolas "Pixel" Noble5937b5b2015-06-26 02:04:12 +02001217argp.add_argument('-x', '--xml_report', default=None, type=str,
1218 help='Generates a JUnit-compatible XML report')
Jan Tattermuschcfcc0752016-10-09 17:02:34 +02001219argp.add_argument('--report_suite_name', default='tests', type=str,
1220 help='Test suite name to use in generated JUnit XML report')
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001221argp.add_argument('--quiet_success',
1222 default=False,
1223 action='store_const',
1224 const=True,
Matt Kwong52ff9862017-04-17 13:56:51 -07001225 help='Don\'t print anything when a test passes. Passing tests also will not be reported in XML report. ' +
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001226 'Useful when running many iterations of each test (argument -n).')
Craig Tiller123f1372016-06-15 15:06:14 -07001227argp.add_argument('--force_default_poller', default=False, action='store_const', const=True,
Matt Kwong52ff9862017-04-17 13:56:51 -07001228 help='Don\'t try to iterate over many polling strategies when they exist')
Craig Tillera1ac2a12017-04-21 07:20:38 -07001229argp.add_argument('--max_time', default=-1, type=int, help='Maximum test runtime in seconds')
Matt Kwong52ff9862017-04-17 13:56:51 -07001230argp.add_argument('--bq_result_table',
1231 default='',
1232 type=str,
1233 nargs='?',
1234 help='Upload test results to a specified BQ table.')
Craig Tiller6af65942017-06-07 09:09:14 -07001235argp.add_argument('--auto_set_flakes', default=True, type=bool,
1236 help='Set flakiness data from historic data')
Nicolas Nobleddef2462015-01-06 18:08:25 -08001237args = argp.parse_args()
1238
Craig Tillerd16abf82017-06-07 08:58:55 -07001239flaky_tests = set()
1240if args.auto_set_flakes:
1241 try:
1242 flaky_tests = set(get_flaky_tests())
1243 except:
1244 print("Unexpected error getting flaky tests:", sys.exc_info()[0])
David Garcia Quintasfaafa4d2017-06-06 14:52:17 -07001245
Craig Tiller123f1372016-06-15 15:06:14 -07001246if args.force_default_poller:
1247 _POLLING_STRATEGIES = {}
1248
Craig Tiller5f735a62016-01-20 09:31:15 -08001249jobset.measure_cpu_costs = args.measure_cpu_costs
1250
Craig Tiller1676f912016-01-05 10:49:44 -08001251# update submodules if necessary
Craig Tillerb361b4e2016-01-06 11:44:17 -08001252need_to_regenerate_projects = False
1253for spec in args.update_submodules:
1254 spec = spec.split(':', 1)
1255 if len(spec) == 1:
1256 submodule = spec[0]
1257 branch = 'master'
1258 elif len(spec) == 2:
1259 submodule = spec[0]
1260 branch = spec[1]
1261 cwd = 'third_party/%s' % submodule
1262 def git(cmd, cwd=cwd):
siddharthshukla0589e532016-07-07 16:08:01 +02001263 print('in %s: git %s' % (cwd, cmd))
David Garcia Quintas03920252017-02-15 12:51:21 -08001264 run_shell_command('git %s' % cmd, cwd=cwd)
Craig Tillerb361b4e2016-01-06 11:44:17 -08001265 git('fetch')
1266 git('checkout %s' % branch)
1267 git('pull origin %s' % branch)
1268 if os.path.exists('src/%s/gen_build_yaml.py' % submodule):
1269 need_to_regenerate_projects = True
1270if need_to_regenerate_projects:
1271 if jobset.platform_string() == 'linux':
David Garcia Quintas03920252017-02-15 12:51:21 -08001272 run_shell_command('tools/buildgen/generate_projects.sh')
Craig Tillerb361b4e2016-01-06 11:44:17 -08001273 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001274 print('WARNING: may need to regenerate projects, but since we are not on')
1275 print(' Linux this step is being skipped. Compilation MAY fail.')
Craig Tiller1676f912016-01-05 10:49:44 -08001276
1277
Nicolas Nobleddef2462015-01-06 18:08:25 -08001278# grab config
Jan Tattermusch77db4322016-02-20 20:19:35 -08001279run_config = _CONFIGS[args.config]
1280build_config = run_config.build_config
Craig Tillerf1973b02015-01-16 12:32:13 -08001281
Craig Tiller06805272015-06-11 14:46:47 -07001282if args.travis:
murgatroid99d3b5b7f2015-10-06 17:02:03 -07001283 _FORCE_ENVIRON_FOR_WRAPPERS = {'GRPC_TRACE': 'api'}
Craig Tiller06805272015-06-11 14:46:47 -07001284
Adele Zhou6b9527c2015-11-20 15:56:35 -08001285if 'all' in args.language:
Craig Tiller1676f912016-01-05 10:49:44 -08001286 lang_list = _LANGUAGES.keys()
Adele Zhou6b9527c2015-11-20 15:56:35 -08001287else:
1288 lang_list = args.language
Craig Tiller16900662016-01-07 19:30:54 -08001289# We don't support code coverage on some languages
1290if 'gcov' in args.config:
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001291 for bad in ['objc', 'sanity']:
Craig Tiller16900662016-01-07 19:30:54 -08001292 if bad in lang_list:
1293 lang_list.remove(bad)
Adele Zhou6b9527c2015-11-20 15:56:35 -08001294
1295languages = set(_LANGUAGES[l] for l in lang_list)
Jan Tattermusch77db4322016-02-20 20:19:35 -08001296for l in languages:
1297 l.configure(run_config, args)
murgatroid99132ce6a2015-03-04 17:29:14 -08001298
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001299language_make_options=[]
1300if any(language.make_options() for language in languages):
Adele Zhou3b6ab812016-05-18 17:04:20 -07001301 if not 'gcov' in args.config and len(languages) != 1:
siddharthshukla0589e532016-07-07 16:08:01 +02001302 print('languages with custom make options cannot be built simultaneously with other languages')
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001303 sys.exit(1)
1304 else:
Matt Kwong0ff19572017-03-30 18:22:21 -07001305 # Combining make options is not clean and just happens to work. It allows C/C++ and C# to build
1306 # together, and is only used under gcov. All other configs should build languages individually.
1307 language_make_options = list(set([make_option for lang in languages for make_option in lang.make_options()]))
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001308
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001309if args.use_docker:
1310 if not args.travis:
siddharthshukla0589e532016-07-07 16:08:01 +02001311 print('Seen --use_docker flag, will run tests under docker.')
1312 print('')
1313 print('IMPORTANT: The changes you are testing need to be locally committed')
1314 print('because only the committed changes in the current branch will be')
1315 print('copied to the docker environment.')
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001316 time.sleep(5)
1317
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001318 dockerfile_dirs = set([l.dockerfile_dir() for l in languages])
1319 if len(dockerfile_dirs) > 1:
Adele Zhou9506ef22016-03-02 13:53:34 -08001320 if 'gcov' in args.config:
1321 dockerfile_dir = 'tools/dockerfile/test/multilang_jessie_x64'
1322 print ('Using multilang_jessie_x64 docker image for code coverage for '
1323 'all languages.')
1324 else:
1325 print ('Languages to be tested require running under different docker '
1326 'images.')
1327 sys.exit(1)
1328 else:
1329 dockerfile_dir = next(iter(dockerfile_dirs))
Craig Tillerde7edf82016-03-20 09:12:16 -07001330
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001331 child_argv = [ arg for arg in sys.argv if not arg == '--use_docker' ]
Jan Tattermusched342b12016-01-26 14:40:31 -08001332 run_tests_cmd = 'python tools/run_tests/run_tests.py %s' % ' '.join(child_argv[1:])
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001333
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001334 env = os.environ.copy()
1335 env['RUN_TESTS_COMMAND'] = run_tests_cmd
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001336 env['DOCKERFILE_DIR'] = dockerfile_dir
Jan Tattermusch9835d4b2016-04-29 15:05:05 -07001337 env['DOCKER_RUN_SCRIPT'] = 'tools/run_tests/dockerize/docker_run_tests.sh'
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001338 if args.xml_report:
1339 env['XML_REPORT'] = args.xml_report
1340 if not args.travis:
1341 env['TTY_FLAG'] = '-t' # enables Ctrl-C when not on Jenkins.
1342
Jan Tattermusch3464f702017-03-08 20:50:57 +01001343 subprocess.check_call('tools/run_tests/dockerize/build_docker_and_run_tests.sh',
1344 shell=True,
1345 env=env)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001346 sys.exit(0)
Jan Tattermusch788ee232016-01-26 12:19:44 -08001347
Jan Tattermuschf08018a2016-01-26 08:22:09 -08001348_check_arch_option(args.arch)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001349
Jan Tattermuschfba65302016-01-25 18:21:14 -08001350def make_jobspec(cfg, targets, makefile='Makefile'):
1351 if platform_string() == 'windows':
Jan Tattermuschc98bde62017-01-25 19:12:11 +01001352 if makefile.startswith('cmake/build/'):
1353 return [jobset.JobSpec(['cmake', '--build', '.',
1354 '--target', '%s' % target,
1355 '--config', _MSBUILD_CONFIG[cfg]],
Jan Tattermusche7f0b852017-02-08 19:06:10 -08001356 cwd=os.path.dirname(makefile),
Jan Tattermuschc98bde62017-01-25 19:12:11 +01001357 timeout_seconds=None) for target in targets]
Craig Tillerfc3c0c42015-09-01 16:47:54 -07001358 extra_args = []
Craig Tillerb5391e12015-09-03 14:35:18 -07001359 # better do parallel compilation
Jan Tattermusch47eeb2b2015-10-07 14:09:18 -07001360 # empirically /m:2 gives the best performance/price and should prevent
1361 # overloading the windows workers.
Adele Zhoue4c35612015-10-16 15:34:23 -07001362 extra_args.extend(['/m:2'])
Craig Tillerb5391e12015-09-03 14:35:18 -07001363 # disable PDB generation: it's broken, and we don't need it during CI
Adele Zhoue4c35612015-10-16 15:34:23 -07001364 extra_args.extend(['/p:Jenkins=true'])
Craig Tiller6fd23842015-09-01 07:36:31 -07001365 return [
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001366 jobset.JobSpec([_windows_build_bat(args.compiler),
murgatroid99cf08daf2015-09-21 15:33:16 -07001367 'vsprojects\\%s.sln' % target,
Jan Tattermuscha2d964c2016-02-22 17:33:09 -08001368 '/p:Configuration=%s' % _MSBUILD_CONFIG[cfg]] +
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001369 extra_args +
1370 language_make_options,
Craig Tiller590105a2016-01-19 13:03:46 -08001371 shell=True, timeout_seconds=None)
Craig Tiller6fd23842015-09-01 07:36:31 -07001372 for target in targets]
Jan Tattermuschfba65302016-01-25 18:21:14 -08001373 else:
Jan Tattermuschc98bde62017-01-25 19:12:11 +01001374 if targets and makefile.startswith('cmake/build/'):
1375 # With cmake, we've passed all the build configuration in the pre-build step already
1376 return [jobset.JobSpec([os.getenv('MAKE', 'make'),
1377 '-j', '%d' % args.jobs] +
1378 targets,
1379 cwd='cmake/build',
1380 timeout_seconds=None)]
murgatroid998ae409f2015-10-26 16:39:00 -07001381 if targets:
1382 return [jobset.JobSpec([os.getenv('MAKE', 'make'),
1383 '-f', makefile,
Craig Tillerdd6f7ed2016-01-21 12:54:42 -08001384 '-j', '%d' % args.jobs,
Craig Tiller71a86042016-01-15 14:59:58 -08001385 'EXTRA_DEFINES=GRPC_TEST_SLOWDOWN_MACHINE_FACTOR=%f' % args.slowdown,
Craig Tiller90ce7232017-04-12 06:53:09 -07001386 'CONFIG=%s' % cfg,
1387 'Q='] +
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001388 language_make_options +
Craig Tiller71a86042016-01-15 14:59:58 -08001389 ([] if not args.travis else ['JENKINS_BUILD=1']) +
1390 targets,
Craig Tiller590105a2016-01-19 13:03:46 -08001391 timeout_seconds=None)]
murgatroid998ae409f2015-10-26 16:39:00 -07001392 else:
1393 return []
Jan Tattermuschfba65302016-01-25 18:21:14 -08001394
murgatroid99a3e244f2015-09-22 11:25:53 -07001395make_targets = {}
1396for l in languages:
1397 makefile = l.makefile_name()
1398 make_targets[makefile] = make_targets.get(makefile, set()).union(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001399 set(l.make_targets()))
Craig Tiller5058c692015-04-08 09:42:04 -07001400
Jan Tattermusche4a69182015-12-15 09:53:01 -08001401def build_step_environ(cfg):
1402 environ = {'CONFIG': cfg}
Jan Tattermuscha2d964c2016-02-22 17:33:09 -08001403 msbuild_cfg = _MSBUILD_CONFIG.get(cfg)
Jan Tattermusche4a69182015-12-15 09:53:01 -08001404 if msbuild_cfg:
1405 environ['MSBUILD_CONFIG'] = msbuild_cfg
1406 return environ
1407
murgatroid99fddac962015-09-22 09:20:11 -07001408build_steps = list(set(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001409 jobset.JobSpec(cmdline, environ=build_step_environ(build_config), flake_retries=5)
murgatroid99256d3df2015-09-21 16:58:02 -07001410 for l in languages
1411 for cmdline in l.pre_build_steps()))
Craig Tillerbd4e3782015-09-01 06:48:55 -07001412if make_targets:
siddharthshukla0589e532016-07-07 16:08:01 +02001413 make_commands = itertools.chain.from_iterable(make_jobspec(build_config, list(targets), makefile) for (makefile, targets) in make_targets.items())
Craig Tiller6fd23842015-09-01 07:36:31 -07001414 build_steps.extend(set(make_commands))
Craig Tiller5058c692015-04-08 09:42:04 -07001415build_steps.extend(set(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001416 jobset.JobSpec(cmdline, environ=build_step_environ(build_config), timeout_seconds=None)
Craig Tiller547db2b2015-01-30 14:08:39 -08001417 for l in languages
Craig Tiller533b1a22015-05-29 08:41:29 -07001418 for cmdline in l.build_steps()))
Craig Tillerf1973b02015-01-16 12:32:13 -08001419
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +02001420post_tests_steps = list(set(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001421 jobset.JobSpec(cmdline, environ=build_step_environ(build_config))
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +02001422 for l in languages
1423 for cmdline in l.post_tests_steps()))
Nicolas Nobleddef2462015-01-06 18:08:25 -08001424runs_per_test = args.runs_per_test
ctiller3040cb72015-01-07 12:13:17 -08001425forever = args.forever
Nicolas Nobleddef2462015-01-06 18:08:25 -08001426
Nicolas Nobleddef2462015-01-06 18:08:25 -08001427
Ken Paysonfa51de52016-06-30 23:50:48 -07001428def _shut_down_legacy_server(legacy_server_port):
1429 try:
siddharthshukla0589e532016-07-07 16:08:01 +02001430 version = int(urllib.request.urlopen(
Ken Paysonfa51de52016-06-30 23:50:48 -07001431 'http://localhost:%d/version_number' % legacy_server_port,
1432 timeout=10).read())
1433 except:
1434 pass
1435 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001436 urllib.request.urlopen(
Ken Paysonfa51de52016-06-30 23:50:48 -07001437 'http://localhost:%d/quitquitquit' % legacy_server_port).read()
1438
1439
Adele Zhoud5fffa52015-10-23 15:51:42 -07001440def _calculate_num_runs_failures(list_of_results):
1441 """Caculate number of runs and failures for a particular test.
1442
1443 Args:
1444 list_of_results: (List) of JobResult object.
1445 Returns:
1446 A tuple of total number of runs and failures.
1447 """
1448 num_runs = len(list_of_results) # By default, there is 1 run per JobResult.
1449 num_failures = 0
1450 for jobresult in list_of_results:
1451 if jobresult.retries > 0:
1452 num_runs += jobresult.retries
1453 if jobresult.num_failures > 0:
1454 num_failures += jobresult.num_failures
1455 return num_runs, num_failures
1456
Adele Zhou6b9527c2015-11-20 15:56:35 -08001457
Craig Tillereb9de8b2016-01-08 08:57:41 -08001458# _build_and_run results
1459class BuildAndRunError(object):
1460
1461 BUILD = object()
1462 TEST = object()
1463 POST_TEST = object()
1464
1465
Craig Tiller819cd882017-04-25 13:18:22 -07001466def _has_epollexclusive():
1467 try:
1468 subprocess.check_call('bins/%s/check_epollexclusive' % args.config)
1469 return True
1470 except subprocess.CalledProcessError, e:
1471 return False
Jan Tattermusch9883e0b2017-05-18 11:00:27 +02001472 except OSError, e:
1473 # For languages other than C and Windows the binary won't exist
1474 return False
Craig Tiller819cd882017-04-25 13:18:22 -07001475
1476
Craig Tillereb9de8b2016-01-08 08:57:41 -08001477# returns a list of things that failed (or an empty list on success)
Craig Tillerf53d9c82015-08-04 14:19:43 -07001478def _build_and_run(
Craig Tiller74189cd2016-06-23 15:39:06 -07001479 check_cancelled, newline_on_success, xml_report=None, build_only=False):
ctiller3040cb72015-01-07 12:13:17 -08001480 """Do one pass of building & running tests."""
murgatroid99666450e2015-01-26 13:03:31 -08001481 # build latest sequentially
Jan Tattermuschaab1e512016-01-28 09:30:44 -08001482 num_failures, resultset = jobset.run(
Adele Zhoue4c35612015-10-16 15:34:23 -07001483 build_steps, maxjobs=1, stop_on_failure=True,
Craig Tiller883064c2015-11-04 10:06:10 -08001484 newline_on_success=newline_on_success, travis=args.travis)
Adele Zhoue4c35612015-10-16 15:34:23 -07001485 if num_failures:
Craig Tillereb9de8b2016-01-08 08:57:41 -08001486 return [BuildAndRunError.BUILD]
Craig Tillerb361b4e2016-01-06 11:44:17 -08001487
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001488 if build_only:
Jan Tattermuschaab1e512016-01-28 09:30:44 -08001489 if xml_report:
Jan Tattermuschcfcc0752016-10-09 17:02:34 +02001490 report_utils.render_junit_xml_report(resultset, xml_report,
1491 suite_name=args.report_suite_name)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001492 return []
ctiller3040cb72015-01-07 12:13:17 -08001493
Craig Tillerb2574bf2017-05-17 09:01:28 -07001494 if not args.travis and not _has_epollexclusive() and platform_string() in _POLLING_STRATEGIES and 'epollex' in _POLLING_STRATEGIES[platform_string()]:
Craig Tiller819cd882017-04-25 13:18:22 -07001495 print('\n\nOmitting EPOLLEXCLUSIVE tests\n\n')
1496 _POLLING_STRATEGIES[platform_string()].remove('epollex')
1497
Craig Tiller234b6e72015-05-23 10:12:40 -07001498 # start antagonists
Jan Tattermusch5c79a312016-12-20 11:02:50 +01001499 antagonists = [subprocess.Popen(['tools/run_tests/python_utils/antagonist.py'])
Craig Tiller234b6e72015-05-23 10:12:40 -07001500 for _ in range(0, args.antagonists)]
Craig Tillercba864b2017-02-17 10:27:56 -08001501 start_port_server.start_port_server()
Adele Zhou7cf72112015-11-04 11:18:43 -08001502 resultset = None
Adele Zhou803af152015-11-30 15:16:16 -08001503 num_test_failures = 0
Craig Tiller234b6e72015-05-23 10:12:40 -07001504 try:
David Garcia Quintase90cd372015-05-31 18:15:26 -07001505 infinite_runs = runs_per_test == 0
yang-g6c1fdc62015-08-18 11:57:42 -07001506 one_run = set(
1507 spec
yang-g6c1fdc62015-08-18 11:57:42 -07001508 for language in languages
Jan Tattermusch77db4322016-02-20 20:19:35 -08001509 for spec in language.test_specs()
Vijay Pai488fd0e2016-06-13 12:37:12 -07001510 if (re.search(args.regex, spec.shortname) and
1511 (args.regex_exclude == '' or
1512 not re.search(args.regex_exclude, spec.shortname))))
David Garcia Quintas79e389f2015-06-02 17:49:42 -07001513 # When running on travis, we want out test runs to be as similar as possible
1514 # for reproducibility purposes.
Craig Tillera1ac2a12017-04-21 07:20:38 -07001515 if args.travis and args.max_time <= 0:
David Garcia Quintas79e389f2015-06-02 17:49:42 -07001516 massaged_one_run = sorted(one_run, key=lambda x: x.shortname)
1517 else:
1518 # whereas otherwise, we want to shuffle things up to give all tests a
1519 # chance to run.
David Garcia Quintas95b37b72017-02-15 16:49:49 -08001520 massaged_one_run = list(one_run) # random.sample needs an indexable seq.
1521 num_jobs = len(massaged_one_run)
1522 # for a random sample, get as many as indicated by the 'sample_percent'
1523 # argument. By default this arg is 100, resulting in a shuffle of all
1524 # jobs.
1525 sample_size = int(num_jobs * args.sample_percent/100.0)
1526 massaged_one_run = random.sample(massaged_one_run, sample_size)
1527 if not isclose(args.sample_percent, 100.0):
David Garcia Quintase2bdbfe2017-03-08 11:55:07 -08001528 assert args.runs_per_test == 1, "Can't do sampling (-p) over multiple runs (-n)."
David Garcia Quintas95b37b72017-02-15 16:49:49 -08001529 print("Running %d tests out of %d (~%d%%)" %
1530 (sample_size, num_jobs, args.sample_percent))
Craig Tillerf7b7c892015-06-22 14:33:25 -07001531 if infinite_runs:
1532 assert len(massaged_one_run) > 0, 'Must have at least one test for a -n inf run'
David Garcia Quintas79e389f2015-06-02 17:49:42 -07001533 runs_sequence = (itertools.repeat(massaged_one_run) if infinite_runs
1534 else itertools.repeat(massaged_one_run, runs_per_test))
David Garcia Quintase90cd372015-05-31 18:15:26 -07001535 all_runs = itertools.chain.from_iterable(runs_sequence)
Nicolas "Pixel" Noble5937b5b2015-06-26 02:04:12 +02001536
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001537 if args.quiet_success:
1538 jobset.message('START', 'Running tests quietly, only failing tests will be reported', do_newline=True)
Adele Zhou803af152015-11-30 15:16:16 -08001539 num_test_failures, resultset = jobset.run(
Adele Zhou2271ab52015-10-28 13:59:14 -07001540 all_runs, check_cancelled, newline_on_success=newline_on_success,
David Garcia Quintas95b37b72017-02-15 16:49:49 -08001541 travis=args.travis, maxjobs=args.jobs,
murgatroid998ae409f2015-10-26 16:39:00 -07001542 stop_on_failure=args.stop_on_failure,
Craig Tillera1ac2a12017-04-21 07:20:38 -07001543 quiet_success=args.quiet_success, max_time=args.max_time)
Adele Zhoud5fffa52015-10-23 15:51:42 -07001544 if resultset:
Craig Tiller2b59dbc2016-05-13 15:59:09 -07001545 for k, v in sorted(resultset.items()):
Adele Zhoud5fffa52015-10-23 15:51:42 -07001546 num_runs, num_failures = _calculate_num_runs_failures(v)
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001547 if num_failures > 0:
1548 if num_failures == num_runs: # what about infinite_runs???
1549 jobset.message('FAILED', k, do_newline=True)
1550 else:
1551 jobset.message(
1552 'FLAKE', '%s [%d/%d runs flaked]' % (k, num_failures, num_runs),
1553 do_newline=True)
Craig Tiller234b6e72015-05-23 10:12:40 -07001554 finally:
1555 for antagonist in antagonists:
1556 antagonist.kill()
Matt Kwong52ff9862017-04-17 13:56:51 -07001557 if args.bq_result_table and resultset:
1558 upload_results_to_bq(resultset, args.bq_result_table, args, platform_string())
Adele Zhou7cf72112015-11-04 11:18:43 -08001559 if xml_report and resultset:
Jan Tattermuschcfcc0752016-10-09 17:02:34 +02001560 report_utils.render_junit_xml_report(resultset, xml_report,
1561 suite_name=args.report_suite_name)
Craig Tillerd86a3942015-01-14 12:48:54 -08001562
Adele Zhouf2ca7bc2015-10-23 15:38:00 -07001563 number_failures, _ = jobset.run(
1564 post_tests_steps, maxjobs=1, stop_on_failure=True,
Craig Tiller883064c2015-11-04 10:06:10 -08001565 newline_on_success=newline_on_success, travis=args.travis)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001566
1567 out = []
1568 if number_failures:
1569 out.append(BuildAndRunError.POST_TEST)
1570 if num_test_failures:
1571 out.append(BuildAndRunError.TEST)
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +02001572
Craig Tillereb9de8b2016-01-08 08:57:41 -08001573 return out
ctiller3040cb72015-01-07 12:13:17 -08001574
1575
1576if forever:
Nicolas Noble044db742015-01-14 16:57:24 -08001577 success = True
ctiller3040cb72015-01-07 12:13:17 -08001578 while True:
Craig Tiller42bc87c2015-02-23 08:50:19 -08001579 dw = watch_dirs.DirWatcher(['src', 'include', 'test', 'examples'])
ctiller3040cb72015-01-07 12:13:17 -08001580 initial_time = dw.most_recent_change()
1581 have_files_changed = lambda: dw.most_recent_change() != initial_time
Nicolas Noble044db742015-01-14 16:57:24 -08001582 previous_success = success
Craig Tillereb9de8b2016-01-08 08:57:41 -08001583 errors = _build_and_run(check_cancelled=have_files_changed,
1584 newline_on_success=False,
Craig Tillereb9de8b2016-01-08 08:57:41 -08001585 build_only=args.build_only) == 0
1586 if not previous_success and not errors:
Nicolas Nobleb09078f2015-01-14 18:06:05 -08001587 jobset.message('SUCCESS',
1588 'All tests are now passing properly',
1589 do_newline=True)
Nicolas Noble044db742015-01-14 16:57:24 -08001590 jobset.message('IDLE', 'No change detected')
ctiller3040cb72015-01-07 12:13:17 -08001591 while not have_files_changed():
1592 time.sleep(1)
1593else:
Craig Tillereb9de8b2016-01-08 08:57:41 -08001594 errors = _build_and_run(check_cancelled=lambda: False,
Craig Tiller71735182015-01-15 17:07:13 -08001595 newline_on_success=args.newline_on_success,
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001596 xml_report=args.xml_report,
1597 build_only=args.build_only)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001598 if not errors:
Nicolas Nobleb09078f2015-01-14 18:06:05 -08001599 jobset.message('SUCCESS', 'All tests passed', do_newline=True)
1600 else:
1601 jobset.message('FAILED', 'Some tests failed', do_newline=True)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001602 exit_code = 0
1603 if BuildAndRunError.BUILD in errors:
1604 exit_code |= 1
Jan Tattermusche480a6a2016-10-07 12:59:08 +02001605 if BuildAndRunError.TEST in errors:
Craig Tillereb9de8b2016-01-08 08:57:41 -08001606 exit_code |= 2
Craig Tiller4f2be362016-01-08 08:59:20 -08001607 if BuildAndRunError.POST_TEST in errors:
1608 exit_code |= 4
Craig Tillereb9de8b2016-01-08 08:57:41 -08001609 sys.exit(exit_code)