blob: d0ab5b2e7ca468e1e8cd9fc378bd7bc78248596f [file] [log] [blame]
Nathaniel Manistaae4fbcd2015-09-23 16:29:44 +00001#!/usr/bin/env python2.7
Craig Tiller6169d5f2016-03-31 07:46:18 -07002# Copyright 2015, Google Inc.
Craig Tillerc2c79212015-02-16 12:00:01 -08003# All rights reserved.
4#
5# Redistribution and use in source and binary forms, with or without
6# modification, are permitted provided that the following conditions are
7# met:
8#
9# * Redistributions of source code must retain the above copyright
10# notice, this list of conditions and the following disclaimer.
11# * Redistributions in binary form must reproduce the above
12# copyright notice, this list of conditions and the following disclaimer
13# in the documentation and/or other materials provided with the
14# distribution.
15# * Neither the name of Google Inc. nor the names of its
16# contributors may be used to endorse or promote products derived from
17# this software without specific prior written permission.
18#
19# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
Nicolas Nobleddef2462015-01-06 18:08:25 -080031"""Run tests in parallel."""
32
siddharthshukla0589e532016-07-07 16:08:01 +020033from __future__ import print_function
34
Nicolas Nobleddef2462015-01-06 18:08:25 -080035import argparse
Craig Tiller9279ac22016-01-20 17:05:23 -080036import ast
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -070037import collections
Nicolas Nobleddef2462015-01-06 18:08:25 -080038import glob
39import itertools
Craig Tiller261dd982015-01-16 16:41:45 -080040import json
Nicolas Nobleddef2462015-01-06 18:08:25 -080041import multiprocessing
Craig Tiller1cc11db2015-01-15 22:50:50 -080042import os
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -070043import os.path
Craig Tiller38fb8de2016-07-13 08:23:32 -070044import pipes
David Garcia Quintas79e389f2015-06-02 17:49:42 -070045import platform
46import random
Craig Tillerfe406ec2015-02-24 13:55:12 -080047import re
Craig Tiller82875232015-09-25 13:57:34 -070048import socket
David Garcia Quintas79e389f2015-06-02 17:49:42 -070049import subprocess
Nicolas Nobleddef2462015-01-06 18:08:25 -080050import sys
Craig Tillerf0a293e2015-10-12 10:05:50 -070051import tempfile
52import traceback
ctiller3040cb72015-01-07 12:13:17 -080053import time
siddharthshukla0589e532016-07-07 16:08:01 +020054from six.moves import urllib
Jan Tattermusch03c01062015-12-11 14:28:56 -080055import uuid
Nicolas Nobleddef2462015-01-06 18:08:25 -080056
Jan Tattermusch5c79a312016-12-20 11:02:50 +010057import python_utils.jobset as jobset
58import python_utils.report_utils as report_utils
59import python_utils.watch_dirs as watch_dirs
Nicolas Nobleddef2462015-01-06 18:08:25 -080060
Craig Tillerb361b4e2016-01-06 11:44:17 -080061
Jan Tattermusch3b5121b2016-02-22 17:41:05 -080062_ROOT = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../..'))
63os.chdir(_ROOT)
Craig Tiller2cc2b842015-02-27 11:38:31 -080064
65
Craig Tiller8f18ee62016-07-18 08:00:33 -070066_FORCE_ENVIRON_FOR_WRAPPERS = {
67 'GRPC_VERBOSITY': 'DEBUG',
68}
Craig Tiller06805272015-06-11 14:46:47 -070069
70
Craig Tiller123f1372016-06-15 15:06:14 -070071_POLLING_STRATEGIES = {
Craig Tillere6684f42016-11-08 08:17:43 -080072 'linux': ['epoll', 'poll', 'poll-cv']
Craig Tiller123f1372016-06-15 15:06:14 -070073}
74
75
Craig Tillerd50993d2015-08-05 08:04:36 -070076def platform_string():
Nicolas "Pixel" Noblef72d7b52015-12-03 03:07:43 +010077 return jobset.platform_string()
Craig Tillerd50993d2015-08-05 08:04:36 -070078
79
Craig Tiller38fb8de2016-07-13 08:23:32 -070080_DEFAULT_TIMEOUT_SECONDS = 5 * 60
81
82
Craig Tiller738c3342015-01-12 14:28:33 -080083# SimpleConfig: just compile with CONFIG=config, and run the binary to test
Craig Tillera0f85172016-01-20 15:56:06 -080084class Config(object):
Craig Tillerb50d1662015-01-15 17:28:21 -080085
murgatroid99c36f6ea2016-10-03 09:24:09 -070086 def __init__(self, config, environ=None, timeout_multiplier=1, tool_prefix=[], iomgr_platform='native'):
murgatroid99132ce6a2015-03-04 17:29:14 -080087 if environ is None:
88 environ = {}
Craig Tiller738c3342015-01-12 14:28:33 -080089 self.build_config = config
Craig Tiller547db2b2015-01-30 14:08:39 -080090 self.environ = environ
murgatroid99132ce6a2015-03-04 17:29:14 -080091 self.environ['CONFIG'] = config
Craig Tillera0f85172016-01-20 15:56:06 -080092 self.tool_prefix = tool_prefix
Masood Malekghassemi26ea9e22015-10-09 15:19:17 -070093 self.timeout_multiplier = timeout_multiplier
murgatroid99c36f6ea2016-10-03 09:24:09 -070094 self.iomgr_platform = iomgr_platform
Craig Tiller738c3342015-01-12 14:28:33 -080095
Craig Tiller38fb8de2016-07-13 08:23:32 -070096 def job_spec(self, cmdline, timeout_seconds=_DEFAULT_TIMEOUT_SECONDS,
Craig Tillerde7edf82016-03-20 09:12:16 -070097 shortname=None, environ={}, cpu_cost=1.0, flaky=False):
Craig Tiller49f61322015-03-03 13:02:11 -080098 """Construct a jobset.JobSpec for a test under this config
99
100 Args:
101 cmdline: a list of strings specifying the command line the test
102 would like to run
Craig Tiller49f61322015-03-03 13:02:11 -0800103 """
Craig Tiller4fc90032015-05-21 10:39:52 -0700104 actual_environ = self.environ.copy()
siddharthshukla0589e532016-07-07 16:08:01 +0200105 for k, v in environ.items():
Craig Tiller4fc90032015-05-21 10:39:52 -0700106 actual_environ[k] = v
Craig Tillera0f85172016-01-20 15:56:06 -0800107 return jobset.JobSpec(cmdline=self.tool_prefix + cmdline,
Jan Tattermusch9a7d30c2015-04-23 16:12:55 -0700108 shortname=shortname,
Craig Tiller4fc90032015-05-21 10:39:52 -0700109 environ=actual_environ,
Craig Tiller56c6b6a2016-01-20 08:27:37 -0800110 cpu_cost=cpu_cost,
Craig Tiller94d04a52016-01-20 10:58:23 -0800111 timeout_seconds=(self.timeout_multiplier * timeout_seconds if timeout_seconds else None),
Craig Tillerde7edf82016-03-20 09:12:16 -0700112 flake_retries=5 if flaky or args.allow_flakes else 0,
Craig Tiller35505de2015-10-08 13:31:33 -0700113 timeout_retries=3 if args.allow_flakes else 0)
Craig Tiller738c3342015-01-12 14:28:33 -0800114
115
murgatroid99cf08daf2015-09-21 15:33:16 -0700116def get_c_tests(travis, test_lang) :
117 out = []
118 platforms_str = 'ci_platforms' if travis else 'platforms'
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100119 with open('tools/run_tests/generated/tests.json') as f:
murgatroid9989899b12015-09-22 09:14:48 -0700120 js = json.load(f)
murgatroid99a3e244f2015-09-22 11:25:53 -0700121 return [tgt
122 for tgt in js
123 if tgt['language'] == test_lang and
124 platform_string() in tgt[platforms_str] and
125 not (travis and tgt['flaky'])]
murgatroid99cf08daf2015-09-21 15:33:16 -0700126
murgatroid99fafeeb32015-09-22 09:13:03 -0700127
Jan Tattermusch77db4322016-02-20 20:19:35 -0800128def _check_compiler(compiler, supported_compilers):
129 if compiler not in supported_compilers:
Jan Tattermuschb2531e22016-03-25 16:14:41 -0700130 raise Exception('Compiler %s not supported (on this platform).' % compiler)
131
132
133def _check_arch(arch, supported_archs):
134 if arch not in supported_archs:
135 raise Exception('Architecture %s not supported.' % arch)
Jan Tattermusch77db4322016-02-20 20:19:35 -0800136
137
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800138def _is_use_docker_child():
139 """Returns True if running running as a --use_docker child."""
140 return True if os.getenv('RUN_TESTS_COMMAND') else False
141
142
siddharthshukla2135a1b2016-08-04 02:11:53 +0200143_PythonConfigVars = collections.namedtuple(
144 '_ConfigVars', ['shell', 'builder', 'builder_prefix_arguments',
145 'venv_relative_python', 'toolchain', 'runner'])
146
147
148def _python_config_generator(name, major, minor, bits, config_vars):
149 return PythonConfig(
150 name,
151 config_vars.shell + config_vars.builder + config_vars.builder_prefix_arguments + [
152 _python_pattern_function(major=major, minor=minor, bits=bits)] + [
153 name] + config_vars.venv_relative_python + config_vars.toolchain,
154 config_vars.shell + config_vars.runner + [
155 os.path.join(name, config_vars.venv_relative_python[0])])
156
157
158def _pypy_config_generator(name, major, config_vars):
159 return PythonConfig(
160 name,
161 config_vars.shell + config_vars.builder + config_vars.builder_prefix_arguments + [
162 _pypy_pattern_function(major=major)] + [
163 name] + config_vars.venv_relative_python + config_vars.toolchain,
164 config_vars.shell + config_vars.runner + [
165 os.path.join(name, config_vars.venv_relative_python[0])])
166
167
168def _python_pattern_function(major, minor, bits):
169 # Bit-ness is handled by the test machine's environment
170 if os.name == "nt":
171 if bits == "64":
172 return '/c/Python{major}{minor}/python.exe'.format(
173 major=major, minor=minor, bits=bits)
174 else:
175 return '/c/Python{major}{minor}_{bits}bits/python.exe'.format(
176 major=major, minor=minor, bits=bits)
177 else:
178 return 'python{major}.{minor}'.format(major=major, minor=minor)
179
180
181def _pypy_pattern_function(major):
182 if major == '2':
183 return 'pypy'
184 elif major == '3':
185 return 'pypy3'
186 else:
187 raise ValueError("Unknown PyPy major version")
188
189
Craig Tillerc7449162015-01-16 14:42:10 -0800190class CLanguage(object):
191
Craig Tillere9c959d2015-01-18 10:23:26 -0800192 def __init__(self, make_target, test_lang):
Craig Tillerc7449162015-01-16 14:42:10 -0800193 self.make_target = make_target
Craig Tillerd50993d2015-08-05 08:04:36 -0700194 self.platform = platform_string()
Craig Tiller711bbe62015-08-19 12:35:16 -0700195 self.test_lang = test_lang
Craig Tillerc7449162015-01-16 14:42:10 -0800196
Jan Tattermusch77db4322016-02-20 20:19:35 -0800197 def configure(self, config, args):
198 self.config = config
199 self.args = args
200 if self.platform == 'windows':
Jan Tattermuschc96caf82016-02-22 17:31:02 -0800201 self._make_options = [_windows_toolset_option(self.args.compiler),
202 _windows_arch_option(self.args.arch)]
Jan Tattermusch77db4322016-02-20 20:19:35 -0800203 else:
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800204 self._docker_distro, self._make_options = self._compiler_options(self.args.use_docker,
205 self.args.compiler)
murgatroid99c36f6ea2016-10-03 09:24:09 -0700206 if args.iomgr_platform == "uv":
207 cflags = '-DGRPC_UV '
208 try:
209 cflags += subprocess.check_output(['pkg-config', '--cflags', 'libuv']).strip() + ' '
murgatroid991687cab2016-10-11 11:42:01 -0700210 except (subprocess.CalledProcessError, OSError):
murgatroid99c36f6ea2016-10-03 09:24:09 -0700211 pass
212 try:
213 ldflags = subprocess.check_output(['pkg-config', '--libs', 'libuv']).strip() + ' '
murgatroid991687cab2016-10-11 11:42:01 -0700214 except (subprocess.CalledProcessError, OSError):
murgatroid99c36f6ea2016-10-03 09:24:09 -0700215 ldflags = '-luv '
216 self._make_options += ['EXTRA_CPPFLAGS={}'.format(cflags),
217 'EXTRA_LDLIBS={}'.format(ldflags)]
Jan Tattermusch77db4322016-02-20 20:19:35 -0800218
219 def test_specs(self):
Craig Tiller547db2b2015-01-30 14:08:39 -0800220 out = []
Jan Tattermusch77db4322016-02-20 20:19:35 -0800221 binaries = get_c_tests(self.args.travis, self.test_lang)
Craig Tiller946ce7a2016-04-06 10:35:58 -0700222 for target in binaries:
Craig Tiller123f1372016-06-15 15:06:14 -0700223 polling_strategies = (_POLLING_STRATEGIES.get(self.platform, ['all'])
Craig Tiller946ce7a2016-04-06 10:35:58 -0700224 if target.get('uses_polling', True)
225 else ['all'])
murgatroid992c287ca2016-10-07 09:55:35 -0700226 if self.args.iomgr_platform == 'uv':
227 polling_strategies = ['all']
Craig Tiller946ce7a2016-04-06 10:35:58 -0700228 for polling_strategy in polling_strategies:
229 env={'GRPC_DEFAULT_SSL_ROOTS_FILE_PATH':
230 _ROOT + '/src/core/lib/tsi/test_creds/ca.pem',
Craig Tiller8f18ee62016-07-18 08:00:33 -0700231 'GRPC_POLL_STRATEGY': polling_strategy,
232 'GRPC_VERBOSITY': 'DEBUG'}
Craig Tiller38fb8de2016-07-13 08:23:32 -0700233 shortname_ext = '' if polling_strategy=='all' else ' GRPC_POLL_STRATEGY=%s' % polling_strategy
Craig Tillerbc28cd62016-12-02 13:52:21 -0800234 timeout_scaling = 1
235 if polling_strategy == 'poll-cv':
236 timeout_scaling *= 5
Craig Tillerb38197e2016-02-26 10:14:54 -0800237 if self.config.build_config in target['exclude_configs']:
238 continue
murgatroid99c36f6ea2016-10-03 09:24:09 -0700239 if self.args.iomgr_platform in target.get('exclude_iomgrs', []):
240 continue
Craig Tillerb38197e2016-02-26 10:14:54 -0800241 if self.platform == 'windows':
242 binary = 'vsprojects/%s%s/%s.exe' % (
243 'x64/' if self.args.arch == 'x64' else '',
244 _MSBUILD_CONFIG[self.config.build_config],
245 target['name'])
Craig Tillerca62ff02016-02-24 22:22:57 -0800246 else:
Craig Tillerb38197e2016-02-26 10:14:54 -0800247 binary = 'bins/%s/%s' % (self.config.build_config, target['name'])
Craig Tillerab34b122016-11-28 13:19:12 -0800248 cpu_cost = target['cpu_cost']
249 if cpu_cost == 'capacity':
250 cpu_cost = multiprocessing.cpu_count()
Craig Tillerb38197e2016-02-26 10:14:54 -0800251 if os.path.isfile(binary):
252 if 'gtest' in target and target['gtest']:
253 # here we parse the output of --gtest_list_tests to build up a
254 # complete list of the tests contained in a binary
255 # for each test, we then add a job to run, filtering for just that
256 # test
257 with open(os.devnull, 'w') as fnull:
258 tests = subprocess.check_output([binary, '--gtest_list_tests'],
259 stderr=fnull)
260 base = None
261 for line in tests.split('\n'):
262 i = line.find('#')
263 if i >= 0: line = line[:i]
264 if not line: continue
265 if line[0] != ' ':
266 base = line.strip()
267 else:
268 assert base is not None
269 assert line[1] == ' '
270 test = base + line.strip()
271 cmdline = [binary] + ['--gtest_filter=%s' % test]
Craig Tiller38fb8de2016-07-13 08:23:32 -0700272 out.append(self.config.job_spec(cmdline,
273 shortname='%s --gtest_filter=%s %s' % (binary, test, shortname_ext),
Craig Tillerab34b122016-11-28 13:19:12 -0800274 cpu_cost=cpu_cost,
Craig Tillerbc28cd62016-12-02 13:52:21 -0800275 timeout_seconds=_DEFAULT_TIMEOUT_SECONDS * timeout_scaling,
Craig Tillerb38197e2016-02-26 10:14:54 -0800276 environ=env))
277 else:
278 cmdline = [binary] + target['args']
Craig Tiller38fb8de2016-07-13 08:23:32 -0700279 out.append(self.config.job_spec(cmdline,
280 shortname=' '.join(
281 pipes.quote(arg)
282 for arg in cmdline) +
283 shortname_ext,
Craig Tillerab34b122016-11-28 13:19:12 -0800284 cpu_cost=cpu_cost,
Craig Tillerc2278152016-03-21 08:59:54 -0700285 flaky=target.get('flaky', False),
Craig Tillerbc28cd62016-12-02 13:52:21 -0800286 timeout_seconds=target.get('timeout_seconds', _DEFAULT_TIMEOUT_SECONDS) * timeout_scaling,
Craig Tillerb38197e2016-02-26 10:14:54 -0800287 environ=env))
288 elif self.args.regex == '.*' or self.platform == 'windows':
siddharthshukla0589e532016-07-07 16:08:01 +0200289 print('\nWARNING: binary not found, skipping', binary)
Nicolas Noblee1445362015-05-11 17:40:26 -0700290 return sorted(out)
Craig Tillerc7449162015-01-16 14:42:10 -0800291
Jan Tattermusch77db4322016-02-20 20:19:35 -0800292 def make_targets(self):
Jan Tattermusch77db4322016-02-20 20:19:35 -0800293 if self.platform == 'windows':
Craig Tiller7bb3efd2015-09-01 08:04:03 -0700294 # don't build tools on windows just yet
295 return ['buildtests_%s' % self.make_target]
Craig Tiller7552f0f2015-06-19 17:46:20 -0700296 return ['buildtests_%s' % self.make_target, 'tools_%s' % self.make_target]
Craig Tillerc7449162015-01-16 14:42:10 -0800297
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800298 def make_options(self):
Jan Tattermuschc96caf82016-02-22 17:31:02 -0800299 return self._make_options;
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800300
murgatroid99256d3df2015-09-21 16:58:02 -0700301 def pre_build_steps(self):
Jan Tattermusch874aec02015-10-07 19:26:19 -0700302 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100303 return [['tools\\run_tests\\helper_scripts\\pre_build_c.bat']]
Jan Tattermusch874aec02015-10-07 19:26:19 -0700304 else:
305 return []
murgatroid99256d3df2015-09-21 16:58:02 -0700306
Craig Tillerc7449162015-01-16 14:42:10 -0800307 def build_steps(self):
308 return []
309
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200310 def post_tests_steps(self):
311 if self.platform == 'windows':
312 return []
313 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100314 return [['tools/run_tests/helper_scripts/post_tests_c.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200315
murgatroid99a3e244f2015-09-22 11:25:53 -0700316 def makefile_name(self):
317 return 'Makefile'
318
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700319 def _clang_make_options(self, version_suffix=''):
320 return ['CC=clang%s' % version_suffix,
321 'CXX=clang++%s' % version_suffix,
322 'LD=clang%s' % version_suffix,
323 'LDXX=clang++%s' % version_suffix]
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800324
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700325 def _gcc_make_options(self, version_suffix):
326 return ['CC=gcc%s' % version_suffix,
327 'CXX=g++%s' % version_suffix,
328 'LD=gcc%s' % version_suffix,
329 'LDXX=g++%s' % version_suffix]
Jan Tattermusch9bb70622016-03-18 10:28:54 -0700330
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800331 def _compiler_options(self, use_docker, compiler):
332 """Returns docker distro and make options to use for given compiler."""
Jan Tattermuschfd3857b2016-06-03 12:24:03 -0700333 if not use_docker and not _is_use_docker_child():
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800334 _check_compiler(compiler, ['default'])
335
336 if compiler == 'gcc4.9' or compiler == 'default':
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800337 return ('jessie', [])
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800338 elif compiler == 'gcc4.4':
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700339 return ('wheezy', self._gcc_make_options(version_suffix='-4.4'))
340 elif compiler == 'gcc4.6':
341 return ('wheezy', self._gcc_make_options(version_suffix='-4.6'))
Matt Kwong029ed102016-11-01 18:04:47 -0700342 elif compiler == 'gcc4.8':
Matt Kwong1347e402016-11-02 18:07:40 -0700343 return ('jessie', self._gcc_make_options(version_suffix='-4.8'))
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800344 elif compiler == 'gcc5.3':
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800345 return ('ubuntu1604', [])
346 elif compiler == 'clang3.4':
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700347 # on ubuntu1404, clang-3.4 alias doesn't exist, just use 'clang'
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800348 return ('ubuntu1404', self._clang_make_options())
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700349 elif compiler == 'clang3.5':
350 return ('jessie', self._clang_make_options(version_suffix='-3.5'))
Jan Tattermuschd4726c12016-02-23 16:57:36 -0800351 elif compiler == 'clang3.6':
Jan Tattermusch6d258c52016-06-10 09:36:51 -0700352 return ('ubuntu1604', self._clang_make_options(version_suffix='-3.6'))
353 elif compiler == 'clang3.7':
354 return ('ubuntu1604', self._clang_make_options(version_suffix='-3.7'))
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800355 else:
356 raise Exception('Compiler %s not supported.' % compiler)
357
Jan Tattermusch77db4322016-02-20 20:19:35 -0800358 def dockerfile_dir(self):
Jan Tattermuschc4cbe392016-02-22 19:29:38 -0800359 return 'tools/dockerfile/test/cxx_%s_%s' % (self._docker_distro,
360 _docker_arch_suffix(self.args.arch))
Jan Tattermusch788ee232016-01-26 12:19:44 -0800361
murgatroid99132ce6a2015-03-04 17:29:14 -0800362 def __str__(self):
363 return self.make_target
364
Craig Tillercc0535d2015-12-08 15:14:47 -0800365
murgatroid992c8d5162015-01-26 10:41:21 -0800366class NodeLanguage(object):
367
Jan Tattermusche477b842016-02-06 22:19:01 -0800368 def __init__(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800369 self.platform = platform_string()
Jan Tattermusche477b842016-02-06 22:19:01 -0800370
Jan Tattermusch77db4322016-02-20 20:19:35 -0800371 def configure(self, config, args):
372 self.config = config
373 self.args = args
murgatroid99eaf79642016-11-01 11:05:02 -0700374 # Note: electron ABI only depends on major and minor version, so that's all
375 # we should specify in the compiler argument
murgatroid999fab4382016-04-29 15:05:00 -0700376 _check_compiler(self.args.compiler, ['default', 'node0.12',
murgatroid99eaf79642016-11-01 11:05:02 -0700377 'node4', 'node5', 'node6',
murgatroid99fdbf7332016-11-01 11:13:58 -0700378 'node7', 'electron1.3'])
murgatroid991191b722017-02-08 11:56:52 -0800379 if args.iomgr_platform == "uv":
380 self.use_uv = True
381 else:
382 self.use_uv = False
murgatroid999fab4382016-04-29 15:05:00 -0700383 if self.args.compiler == 'default':
murgatroid99c5181982017-01-12 12:40:11 -0800384 self.runtime = 'node'
murgatroid991191b722017-02-08 11:56:52 -0800385 self.node_version = '7'
murgatroid999fab4382016-04-29 15:05:00 -0700386 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700387 if self.args.compiler.startswith('electron'):
388 self.runtime = 'electron'
389 self.node_version = self.args.compiler[8:]
390 else:
391 self.runtime = 'node'
392 # Take off the word "node"
393 self.node_version = self.args.compiler[4:]
Jan Tattermusch77db4322016-02-20 20:19:35 -0800394
395 def test_specs(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800396 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100397 return [self.config.job_spec(['tools\\run_tests\\helper_scripts\\run_node.bat'])]
Michael Lumishaaa876a2016-02-10 15:27:58 -0800398 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700399 run_script = 'run_node'
400 if self.runtime == 'electron':
401 run_script += '_electron'
murgatroid99c34cac22017-01-04 15:43:02 -0800402 return [self.config.job_spec(['tools/run_tests/helper_scripts/{}.sh'.format(run_script),
murgatroid99eaf79642016-11-01 11:05:02 -0700403 self.node_version],
Jan Tattermusch77db4322016-02-20 20:19:35 -0800404 None,
405 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
murgatroid992c8d5162015-01-26 10:41:21 -0800406
murgatroid99256d3df2015-09-21 16:58:02 -0700407 def pre_build_steps(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800408 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100409 return [['tools\\run_tests\\helper_scripts\\pre_build_node.bat']]
Michael Lumishaaa876a2016-02-10 15:27:58 -0800410 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700411 build_script = 'pre_build_node'
412 if self.runtime == 'electron':
413 build_script += '_electron'
murgatroid991191b722017-02-08 11:56:52 -0800414 return [['tools/run_tests/helper_scripts/{}.sh'.format(build_script),
415 self.node_version]]
murgatroid99256d3df2015-09-21 16:58:02 -0700416
Jan Tattermusch77db4322016-02-20 20:19:35 -0800417 def make_targets(self):
murgatroid99db5b1602015-10-01 13:20:11 -0700418 return []
murgatroid992c8d5162015-01-26 10:41:21 -0800419
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800420 def make_options(self):
421 return []
422
murgatroid992c8d5162015-01-26 10:41:21 -0800423 def build_steps(self):
Michael Lumishaaa876a2016-02-10 15:27:58 -0800424 if self.platform == 'windows':
murgatroid991191b722017-02-08 11:56:52 -0800425 if config == 'dbg':
426 config_flag = '--debug'
427 else:
428 config_flag = '--release'
429 return [['tools\\run_tests\\helper_scripts\\build_node.bat',
430 '--grpc_uv={}'.format('true' if self.use_uv else 'false'),
431 config_flag]]
Michael Lumishaaa876a2016-02-10 15:27:58 -0800432 else:
murgatroid99eaf79642016-11-01 11:05:02 -0700433 build_script = 'build_node'
434 if self.runtime == 'electron':
435 build_script += '_electron'
436 # building for electron requires a patch version
437 self.node_version += '.0'
murgatroid991191b722017-02-08 11:56:52 -0800438 return [['tools/run_tests/helper_scripts/{}.sh'.format(build_script),
439 self.node_version,
440 '--grpc_uv={}'.format('true' if self.use_uv else 'false')]]
Craig Tillerc7449162015-01-16 14:42:10 -0800441
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200442 def post_tests_steps(self):
443 return []
444
murgatroid99a3e244f2015-09-22 11:25:53 -0700445 def makefile_name(self):
446 return 'Makefile'
447
Jan Tattermusch77db4322016-02-20 20:19:35 -0800448 def dockerfile_dir(self):
449 return 'tools/dockerfile/test/node_jessie_%s' % _docker_arch_suffix(self.args.arch)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800450
murgatroid99132ce6a2015-03-04 17:29:14 -0800451 def __str__(self):
452 return 'node'
453
Craig Tiller99775822015-01-30 13:07:16 -0800454
Craig Tillerc7449162015-01-16 14:42:10 -0800455class PhpLanguage(object):
456
Jan Tattermusch77db4322016-02-20 20:19:35 -0800457 def configure(self, config, args):
458 self.config = config
459 self.args = args
460 _check_compiler(self.args.compiler, ['default'])
461
462 def test_specs(self):
Jan Tattermuschfffb2672016-12-19 15:24:45 +0100463 return [self.config.job_spec(['src/php/bin/run_tests.sh'],
Jan Tattermusch77db4322016-02-20 20:19:35 -0800464 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
Craig Tillerc7449162015-01-16 14:42:10 -0800465
murgatroid99256d3df2015-09-21 16:58:02 -0700466 def pre_build_steps(self):
467 return []
468
Jan Tattermusch77db4322016-02-20 20:19:35 -0800469 def make_targets(self):
Craig Tilleraf7cf542015-05-22 10:07:34 -0700470 return ['static_c', 'shared_c']
Craig Tillerc7449162015-01-16 14:42:10 -0800471
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800472 def make_options(self):
473 return []
474
Craig Tillerc7449162015-01-16 14:42:10 -0800475 def build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100476 return [['tools/run_tests/helper_scripts/build_php.sh']]
Craig Tillerc7449162015-01-16 14:42:10 -0800477
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200478 def post_tests_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100479 return [['tools/run_tests/helper_scripts/post_tests_php.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200480
murgatroid99a3e244f2015-09-22 11:25:53 -0700481 def makefile_name(self):
482 return 'Makefile'
483
Jan Tattermusch77db4322016-02-20 20:19:35 -0800484 def dockerfile_dir(self):
485 return 'tools/dockerfile/test/php_jessie_%s' % _docker_arch_suffix(self.args.arch)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800486
murgatroid99132ce6a2015-03-04 17:29:14 -0800487 def __str__(self):
488 return 'php'
489
Craig Tillerc7449162015-01-16 14:42:10 -0800490
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700491class Php7Language(object):
492
493 def configure(self, config, args):
494 self.config = config
495 self.args = args
496 _check_compiler(self.args.compiler, ['default'])
497
498 def test_specs(self):
Jan Tattermuschfffb2672016-12-19 15:24:45 +0100499 return [self.config.job_spec(['src/php/bin/run_tests.sh'],
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700500 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
501
502 def pre_build_steps(self):
503 return []
504
505 def make_targets(self):
506 return ['static_c', 'shared_c']
507
508 def make_options(self):
509 return []
510
511 def build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100512 return [['tools/run_tests/helper_scripts/build_php.sh']]
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700513
514 def post_tests_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100515 return [['tools/run_tests/helper_scripts/post_tests_php.sh']]
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700516
517 def makefile_name(self):
518 return 'Makefile'
519
520 def dockerfile_dir(self):
521 return 'tools/dockerfile/test/php7_jessie_%s' % _docker_arch_suffix(self.args.arch)
522
523 def __str__(self):
524 return 'php7'
525
526
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700527class PythonConfig(collections.namedtuple('PythonConfig', [
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700528 'name', 'build', 'run'])):
529 """Tuple of commands (named s.t. 'what it says on the tin' applies)"""
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700530
Nathaniel Manista840615e2015-01-22 20:31:47 +0000531class PythonLanguage(object):
532
Jan Tattermusch77db4322016-02-20 20:19:35 -0800533 def configure(self, config, args):
534 self.config = config
535 self.args = args
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700536 self.pythons = self._get_pythons(self.args)
Jan Tattermusch77db4322016-02-20 20:19:35 -0800537
538 def test_specs(self):
Jan Tattermusch072ebaa2016-03-01 18:33:12 -0800539 # load list of known test suites
Masood Malekghassemi1ff429d2016-06-02 16:39:20 -0700540 with open('src/python/grpcio_tests/tests/tests.json') as tests_json_file:
Jan Tattermusch072ebaa2016-03-01 18:33:12 -0800541 tests_json = json.load(tests_json_file)
Masood Malekghassemi2b841622015-07-28 17:39:02 -0700542 environment = dict(_FORCE_ENVIRON_FOR_WRAPPERS)
Masood Malekghassemi1c062bd2016-06-13 18:41:36 -0700543 return [self.config.job_spec(
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700544 config.run,
Masood Malekghassemie6a23e22016-06-28 13:58:42 -0700545 timeout_seconds=5*60,
siddharthshukla0589e532016-07-07 16:08:01 +0200546 environ=dict(list(environment.items()) +
Masood Malekghassemic4f5a2e2016-12-05 15:54:56 -0800547 [('GRPC_PYTHON_TESTRUNNER_FILTER', str(suite_name))]),
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700548 shortname='%s.test.%s' % (config.name, suite_name),)
Masood Malekghassemi1c062bd2016-06-13 18:41:36 -0700549 for suite_name in tests_json
550 for config in self.pythons]
Nathaniel Manista840615e2015-01-22 20:31:47 +0000551
murgatroid99256d3df2015-09-21 16:58:02 -0700552 def pre_build_steps(self):
553 return []
554
Jan Tattermusch77db4322016-02-20 20:19:35 -0800555 def make_targets(self):
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700556 return []
Nathaniel Manista840615e2015-01-22 20:31:47 +0000557
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800558 def make_options(self):
559 return []
560
Nathaniel Manista840615e2015-01-22 20:31:47 +0000561 def build_steps(self):
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700562 return [config.build for config in self.pythons]
Nathaniel Manista840615e2015-01-22 20:31:47 +0000563
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200564 def post_tests_steps(self):
565 return []
566
murgatroid99a3e244f2015-09-22 11:25:53 -0700567 def makefile_name(self):
568 return 'Makefile'
569
Jan Tattermusch77db4322016-02-20 20:19:35 -0800570 def dockerfile_dir(self):
siddharthshuklac4782142016-06-28 18:48:47 +0200571 return 'tools/dockerfile/test/python_%s_%s' % (self.python_manager_name(), _docker_arch_suffix(self.args.arch))
572
573 def python_manager_name(self):
574 return 'pyenv' if self.args.compiler in ['python3.5', 'python3.6'] else 'jessie'
Jan Tattermusch788ee232016-01-26 12:19:44 -0800575
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700576 def _get_pythons(self, args):
577 if args.arch == 'x86':
578 bits = '32'
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700579 else:
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700580 bits = '64'
siddharthshukla2135a1b2016-08-04 02:11:53 +0200581
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700582 if os.name == 'nt':
583 shell = ['bash']
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100584 builder = [os.path.abspath('tools/run_tests/helper_scripts/build_python_msys2.sh')]
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700585 builder_prefix_arguments = ['MINGW{}'.format(bits)]
586 venv_relative_python = ['Scripts/python.exe']
587 toolchain = ['mingw32']
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700588 else:
589 shell = []
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100590 builder = [os.path.abspath('tools/run_tests/helper_scripts/build_python.sh')]
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700591 builder_prefix_arguments = []
592 venv_relative_python = ['bin/python']
593 toolchain = ['unix']
siddharthshukla2135a1b2016-08-04 02:11:53 +0200594
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100595 runner = [os.path.abspath('tools/run_tests/helper_scripts/run_python.sh')]
siddharthshukla2135a1b2016-08-04 02:11:53 +0200596 config_vars = _PythonConfigVars(shell, builder, builder_prefix_arguments,
597 venv_relative_python, toolchain, runner)
598 python27_config = _python_config_generator(name='py27', major='2',
599 minor='7', bits=bits,
600 config_vars=config_vars)
601 python34_config = _python_config_generator(name='py34', major='3',
602 minor='4', bits=bits,
603 config_vars=config_vars)
604 python35_config = _python_config_generator(name='py35', major='3',
605 minor='5', bits=bits,
606 config_vars=config_vars)
607 python36_config = _python_config_generator(name='py36', major='3',
608 minor='6', bits=bits,
609 config_vars=config_vars)
610 pypy27_config = _pypy_config_generator(name='pypy', major='2',
611 config_vars=config_vars)
612 pypy32_config = _pypy_config_generator(name='pypy3', major='3',
613 config_vars=config_vars)
614
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700615 if args.compiler == 'default':
616 if os.name == 'nt':
617 return (python27_config,)
618 else:
619 return (python27_config, python34_config,)
620 elif args.compiler == 'python2.7':
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700621 return (python27_config,)
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700622 elif args.compiler == 'python3.4':
Masood Malekghassemi3b5b2062016-06-02 20:27:20 -0700623 return (python34_config,)
siddharthshuklac4782142016-06-28 18:48:47 +0200624 elif args.compiler == 'python3.5':
625 return (python35_config,)
626 elif args.compiler == 'python3.6':
627 return (python36_config,)
siddharthshukla2135a1b2016-08-04 02:11:53 +0200628 elif args.compiler == 'pypy':
629 return (pypy27_config,)
630 elif args.compiler == 'pypy3':
631 return (pypy32_config,)
Jan Tattermusch825471c2016-04-25 16:52:25 -0700632 else:
Masood Malekghassemicab9d4f2016-06-28 09:09:31 -0700633 raise Exception('Compiler %s not supported.' % args.compiler)
Jan Tattermusch825471c2016-04-25 16:52:25 -0700634
murgatroid99132ce6a2015-03-04 17:29:14 -0800635 def __str__(self):
636 return 'python'
637
Craig Tillerd625d812015-04-08 15:52:35 -0700638
murgatroid996a4c4fa2015-02-27 12:08:57 -0800639class RubyLanguage(object):
640
Jan Tattermusch77db4322016-02-20 20:19:35 -0800641 def configure(self, config, args):
642 self.config = config
643 self.args = args
644 _check_compiler(self.args.compiler, ['default'])
645
646 def test_specs(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100647 return [self.config.job_spec(['tools/run_tests/helper_scripts/run_ruby.sh'],
Jan Tattermusch77db4322016-02-20 20:19:35 -0800648 timeout_seconds=10*60,
649 environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
murgatroid996a4c4fa2015-02-27 12:08:57 -0800650
murgatroid99256d3df2015-09-21 16:58:02 -0700651 def pre_build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100652 return [['tools/run_tests/helper_scripts/pre_build_ruby.sh']]
murgatroid99256d3df2015-09-21 16:58:02 -0700653
Jan Tattermusch4651bef2016-02-23 08:31:25 -0800654 def make_targets(self):
murgatroid997d243df2016-02-18 09:58:05 -0800655 return []
murgatroid996a4c4fa2015-02-27 12:08:57 -0800656
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800657 def make_options(self):
658 return []
659
murgatroid996a4c4fa2015-02-27 12:08:57 -0800660 def build_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100661 return [['tools/run_tests/helper_scripts/build_ruby.sh']]
murgatroid996a4c4fa2015-02-27 12:08:57 -0800662
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200663 def post_tests_steps(self):
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100664 return [['tools/run_tests/helper_scripts/post_tests_ruby.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200665
murgatroid99a3e244f2015-09-22 11:25:53 -0700666 def makefile_name(self):
667 return 'Makefile'
668
Jan Tattermusch77db4322016-02-20 20:19:35 -0800669 def dockerfile_dir(self):
670 return 'tools/dockerfile/test/ruby_jessie_%s' % _docker_arch_suffix(self.args.arch)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800671
murgatroid99132ce6a2015-03-04 17:29:14 -0800672 def __str__(self):
673 return 'ruby'
674
Craig Tillerd625d812015-04-08 15:52:35 -0700675
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800676class CSharpLanguage(object):
Jan Tattermusch77db4322016-02-20 20:19:35 -0800677
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700678 def __init__(self):
Craig Tillerd50993d2015-08-05 08:04:36 -0700679 self.platform = platform_string()
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700680
Jan Tattermusch77db4322016-02-20 20:19:35 -0800681 def configure(self, config, args):
682 self.config = config
683 self.args = args
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700684 if self.platform == 'windows':
Jan Tattermuschb2531e22016-03-25 16:14:41 -0700685 # Explicitly choosing between x86 and x64 arch doesn't work yet
686 _check_arch(self.args.arch, ['default'])
Jan Tattermusch6e2f88c2016-06-21 10:52:40 -0700687 # CoreCLR use 64bit runtime by default.
688 arch_option = 'x64' if self.args.compiler == 'coreclr' else self.args.arch
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700689 self._make_options = [_windows_toolset_option(self.args.compiler),
Jan Tattermusch6e2f88c2016-06-21 10:52:40 -0700690 _windows_arch_option(arch_option)]
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700691 else:
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700692 _check_compiler(self.args.compiler, ['default', 'coreclr'])
693 if self.platform == 'linux' and self.args.compiler == 'coreclr':
694 self._docker_distro = 'coreclr'
Jan Tattermusch743decd2016-06-21 11:40:47 -0700695 else:
696 self._docker_distro = 'jessie'
Jan Tattermusch76511a52016-06-17 14:00:57 -0700697
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700698 if self.platform == 'mac':
Jan Tattermusch2a322c22016-03-30 13:55:07 -0700699 # TODO(jtattermusch): EMBED_ZLIB=true currently breaks the mac build
Jan Tattermusch6d082202016-06-21 10:03:38 -0700700 self._make_options = ['EMBED_OPENSSL=true']
701 if self.args.compiler != 'coreclr':
702 # On Mac, official distribution of mono is 32bit.
703 self._make_options += ['CFLAGS=-m32', 'LDFLAGS=-m32']
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700704 else:
705 self._make_options = ['EMBED_OPENSSL=true', 'EMBED_ZLIB=true']
Jan Tattermusch77db4322016-02-20 20:19:35 -0800706
707 def test_specs(self):
Jan Tattermusch03c01062015-12-11 14:28:56 -0800708 with open('src/csharp/tests.json') as f:
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700709 tests_by_assembly = json.load(f)
Jan Tattermusch03c01062015-12-11 14:28:56 -0800710
Jan Tattermuscha2d964c2016-02-22 17:33:09 -0800711 msbuild_config = _MSBUILD_CONFIG[self.config.build_config]
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700712 nunit_args = ['--labels=All']
Jan Tattermusch76511a52016-06-17 14:00:57 -0700713 assembly_subdir = 'bin/%s' % msbuild_config
714 assembly_extension = '.exe'
715
716 if self.args.compiler == 'coreclr':
Jan Tattermusch1f7ce192016-09-08 16:21:16 +0200717 assembly_subdir += '/netcoreapp1.0'
718 runtime_cmd = ['dotnet', 'exec']
719 assembly_extension = '.dll'
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700720 else:
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700721 nunit_args += ['--noresult', '--workers=1']
722 if self.platform == 'windows':
723 runtime_cmd = []
724 else:
725 runtime_cmd = ['mono']
Jan Tattermuschbf3b1532015-10-26 10:24:42 -0700726
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700727 specs = []
728 for assembly in tests_by_assembly.iterkeys():
Jan Tattermusch76511a52016-06-17 14:00:57 -0700729 assembly_file = 'src/csharp/%s/%s/%s%s' % (assembly,
730 assembly_subdir,
731 assembly,
732 assembly_extension)
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700733 if self.config.build_config != 'gcov' or self.platform != 'windows':
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700734 # normally, run each test as a separate process
735 for test in tests_by_assembly[assembly]:
736 cmdline = runtime_cmd + [assembly_file, '--test=%s' % test] + nunit_args
737 specs.append(self.config.job_spec(cmdline,
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700738 shortname='csharp.%s' % test,
739 environ=_FORCE_ENVIRON_FOR_WRAPPERS))
740 else:
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700741 # For C# test coverage, run all tests from the same assembly at once
742 # using OpenCover.Console (only works on Windows).
743 cmdline = ['src\\csharp\\packages\\OpenCover.4.6.519\\tools\\OpenCover.Console.exe',
744 '-target:%s' % assembly_file,
745 '-targetdir:src\\csharp',
746 '-targetargs:%s' % ' '.join(nunit_args),
747 '-filter:+[Grpc.Core]*',
748 '-register:user',
749 '-output:src\\csharp\\coverage_csharp_%s.xml' % assembly]
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700750
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700751 # set really high cpu_cost to make sure instances of OpenCover.Console run exclusively
752 # to prevent problems with registering the profiler.
753 run_exclusive = 1000000
Jan Tattermusch35e608f2016-04-09 16:35:06 -0700754 specs.append(self.config.job_spec(cmdline,
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700755 shortname='csharp.coverage.%s' % assembly,
Jan Tattermuscha5f1f122016-04-11 15:49:56 -0700756 cpu_cost=run_exclusive,
Jan Tattermusch77db4322016-02-20 20:19:35 -0800757 environ=_FORCE_ENVIRON_FOR_WRAPPERS))
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700758 return specs
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800759
murgatroid99256d3df2015-09-21 16:58:02 -0700760 def pre_build_steps(self):
Jan Tattermusch48423fc2015-10-07 18:59:16 -0700761 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100762 return [['tools\\run_tests\\helper_scripts\\pre_build_csharp.bat']]
Jan Tattermusch48423fc2015-10-07 18:59:16 -0700763 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100764 return [['tools/run_tests/helper_scripts/pre_build_csharp.sh']]
murgatroid99256d3df2015-09-21 16:58:02 -0700765
Jan Tattermusch77db4322016-02-20 20:19:35 -0800766 def make_targets(self):
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700767 return ['grpc_csharp_ext']
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800768
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800769 def make_options(self):
Jan Tattermusch6d28d352016-03-25 15:07:22 -0700770 return self._make_options;
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800771
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800772 def build_steps(self):
Jan Tattermusch76511a52016-06-17 14:00:57 -0700773 if self.args.compiler == 'coreclr':
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700774 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100775 return [['tools\\run_tests\\helper_scripts\\build_csharp_coreclr.bat']]
Jan Tattermuschbc98af12016-06-17 18:38:27 -0700776 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100777 return [['tools/run_tests/helper_scripts/build_csharp_coreclr.sh']]
Jan Tattermuschb00aa672015-06-01 15:48:03 -0700778 else:
Jan Tattermusch76511a52016-06-17 14:00:57 -0700779 if self.platform == 'windows':
780 return [[_windows_build_bat(self.args.compiler),
781 'src/csharp/Grpc.sln',
782 '/p:Configuration=%s' % _MSBUILD_CONFIG[self.config.build_config]]]
783 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100784 return [['tools/run_tests/helper_scripts/build_csharp.sh']]
Nathaniel Manista840615e2015-01-22 20:31:47 +0000785
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200786 def post_tests_steps(self):
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700787 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100788 return [['tools\\run_tests\\helper_scripts\\post_tests_csharp.bat']]
Jan Tattermusch38ed2cf2016-04-09 16:24:16 -0700789 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100790 return [['tools/run_tests/helper_scripts/post_tests_csharp.sh']]
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200791
murgatroid99a3e244f2015-09-22 11:25:53 -0700792 def makefile_name(self):
793 return 'Makefile'
794
Jan Tattermusch77db4322016-02-20 20:19:35 -0800795 def dockerfile_dir(self):
Jan Tattermusch76511a52016-06-17 14:00:57 -0700796 return 'tools/dockerfile/test/csharp_%s_%s' % (self._docker_distro,
797 _docker_arch_suffix(self.args.arch))
Jan Tattermusch788ee232016-01-26 12:19:44 -0800798
murgatroid99132ce6a2015-03-04 17:29:14 -0800799 def __str__(self):
800 return 'csharp'
801
Craig Tillerd625d812015-04-08 15:52:35 -0700802
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700803class ObjCLanguage(object):
804
Jan Tattermusch77db4322016-02-20 20:19:35 -0800805 def configure(self, config, args):
806 self.config = config
807 self.args = args
808 _check_compiler(self.args.compiler, ['default'])
809
810 def test_specs(self):
Jorge Canizales8a9fe2a2016-07-29 15:44:47 -0700811 return [
812 self.config.job_spec(['src/objective-c/tests/run_tests.sh'],
Jan Tattermuschfffb2672016-12-19 15:24:45 +0100813 timeout_seconds=60*60,
Jorge Canizales8a9fe2a2016-07-29 15:44:47 -0700814 shortname='objc-tests',
815 environ=_FORCE_ENVIRON_FOR_WRAPPERS),
816 self.config.job_spec(['src/objective-c/tests/build_example_test.sh'],
Nicolas "Pixel" Noble7f074e02016-08-11 21:00:08 +0200817 timeout_seconds=30*60,
Jorge Canizales8a9fe2a2016-07-29 15:44:47 -0700818 shortname='objc-examples-build',
819 environ=_FORCE_ENVIRON_FOR_WRAPPERS),
820 ]
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700821
murgatroid99256d3df2015-09-21 16:58:02 -0700822 def pre_build_steps(self):
823 return []
824
Jan Tattermusch77db4322016-02-20 20:19:35 -0800825 def make_targets(self):
Jorge Canizales6eade6d2016-07-11 00:34:14 -0700826 return ['interop_server']
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700827
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800828 def make_options(self):
829 return []
830
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700831 def build_steps(self):
Jorge Canizalesd0b32e92015-07-30 23:08:43 -0700832 return [['src/objective-c/tests/build_tests.sh']]
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700833
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +0200834 def post_tests_steps(self):
835 return []
836
murgatroid99a3e244f2015-09-22 11:25:53 -0700837 def makefile_name(self):
838 return 'Makefile'
839
Jan Tattermusch77db4322016-02-20 20:19:35 -0800840 def dockerfile_dir(self):
Jan Tattermusch788ee232016-01-26 12:19:44 -0800841 return None
842
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700843 def __str__(self):
844 return 'objc'
845
846
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100847class Sanity(object):
848
Jan Tattermusch77db4322016-02-20 20:19:35 -0800849 def configure(self, config, args):
850 self.config = config
851 self.args = args
852 _check_compiler(self.args.compiler, ['default'])
853
854 def test_specs(self):
Craig Tiller94d04a52016-01-20 10:58:23 -0800855 import yaml
Jan Tattermusch788ee232016-01-26 12:19:44 -0800856 with open('tools/run_tests/sanity/sanity_tests.yaml', 'r') as f:
Jan Tattermusch7dd2cc62016-12-20 17:15:39 +0100857 environ={'TEST': 'true'}
858 if _is_use_docker_child():
859 environ['CLANG_FORMAT_SKIP_DOCKER'] = 'true'
Craig Tiller34226af2016-06-24 16:46:25 -0700860 return [self.config.job_spec(cmd['script'].split(),
Jan Tattermusch7dd2cc62016-12-20 17:15:39 +0100861 timeout_seconds=30*60,
862 environ=environ,
Jan Tattermusch77db4322016-02-20 20:19:35 -0800863 cpu_cost=cmd.get('cpu_cost', 1))
Craig Tiller94d04a52016-01-20 10:58:23 -0800864 for cmd in yaml.load(f)]
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100865
murgatroid99256d3df2015-09-21 16:58:02 -0700866 def pre_build_steps(self):
867 return []
868
Jan Tattermusch77db4322016-02-20 20:19:35 -0800869 def make_targets(self):
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100870 return ['run_dep_checks']
871
Jan Tattermuschc895fe02016-01-20 09:13:09 -0800872 def make_options(self):
873 return []
874
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100875 def build_steps(self):
876 return []
877
Nicolas "Pixel" Noble87879b32015-10-12 23:28:53 +0200878 def post_tests_steps(self):
879 return []
880
murgatroid99a3e244f2015-09-22 11:25:53 -0700881 def makefile_name(self):
882 return 'Makefile'
883
Jan Tattermusch77db4322016-02-20 20:19:35 -0800884 def dockerfile_dir(self):
Jan Tattermusche70b3c52016-02-07 20:21:02 -0800885 return 'tools/dockerfile/test/sanity'
Jan Tattermusch788ee232016-01-26 12:19:44 -0800886
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100887 def __str__(self):
888 return 'sanity'
889
murgatroid99b53e5d12016-10-18 09:55:28 -0700890class NodeExpressLanguage(object):
891 """Dummy Node express test target to enable running express performance
892 benchmarks"""
893
894 def __init__(self):
895 self.platform = platform_string()
896
897 def configure(self, config, args):
898 self.config = config
899 self.args = args
900 _check_compiler(self.args.compiler, ['default', 'node0.12',
901 'node4', 'node5', 'node6'])
902 if self.args.compiler == 'default':
903 self.node_version = '4'
904 else:
905 # Take off the word "node"
906 self.node_version = self.args.compiler[4:]
907
908 def test_specs(self):
909 return []
910
911 def pre_build_steps(self):
912 if self.platform == 'windows':
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100913 return [['tools\\run_tests\\helper_scripts\\pre_build_node.bat']]
murgatroid99b53e5d12016-10-18 09:55:28 -0700914 else:
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100915 return [['tools/run_tests/helper_scripts/pre_build_node.sh', self.node_version]]
murgatroid99b53e5d12016-10-18 09:55:28 -0700916
917 def make_targets(self):
918 return []
919
920 def make_options(self):
921 return []
922
923 def build_steps(self):
924 return []
925
926 def post_tests_steps(self):
927 return []
928
929 def makefile_name(self):
930 return 'Makefile'
931
932 def dockerfile_dir(self):
933 return 'tools/dockerfile/test/node_jessie_%s' % _docker_arch_suffix(self.args.arch)
934
935 def __str__(self):
936 return 'node_express'
Nicolas "Pixel" Noblee55cd7f2015-04-14 17:59:13 +0200937
Craig Tiller738c3342015-01-12 14:28:33 -0800938# different configurations we can run under
Jan Tattermusch5c79a312016-12-20 11:02:50 +0100939with open('tools/run_tests/generated/configs.json') as f:
Craig Tiller1dce9062016-01-20 17:01:56 -0800940 _CONFIGS = dict((cfg['config'], Config(**cfg)) for cfg in ast.literal_eval(f.read()))
Craig Tiller738c3342015-01-12 14:28:33 -0800941
942
Craig Tillerc7449162015-01-16 14:42:10 -0800943_LANGUAGES = {
Craig Tillere9c959d2015-01-18 10:23:26 -0800944 'c++': CLanguage('cxx', 'c++'),
945 'c': CLanguage('c', 'c'),
murgatroid992c8d5162015-01-26 10:41:21 -0800946 'node': NodeLanguage(),
murgatroid99b53e5d12016-10-18 09:55:28 -0700947 'node_express': NodeExpressLanguage(),
Nathaniel Manista840615e2015-01-22 20:31:47 +0000948 'php': PhpLanguage(),
Stanley Cheung2e2cdff2016-07-23 19:07:36 -0700949 'php7': Php7Language(),
Nathaniel Manista840615e2015-01-22 20:31:47 +0000950 'python': PythonLanguage(),
Jan Tattermusch1970a5b2015-03-03 15:17:25 -0800951 'ruby': RubyLanguage(),
Nicolas "Pixel" Noble9f728642015-03-24 18:50:30 +0100952 'csharp': CSharpLanguage(),
Jorge Canizalesa0b3bfa2015-07-30 19:25:52 -0700953 'objc' : ObjCLanguage(),
Jan Tattermusch70a57e42016-02-20 18:50:27 -0800954 'sanity': Sanity()
Craig Tillereb272bc2015-01-30 13:13:14 -0800955 }
Nicolas Nobleddef2462015-01-06 18:08:25 -0800956
Jan Tattermusch77db4322016-02-20 20:19:35 -0800957
Jan Tattermuscha2d964c2016-02-22 17:33:09 -0800958_MSBUILD_CONFIG = {
Craig Tiller7bb3efd2015-09-01 08:04:03 -0700959 'dbg': 'Debug',
960 'opt': 'Release',
Jan Tattermusche4a69182015-12-15 09:53:01 -0800961 'gcov': 'Debug',
Craig Tiller7bb3efd2015-09-01 08:04:03 -0700962 }
963
David Garcia Quintase90cd372015-05-31 18:15:26 -0700964
Jan Tattermusch2dd156e2015-12-04 18:26:17 -0800965def _windows_arch_option(arch):
966 """Returns msbuild cmdline option for selected architecture."""
Jan Tattermusch9be594f2016-01-25 18:08:47 -0800967 if arch == 'default' or arch == 'x86':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -0800968 return '/p:Platform=Win32'
Jan Tattermusch9be594f2016-01-25 18:08:47 -0800969 elif arch == 'x64':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -0800970 return '/p:Platform=x64'
971 else:
siddharthshukla0589e532016-07-07 16:08:01 +0200972 print('Architecture %s not supported.' % arch)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -0800973 sys.exit(1)
Jan Tattermusch788ee232016-01-26 12:19:44 -0800974
Jan Tattermuschf08018a2016-01-26 08:22:09 -0800975
976def _check_arch_option(arch):
977 """Checks that architecture option is valid."""
978 if platform_string() == 'windows':
979 _windows_arch_option(arch)
980 elif platform_string() == 'linux':
981 # On linux, we need to be running under docker with the right architecture.
Jan Tattermusch07fb0422016-01-26 10:46:56 -0800982 runtime_arch = platform.architecture()[0]
Jan Tattermuschf08018a2016-01-26 08:22:09 -0800983 if arch == 'default':
984 return
985 elif runtime_arch == '64bit' and arch == 'x64':
986 return
987 elif runtime_arch == '32bit' and arch == 'x86':
988 return
989 else:
siddharthshukla0589e532016-07-07 16:08:01 +0200990 print('Architecture %s does not match current runtime architecture.' % arch)
Jan Tattermuschf08018a2016-01-26 08:22:09 -0800991 sys.exit(1)
992 else:
993 if args.arch != 'default':
siddharthshukla0589e532016-07-07 16:08:01 +0200994 print('Architecture %s not supported on current platform.' % args.arch)
Jan Tattermuschf08018a2016-01-26 08:22:09 -0800995 sys.exit(1)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -0800996
Jan Tattermusch4dc9e722016-01-25 17:00:54 -0800997
Jan Tattermusch2dd156e2015-12-04 18:26:17 -0800998def _windows_build_bat(compiler):
999 """Returns name of build.bat for selected compiler."""
Jan Tattermuschbc98af12016-06-17 18:38:27 -07001000 # For CoreCLR, fall back to the default compiler for C core
1001 if compiler == 'default' or compiler == 'vs2013' or compiler == 'coreclr':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001002 return 'vsprojects\\build_vs2013.bat'
1003 elif compiler == 'vs2015':
1004 return 'vsprojects\\build_vs2015.bat'
1005 elif compiler == 'vs2010':
1006 return 'vsprojects\\build_vs2010.bat'
1007 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001008 print('Compiler %s not supported.' % compiler)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001009 sys.exit(1)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001010
1011
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001012def _windows_toolset_option(compiler):
1013 """Returns msbuild PlatformToolset for selected compiler."""
Jan Tattermuschbc98af12016-06-17 18:38:27 -07001014 # For CoreCLR, fall back to the default compiler for C core
1015 if compiler == 'default' or compiler == 'vs2013' or compiler == 'coreclr':
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001016 return '/p:PlatformToolset=v120'
1017 elif compiler == 'vs2015':
1018 return '/p:PlatformToolset=v140'
1019 elif compiler == 'vs2010':
1020 return '/p:PlatformToolset=v100'
1021 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001022 print('Compiler %s not supported.' % compiler)
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001023 sys.exit(1)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001024
1025
Jan Tattermusche70b3c52016-02-07 20:21:02 -08001026def _docker_arch_suffix(arch):
1027 """Returns suffix to dockerfile dir to use."""
1028 if arch == 'default' or arch == 'x64':
1029 return 'x64'
1030 elif arch == 'x86':
1031 return 'x86'
1032 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001033 print('Architecture %s not supported with current settings.' % arch)
Jan Tattermusche70b3c52016-02-07 20:21:02 -08001034 sys.exit(1)
1035
1036
David Garcia Quintase90cd372015-05-31 18:15:26 -07001037def runs_per_test_type(arg_str):
1038 """Auxilary function to parse the "runs_per_test" flag.
1039
1040 Returns:
1041 A positive integer or 0, the latter indicating an infinite number of
1042 runs.
1043
1044 Raises:
1045 argparse.ArgumentTypeError: Upon invalid input.
1046 """
1047 if arg_str == 'inf':
1048 return 0
1049 try:
1050 n = int(arg_str)
1051 if n <= 0: raise ValueError
Craig Tiller50e53e22015-06-01 20:18:21 -07001052 return n
David Garcia Quintase90cd372015-05-31 18:15:26 -07001053 except:
Adele Zhoue4c35612015-10-16 15:34:23 -07001054 msg = '\'{}\' is not a positive integer or \'inf\''.format(arg_str)
David Garcia Quintase90cd372015-05-31 18:15:26 -07001055 raise argparse.ArgumentTypeError(msg)
Jan Tattermuschc95eead2015-09-18 13:03:50 -07001056
siddharthshukla2135a1b2016-08-04 02:11:53 +02001057
Jan Tattermuschc95eead2015-09-18 13:03:50 -07001058# parse command line
1059argp = argparse.ArgumentParser(description='Run grpc tests.')
1060argp.add_argument('-c', '--config',
Jan Tattermusch77db4322016-02-20 20:19:35 -08001061 choices=sorted(_CONFIGS.keys()),
1062 default='opt')
David Garcia Quintase90cd372015-05-31 18:15:26 -07001063argp.add_argument('-n', '--runs_per_test', default=1, type=runs_per_test_type,
1064 help='A positive integer or "inf". If "inf", all tests will run in an '
1065 'infinite loop. Especially useful in combination with "-f"')
Craig Tillerfe406ec2015-02-24 13:55:12 -08001066argp.add_argument('-r', '--regex', default='.*', type=str)
Vijay Pai488fd0e2016-06-13 12:37:12 -07001067argp.add_argument('--regex_exclude', default='', type=str)
Craig Tiller5f735a62016-01-20 09:31:15 -08001068argp.add_argument('-j', '--jobs', default=multiprocessing.cpu_count(), type=int)
Craig Tiller8451e872015-02-27 09:25:51 -08001069argp.add_argument('-s', '--slowdown', default=1.0, type=float)
ctiller3040cb72015-01-07 12:13:17 -08001070argp.add_argument('-f', '--forever',
1071 default=False,
1072 action='store_const',
1073 const=True)
Nicolas "Pixel" Noblea7df3f92015-02-26 22:07:04 +01001074argp.add_argument('-t', '--travis',
1075 default=False,
1076 action='store_const',
1077 const=True)
Nicolas Noble044db742015-01-14 16:57:24 -08001078argp.add_argument('--newline_on_success',
1079 default=False,
1080 action='store_const',
1081 const=True)
Craig Tiller686fb262015-01-15 07:39:09 -08001082argp.add_argument('-l', '--language',
Craig Tiller60f15e62015-05-13 09:05:17 -07001083 choices=['all'] + sorted(_LANGUAGES.keys()),
Craig Tiller686fb262015-01-15 07:39:09 -08001084 nargs='+',
Craig Tiller60f15e62015-05-13 09:05:17 -07001085 default=['all'])
Craig Tillercd43da82015-05-29 08:41:29 -07001086argp.add_argument('-S', '--stop_on_failure',
1087 default=False,
1088 action='store_const',
1089 const=True)
Jan Tattermuschc95eead2015-09-18 13:03:50 -07001090argp.add_argument('--use_docker',
1091 default=False,
1092 action='store_const',
1093 const=True,
Adele Zhoue4c35612015-10-16 15:34:23 -07001094 help='Run all the tests under docker. That provides ' +
1095 'additional isolation and prevents the need to install ' +
1096 'language specific prerequisites. Only available on Linux.')
Craig Tillerd4509a12015-09-28 09:18:40 -07001097argp.add_argument('--allow_flakes',
1098 default=False,
1099 action='store_const',
1100 const=True,
Adele Zhoue4c35612015-10-16 15:34:23 -07001101 help='Allow flaky tests to show as passing (re-runs failed tests up to five times)')
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001102argp.add_argument('--arch',
Jan Tattermusch9be594f2016-01-25 18:08:47 -08001103 choices=['default', 'x86', 'x64'],
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001104 default='default',
1105 help='Selects architecture to target. For some platforms "default" is the only supported choice.')
1106argp.add_argument('--compiler',
Jan Tattermuschc4cbe392016-02-22 19:29:38 -08001107 choices=['default',
Matt Kwong029ed102016-11-01 18:04:47 -07001108 'gcc4.4', 'gcc4.6', 'gcc4.8', 'gcc4.9', 'gcc5.3',
Jan Tattermusch6d258c52016-06-10 09:36:51 -07001109 'clang3.4', 'clang3.5', 'clang3.6', 'clang3.7',
Jan Tattermusch825471c2016-04-25 16:52:25 -07001110 'vs2010', 'vs2013', 'vs2015',
siddharthshukla2135a1b2016-08-04 02:11:53 +02001111 'python2.7', 'python3.4', 'python3.5', 'python3.6', 'pypy', 'pypy3',
murgatroid995a3f8622016-10-26 13:45:04 -07001112 'node0.12', 'node4', 'node5', 'node6', 'node7',
murgatroid99eaf79642016-11-01 11:05:02 -07001113 'electron1.3',
Jan Tattermusch76511a52016-06-17 14:00:57 -07001114 'coreclr'],
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001115 default='default',
Jan Tattermusch77db4322016-02-20 20:19:35 -08001116 help='Selects compiler to use. Allowed values depend on the platform and language.')
murgatroid99c36f6ea2016-10-03 09:24:09 -07001117argp.add_argument('--iomgr_platform',
1118 choices=['native', 'uv'],
1119 default='native',
1120 help='Selects iomgr platform to build on')
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001121argp.add_argument('--build_only',
1122 default=False,
1123 action='store_const',
1124 const=True,
1125 help='Perform all the build steps but dont run any tests.')
Craig Tiller5f735a62016-01-20 09:31:15 -08001126argp.add_argument('--measure_cpu_costs', default=False, action='store_const', const=True,
1127 help='Measure the cpu costs of tests')
Craig Tiller1676f912016-01-05 10:49:44 -08001128argp.add_argument('--update_submodules', default=[], nargs='*',
1129 help='Update some submodules before building. If any are updated, also run generate_projects. ' +
1130 'Submodules are specified as SUBMODULE_NAME:BRANCH; if BRANCH is omitted, master is assumed.')
Craig Tiller234b6e72015-05-23 10:12:40 -07001131argp.add_argument('-a', '--antagonists', default=0, type=int)
Nicolas "Pixel" Noble5937b5b2015-06-26 02:04:12 +02001132argp.add_argument('-x', '--xml_report', default=None, type=str,
1133 help='Generates a JUnit-compatible XML report')
Jan Tattermuschcfcc0752016-10-09 17:02:34 +02001134argp.add_argument('--report_suite_name', default='tests', type=str,
1135 help='Test suite name to use in generated JUnit XML report')
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001136argp.add_argument('--quiet_success',
1137 default=False,
1138 action='store_const',
1139 const=True,
1140 help='Dont print anything when a test passes. Passing tests also will not be reported in XML report. ' +
1141 'Useful when running many iterations of each test (argument -n).')
Craig Tiller123f1372016-06-15 15:06:14 -07001142argp.add_argument('--force_default_poller', default=False, action='store_const', const=True,
1143 help='Dont try to iterate over many polling strategies when they exist')
Nicolas Nobleddef2462015-01-06 18:08:25 -08001144args = argp.parse_args()
1145
Craig Tiller123f1372016-06-15 15:06:14 -07001146if args.force_default_poller:
1147 _POLLING_STRATEGIES = {}
1148
Craig Tiller5f735a62016-01-20 09:31:15 -08001149jobset.measure_cpu_costs = args.measure_cpu_costs
1150
Craig Tiller1676f912016-01-05 10:49:44 -08001151# update submodules if necessary
Craig Tillerb361b4e2016-01-06 11:44:17 -08001152need_to_regenerate_projects = False
1153for spec in args.update_submodules:
1154 spec = spec.split(':', 1)
1155 if len(spec) == 1:
1156 submodule = spec[0]
1157 branch = 'master'
1158 elif len(spec) == 2:
1159 submodule = spec[0]
1160 branch = spec[1]
1161 cwd = 'third_party/%s' % submodule
1162 def git(cmd, cwd=cwd):
siddharthshukla0589e532016-07-07 16:08:01 +02001163 print('in %s: git %s' % (cwd, cmd))
Craig Tillerb361b4e2016-01-06 11:44:17 -08001164 subprocess.check_call('git %s' % cmd, cwd=cwd, shell=True)
1165 git('fetch')
1166 git('checkout %s' % branch)
1167 git('pull origin %s' % branch)
1168 if os.path.exists('src/%s/gen_build_yaml.py' % submodule):
1169 need_to_regenerate_projects = True
1170if need_to_regenerate_projects:
1171 if jobset.platform_string() == 'linux':
1172 subprocess.check_call('tools/buildgen/generate_projects.sh', shell=True)
1173 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001174 print('WARNING: may need to regenerate projects, but since we are not on')
1175 print(' Linux this step is being skipped. Compilation MAY fail.')
Craig Tiller1676f912016-01-05 10:49:44 -08001176
1177
Nicolas Nobleddef2462015-01-06 18:08:25 -08001178# grab config
Jan Tattermusch77db4322016-02-20 20:19:35 -08001179run_config = _CONFIGS[args.config]
1180build_config = run_config.build_config
Craig Tillerf1973b02015-01-16 12:32:13 -08001181
Craig Tiller06805272015-06-11 14:46:47 -07001182if args.travis:
murgatroid99d3b5b7f2015-10-06 17:02:03 -07001183 _FORCE_ENVIRON_FOR_WRAPPERS = {'GRPC_TRACE': 'api'}
Craig Tiller06805272015-06-11 14:46:47 -07001184
Adele Zhou6b9527c2015-11-20 15:56:35 -08001185if 'all' in args.language:
Craig Tiller1676f912016-01-05 10:49:44 -08001186 lang_list = _LANGUAGES.keys()
Adele Zhou6b9527c2015-11-20 15:56:35 -08001187else:
1188 lang_list = args.language
Craig Tiller16900662016-01-07 19:30:54 -08001189# We don't support code coverage on some languages
1190if 'gcov' in args.config:
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001191 for bad in ['objc', 'sanity']:
Craig Tiller16900662016-01-07 19:30:54 -08001192 if bad in lang_list:
1193 lang_list.remove(bad)
Adele Zhou6b9527c2015-11-20 15:56:35 -08001194
1195languages = set(_LANGUAGES[l] for l in lang_list)
Jan Tattermusch77db4322016-02-20 20:19:35 -08001196for l in languages:
1197 l.configure(run_config, args)
murgatroid99132ce6a2015-03-04 17:29:14 -08001198
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001199language_make_options=[]
1200if any(language.make_options() for language in languages):
Adele Zhou3b6ab812016-05-18 17:04:20 -07001201 if not 'gcov' in args.config and len(languages) != 1:
siddharthshukla0589e532016-07-07 16:08:01 +02001202 print('languages with custom make options cannot be built simultaneously with other languages')
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001203 sys.exit(1)
1204 else:
1205 language_make_options = next(iter(languages)).make_options()
1206
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001207if args.use_docker:
1208 if not args.travis:
siddharthshukla0589e532016-07-07 16:08:01 +02001209 print('Seen --use_docker flag, will run tests under docker.')
1210 print('')
1211 print('IMPORTANT: The changes you are testing need to be locally committed')
1212 print('because only the committed changes in the current branch will be')
1213 print('copied to the docker environment.')
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001214 time.sleep(5)
1215
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001216 dockerfile_dirs = set([l.dockerfile_dir() for l in languages])
1217 if len(dockerfile_dirs) > 1:
Adele Zhou9506ef22016-03-02 13:53:34 -08001218 if 'gcov' in args.config:
1219 dockerfile_dir = 'tools/dockerfile/test/multilang_jessie_x64'
1220 print ('Using multilang_jessie_x64 docker image for code coverage for '
1221 'all languages.')
1222 else:
1223 print ('Languages to be tested require running under different docker '
1224 'images.')
1225 sys.exit(1)
1226 else:
1227 dockerfile_dir = next(iter(dockerfile_dirs))
Craig Tillerde7edf82016-03-20 09:12:16 -07001228
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001229 child_argv = [ arg for arg in sys.argv if not arg == '--use_docker' ]
Jan Tattermusched342b12016-01-26 14:40:31 -08001230 run_tests_cmd = 'python tools/run_tests/run_tests.py %s' % ' '.join(child_argv[1:])
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001231
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001232 env = os.environ.copy()
1233 env['RUN_TESTS_COMMAND'] = run_tests_cmd
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001234 env['DOCKERFILE_DIR'] = dockerfile_dir
Jan Tattermusch9835d4b2016-04-29 15:05:05 -07001235 env['DOCKER_RUN_SCRIPT'] = 'tools/run_tests/dockerize/docker_run_tests.sh'
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001236 if args.xml_report:
1237 env['XML_REPORT'] = args.xml_report
1238 if not args.travis:
1239 env['TTY_FLAG'] = '-t' # enables Ctrl-C when not on Jenkins.
1240
Jan Tattermusch9835d4b2016-04-29 15:05:05 -07001241 subprocess.check_call(['tools/run_tests/dockerize/build_docker_and_run_tests.sh'],
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001242 shell=True,
1243 env=env)
1244 sys.exit(0)
Jan Tattermusch788ee232016-01-26 12:19:44 -08001245
Jan Tattermuschf08018a2016-01-26 08:22:09 -08001246_check_arch_option(args.arch)
Jan Tattermusch4dc9e722016-01-25 17:00:54 -08001247
Jan Tattermuschfba65302016-01-25 18:21:14 -08001248def make_jobspec(cfg, targets, makefile='Makefile'):
1249 if platform_string() == 'windows':
Craig Tillerfc3c0c42015-09-01 16:47:54 -07001250 extra_args = []
Craig Tillerb5391e12015-09-03 14:35:18 -07001251 # better do parallel compilation
Jan Tattermusch47eeb2b2015-10-07 14:09:18 -07001252 # empirically /m:2 gives the best performance/price and should prevent
1253 # overloading the windows workers.
Adele Zhoue4c35612015-10-16 15:34:23 -07001254 extra_args.extend(['/m:2'])
Craig Tillerb5391e12015-09-03 14:35:18 -07001255 # disable PDB generation: it's broken, and we don't need it during CI
Adele Zhoue4c35612015-10-16 15:34:23 -07001256 extra_args.extend(['/p:Jenkins=true'])
Craig Tiller6fd23842015-09-01 07:36:31 -07001257 return [
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001258 jobset.JobSpec([_windows_build_bat(args.compiler),
murgatroid99cf08daf2015-09-21 15:33:16 -07001259 'vsprojects\\%s.sln' % target,
Jan Tattermuscha2d964c2016-02-22 17:33:09 -08001260 '/p:Configuration=%s' % _MSBUILD_CONFIG[cfg]] +
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001261 extra_args +
1262 language_make_options,
Craig Tiller590105a2016-01-19 13:03:46 -08001263 shell=True, timeout_seconds=None)
Craig Tiller6fd23842015-09-01 07:36:31 -07001264 for target in targets]
Jan Tattermuschfba65302016-01-25 18:21:14 -08001265 else:
murgatroid998ae409f2015-10-26 16:39:00 -07001266 if targets:
1267 return [jobset.JobSpec([os.getenv('MAKE', 'make'),
1268 '-f', makefile,
Craig Tillerdd6f7ed2016-01-21 12:54:42 -08001269 '-j', '%d' % args.jobs,
Craig Tiller71a86042016-01-15 14:59:58 -08001270 'EXTRA_DEFINES=GRPC_TEST_SLOWDOWN_MACHINE_FACTOR=%f' % args.slowdown,
1271 'CONFIG=%s' % cfg] +
Jan Tattermuschc895fe02016-01-20 09:13:09 -08001272 language_make_options +
Craig Tiller71a86042016-01-15 14:59:58 -08001273 ([] if not args.travis else ['JENKINS_BUILD=1']) +
1274 targets,
Craig Tiller590105a2016-01-19 13:03:46 -08001275 timeout_seconds=None)]
murgatroid998ae409f2015-10-26 16:39:00 -07001276 else:
1277 return []
Jan Tattermuschfba65302016-01-25 18:21:14 -08001278
murgatroid99a3e244f2015-09-22 11:25:53 -07001279make_targets = {}
1280for l in languages:
1281 makefile = l.makefile_name()
1282 make_targets[makefile] = make_targets.get(makefile, set()).union(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001283 set(l.make_targets()))
Craig Tiller5058c692015-04-08 09:42:04 -07001284
Jan Tattermusche4a69182015-12-15 09:53:01 -08001285def build_step_environ(cfg):
1286 environ = {'CONFIG': cfg}
Jan Tattermuscha2d964c2016-02-22 17:33:09 -08001287 msbuild_cfg = _MSBUILD_CONFIG.get(cfg)
Jan Tattermusche4a69182015-12-15 09:53:01 -08001288 if msbuild_cfg:
1289 environ['MSBUILD_CONFIG'] = msbuild_cfg
1290 return environ
1291
murgatroid99fddac962015-09-22 09:20:11 -07001292build_steps = list(set(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001293 jobset.JobSpec(cmdline, environ=build_step_environ(build_config), flake_retries=5)
murgatroid99256d3df2015-09-21 16:58:02 -07001294 for l in languages
1295 for cmdline in l.pre_build_steps()))
Craig Tillerbd4e3782015-09-01 06:48:55 -07001296if make_targets:
siddharthshukla0589e532016-07-07 16:08:01 +02001297 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 -07001298 build_steps.extend(set(make_commands))
Craig Tiller5058c692015-04-08 09:42:04 -07001299build_steps.extend(set(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001300 jobset.JobSpec(cmdline, environ=build_step_environ(build_config), timeout_seconds=None)
Craig Tiller547db2b2015-01-30 14:08:39 -08001301 for l in languages
Craig Tiller533b1a22015-05-29 08:41:29 -07001302 for cmdline in l.build_steps()))
Craig Tillerf1973b02015-01-16 12:32:13 -08001303
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +02001304post_tests_steps = list(set(
Jan Tattermusch77db4322016-02-20 20:19:35 -08001305 jobset.JobSpec(cmdline, environ=build_step_environ(build_config))
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +02001306 for l in languages
1307 for cmdline in l.post_tests_steps()))
Nicolas Nobleddef2462015-01-06 18:08:25 -08001308runs_per_test = args.runs_per_test
ctiller3040cb72015-01-07 12:13:17 -08001309forever = args.forever
Nicolas Nobleddef2462015-01-06 18:08:25 -08001310
Nicolas Nobleddef2462015-01-06 18:08:25 -08001311
Ken Paysonfa51de52016-06-30 23:50:48 -07001312def _shut_down_legacy_server(legacy_server_port):
1313 try:
siddharthshukla0589e532016-07-07 16:08:01 +02001314 version = int(urllib.request.urlopen(
Ken Paysonfa51de52016-06-30 23:50:48 -07001315 'http://localhost:%d/version_number' % legacy_server_port,
1316 timeout=10).read())
1317 except:
1318 pass
1319 else:
siddharthshukla0589e532016-07-07 16:08:01 +02001320 urllib.request.urlopen(
Ken Paysonfa51de52016-06-30 23:50:48 -07001321 'http://localhost:%d/quitquitquit' % legacy_server_port).read()
1322
1323
Craig Tillerf53d9c82015-08-04 14:19:43 -07001324def _start_port_server(port_server_port):
1325 # check if a compatible port server is running
1326 # if incompatible (version mismatch) ==> start a new one
1327 # if not running ==> start a new one
1328 # otherwise, leave it up
1329 try:
siddharthshukla0589e532016-07-07 16:08:01 +02001330 version = int(urllib.request.urlopen(
Craig Tillerfe4939f2015-10-06 12:55:36 -07001331 'http://localhost:%d/version_number' % port_server_port,
Jan Tattermusch292d0102016-06-28 10:29:41 -07001332 timeout=10).read())
siddharthshukla0589e532016-07-07 16:08:01 +02001333 print('detected port server running version %d' % version)
Craig Tillerf53d9c82015-08-04 14:19:43 -07001334 running = True
Craig Tillerfe4939f2015-10-06 12:55:36 -07001335 except Exception as e:
siddharthshukla0589e532016-07-07 16:08:01 +02001336 print('failed to detect port server: %s' % sys.exc_info()[0])
1337 print(e.strerror)
Craig Tillerf53d9c82015-08-04 14:19:43 -07001338 running = False
1339 if running:
Craig Tillerfe4939f2015-10-06 12:55:36 -07001340 current_version = int(subprocess.check_output(
Jan Tattermusch5c79a312016-12-20 11:02:50 +01001341 [sys.executable, os.path.abspath('tools/run_tests/python_utils/port_server.py'),
Jan Tattermusch3bd08272015-11-04 19:24:37 -08001342 'dump_version']))
siddharthshukla0589e532016-07-07 16:08:01 +02001343 print('my port server is version %d' % current_version)
Craig Tillerfe4939f2015-10-06 12:55:36 -07001344 running = (version >= current_version)
1345 if not running:
siddharthshukla0589e532016-07-07 16:08:01 +02001346 print('port_server version mismatch: killing the old one')
1347 urllib.request.urlopen('http://localhost:%d/quitquitquit' % port_server_port).read()
Craig Tillerfe4939f2015-10-06 12:55:36 -07001348 time.sleep(1)
Craig Tillerf53d9c82015-08-04 14:19:43 -07001349 if not running:
Craig Tillerf0a293e2015-10-12 10:05:50 -07001350 fd, logfile = tempfile.mkstemp()
1351 os.close(fd)
siddharthshukla0589e532016-07-07 16:08:01 +02001352 print('starting port_server, with log file %s' % logfile)
Jan Tattermusch5c79a312016-12-20 11:02:50 +01001353 args = [sys.executable, os.path.abspath('tools/run_tests/python_utils/port_server.py'),
Jan Tattermusch3bd08272015-11-04 19:24:37 -08001354 '-p', '%d' % port_server_port, '-l', logfile]
Craig Tiller367d41d2015-10-12 13:00:22 -07001355 env = dict(os.environ)
1356 env['BUILD_ID'] = 'pleaseDontKillMeJenkins'
Nicolas "Pixel" Noblef72d7b52015-12-03 03:07:43 +01001357 if platform_string() == 'windows':
Jan Tattermusch3bd08272015-11-04 19:24:37 -08001358 # Working directory of port server needs to be outside of Jenkins
1359 # workspace to prevent file lock issues.
1360 tempdir = tempfile.mkdtemp()
Craig Tillerd2c39712015-10-12 11:08:49 -07001361 port_server = subprocess.Popen(
Craig Tiller367d41d2015-10-12 13:00:22 -07001362 args,
1363 env=env,
Jan Tattermusch3bd08272015-11-04 19:24:37 -08001364 cwd=tempdir,
Craig Tiller367d41d2015-10-12 13:00:22 -07001365 creationflags = 0x00000008, # detached process
1366 close_fds=True)
Craig Tillerd2c39712015-10-12 11:08:49 -07001367 else:
1368 port_server = subprocess.Popen(
1369 args,
Craig Tiller367d41d2015-10-12 13:00:22 -07001370 env=env,
Craig Tillerd2c39712015-10-12 11:08:49 -07001371 preexec_fn=os.setsid,
1372 close_fds=True)
Craig Tillerf0a293e2015-10-12 10:05:50 -07001373 time.sleep(1)
Craig Tiller8b5f4dc2015-08-26 08:02:01 -07001374 # ensure port server is up
Craig Tillerabd37fd2015-08-26 07:54:01 -07001375 waits = 0
Craig Tillerf53d9c82015-08-04 14:19:43 -07001376 while True:
Craig Tillerabd37fd2015-08-26 07:54:01 -07001377 if waits > 10:
siddharthshukla0589e532016-07-07 16:08:01 +02001378 print('killing port server due to excessive start up waits')
Craig Tillerabd37fd2015-08-26 07:54:01 -07001379 port_server.kill()
Craig Tillera2f38b02015-09-24 11:19:17 -07001380 if port_server.poll() is not None:
siddharthshukla0589e532016-07-07 16:08:01 +02001381 print('port_server failed to start')
Craig Tillerf0a293e2015-10-12 10:05:50 -07001382 # try one final time: maybe another build managed to start one
1383 time.sleep(1)
1384 try:
siddharthshukla0589e532016-07-07 16:08:01 +02001385 urllib.request.urlopen('http://localhost:%d/get' % port_server_port,
Craig Tillerf0a293e2015-10-12 10:05:50 -07001386 timeout=1).read()
siddharthshukla0589e532016-07-07 16:08:01 +02001387 print('last ditch attempt to contact port server succeeded')
Craig Tillerf0a293e2015-10-12 10:05:50 -07001388 break
1389 except:
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001390 traceback.print_exc()
Craig Tillerf0a293e2015-10-12 10:05:50 -07001391 port_log = open(logfile, 'r').read()
siddharthshukla0589e532016-07-07 16:08:01 +02001392 print(port_log)
Craig Tillerf0a293e2015-10-12 10:05:50 -07001393 sys.exit(1)
Craig Tillerf53d9c82015-08-04 14:19:43 -07001394 try:
siddharthshukla0589e532016-07-07 16:08:01 +02001395 urllib.request.urlopen('http://localhost:%d/get' % port_server_port,
Craig Tillerabd37fd2015-08-26 07:54:01 -07001396 timeout=1).read()
siddharthshukla0589e532016-07-07 16:08:01 +02001397 print('port server is up and ready')
Craig Tillerf53d9c82015-08-04 14:19:43 -07001398 break
Craig Tiller31fdaa42015-09-25 13:09:59 -07001399 except socket.timeout:
siddharthshukla0589e532016-07-07 16:08:01 +02001400 print('waiting for port_server: timeout')
Craig Tillerf0a293e2015-10-12 10:05:50 -07001401 traceback.print_exc();
1402 time.sleep(1)
Craig Tiller31fdaa42015-09-25 13:09:59 -07001403 waits += 1
siddharthshukla0589e532016-07-07 16:08:01 +02001404 except urllib.error.URLError:
1405 print('waiting for port_server: urlerror')
Craig Tillerf0a293e2015-10-12 10:05:50 -07001406 traceback.print_exc();
1407 time.sleep(1)
Craig Tillerabd37fd2015-08-26 07:54:01 -07001408 waits += 1
Craig Tillerf53d9c82015-08-04 14:19:43 -07001409 except:
Jan Tattermusch3b5121b2016-02-22 17:41:05 -08001410 traceback.print_exc()
Craig Tillerf53d9c82015-08-04 14:19:43 -07001411 port_server.kill()
1412 raise
1413
1414
Adele Zhoud5fffa52015-10-23 15:51:42 -07001415def _calculate_num_runs_failures(list_of_results):
1416 """Caculate number of runs and failures for a particular test.
1417
1418 Args:
1419 list_of_results: (List) of JobResult object.
1420 Returns:
1421 A tuple of total number of runs and failures.
1422 """
1423 num_runs = len(list_of_results) # By default, there is 1 run per JobResult.
1424 num_failures = 0
1425 for jobresult in list_of_results:
1426 if jobresult.retries > 0:
1427 num_runs += jobresult.retries
1428 if jobresult.num_failures > 0:
1429 num_failures += jobresult.num_failures
1430 return num_runs, num_failures
1431
Adele Zhou6b9527c2015-11-20 15:56:35 -08001432
Craig Tillereb9de8b2016-01-08 08:57:41 -08001433# _build_and_run results
1434class BuildAndRunError(object):
1435
1436 BUILD = object()
1437 TEST = object()
1438 POST_TEST = object()
1439
1440
1441# returns a list of things that failed (or an empty list on success)
Craig Tillerf53d9c82015-08-04 14:19:43 -07001442def _build_and_run(
Craig Tiller74189cd2016-06-23 15:39:06 -07001443 check_cancelled, newline_on_success, xml_report=None, build_only=False):
ctiller3040cb72015-01-07 12:13:17 -08001444 """Do one pass of building & running tests."""
murgatroid99666450e2015-01-26 13:03:31 -08001445 # build latest sequentially
Jan Tattermuschaab1e512016-01-28 09:30:44 -08001446 num_failures, resultset = jobset.run(
Adele Zhoue4c35612015-10-16 15:34:23 -07001447 build_steps, maxjobs=1, stop_on_failure=True,
Craig Tiller883064c2015-11-04 10:06:10 -08001448 newline_on_success=newline_on_success, travis=args.travis)
Adele Zhoue4c35612015-10-16 15:34:23 -07001449 if num_failures:
Craig Tillereb9de8b2016-01-08 08:57:41 -08001450 return [BuildAndRunError.BUILD]
Craig Tillerb361b4e2016-01-06 11:44:17 -08001451
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001452 if build_only:
Jan Tattermuschaab1e512016-01-28 09:30:44 -08001453 if xml_report:
Jan Tattermuschcfcc0752016-10-09 17:02:34 +02001454 report_utils.render_junit_xml_report(resultset, xml_report,
1455 suite_name=args.report_suite_name)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001456 return []
ctiller3040cb72015-01-07 12:13:17 -08001457
Craig Tiller234b6e72015-05-23 10:12:40 -07001458 # start antagonists
Jan Tattermusch5c79a312016-12-20 11:02:50 +01001459 antagonists = [subprocess.Popen(['tools/run_tests/python_utils/antagonist.py'])
Craig Tiller234b6e72015-05-23 10:12:40 -07001460 for _ in range(0, args.antagonists)]
Ken Paysonfa51de52016-06-30 23:50:48 -07001461 port_server_port = 32766
Craig Tillerf53d9c82015-08-04 14:19:43 -07001462 _start_port_server(port_server_port)
Adele Zhou7cf72112015-11-04 11:18:43 -08001463 resultset = None
Adele Zhou803af152015-11-30 15:16:16 -08001464 num_test_failures = 0
Craig Tiller234b6e72015-05-23 10:12:40 -07001465 try:
David Garcia Quintase90cd372015-05-31 18:15:26 -07001466 infinite_runs = runs_per_test == 0
yang-g6c1fdc62015-08-18 11:57:42 -07001467 one_run = set(
1468 spec
yang-g6c1fdc62015-08-18 11:57:42 -07001469 for language in languages
Jan Tattermusch77db4322016-02-20 20:19:35 -08001470 for spec in language.test_specs()
Vijay Pai488fd0e2016-06-13 12:37:12 -07001471 if (re.search(args.regex, spec.shortname) and
1472 (args.regex_exclude == '' or
1473 not re.search(args.regex_exclude, spec.shortname))))
David Garcia Quintas79e389f2015-06-02 17:49:42 -07001474 # When running on travis, we want out test runs to be as similar as possible
1475 # for reproducibility purposes.
Craig Tiller883064c2015-11-04 10:06:10 -08001476 if args.travis:
David Garcia Quintas79e389f2015-06-02 17:49:42 -07001477 massaged_one_run = sorted(one_run, key=lambda x: x.shortname)
1478 else:
1479 # whereas otherwise, we want to shuffle things up to give all tests a
1480 # chance to run.
1481 massaged_one_run = list(one_run) # random.shuffle needs an indexable seq.
1482 random.shuffle(massaged_one_run) # which it modifies in-place.
Craig Tillerf7b7c892015-06-22 14:33:25 -07001483 if infinite_runs:
1484 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 -07001485 runs_sequence = (itertools.repeat(massaged_one_run) if infinite_runs
1486 else itertools.repeat(massaged_one_run, runs_per_test))
David Garcia Quintase90cd372015-05-31 18:15:26 -07001487 all_runs = itertools.chain.from_iterable(runs_sequence)
Nicolas "Pixel" Noble5937b5b2015-06-26 02:04:12 +02001488
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001489 if args.quiet_success:
1490 jobset.message('START', 'Running tests quietly, only failing tests will be reported', do_newline=True)
Adele Zhou803af152015-11-30 15:16:16 -08001491 num_test_failures, resultset = jobset.run(
Adele Zhou2271ab52015-10-28 13:59:14 -07001492 all_runs, check_cancelled, newline_on_success=newline_on_success,
Craig Tiller883064c2015-11-04 10:06:10 -08001493 travis=args.travis, infinite_runs=infinite_runs, maxjobs=args.jobs,
murgatroid998ae409f2015-10-26 16:39:00 -07001494 stop_on_failure=args.stop_on_failure,
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001495 add_env={'GRPC_TEST_PORT_SERVER': 'localhost:%d' % port_server_port},
1496 quiet_success=args.quiet_success)
Adele Zhoud5fffa52015-10-23 15:51:42 -07001497 if resultset:
Craig Tiller2b59dbc2016-05-13 15:59:09 -07001498 for k, v in sorted(resultset.items()):
Adele Zhoud5fffa52015-10-23 15:51:42 -07001499 num_runs, num_failures = _calculate_num_runs_failures(v)
Jan Tattermusch68e27bf2016-12-16 14:09:03 +01001500 if num_failures > 0:
1501 if num_failures == num_runs: # what about infinite_runs???
1502 jobset.message('FAILED', k, do_newline=True)
1503 else:
1504 jobset.message(
1505 'FLAKE', '%s [%d/%d runs flaked]' % (k, num_failures, num_runs),
1506 do_newline=True)
Craig Tiller234b6e72015-05-23 10:12:40 -07001507 finally:
1508 for antagonist in antagonists:
1509 antagonist.kill()
Adele Zhou7cf72112015-11-04 11:18:43 -08001510 if xml_report and resultset:
Jan Tattermuschcfcc0752016-10-09 17:02:34 +02001511 report_utils.render_junit_xml_report(resultset, xml_report,
1512 suite_name=args.report_suite_name)
Craig Tillerd86a3942015-01-14 12:48:54 -08001513
Adele Zhouf2ca7bc2015-10-23 15:38:00 -07001514 number_failures, _ = jobset.run(
1515 post_tests_steps, maxjobs=1, stop_on_failure=True,
Craig Tiller883064c2015-11-04 10:06:10 -08001516 newline_on_success=newline_on_success, travis=args.travis)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001517
1518 out = []
1519 if number_failures:
1520 out.append(BuildAndRunError.POST_TEST)
1521 if num_test_failures:
1522 out.append(BuildAndRunError.TEST)
Nicolas "Pixel" Noble3fcd3bf2015-10-10 02:30:38 +02001523
Craig Tillereb9de8b2016-01-08 08:57:41 -08001524 return out
ctiller3040cb72015-01-07 12:13:17 -08001525
1526
1527if forever:
Nicolas Noble044db742015-01-14 16:57:24 -08001528 success = True
ctiller3040cb72015-01-07 12:13:17 -08001529 while True:
Craig Tiller42bc87c2015-02-23 08:50:19 -08001530 dw = watch_dirs.DirWatcher(['src', 'include', 'test', 'examples'])
ctiller3040cb72015-01-07 12:13:17 -08001531 initial_time = dw.most_recent_change()
1532 have_files_changed = lambda: dw.most_recent_change() != initial_time
Nicolas Noble044db742015-01-14 16:57:24 -08001533 previous_success = success
Craig Tillereb9de8b2016-01-08 08:57:41 -08001534 errors = _build_and_run(check_cancelled=have_files_changed,
1535 newline_on_success=False,
Craig Tillereb9de8b2016-01-08 08:57:41 -08001536 build_only=args.build_only) == 0
1537 if not previous_success and not errors:
Nicolas Nobleb09078f2015-01-14 18:06:05 -08001538 jobset.message('SUCCESS',
1539 'All tests are now passing properly',
1540 do_newline=True)
Nicolas Noble044db742015-01-14 16:57:24 -08001541 jobset.message('IDLE', 'No change detected')
ctiller3040cb72015-01-07 12:13:17 -08001542 while not have_files_changed():
1543 time.sleep(1)
1544else:
Craig Tillereb9de8b2016-01-08 08:57:41 -08001545 errors = _build_and_run(check_cancelled=lambda: False,
Craig Tiller71735182015-01-15 17:07:13 -08001546 newline_on_success=args.newline_on_success,
Jan Tattermusch2dd156e2015-12-04 18:26:17 -08001547 xml_report=args.xml_report,
1548 build_only=args.build_only)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001549 if not errors:
Nicolas Nobleb09078f2015-01-14 18:06:05 -08001550 jobset.message('SUCCESS', 'All tests passed', do_newline=True)
1551 else:
1552 jobset.message('FAILED', 'Some tests failed', do_newline=True)
Craig Tillereb9de8b2016-01-08 08:57:41 -08001553 exit_code = 0
1554 if BuildAndRunError.BUILD in errors:
1555 exit_code |= 1
Jan Tattermusche480a6a2016-10-07 12:59:08 +02001556 if BuildAndRunError.TEST in errors:
Craig Tillereb9de8b2016-01-08 08:57:41 -08001557 exit_code |= 2
Craig Tiller4f2be362016-01-08 08:59:20 -08001558 if BuildAndRunError.POST_TEST in errors:
1559 exit_code |= 4
Craig Tillereb9de8b2016-01-08 08:57:41 -08001560 sys.exit(exit_code)