Skip to content
Snippets Groups Projects
run_tests.py 58.5 KiB
Newer Older
#!/usr/bin/env python
# Copyright 2015 gRPC authors.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#     http://www.apache.org/licenses/LICENSE-2.0
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Run tests in parallel."""

from __future__ import print_function

Craig Tiller's avatar
Craig Tiller committed
import ast
import glob
import itertools
Craig Tiller's avatar
Craig Tiller committed
import json
import multiprocessing
import platform
import random
Craig Tiller's avatar
Craig Tiller committed
import socket
import tempfile
import traceback
from six.moves import urllib
import python_utils.jobset as jobset
import python_utils.report_utils as report_utils
import python_utils.watch_dirs as watch_dirs
import python_utils.start_port_server as start_port_server
try:
  from python_utils.upload_test_results import upload_results_to_bq
except (ImportError):
  pass # It's ok to not import because this is only necessary to upload results to BQ.
gcp_utils_dir = os.path.abspath(os.path.join(
        os.path.dirname(__file__), '../gcp/utils'))
sys.path.append(gcp_utils_dir)

Jan Tattermusch's avatar
Jan Tattermusch committed
_ROOT = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../..'))
os.chdir(_ROOT)
_FORCE_ENVIRON_FOR_WRAPPERS = {
  'GRPC_VERBOSITY': 'DEBUG',
}
  'linux': ['epollsig', 'poll', 'poll-cv'],
# TODO(ctiller, sreecha): enable epoll1, epollex, epoll-thread-pool
Vijay Pai's avatar
Vijay Pai committed
  'mac': ['poll'],
def get_flaky_tests(limit=None):
Craig Tiller's avatar
Craig Tiller committed
  import big_query_utils

  bq = big_query_utils.create_big_query()
SELECT
  filtered_test_name,
  FROM (
  SELECT
    REGEXP_REPLACE(test_name, r'/\d+', '') AS filtered_test_name,
    result
  FROM
    [grpc-testing:jenkins_test_results.aggregate_results]
  WHERE
    timestamp >= DATE_ADD(CURRENT_DATE(), -1, "WEEK")
    AND NOT REGEXP_MATCH(job_name, '.*portability.*') )
GROUP BY
  filtered_test_name
HAVING
  SUM(result != 'PASSED' AND result != 'SKIPPED') > 0"""
  if limit:
    query += " limit {}".format(limit)
Craig Tiller's avatar
Craig Tiller committed
  query_job = big_query_utils.sync_query_job(bq, 'grpc-testing', query)
  page = bq.jobs().getQueryResults(
      pageToken=None,
      **query_job['jobReference']).execute(num_retries=3)
  flake_names = [row['f'][0]['v'] for row in page['rows']]
  return flake_names


  return jobset.platform_string()
_DEFAULT_TIMEOUT_SECONDS = 5 * 60

def run_shell_command(cmd, env=None, cwd=None):
  try:
    subprocess.check_output(cmd, shell=True, env=env, cwd=cwd)
  except subprocess.CalledProcessError as e:
    logging.exception("Error while running command '%s'. Exit status %d. Output:\n%s",
                       e.cmd, e.returncode, e.output)
# 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=[], iomgr_platform='native'):
    self.build_config = config
    self.environ = environ
    self.tool_prefix = tool_prefix
    self.timeout_multiplier = timeout_multiplier
    self.iomgr_platform = iomgr_platform
  def job_spec(self, cmdline, timeout_seconds=_DEFAULT_TIMEOUT_SECONDS,
               shortname=None, environ={}, cpu_cost=1.0, flaky=False):
    """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
    """
    actual_environ = self.environ.copy()
    for k, v in environ.items():
    if not flaky and shortname and shortname in flaky_tests:
      print('Setting %s to flaky' % shortname)
      flaky = True
    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),
                          flake_retries=5 if flaky or args.allow_flakes else 0,
                          timeout_retries=3 if flaky or 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/generated/tests.json') as f:
murgatroid99's avatar
murgatroid99 committed
    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'])]
