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)
|
|