| Index: third_party/grpc/tools/run_tests/run_tests.py
|
| diff --git a/third_party/grpc/tools/run_tests/run_tests.py b/third_party/grpc/tools/run_tests/run_tests.py
|
| new file mode 100755
|
| index 0000000000000000000000000000000000000000..2df82adf0a7ed54a99ec3939d3ff371584848b0a
|
| --- /dev/null
|
| +++ b/third_party/grpc/tools/run_tests/run_tests.py
|
| @@ -0,0 +1,1229 @@
|
| +#!/usr/bin/env python2.7
|
| +# Copyright 2015-2016, Google Inc.
|
| +# All rights reserved.
|
| +#
|
| +# Redistribution and use in source and binary forms, with or without
|
| +# modification, are permitted provided that the following conditions are
|
| +# met:
|
| +#
|
| +# * Redistributions of source code must retain the above copyright
|
| +# notice, this list of conditions and the following disclaimer.
|
| +# * Redistributions in binary form must reproduce the above
|
| +# copyright notice, this list of conditions and the following disclaimer
|
| +# in the documentation and/or other materials provided with the
|
| +# distribution.
|
| +# * Neither the name of Google Inc. nor the names of its
|
| +# contributors may be used to endorse or promote products derived from
|
| +# this software without specific prior written permission.
|
| +#
|
| +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| +# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| +# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| +# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| +# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| +# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| +# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| +# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| +
|
| +"""Run tests in parallel."""
|
| +
|
| +import argparse
|
| +import ast
|
| +import glob
|
| +import hashlib
|
| +import itertools
|
| +import json
|
| +import multiprocessing
|
| +import os
|
| +import platform
|
| +import random
|
| +import re
|
| +import socket
|
| +import subprocess
|
| +import sys
|
| +import tempfile
|
| +import traceback
|
| +import time
|
| +import urllib2
|
| +import uuid
|
| +
|
| +import jobset
|
| +import report_utils
|
| +import watch_dirs
|
| +
|
| +
|
| +_ROOT = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../..'))
|
| +os.chdir(_ROOT)
|
| +
|
| +
|
| +_FORCE_ENVIRON_FOR_WRAPPERS = {}
|
| +
|
| +
|
| +def platform_string():
|
| + return jobset.platform_string()
|
| +
|
| +
|
| +# SimpleConfig: just compile with CONFIG=config, and run the binary to test
|
| +class Config(object):
|
| +
|
| + def __init__(self, config, environ=None, timeout_multiplier=1, tool_prefix=[]):
|
| + if environ is None:
|
| + environ = {}
|
| + self.build_config = config
|
| + self.allow_hashing = (config != 'gcov')
|
| + self.environ = environ
|
| + self.environ['CONFIG'] = config
|
| + self.tool_prefix = tool_prefix
|
| + self.timeout_multiplier = timeout_multiplier
|
| +
|
| + def job_spec(self, cmdline, hash_targets, timeout_seconds=5*60,
|
| + shortname=None, environ={}, cpu_cost=1.0):
|
| + """Construct a jobset.JobSpec for a test under this config
|
| +
|
| + Args:
|
| + cmdline: a list of strings specifying the command line the test
|
| + would like to run
|
| + hash_targets: either None (don't do caching of test results), or
|
| + a list of strings specifying files to include in a
|
| + binary hash to check if a test has changed
|
| + -- if used, all artifacts needed to run the test must
|
| + be listed
|
| + """
|
| + actual_environ = self.environ.copy()
|
| + for k, v in environ.iteritems():
|
| + actual_environ[k] = v
|
| + return jobset.JobSpec(cmdline=self.tool_prefix + cmdline,
|
| + shortname=shortname,
|
| + environ=actual_environ,
|
| + cpu_cost=cpu_cost,
|
| + timeout_seconds=(self.timeout_multiplier * timeout_seconds if timeout_seconds else None),
|
| + hash_targets=hash_targets
|
| + if self.allow_hashing else None,
|
| + flake_retries=5 if args.allow_flakes else 0,
|
| + timeout_retries=3 if args.allow_flakes else 0)
|
| +
|
| +
|
| +def get_c_tests(travis, test_lang) :
|
| + out = []
|
| + platforms_str = 'ci_platforms' if travis else 'platforms'
|
| + with open('tools/run_tests/tests.json') as f:
|
| + js = json.load(f)
|
| + return [tgt
|
| + for tgt in js
|
| + if tgt['language'] == test_lang and
|
| + platform_string() in tgt[platforms_str] and
|
| + not (travis and tgt['flaky'])]
|
| +
|
| +
|
| +def _check_compiler(compiler, supported_compilers):
|
| + if compiler not in supported_compilers:
|
| + raise Exception('Compiler %s not supported.' % compiler)
|
| +
|
| +
|
| +def _is_use_docker_child():
|
| + """Returns True if running running as a --use_docker child."""
|
| + return True if os.getenv('RUN_TESTS_COMMAND') else False
|
| +
|
| +
|
| +class CLanguage(object):
|
| +
|
| + def __init__(self, make_target, test_lang):
|
| + self.make_target = make_target
|
| + self.platform = platform_string()
|
| + self.test_lang = test_lang
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + if self.platform == 'windows':
|
| + self._make_options = [_windows_toolset_option(self.args.compiler),
|
| + _windows_arch_option(self.args.arch)]
|
| + else:
|
| + self._make_options = []
|
| + self._docker_distro = self._get_docker_distro(self.args.use_docker,
|
| + self.args.compiler)
|
| +
|
| + def test_specs(self):
|
| + out = []
|
| + binaries = get_c_tests(self.args.travis, self.test_lang)
|
| + for target in binaries:
|
| + if self.config.build_config in target['exclude_configs']:
|
| + continue
|
| + if self.platform == 'windows':
|
| + binary = 'vsprojects/%s%s/%s.exe' % (
|
| + 'x64/' if self.args.arch == 'x64' else '',
|
| + _MSBUILD_CONFIG[self.config.build_config],
|
| + target['name'])
|
| + else:
|
| + binary = 'bins/%s/%s' % (self.config.build_config, target['name'])
|
| + if os.path.isfile(binary):
|
| + if 'gtest' in target and target['gtest']:
|
| + # here we parse the output of --gtest_list_tests to build up a
|
| + # complete list of the tests contained in a binary
|
| + # for each test, we then add a job to run, filtering for just that
|
| + # test
|
| + with open(os.devnull, 'w') as fnull:
|
| + tests = subprocess.check_output([binary, '--gtest_list_tests'],
|
| + stderr=fnull)
|
| + base = None
|
| + for line in tests.split('\n'):
|
| + i = line.find('#')
|
| + if i >= 0: line = line[:i]
|
| + if not line: continue
|
| + if line[0] != ' ':
|
| + base = line.strip()
|
| + else:
|
| + assert base is not None
|
| + assert line[1] == ' '
|
| + test = base + line.strip()
|
| + cmdline = [binary] + ['--gtest_filter=%s' % test]
|
| + out.append(self.config.job_spec(cmdline, [binary],
|
| + shortname='%s:%s' % (binary, test),
|
| + cpu_cost=target['cpu_cost'],
|
| + environ={'GRPC_DEFAULT_SSL_ROOTS_FILE_PATH':
|
| + _ROOT + '/src/core/tsi/test_creds/ca.pem'}))
|
| + else:
|
| + cmdline = [binary] + target['args']
|
| + out.append(self.config.job_spec(cmdline, [binary],
|
| + shortname=' '.join(cmdline),
|
| + cpu_cost=target['cpu_cost'],
|
| + environ={'GRPC_DEFAULT_SSL_ROOTS_FILE_PATH':
|
| + _ROOT + '/src/core/tsi/test_creds/ca.pem'}))
|
| + elif self.args.regex == '.*' or self.platform == 'windows':
|
| + print '\nWARNING: binary not found, skipping', binary
|
| + return sorted(out)
|
| +
|
| + def make_targets(self):
|
| + test_regex = self.args.regex
|
| + if self.platform != 'windows' and self.args.regex != '.*':
|
| + # use the regex to minimize the number of things to build
|
| + return [os.path.basename(target['name'])
|
| + for target in get_c_tests(False, self.test_lang)
|
| + if re.search(test_regex, '/' + target['name'])]
|
| + if self.platform == 'windows':
|
| + # don't build tools on windows just yet
|
| + return ['buildtests_%s' % self.make_target]
|
| + return ['buildtests_%s' % self.make_target, 'tools_%s' % self.make_target]
|
| +
|
| + def make_options(self):
|
| + return self._make_options;
|
| +
|
| + def pre_build_steps(self):
|
| + if self.platform == 'windows':
|
| + return [['tools\\run_tests\\pre_build_c.bat']]
|
| + else:
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + return []
|
| +
|
| + def post_tests_steps(self):
|
| + if self.platform == 'windows':
|
| + return []
|
| + else:
|
| + return [['tools/run_tests/post_tests_c.sh']]
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def _get_docker_distro(self, use_docker, compiler):
|
| + if _is_use_docker_child():
|
| + return "already_under_docker"
|
| + if not use_docker:
|
| + _check_compiler(compiler, ['default'])
|
| +
|
| + if compiler == 'gcc4.9' or compiler == 'default':
|
| + return 'jessie'
|
| + elif compiler == 'gcc4.4':
|
| + return 'squeeze'
|
| + elif compiler == 'gcc5.3':
|
| + return 'ubuntu1604'
|
| + else:
|
| + raise Exception('Compiler %s not supported.' % compiler)
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/cxx_%s_%s' % (self._docker_distro,
|
| + _docker_arch_suffix(self.args.arch))
|
| +
|
| + def __str__(self):
|
| + return self.make_target
|
| +
|
| +
|
| +class NodeLanguage(object):
|
| +
|
| + def __init__(self):
|
| + self.platform = platform_string()
|
| + self.node_version = '0.12'
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + if self.platform == 'windows':
|
| + return [self.config.job_spec(['tools\\run_tests\\run_node.bat'], None)]
|
| + else:
|
| + return [self.config.job_spec(['tools/run_tests/run_node.sh', self.node_version],
|
| + None,
|
| + environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
|
| +
|
| + def pre_build_steps(self):
|
| + if self.platform == 'windows':
|
| + return [['tools\\run_tests\\pre_build_node.bat']]
|
| + else:
|
| + return [['tools/run_tests/pre_build_node.sh', self.node_version]]
|
| +
|
| + def make_targets(self):
|
| + return []
|
| +
|
| + def make_options(self):
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + if self.platform == 'windows':
|
| + return [['tools\\run_tests\\build_node.bat']]
|
| + else:
|
| + return [['tools/run_tests/build_node.sh', self.node_version]]
|
| +
|
| + def post_tests_steps(self):
|
| + return []
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/node_jessie_%s' % _docker_arch_suffix(self.args.arch)
|
| +
|
| + def __str__(self):
|
| + return 'node'
|
| +
|
| +
|
| +class PhpLanguage(object):
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + return [self.config.job_spec(['src/php/bin/run_tests.sh'], None,
|
| + environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
|
| +
|
| + def pre_build_steps(self):
|
| + return []
|
| +
|
| + def make_targets(self):
|
| + return ['static_c', 'shared_c']
|
| +
|
| + def make_options(self):
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + return [['tools/run_tests/build_php.sh']]
|
| +
|
| + def post_tests_steps(self):
|
| + return [['tools/run_tests/post_tests_php.sh']]
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/php_jessie_%s' % _docker_arch_suffix(self.args.arch)
|
| +
|
| + def __str__(self):
|
| + return 'php'
|
| +
|
| +
|
| +class PythonLanguage(object):
|
| +
|
| + def __init__(self):
|
| + self._build_python_versions = ['2.7']
|
| + self._has_python_versions = []
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + # load list of known test suites
|
| + with open('src/python/grpcio/tests/tests.json') as tests_json_file:
|
| + tests_json = json.load(tests_json_file)
|
| + environment = dict(_FORCE_ENVIRON_FOR_WRAPPERS)
|
| + environment['PYVER'] = '2.7'
|
| + if self.config.build_config != 'gcov':
|
| + return [self.config.job_spec(
|
| + ['tools/run_tests/run_python.sh'],
|
| + None,
|
| + environ=dict(environment.items() +
|
| + [('GRPC_PYTHON_TESTRUNNER_FILTER', suite_name)]),
|
| + shortname='py.test.%s' % suite_name,
|
| + timeout_seconds=5*60)
|
| + for suite_name in tests_json]
|
| + else:
|
| + return [self.config.job_spec(['tools/run_tests/run_python.sh'],
|
| + None,
|
| + environ=environment,
|
| + shortname='py.test.coverage',
|
| + timeout_seconds=15*60)]
|
| +
|
| +
|
| + def pre_build_steps(self):
|
| + return []
|
| +
|
| + def make_targets(self):
|
| + return ['static_c', 'grpc_python_plugin', 'shared_c']
|
| +
|
| + def make_options(self):
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + commands = []
|
| + for python_version in self._build_python_versions:
|
| + try:
|
| + with open(os.devnull, 'w') as output:
|
| + subprocess.check_call(['which', 'python' + python_version],
|
| + stdout=output, stderr=output)
|
| + commands.append(['tools/run_tests/build_python.sh', python_version])
|
| + self._has_python_versions.append(python_version)
|
| + except:
|
| + jobset.message('WARNING', 'Missing Python ' + python_version,
|
| + do_newline=True)
|
| + return commands
|
| +
|
| + def post_tests_steps(self):
|
| + return []
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/python_jessie_%s' % _docker_arch_suffix(self.args.arch)
|
| +
|
| + def __str__(self):
|
| + return 'python'
|
| +
|
| +
|
| +class RubyLanguage(object):
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + return [self.config.job_spec(['tools/run_tests/run_ruby.sh'], None,
|
| + timeout_seconds=10*60,
|
| + environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
|
| +
|
| + def pre_build_steps(self):
|
| + return [['tools/run_tests/pre_build_ruby.sh']]
|
| +
|
| + def make_targets(self):
|
| + return []
|
| +
|
| + def make_options(self):
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + return [['tools/run_tests/build_ruby.sh']]
|
| +
|
| + def post_tests_steps(self):
|
| + return [['tools/run_tests/post_tests_ruby.sh']]
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/ruby_jessie_%s' % _docker_arch_suffix(self.args.arch)
|
| +
|
| + def __str__(self):
|
| + return 'ruby'
|
| +
|
| +
|
| +class CSharpLanguage(object):
|
| +
|
| + def __init__(self):
|
| + self.platform = platform_string()
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + with open('src/csharp/tests.json') as f:
|
| + tests_json = json.load(f)
|
| + assemblies = tests_json['assemblies']
|
| + tests = tests_json['tests']
|
| +
|
| + msbuild_config = _MSBUILD_CONFIG[self.config.build_config]
|
| + assembly_files = ['%s/bin/%s/%s.dll' % (a, msbuild_config, a)
|
| + for a in assemblies]
|
| +
|
| + extra_args = ['-labels'] + assembly_files
|
| +
|
| + if self.platform == 'windows':
|
| + script_name = 'tools\\run_tests\\run_csharp.bat'
|
| + extra_args += ['-domain=None']
|
| + else:
|
| + script_name = 'tools/run_tests/run_csharp.sh'
|
| +
|
| + if self.config.build_config == 'gcov':
|
| + # On Windows, we only collect C# code coverage.
|
| + # On Linux, we only collect coverage for native extension.
|
| + # For code coverage all tests need to run as one suite.
|
| + return [self.config.job_spec([script_name] + extra_args, None,
|
| + shortname='csharp.coverage',
|
| + environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
|
| + else:
|
| + specs = []
|
| + for test in tests:
|
| + cmdline = [script_name, '-run=%s' % test] + extra_args
|
| + if self.platform == 'windows':
|
| + # use different output directory for each test to prevent
|
| + # TestResult.xml clash between parallel test runs.
|
| + cmdline += ['-work=test-result/%s' % uuid.uuid4()]
|
| + specs.append(self.config.job_spec(cmdline, None,
|
| + shortname='csharp.%s' % test,
|
| + environ=_FORCE_ENVIRON_FOR_WRAPPERS))
|
| + return specs
|
| +
|
| + def pre_build_steps(self):
|
| + if self.platform == 'windows':
|
| + return [['tools\\run_tests\\pre_build_csharp.bat']]
|
| + else:
|
| + return [['tools/run_tests/pre_build_csharp.sh']]
|
| +
|
| + def make_targets(self):
|
| + # For Windows, this target doesn't really build anything,
|
| + # everything is build by buildall script later.
|
| + if self.platform == 'windows':
|
| + return []
|
| + else:
|
| + return ['grpc_csharp_ext']
|
| +
|
| + def make_options(self):
|
| + if self.platform == 'mac':
|
| + # On Mac, official distribution of mono is 32bit.
|
| + return ['CFLAGS=-arch i386', 'LDFLAGS=-arch i386']
|
| + else:
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + if self.platform == 'windows':
|
| + return [['src\\csharp\\buildall.bat']]
|
| + else:
|
| + return [['tools/run_tests/build_csharp.sh']]
|
| +
|
| + def post_tests_steps(self):
|
| + return []
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/csharp_jessie_%s' % _docker_arch_suffix(self.args.arch)
|
| +
|
| + def __str__(self):
|
| + return 'csharp'
|
| +
|
| +
|
| +class ObjCLanguage(object):
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + return [self.config.job_spec(['src/objective-c/tests/run_tests.sh'], None,
|
| + environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
|
| +
|
| + def pre_build_steps(self):
|
| + return []
|
| +
|
| + def make_targets(self):
|
| + return ['grpc_objective_c_plugin', 'interop_server']
|
| +
|
| + def make_options(self):
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + return [['src/objective-c/tests/build_tests.sh']]
|
| +
|
| + def post_tests_steps(self):
|
| + return []
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return None
|
| +
|
| + def __str__(self):
|
| + return 'objc'
|
| +
|
| +
|
| +class Sanity(object):
|
| +
|
| + def configure(self, config, args):
|
| + self.config = config
|
| + self.args = args
|
| + _check_compiler(self.args.compiler, ['default'])
|
| +
|
| + def test_specs(self):
|
| + import yaml
|
| + with open('tools/run_tests/sanity/sanity_tests.yaml', 'r') as f:
|
| + return [self.config.job_spec(cmd['script'].split(), None,
|
| + timeout_seconds=None, environ={'TEST': 'true'},
|
| + cpu_cost=cmd.get('cpu_cost', 1))
|
| + for cmd in yaml.load(f)]
|
| +
|
| + def pre_build_steps(self):
|
| + return []
|
| +
|
| + def make_targets(self):
|
| + return ['run_dep_checks']
|
| +
|
| + def make_options(self):
|
| + return []
|
| +
|
| + def build_steps(self):
|
| + return []
|
| +
|
| + def post_tests_steps(self):
|
| + return []
|
| +
|
| + def makefile_name(self):
|
| + return 'Makefile'
|
| +
|
| + def dockerfile_dir(self):
|
| + return 'tools/dockerfile/test/sanity'
|
| +
|
| + def __str__(self):
|
| + return 'sanity'
|
| +
|
| +
|
| +# different configurations we can run under
|
| +with open('tools/run_tests/configs.json') as f:
|
| + _CONFIGS = dict((cfg['config'], Config(**cfg)) for cfg in ast.literal_eval(f.read()))
|
| +
|
| +
|
| +_LANGUAGES = {
|
| + 'c++': CLanguage('cxx', 'c++'),
|
| + 'c': CLanguage('c', 'c'),
|
| + 'node': NodeLanguage(),
|
| + 'php': PhpLanguage(),
|
| + 'python': PythonLanguage(),
|
| + 'ruby': RubyLanguage(),
|
| + 'csharp': CSharpLanguage(),
|
| + 'objc' : ObjCLanguage(),
|
| + 'sanity': Sanity()
|
| + }
|
| +
|
| +
|
| +_MSBUILD_CONFIG = {
|
| + 'dbg': 'Debug',
|
| + 'opt': 'Release',
|
| + 'gcov': 'Debug',
|
| + }
|
| +
|
| +
|
| +def _windows_arch_option(arch):
|
| + """Returns msbuild cmdline option for selected architecture."""
|
| + if arch == 'default' or arch == 'x86':
|
| + return '/p:Platform=Win32'
|
| + elif arch == 'x64':
|
| + return '/p:Platform=x64'
|
| + else:
|
| + print 'Architecture %s not supported.' % arch
|
| + sys.exit(1)
|
| +
|
| +
|
| +def _check_arch_option(arch):
|
| + """Checks that architecture option is valid."""
|
| + if platform_string() == 'windows':
|
| + _windows_arch_option(arch)
|
| + elif platform_string() == 'linux':
|
| + # On linux, we need to be running under docker with the right architecture.
|
| + runtime_arch = platform.architecture()[0]
|
| + if arch == 'default':
|
| + return
|
| + elif runtime_arch == '64bit' and arch == 'x64':
|
| + return
|
| + elif runtime_arch == '32bit' and arch == 'x86':
|
| + return
|
| + else:
|
| + print 'Architecture %s does not match current runtime architecture.' % arch
|
| + sys.exit(1)
|
| + else:
|
| + if args.arch != 'default':
|
| + print 'Architecture %s not supported on current platform.' % args.arch
|
| + sys.exit(1)
|
| +
|
| +
|
| +def _windows_build_bat(compiler):
|
| + """Returns name of build.bat for selected compiler."""
|
| + if compiler == 'default' or compiler == 'vs2013':
|
| + return 'vsprojects\\build_vs2013.bat'
|
| + elif compiler == 'vs2015':
|
| + return 'vsprojects\\build_vs2015.bat'
|
| + elif compiler == 'vs2010':
|
| + return 'vsprojects\\build_vs2010.bat'
|
| + else:
|
| + print 'Compiler %s not supported.' % compiler
|
| + sys.exit(1)
|
| +
|
| +
|
| +def _windows_toolset_option(compiler):
|
| + """Returns msbuild PlatformToolset for selected compiler."""
|
| + if compiler == 'default' or compiler == 'vs2013':
|
| + return '/p:PlatformToolset=v120'
|
| + elif compiler == 'vs2015':
|
| + return '/p:PlatformToolset=v140'
|
| + elif compiler == 'vs2010':
|
| + return '/p:PlatformToolset=v100'
|
| + else:
|
| + print 'Compiler %s not supported.' % compiler
|
| + sys.exit(1)
|
| +
|
| +
|
| +def _docker_arch_suffix(arch):
|
| + """Returns suffix to dockerfile dir to use."""
|
| + if arch == 'default' or arch == 'x64':
|
| + return 'x64'
|
| + elif arch == 'x86':
|
| + return 'x86'
|
| + else:
|
| + print 'Architecture %s not supported with current settings.' % arch
|
| + sys.exit(1)
|
| +
|
| +
|
| +def runs_per_test_type(arg_str):
|
| + """Auxilary function to parse the "runs_per_test" flag.
|
| +
|
| + Returns:
|
| + A positive integer or 0, the latter indicating an infinite number of
|
| + runs.
|
| +
|
| + Raises:
|
| + argparse.ArgumentTypeError: Upon invalid input.
|
| + """
|
| + if arg_str == 'inf':
|
| + return 0
|
| + try:
|
| + n = int(arg_str)
|
| + if n <= 0: raise ValueError
|
| + return n
|
| + except:
|
| + msg = '\'{}\' is not a positive integer or \'inf\''.format(arg_str)
|
| + raise argparse.ArgumentTypeError(msg)
|
| +
|
| +# parse command line
|
| +argp = argparse.ArgumentParser(description='Run grpc tests.')
|
| +argp.add_argument('-c', '--config',
|
| + choices=sorted(_CONFIGS.keys()),
|
| + default='opt')
|
| +argp.add_argument('-n', '--runs_per_test', default=1, type=runs_per_test_type,
|
| + help='A positive integer or "inf". If "inf", all tests will run in an '
|
| + 'infinite loop. Especially useful in combination with "-f"')
|
| +argp.add_argument('-r', '--regex', default='.*', type=str)
|
| +argp.add_argument('-j', '--jobs', default=multiprocessing.cpu_count(), type=int)
|
| +argp.add_argument('-s', '--slowdown', default=1.0, type=float)
|
| +argp.add_argument('-f', '--forever',
|
| + default=False,
|
| + action='store_const',
|
| + const=True)
|
| +argp.add_argument('-t', '--travis',
|
| + default=False,
|
| + action='store_const',
|
| + const=True)
|
| +argp.add_argument('--newline_on_success',
|
| + default=False,
|
| + action='store_const',
|
| + const=True)
|
| +argp.add_argument('-l', '--language',
|
| + choices=['all'] + sorted(_LANGUAGES.keys()),
|
| + nargs='+',
|
| + default=['all'])
|
| +argp.add_argument('-S', '--stop_on_failure',
|
| + default=False,
|
| + action='store_const',
|
| + const=True)
|
| +argp.add_argument('--use_docker',
|
| + default=False,
|
| + action='store_const',
|
| + const=True,
|
| + help='Run all the tests under docker. That provides ' +
|
| + 'additional isolation and prevents the need to install ' +
|
| + 'language specific prerequisites. Only available on Linux.')
|
| +argp.add_argument('--allow_flakes',
|
| + default=False,
|
| + action='store_const',
|
| + const=True,
|
| + help='Allow flaky tests to show as passing (re-runs failed tests up to five times)')
|
| +argp.add_argument('--arch',
|
| + choices=['default', 'x86', 'x64'],
|
| + default='default',
|
| + help='Selects architecture to target. For some platforms "default" is the only supported choice.')
|
| +argp.add_argument('--compiler',
|
| + choices=['default',
|
| + 'gcc4.4', 'gcc4.9', 'gcc5.3',
|
| + 'vs2010', 'vs2013', 'vs2015'],
|
| + default='default',
|
| + help='Selects compiler to use. Allowed values depend on the platform and language.')
|
| +argp.add_argument('--build_only',
|
| + default=False,
|
| + action='store_const',
|
| + const=True,
|
| + help='Perform all the build steps but dont run any tests.')
|
| +argp.add_argument('--measure_cpu_costs', default=False, action='store_const', const=True,
|
| + help='Measure the cpu costs of tests')
|
| +argp.add_argument('--update_submodules', default=[], nargs='*',
|
| + help='Update some submodules before building. If any are updated, also run generate_projects. ' +
|
| + 'Submodules are specified as SUBMODULE_NAME:BRANCH; if BRANCH is omitted, master is assumed.')
|
| +argp.add_argument('-a', '--antagonists', default=0, type=int)
|
| +argp.add_argument('-x', '--xml_report', default=None, type=str,
|
| + help='Generates a JUnit-compatible XML report')
|
| +args = argp.parse_args()
|
| +
|
| +jobset.measure_cpu_costs = args.measure_cpu_costs
|
| +
|
| +# update submodules if necessary
|
| +need_to_regenerate_projects = False
|
| +for spec in args.update_submodules:
|
| + spec = spec.split(':', 1)
|
| + if len(spec) == 1:
|
| + submodule = spec[0]
|
| + branch = 'master'
|
| + elif len(spec) == 2:
|
| + submodule = spec[0]
|
| + branch = spec[1]
|
| + cwd = 'third_party/%s' % submodule
|
| + def git(cmd, cwd=cwd):
|
| + print 'in %s: git %s' % (cwd, cmd)
|
| + subprocess.check_call('git %s' % cmd, cwd=cwd, shell=True)
|
| + git('fetch')
|
| + git('checkout %s' % branch)
|
| + git('pull origin %s' % branch)
|
| + if os.path.exists('src/%s/gen_build_yaml.py' % submodule):
|
| + need_to_regenerate_projects = True
|
| +if need_to_regenerate_projects:
|
| + if jobset.platform_string() == 'linux':
|
| + subprocess.check_call('tools/buildgen/generate_projects.sh', shell=True)
|
| + else:
|
| + print 'WARNING: may need to regenerate projects, but since we are not on'
|
| + print ' Linux this step is being skipped. Compilation MAY fail.'
|
| +
|
| +
|
| +# grab config
|
| +run_config = _CONFIGS[args.config]
|
| +build_config = run_config.build_config
|
| +
|
| +if args.travis:
|
| + _FORCE_ENVIRON_FOR_WRAPPERS = {'GRPC_TRACE': 'api'}
|
| +
|
| +if 'all' in args.language:
|
| + lang_list = _LANGUAGES.keys()
|
| +else:
|
| + lang_list = args.language
|
| +# We don't support code coverage on some languages
|
| +if 'gcov' in args.config:
|
| + for bad in ['objc', 'sanity']:
|
| + if bad in lang_list:
|
| + lang_list.remove(bad)
|
| +
|
| +languages = set(_LANGUAGES[l] for l in lang_list)
|
| +for l in languages:
|
| + l.configure(run_config, args)
|
| +
|
| +language_make_options=[]
|
| +if any(language.make_options() for language in languages):
|
| + if len(languages) != 1:
|
| + print 'languages with custom make options cannot be built simultaneously with other languages'
|
| + sys.exit(1)
|
| + else:
|
| + language_make_options = next(iter(languages)).make_options()
|
| +
|
| +if args.use_docker:
|
| + if not args.travis:
|
| + print 'Seen --use_docker flag, will run tests under docker.'
|
| + print
|
| + print 'IMPORTANT: The changes you are testing need to be locally committed'
|
| + print 'because only the committed changes in the current branch will be'
|
| + print 'copied to the docker environment.'
|
| + time.sleep(5)
|
| +
|
| + dockerfile_dirs = set([l.dockerfile_dir() for l in languages])
|
| + if len(dockerfile_dirs) > 1:
|
| + print 'Languages to be tested require running under different docker images.'
|
| + sys.exit(1)
|
| + dockerfile_dir = next(iter(dockerfile_dirs))
|
| +
|
| + child_argv = [ arg for arg in sys.argv if not arg == '--use_docker' ]
|
| + run_tests_cmd = 'python tools/run_tests/run_tests.py %s' % ' '.join(child_argv[1:])
|
| +
|
| + env = os.environ.copy()
|
| + env['RUN_TESTS_COMMAND'] = run_tests_cmd
|
| + env['DOCKERFILE_DIR'] = dockerfile_dir
|
| + env['DOCKER_RUN_SCRIPT'] = 'tools/jenkins/docker_run_tests.sh'
|
| + if args.xml_report:
|
| + env['XML_REPORT'] = args.xml_report
|
| + if not args.travis:
|
| + env['TTY_FLAG'] = '-t' # enables Ctrl-C when not on Jenkins.
|
| +
|
| + subprocess.check_call(['tools/jenkins/build_docker_and_run_tests.sh'],
|
| + shell=True,
|
| + env=env)
|
| + sys.exit(0)
|
| +
|
| +_check_arch_option(args.arch)
|
| +
|
| +def make_jobspec(cfg, targets, makefile='Makefile'):
|
| + if platform_string() == 'windows':
|
| + extra_args = []
|
| + # better do parallel compilation
|
| + # empirically /m:2 gives the best performance/price and should prevent
|
| + # overloading the windows workers.
|
| + extra_args.extend(['/m:2'])
|
| + # disable PDB generation: it's broken, and we don't need it during CI
|
| + extra_args.extend(['/p:Jenkins=true'])
|
| + return [
|
| + jobset.JobSpec([_windows_build_bat(args.compiler),
|
| + 'vsprojects\\%s.sln' % target,
|
| + '/p:Configuration=%s' % _MSBUILD_CONFIG[cfg]] +
|
| + extra_args +
|
| + language_make_options,
|
| + shell=True, timeout_seconds=None)
|
| + for target in targets]
|
| + else:
|
| + if targets:
|
| + return [jobset.JobSpec([os.getenv('MAKE', 'make'),
|
| + '-f', makefile,
|
| + '-j', '%d' % args.jobs,
|
| + 'EXTRA_DEFINES=GRPC_TEST_SLOWDOWN_MACHINE_FACTOR=%f' % args.slowdown,
|
| + 'CONFIG=%s' % cfg] +
|
| + language_make_options +
|
| + ([] if not args.travis else ['JENKINS_BUILD=1']) +
|
| + targets,
|
| + timeout_seconds=None)]
|
| + else:
|
| + return []
|
| +
|
| +make_targets = {}
|
| +for l in languages:
|
| + makefile = l.makefile_name()
|
| + make_targets[makefile] = make_targets.get(makefile, set()).union(
|
| + set(l.make_targets()))
|
| +
|
| +def build_step_environ(cfg):
|
| + environ = {'CONFIG': cfg}
|
| + msbuild_cfg = _MSBUILD_CONFIG.get(cfg)
|
| + if msbuild_cfg:
|
| + environ['MSBUILD_CONFIG'] = msbuild_cfg
|
| + return environ
|
| +
|
| +build_steps = list(set(
|
| + jobset.JobSpec(cmdline, environ=build_step_environ(build_config), flake_retries=5)
|
| + for l in languages
|
| + for cmdline in l.pre_build_steps()))
|
| +if make_targets:
|
| + make_commands = itertools.chain.from_iterable(make_jobspec(build_config, list(targets), makefile) for (makefile, targets) in make_targets.iteritems())
|
| + build_steps.extend(set(make_commands))
|
| +build_steps.extend(set(
|
| + jobset.JobSpec(cmdline, environ=build_step_environ(build_config), timeout_seconds=None)
|
| + for l in languages
|
| + for cmdline in l.build_steps()))
|
| +
|
| +post_tests_steps = list(set(
|
| + jobset.JobSpec(cmdline, environ=build_step_environ(build_config))
|
| + for l in languages
|
| + for cmdline in l.post_tests_steps()))
|
| +runs_per_test = args.runs_per_test
|
| +forever = args.forever
|
| +
|
| +
|
| +class TestCache(object):
|
| + """Cache for running tests."""
|
| +
|
| + def __init__(self, use_cache_results):
|
| + self._last_successful_run = {}
|
| + self._use_cache_results = use_cache_results
|
| + self._last_save = time.time()
|
| +
|
| + def should_run(self, cmdline, bin_hash):
|
| + if cmdline not in self._last_successful_run:
|
| + return True
|
| + if self._last_successful_run[cmdline] != bin_hash:
|
| + return True
|
| + if not self._use_cache_results:
|
| + return True
|
| + return False
|
| +
|
| + def finished(self, cmdline, bin_hash):
|
| + self._last_successful_run[cmdline] = bin_hash
|
| + if time.time() - self._last_save > 1:
|
| + self.save()
|
| +
|
| + def dump(self):
|
| + return [{'cmdline': k, 'hash': v}
|
| + for k, v in self._last_successful_run.iteritems()]
|
| +
|
| + def parse(self, exdump):
|
| + self._last_successful_run = dict((o['cmdline'], o['hash']) for o in exdump)
|
| +
|
| + def save(self):
|
| + with open('.run_tests_cache', 'w') as f:
|
| + f.write(json.dumps(self.dump()))
|
| + self._last_save = time.time()
|
| +
|
| + def maybe_load(self):
|
| + if os.path.exists('.run_tests_cache'):
|
| + with open('.run_tests_cache') as f:
|
| + self.parse(json.loads(f.read()))
|
| +
|
| +
|
| +def _start_port_server(port_server_port):
|
| + # check if a compatible port server is running
|
| + # if incompatible (version mismatch) ==> start a new one
|
| + # if not running ==> start a new one
|
| + # otherwise, leave it up
|
| + try:
|
| + version = int(urllib2.urlopen(
|
| + 'http://localhost:%d/version_number' % port_server_port,
|
| + timeout=1).read())
|
| + print 'detected port server running version %d' % version
|
| + running = True
|
| + except Exception as e:
|
| + print 'failed to detect port server: %s' % sys.exc_info()[0]
|
| + print e.strerror
|
| + running = False
|
| + if running:
|
| + current_version = int(subprocess.check_output(
|
| + [sys.executable, os.path.abspath('tools/run_tests/port_server.py'),
|
| + 'dump_version']))
|
| + print 'my port server is version %d' % current_version
|
| + running = (version >= current_version)
|
| + if not running:
|
| + print 'port_server version mismatch: killing the old one'
|
| + urllib2.urlopen('http://localhost:%d/quitquitquit' % port_server_port).read()
|
| + time.sleep(1)
|
| + if not running:
|
| + fd, logfile = tempfile.mkstemp()
|
| + os.close(fd)
|
| + print 'starting port_server, with log file %s' % logfile
|
| + args = [sys.executable, os.path.abspath('tools/run_tests/port_server.py'),
|
| + '-p', '%d' % port_server_port, '-l', logfile]
|
| + env = dict(os.environ)
|
| + env['BUILD_ID'] = 'pleaseDontKillMeJenkins'
|
| + if platform_string() == 'windows':
|
| + # Working directory of port server needs to be outside of Jenkins
|
| + # workspace to prevent file lock issues.
|
| + tempdir = tempfile.mkdtemp()
|
| + port_server = subprocess.Popen(
|
| + args,
|
| + env=env,
|
| + cwd=tempdir,
|
| + creationflags = 0x00000008, # detached process
|
| + close_fds=True)
|
| + else:
|
| + port_server = subprocess.Popen(
|
| + args,
|
| + env=env,
|
| + preexec_fn=os.setsid,
|
| + close_fds=True)
|
| + time.sleep(1)
|
| + # ensure port server is up
|
| + waits = 0
|
| + while True:
|
| + if waits > 10:
|
| + print 'killing port server due to excessive start up waits'
|
| + port_server.kill()
|
| + if port_server.poll() is not None:
|
| + print 'port_server failed to start'
|
| + # try one final time: maybe another build managed to start one
|
| + time.sleep(1)
|
| + try:
|
| + urllib2.urlopen('http://localhost:%d/get' % port_server_port,
|
| + timeout=1).read()
|
| + print 'last ditch attempt to contact port server succeeded'
|
| + break
|
| + except:
|
| + traceback.print_exc()
|
| + port_log = open(logfile, 'r').read()
|
| + print port_log
|
| + sys.exit(1)
|
| + try:
|
| + urllib2.urlopen('http://localhost:%d/get' % port_server_port,
|
| + timeout=1).read()
|
| + print 'port server is up and ready'
|
| + break
|
| + except socket.timeout:
|
| + print 'waiting for port_server: timeout'
|
| + traceback.print_exc();
|
| + time.sleep(1)
|
| + waits += 1
|
| + except urllib2.URLError:
|
| + print 'waiting for port_server: urlerror'
|
| + traceback.print_exc();
|
| + time.sleep(1)
|
| + waits += 1
|
| + except:
|
| + traceback.print_exc()
|
| + port_server.kill()
|
| + raise
|
| +
|
| +
|
| +def _calculate_num_runs_failures(list_of_results):
|
| + """Caculate number of runs and failures for a particular test.
|
| +
|
| + Args:
|
| + list_of_results: (List) of JobResult object.
|
| + Returns:
|
| + A tuple of total number of runs and failures.
|
| + """
|
| + num_runs = len(list_of_results) # By default, there is 1 run per JobResult.
|
| + num_failures = 0
|
| + for jobresult in list_of_results:
|
| + if jobresult.retries > 0:
|
| + num_runs += jobresult.retries
|
| + if jobresult.num_failures > 0:
|
| + num_failures += jobresult.num_failures
|
| + return num_runs, num_failures
|
| +
|
| +
|
| +# _build_and_run results
|
| +class BuildAndRunError(object):
|
| +
|
| + BUILD = object()
|
| + TEST = object()
|
| + POST_TEST = object()
|
| +
|
| +
|
| +# returns a list of things that failed (or an empty list on success)
|
| +def _build_and_run(
|
| + check_cancelled, newline_on_success, cache, xml_report=None, build_only=False):
|
| + """Do one pass of building & running tests."""
|
| + # build latest sequentially
|
| + num_failures, resultset = jobset.run(
|
| + build_steps, maxjobs=1, stop_on_failure=True,
|
| + newline_on_success=newline_on_success, travis=args.travis)
|
| + if num_failures:
|
| + return [BuildAndRunError.BUILD]
|
| +
|
| + if build_only:
|
| + if xml_report:
|
| + report_utils.render_junit_xml_report(resultset, xml_report)
|
| + return []
|
| +
|
| + # start antagonists
|
| + antagonists = [subprocess.Popen(['tools/run_tests/antagonist.py'])
|
| + for _ in range(0, args.antagonists)]
|
| + port_server_port = 32767
|
| + _start_port_server(port_server_port)
|
| + resultset = None
|
| + num_test_failures = 0
|
| + try:
|
| + infinite_runs = runs_per_test == 0
|
| + one_run = set(
|
| + spec
|
| + for language in languages
|
| + for spec in language.test_specs()
|
| + if re.search(args.regex, spec.shortname))
|
| + # When running on travis, we want out test runs to be as similar as possible
|
| + # for reproducibility purposes.
|
| + if args.travis:
|
| + massaged_one_run = sorted(one_run, key=lambda x: x.shortname)
|
| + else:
|
| + # whereas otherwise, we want to shuffle things up to give all tests a
|
| + # chance to run.
|
| + massaged_one_run = list(one_run) # random.shuffle needs an indexable seq.
|
| + random.shuffle(massaged_one_run) # which it modifies in-place.
|
| + if infinite_runs:
|
| + assert len(massaged_one_run) > 0, 'Must have at least one test for a -n inf run'
|
| + runs_sequence = (itertools.repeat(massaged_one_run) if infinite_runs
|
| + else itertools.repeat(massaged_one_run, runs_per_test))
|
| + all_runs = itertools.chain.from_iterable(runs_sequence)
|
| +
|
| + num_test_failures, resultset = jobset.run(
|
| + all_runs, check_cancelled, newline_on_success=newline_on_success,
|
| + travis=args.travis, infinite_runs=infinite_runs, maxjobs=args.jobs,
|
| + stop_on_failure=args.stop_on_failure,
|
| + cache=cache if not xml_report else None,
|
| + add_env={'GRPC_TEST_PORT_SERVER': 'localhost:%d' % port_server_port})
|
| + if resultset:
|
| + for k, v in resultset.iteritems():
|
| + num_runs, num_failures = _calculate_num_runs_failures(v)
|
| + if num_failures == num_runs: # what about infinite_runs???
|
| + jobset.message('FAILED', k, do_newline=True)
|
| + elif num_failures > 0:
|
| + jobset.message(
|
| + 'FLAKE', '%s [%d/%d runs flaked]' % (k, num_failures, num_runs),
|
| + do_newline=True)
|
| + else:
|
| + jobset.message('PASSED', k, do_newline=True)
|
| + finally:
|
| + for antagonist in antagonists:
|
| + antagonist.kill()
|
| + if xml_report and resultset:
|
| + report_utils.render_junit_xml_report(resultset, xml_report)
|
| +
|
| + number_failures, _ = jobset.run(
|
| + post_tests_steps, maxjobs=1, stop_on_failure=True,
|
| + newline_on_success=newline_on_success, travis=args.travis)
|
| +
|
| + out = []
|
| + if number_failures:
|
| + out.append(BuildAndRunError.POST_TEST)
|
| + if num_test_failures:
|
| + out.append(BuildAndRunError.TEST)
|
| +
|
| + if cache: cache.save()
|
| +
|
| + return out
|
| +
|
| +
|
| +test_cache = TestCache(runs_per_test == 1)
|
| +test_cache.maybe_load()
|
| +
|
| +if forever:
|
| + success = True
|
| + while True:
|
| + dw = watch_dirs.DirWatcher(['src', 'include', 'test', 'examples'])
|
| + initial_time = dw.most_recent_change()
|
| + have_files_changed = lambda: dw.most_recent_change() != initial_time
|
| + previous_success = success
|
| + errors = _build_and_run(check_cancelled=have_files_changed,
|
| + newline_on_success=False,
|
| + cache=test_cache,
|
| + build_only=args.build_only) == 0
|
| + if not previous_success and not errors:
|
| + jobset.message('SUCCESS',
|
| + 'All tests are now passing properly',
|
| + do_newline=True)
|
| + jobset.message('IDLE', 'No change detected')
|
| + while not have_files_changed():
|
| + time.sleep(1)
|
| +else:
|
| + errors = _build_and_run(check_cancelled=lambda: False,
|
| + newline_on_success=args.newline_on_success,
|
| + cache=test_cache,
|
| + xml_report=args.xml_report,
|
| + build_only=args.build_only)
|
| + if not errors:
|
| + jobset.message('SUCCESS', 'All tests passed', do_newline=True)
|
| + else:
|
| + jobset.message('FAILED', 'Some tests failed', do_newline=True)
|
| + exit_code = 0
|
| + if BuildAndRunError.BUILD in errors:
|
| + exit_code |= 1
|
| + if BuildAndRunError.TEST in errors and not args.travis:
|
| + exit_code |= 2
|
| + if BuildAndRunError.POST_TEST in errors:
|
| + exit_code |= 4
|
| + sys.exit(exit_code)
|
|
|