Jan Tattermusch's avatar
Jan Tattermusch committed
def _check_compiler(compiler, supported_compilers):
  if compiler not in supported_compilers:
    raise Exception('Compiler %s not supported (on this platform).' % compiler)


def _check_arch(arch, supported_archs):
  if arch not in supported_archs:
    raise Exception('Architecture %s not supported.' % arch)
Jan Tattermusch's avatar
Jan Tattermusch committed
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


_PythonConfigVars = collections.namedtuple(
  '_ConfigVars', ['shell', 'builder', 'builder_prefix_arguments',
                  'venv_relative_python', 'toolchain', 'runner'])


def _python_config_generator(name, major, minor, bits, config_vars):
  return PythonConfig(
    name,
    config_vars.shell + config_vars.builder + config_vars.builder_prefix_arguments + [
      _python_pattern_function(major=major, minor=minor, bits=bits)] + [
      name] + config_vars.venv_relative_python + config_vars.toolchain,
    config_vars.shell + config_vars.runner + [
      os.path.join(name, config_vars.venv_relative_python[0])])


def _pypy_config_generator(name, major, config_vars):
  return PythonConfig(
    name,
    config_vars.shell + config_vars.builder + config_vars.builder_prefix_arguments + [
      _pypy_pattern_function(major=major)] + [
      name] + config_vars.venv_relative_python + config_vars.toolchain,
    config_vars.shell + config_vars.runner + [
      os.path.join(name, config_vars.venv_relative_python[0])])


def _python_pattern_function(major, minor, bits):
  # Bit-ness is handled by the test machine's environment
  if os.name == "nt":
    if bits == "64":
      return '/c/Python{major}{minor}/python.exe'.format(
        major=major, minor=minor, bits=bits)
    else:
      return '/c/Python{major}{minor}_{bits}bits/python.exe'.format(
        major=major, minor=minor, bits=bits)
  else:
    return 'python{major}.{minor}'.format(major=major, minor=minor)


def _pypy_pattern_function(major):
  if major == '2':
    return 'pypy'
  elif major == '3':
    return 'pypy3'
  else:
    raise ValueError("Unknown PyPy major version")


class CLanguage(object):

  def __init__(self, make_target, test_lang):
    self.make_target = make_target
    self.platform = platform_string()
    self.test_lang = test_lang
Jan Tattermusch's avatar
Jan Tattermusch committed
  def configure(self, config, args):
    self.config = config
    self.args = args
    if self.args.compiler == 'cmake':
      _check_arch(self.args.arch, ['default'])
      self._use_cmake = True
      self._docker_distro = 'jessie'
      self._make_options = []
    elif self.platform == 'windows':
      self._use_cmake = False
Jan Tattermusch's avatar
Jan Tattermusch committed
      self._make_options = [_windows_toolset_option(self.args.compiler),
                            _windows_arch_option(self.args.arch)]
Jan Tattermusch's avatar
Jan Tattermusch committed
    else:
      self._use_cmake = False
      self._docker_distro, self._make_options = self._compiler_options(self.args.use_docker,
                                                                       self.args.compiler)
    if args.iomgr_platform == "uv":
      cflags = '-DGRPC_UV -DGRPC_UV_THREAD_CHECK'
      try:
        cflags += subprocess.check_output(['pkg-config', '--cflags', 'libuv']).strip() + ' '
      except (subprocess.CalledProcessError, OSError):
        pass
      try:
        ldflags = subprocess.check_output(['pkg-config', '--libs', 'libuv']).strip() + ' '
      except (subprocess.CalledProcessError, OSError):
        ldflags = '-luv '
      self._make_options += ['EXTRA_CPPFLAGS={}'.format(cflags),
                             'EXTRA_LDLIBS={}'.format(ldflags)]
Jan Tattermusch's avatar
Jan Tattermusch committed

  def test_specs(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    binaries = get_c_tests(self.args.travis, self.test_lang)
      if self._use_cmake and target.get('boringssl', False):
        # cmake doesn't build boringssl tests
        continue
      polling_strategies = (_POLLING_STRATEGIES.get(self.platform, ['all'])
                            if target.get('uses_polling', True)
                            else ['all'])
murgatroid99's avatar
murgatroid99 committed
      if self.args.iomgr_platform == 'uv':
        polling_strategies = ['all']
      for polling_strategy in polling_strategies:
        env={'GRPC_DEFAULT_SSL_ROOTS_FILE_PATH':
Craig Tiller's avatar
Craig Tiller committed
                 _ROOT + '/src/core/tsi/test_creds/ca.pem',
             'GRPC_POLL_STRATEGY': polling_strategy,
             'GRPC_VERBOSITY': 'DEBUG'}
        resolver = os.environ.get('GRPC_DNS_RESOLVER', None);
        if resolver:
          env['GRPC_DNS_RESOLVER'] = resolver
        shortname_ext = '' if polling_strategy=='all' else ' GRPC_POLL_STRATEGY=%s' % polling_strategy
        timeout_scaling = 1
        if polling_strategy == 'poll-cv':
          timeout_scaling *= 5
        if polling_strategy in target.get('excluded_poll_engines', []):
Sree Kuchibhotla's avatar
Sree Kuchibhotla committed
          continue
        # Scale overall test timeout if running under various sanitizers.
        config = self.args.config
        if ('asan' in config
            or config == 'msan'
            or config == 'tsan'
            or config == 'ubsan'
            or config == 'helgrind'
            or config == 'memcheck'):
          timeout_scaling *= 20

        if self.config.build_config in target['exclude_configs']:
          continue
        if self.args.iomgr_platform in target.get('exclude_iomgrs', []):
          continue
        if self.platform == 'windows':
          if self._use_cmake:
            binary = 'cmake/build/%s/%s.exe' % (_MSBUILD_CONFIG[self.config.build_config], target['name'])
          else:
            binary = 'vsprojects/%s%s/%s.exe' % (
                'x64/' if self.args.arch == 'x64' else '',
                _MSBUILD_CONFIG[self.config.build_config],
                target['name'])
          if self._use_cmake:
            binary = 'cmake/build/%s' % target['name']
          else:
            binary = 'bins/%s/%s' % (self.config.build_config, target['name'])
        cpu_cost = target['cpu_cost']
        if cpu_cost == 'capacity':
          cpu_cost = multiprocessing.cpu_count()
        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] + target['args']
                out.append(self.config.job_spec(cmdline,
                                                shortname='%s %s' % (' '.join(cmdline), shortname_ext),
                                                timeout_seconds=_DEFAULT_TIMEOUT_SECONDS * timeout_scaling,
                                                environ=env))
          else:
            cmdline = [binary] + target['args']
            out.append(self.config.job_spec(cmdline,
                                            shortname=' '.join(
                                                          pipes.quote(arg)
                                                          for arg in cmdline) +
                                                      shortname_ext,
                                            flaky=target.get('flaky', False),
                                            timeout_seconds=target.get('timeout_seconds', _DEFAULT_TIMEOUT_SECONDS) * timeout_scaling,
                                            environ=env))
        elif self.args.regex == '.*' or self.platform == 'windows':
          print('\nWARNING: binary not found, skipping', binary)
    return sorted(out)
Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
    if self.platform == 'windows':
Craig Tiller's avatar
Craig Tiller committed
      # 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,
            'check_epollexclusive']
  def make_options(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    return self._make_options;
  def pre_build_steps(self):
    if self._use_cmake:
      if self.platform == 'windows':
        return [['tools\\run_tests\\helper_scripts\\pre_build_cmake.bat']]
      else:
        return [['tools/run_tests/helper_scripts/pre_build_cmake.sh']]
      if self.platform == 'windows':
        return [['tools\\run_tests\\helper_scripts\\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/helper_scripts/post_tests_c.sh']]
    if self._use_cmake:
      return 'cmake/build/Makefile'
    else:
      return 'Makefile'
  def _clang_make_options(self, version_suffix=''):
    return ['CC=clang%s' % version_suffix,
            'CXX=clang++%s' % version_suffix,
            'LD=clang%s' % version_suffix,
            'LDXX=clang++%s' % version_suffix]

  def _gcc_make_options(self, version_suffix):
    return ['CC=gcc%s' % version_suffix,
            'CXX=g++%s' % version_suffix,
            'LD=gcc%s' % version_suffix,
            'LDXX=g++%s' % version_suffix]
  def _compiler_options(self, use_docker, compiler):
    """Returns docker distro and make options to use for given compiler."""
    if not use_docker and not _is_use_docker_child():
Jan Tattermusch's avatar
Jan Tattermusch committed
      _check_compiler(compiler, ['default'])

    if compiler == 'gcc4.9' or compiler == 'default':
      return ('jessie', [])
    elif compiler == 'gcc4.8':
      return ('jessie', self._gcc_make_options(version_suffix='-4.8'))
Jan Tattermusch's avatar
Jan Tattermusch committed
    elif compiler == 'gcc5.3':
      return ('ubuntu1604', [])
    elif compiler == 'gcc_musl':
      return ('alpine', [])
    elif compiler == 'clang3.4':
      # on ubuntu1404, clang-3.4 alias doesn't exist, just use 'clang'
      return ('ubuntu1404', self._clang_make_options())
    elif compiler == 'clang3.5':
      return ('jessie', self._clang_make_options(version_suffix='-3.5'))
    elif compiler == 'clang3.6':
      return ('ubuntu1604', self._clang_make_options(version_suffix='-3.6'))
    elif compiler == 'clang3.7':
      return ('ubuntu1604', self._clang_make_options(version_suffix='-3.7'))
Jan Tattermusch's avatar
Jan Tattermusch committed
    else:
      raise Exception('Compiler %s not supported.' % compiler)

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    return 'tools/dockerfile/test/cxx_%s_%s' % (self._docker_distro,
                                                _docker_arch_suffix(self.args.arch))
murgatroid99's avatar
murgatroid99 committed
class NodeLanguage(object):

  def __init__(self):
    self.platform = platform_string()
Jan Tattermusch's avatar
Jan Tattermusch committed
  def configure(self, config, args):
    self.config = config
    self.args = args
    # Note: electron ABI only depends on major and minor version, so that's all
    # we should specify in the compiler argument
    _check_compiler(self.args.compiler, ['default', 'node0.12',
                                         'node4', 'node5', 'node6',
                                         'node7', 'node8',
                                         'electron1.3', 'electron1.6'])
      if self.args.compiler.startswith('electron'):
        self.runtime = 'electron'
        self.node_version = self.args.compiler[8:]
      else:
        self.runtime = 'node'
        # Take off the word "node"
        self.node_version = self.args.compiler[4:]
Jan Tattermusch's avatar
Jan Tattermusch committed

  def test_specs(self):
    if self.platform == 'windows':
      return [self.config.job_spec(['tools\\run_tests\\helper_scripts\\run_node.bat'])]
      run_script = 'run_node'
      if self.runtime == 'electron':
        run_script += '_electron'
      return [self.config.job_spec(['tools/run_tests/helper_scripts/{}.sh'.format(run_script),
                                    self.node_version],
Jan Tattermusch's avatar
Jan Tattermusch committed
                                   None,
                                   environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
  def pre_build_steps(self):
    if self.platform == 'windows':
      return [['tools\\run_tests\\helper_scripts\\pre_build_node.bat']]
      build_script = 'pre_build_node'
      if self.runtime == 'electron':
        build_script += '_electron'
      return [['tools/run_tests/helper_scripts/{}.sh'.format(build_script),
               self.node_version]]
Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
  def make_options(self):
    return []

murgatroid99's avatar
murgatroid99 committed
  def build_steps(self):
    if self.platform == 'windows':
murgatroid99's avatar
murgatroid99 committed
      if self.config == 'dbg':
        config_flag = '--debug'
      else:
        config_flag = '--release'
      return [['tools\\run_tests\\helper_scripts\\build_node.bat',
               config_flag]]
      build_script = 'build_node'
      if self.runtime == 'electron':
        build_script += '_electron'
        # building for electron requires a patch version
        self.node_version += '.0'
      return [['tools/run_tests/helper_scripts/{}.sh'.format(build_script),
               self.node_version]]
  def post_tests_steps(self):
    return []

  def makefile_name(self):
    return 'Makefile'

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
    return 'tools/dockerfile/test/node_jessie_%s' % _docker_arch_suffix(self.args.arch)
Craig Tiller's avatar
Craig Tiller committed

class PhpLanguage(object):

Jan Tattermusch's avatar
Jan Tattermusch committed
  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'],
Jan Tattermusch's avatar
Jan Tattermusch committed
                                  environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
  def pre_build_steps(self):
    return []

Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
    return ['static_c', 'shared_c']
  def make_options(self):
    return []

  def build_steps(self):
    return [['tools/run_tests/helper_scripts/build_php.sh']]
  def post_tests_steps(self):
    return [['tools/run_tests/helper_scripts/post_tests_php.sh']]
  def makefile_name(self):
    return 'Makefile'

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
    return 'tools/dockerfile/test/php_jessie_%s' % _docker_arch_suffix(self.args.arch)
class Php7Language(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'],
                                  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/helper_scripts/build_php.sh']]

  def post_tests_steps(self):
    return [['tools/run_tests/helper_scripts/post_tests_php.sh']]

  def makefile_name(self):
    return 'Makefile'

  def dockerfile_dir(self):
    return 'tools/dockerfile/test/php7_jessie_%s' % _docker_arch_suffix(self.args.arch)

  def __str__(self):
    return 'php7'


class PythonConfig(collections.namedtuple('PythonConfig', [
    'name', 'build', 'run'])):
  """Tuple of commands (named s.t. 'what it says on the tin' applies)"""
class PythonLanguage(object):

Jan Tattermusch's avatar
Jan Tattermusch committed
  def configure(self, config, args):
    self.config = config
    self.args = args
    self.pythons = self._get_pythons(self.args)
Jan Tattermusch's avatar
Jan Tattermusch committed

  def test_specs(self):
    # load list of known test suites
    with open('src/python/grpcio_tests/tests/tests.json') as tests_json_file:
      tests_json = json.load(tests_json_file)
    environment = dict(_FORCE_ENVIRON_FOR_WRAPPERS)
    return [self.config.job_spec(
        environ=dict(list(environment.items()) +
                     [('GRPC_PYTHON_TESTRUNNER_FILTER', str(suite_name))]),
        shortname='%s.test.%s' % (config.name, suite_name),)
        for suite_name in tests_json
        for config in self.pythons]
  def pre_build_steps(self):
    return []

Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
  def make_options(self):
    return []

  def build_steps(self):
    return [config.build for config in self.pythons]
  def post_tests_steps(self):
Matt Kwong's avatar
Matt Kwong committed
    if self.config != 'gcov':
      return []
    else:
      return [['tools/run_tests/helper_scripts/post_tests_python.sh']]
  def makefile_name(self):
    return 'Makefile'

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
    return 'tools/dockerfile/test/python_%s_%s' % (self.python_manager_name(), _docker_arch_suffix(self.args.arch))

  def python_manager_name(self):
    if self.args.compiler in ['python3.5', 'python3.6']:
      return 'pyenv'
    elif self.args.compiler == 'python_alpine':
      return 'alpine'
    else:
      return 'jessie'
  def _get_pythons(self, args):
    if args.arch == 'x86':
      bits = '32'
    else:
      bits = '64'
      shell = ['bash']
      builder = [os.path.abspath('tools/run_tests/helper_scripts/build_python_msys2.sh')]
      builder_prefix_arguments = ['MINGW{}'.format(bits)]
      venv_relative_python = ['Scripts/python.exe']
      toolchain = ['mingw32']
      builder = [os.path.abspath('tools/run_tests/helper_scripts/build_python.sh')]
      builder_prefix_arguments = []
      venv_relative_python = ['bin/python']
      toolchain = ['unix']
    runner = [os.path.abspath('tools/run_tests/helper_scripts/run_python.sh')]
    config_vars = _PythonConfigVars(shell, builder, builder_prefix_arguments,
                              venv_relative_python, toolchain, runner)
    python27_config = _python_config_generator(name='py27', major='2',
                                               minor='7', bits=bits,
                                               config_vars=config_vars)
    python34_config = _python_config_generator(name='py34', major='3',
                                               minor='4', bits=bits,
                                               config_vars=config_vars)
    python35_config = _python_config_generator(name='py35', major='3',
                                               minor='5', bits=bits,
                                               config_vars=config_vars)
    python36_config = _python_config_generator(name='py36', major='3',
                                               minor='6', bits=bits,
                                               config_vars=config_vars)
    pypy27_config = _pypy_config_generator(name='pypy', major='2',
                                           config_vars=config_vars)
    pypy32_config = _pypy_config_generator(name='pypy3', major='3',
                                           config_vars=config_vars)

    if args.compiler == 'default':
      if os.name == 'nt':
        return (python35_config,)
      else:
        return (python27_config, python34_config,)
    elif args.compiler == 'python2.7':
      return (python27_config,)
    elif args.compiler == 'python3.4':
      return (python34_config,)
    elif args.compiler == 'python3.5':
      return (python35_config,)
    elif args.compiler == 'python3.6':
      return (python36_config,)
    elif args.compiler == 'pypy':
      return (pypy27_config,)
    elif args.compiler == 'pypy3':
      return (pypy32_config,)
    elif args.compiler == 'python_alpine':
      return (python27_config,)
      raise Exception('Compiler %s not supported.' % args.compiler)
class RubyLanguage(object):

Jan Tattermusch's avatar
Jan Tattermusch committed
  def configure(self, config, args):
    self.config = config
    self.args = args
    _check_compiler(self.args.compiler, ['default'])

  def test_specs(self):
    tests = [self.config.job_spec(['tools/run_tests/helper_scripts/run_ruby.sh'],
                                  timeout_seconds=10*60,
                                  environ=_FORCE_ENVIRON_FOR_WRAPPERS)]
    tests.append(self.config.job_spec(['tools/run_tests/helper_scripts/run_ruby_end2end_tests.sh'],
                 timeout_seconds=10*60,
                 environ=_FORCE_ENVIRON_FOR_WRAPPERS))
    return tests
  def pre_build_steps(self):
    return [['tools/run_tests/helper_scripts/pre_build_ruby.sh']]
Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
  def make_options(self):
    return []

  def build_steps(self):
    return [['tools/run_tests/helper_scripts/build_ruby.sh']]
  def post_tests_steps(self):
    return [['tools/run_tests/helper_scripts/post_tests_ruby.sh']]
  def makefile_name(self):
    return 'Makefile'

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
    return 'tools/dockerfile/test/ruby_jessie_%s' % _docker_arch_suffix(self.args.arch)
class CSharpLanguage(object):
  def __init__(self):
    self.platform = platform_string()
Jan Tattermusch's avatar
Jan Tattermusch committed
  def configure(self, config, args):
    self.config = config
    self.args = args
Jan Tattermusch's avatar
Jan Tattermusch committed
    if self.platform == 'windows':
      _check_compiler(self.args.compiler, ['coreclr', 'default'])
      _check_arch(self.args.arch, ['default'])
      self._cmake_arch_option = 'x64'
      self._make_options = []
Jan Tattermusch's avatar
Jan Tattermusch committed
    else:
      _check_compiler(self.args.compiler, ['default', 'coreclr'])
      self._docker_distro = 'jessie'
Jan Tattermusch's avatar
Jan Tattermusch committed
      if self.platform == 'mac':
        # TODO(jtattermusch): EMBED_ZLIB=true currently breaks the mac build
        self._make_options = ['EMBED_OPENSSL=true']
        if self.args.compiler != 'coreclr':
          # On Mac, official distribution of mono is 32bit.
Craig Tiller's avatar
Craig Tiller committed
          self._make_options += ['ARCH_FLAGS=-m32', 'LDFLAGS=-m32']
Jan Tattermusch's avatar
Jan Tattermusch committed
      else:
        self._make_options = ['EMBED_OPENSSL=true', 'EMBED_ZLIB=true']
Jan Tattermusch's avatar
Jan Tattermusch committed

  def test_specs(self):
    with open('src/csharp/tests.json') as f:
      tests_by_assembly = json.load(f)
Jan Tattermusch's avatar
Jan Tattermusch committed
    msbuild_config = _MSBUILD_CONFIG[self.config.build_config]
    nunit_args = ['--labels=All', '--noresult', '--workers=1']
    assembly_subdir = 'bin/%s' % msbuild_config
    assembly_extension = '.exe'

    if self.args.compiler == 'coreclr':
      assembly_subdir += '/netcoreapp1.0'
      runtime_cmd = ['dotnet', 'exec']
      assembly_extension = '.dll'
      assembly_subdir += '/net45'
      if self.platform == 'windows':
        runtime_cmd = []
      else:
        runtime_cmd = ['mono']
    for assembly in six.iterkeys(tests_by_assembly):
      assembly_file = 'src/csharp/%s/%s/%s%s' % (assembly,
                                                 assembly_subdir,
                                                 assembly,
                                                 assembly_extension)
      if self.config.build_config != 'gcov' or self.platform != 'windows':
        # normally, run each test as a separate process
        for test in tests_by_assembly[assembly]:
          cmdline = runtime_cmd + [assembly_file, '--test=%s' % test] + nunit_args
          specs.append(self.config.job_spec(cmdline,
                                            shortname='csharp.%s' % test,
                                            environ=_FORCE_ENVIRON_FOR_WRAPPERS))
      else:
        # For C# test coverage, run all tests from the same assembly at once
        # using OpenCover.Console (only works on Windows).
        cmdline = ['src\\csharp\\packages\\OpenCover.4.6.519\\tools\\OpenCover.Console.exe',
                   '-target:%s' % assembly_file,
                   '-targetdir:src\\csharp',
                   '-targetargs:%s' % ' '.join(nunit_args),
                   '-filter:+[Grpc.Core]*',
                   '-register:user',
                   '-output:src\\csharp\\coverage_csharp_%s.xml' % assembly]

        # set really high cpu_cost to make sure instances of OpenCover.Console run exclusively
        # to prevent problems with registering the profiler.
        run_exclusive = 1000000
        specs.append(self.config.job_spec(cmdline,
                                          shortname='csharp.coverage.%s' % assembly,
                                          cpu_cost=run_exclusive,
Jan Tattermusch's avatar
Jan Tattermusch committed
                                          environ=_FORCE_ENVIRON_FOR_WRAPPERS))
  def pre_build_steps(self):
    if self.platform == 'windows':
      return [['tools\\run_tests\\helper_scripts\\pre_build_csharp.bat', self._cmake_arch_option]]
      return [['tools/run_tests/helper_scripts/pre_build_csharp.sh']]
Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    return ['grpc_csharp_ext']
  def make_options(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    return self._make_options;
  def build_steps(self):
    if self.platform == 'windows':
      return [['tools\\run_tests\\helper_scripts\\build_csharp.bat']]
      return [['tools/run_tests/helper_scripts/build_csharp.sh']]
  def post_tests_steps(self):
    if self.platform == 'windows':
      return [['tools\\run_tests\\helper_scripts\\post_tests_csharp.bat']]
      return [['tools/run_tests/helper_scripts/post_tests_csharp.sh']]
    if self.platform == 'windows':
      return 'cmake/build/%s/Makefile' % self._cmake_arch_option
    else:
      return 'Makefile'
Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
    return 'tools/dockerfile/test/csharp_%s_%s' % (self._docker_distro,
                                                   _docker_arch_suffix(self.args.arch))
class ObjCLanguage(object):

Jan Tattermusch's avatar
Jan Tattermusch committed
  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'],
                              timeout_seconds=60*60,
                              shortname='objc-tests',
                              cpu_cost=1e6,
                              environ=_FORCE_ENVIRON_FOR_WRAPPERS),
Jan Tattermusch's avatar
Jan Tattermusch committed
        self.config.job_spec(['src/objective-c/tests/run_plugin_tests.sh'],
                              timeout_seconds=60*60,
                              shortname='objc-plugin-tests',
                              cpu_cost=1e6,
                              environ=_FORCE_ENVIRON_FOR_WRAPPERS),
Jan Tattermusch's avatar
Jan Tattermusch committed
        self.config.job_spec(['src/objective-c/tests/build_one_example.sh'],
                              timeout_seconds=10*60,
                              shortname='objc-build-example-helloworld',
                              cpu_cost=1e6,
Jan Tattermusch's avatar
Jan Tattermusch committed
                              environ={'SCHEME': 'HelloWorld',
                                       'EXAMPLE_PATH': 'examples/objective-c/helloworld'}),
        self.config.job_spec(['src/objective-c/tests/build_one_example.sh'],
                              timeout_seconds=10*60,
                              shortname='objc-build-example-routeguide',
                              cpu_cost=1e6,
Jan Tattermusch's avatar
Jan Tattermusch committed
                              environ={'SCHEME': 'RouteGuideClient',
                                       'EXAMPLE_PATH': 'examples/objective-c/route_guide'}),
        self.config.job_spec(['src/objective-c/tests/build_one_example.sh'],
                              timeout_seconds=10*60,
                              shortname='objc-build-example-authsample',
                              cpu_cost=1e6,
Jan Tattermusch's avatar
Jan Tattermusch committed
                              environ={'SCHEME': 'AuthSample',
                                       'EXAMPLE_PATH': 'examples/objective-c/auth_sample'}),
        self.config.job_spec(['src/objective-c/tests/build_one_example.sh'],
                              timeout_seconds=10*60,
                              shortname='objc-build-example-sample',
                              cpu_cost=1e6,
Jan Tattermusch's avatar
Jan Tattermusch committed
                              environ={'SCHEME': 'Sample',
                                       'EXAMPLE_PATH': 'src/objective-c/examples/Sample'}),
        self.config.job_spec(['src/objective-c/tests/build_one_example.sh'],
                              timeout_seconds=10*60,
                              shortname='objc-build-example-sample-frameworks',
                              cpu_cost=1e6,
Jan Tattermusch's avatar
Jan Tattermusch committed
                              environ={'SCHEME': 'Sample',
                                       'EXAMPLE_PATH': 'src/objective-c/examples/Sample',
                                       'FRAMEWORKS': 'YES'}),
        self.config.job_spec(['src/objective-c/tests/build_one_example.sh'],
                              timeout_seconds=10*60,
                              shortname='objc-build-example-switftsample',
                              cpu_cost=1e6,
Jan Tattermusch's avatar
Jan Tattermusch committed
                              environ={'SCHEME': 'SwiftSample',
                                       'EXAMPLE_PATH': 'src/objective-c/examples/SwiftSample'}),
  def pre_build_steps(self):
    return []

Jan Tattermusch's avatar
Jan Tattermusch committed
  def make_targets(self):
    return ['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'

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    return None

  def __str__(self):
    return 'objc'


class Sanity(object):

Jan Tattermusch's avatar
Jan Tattermusch committed
  def configure(self, config, args):
    self.config = config
    self.args = args
    _check_compiler(self.args.compiler, ['default'])

  def test_specs(self):
Jan Tattermusch's avatar
Jan Tattermusch committed
    with open('tools/run_tests/sanity/sanity_tests.yaml', 'r') as f:
      environ={'TEST': 'true'}
      if _is_use_docker_child():
        environ['CLANG_FORMAT_SKIP_DOCKER'] = 'true'
Craig Tiller's avatar
Craig Tiller committed
      return [self.config.job_spec(cmd['script'].split(),
                                   timeout_seconds=30*60,
                                   environ=environ,
Jan Tattermusch's avatar
Jan Tattermusch committed
                                   cpu_cost=cmd.get('cpu_cost', 1))
              for cmd in yaml.load(f)]
  def pre_build_steps(self):
    return []

Jan Tattermusch's avatar
Jan Tattermusch committed
  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'

Jan Tattermusch's avatar
Jan Tattermusch committed
  def dockerfile_dir(self):
    return 'tools/dockerfile/test/sanity'