mirror of https://github.com/grpc/grpc.git
Yuchen Zeng
8 years ago
commit
dc75e30e87
44 changed files with 1155 additions and 287 deletions
@ -0,0 +1,212 @@ |
||||
//
|
||||
// Copyright 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.
|
||||
//
|
||||
|
||||
// This is similar to the sockaddr resolver, except that it supports a
|
||||
// bunch of query args that are useful for dependency injection in tests.
|
||||
|
||||
#include <stdbool.h> |
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
|
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/host_port.h> |
||||
#include <grpc/support/port_platform.h> |
||||
#include <grpc/support/string_util.h> |
||||
|
||||
#include "src/core/ext/client_config/parse_address.h" |
||||
#include "src/core/ext/client_config/resolver_registry.h" |
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/iomgr/resolve_address.h" |
||||
#include "src/core/lib/iomgr/unix_sockets_posix.h" |
||||
#include "src/core/lib/support/string.h" |
||||
|
||||
//
|
||||
// fake_resolver
|
||||
//
|
||||
|
||||
typedef struct { |
||||
// base class -- must be first
|
||||
grpc_resolver base; |
||||
|
||||
// passed-in parameters
|
||||
char* target_name; // the path component of the uri passed in
|
||||
grpc_lb_addresses* addresses; |
||||
char* lb_policy_name; |
||||
|
||||
// mutex guarding the rest of the state
|
||||
gpr_mu mu; |
||||
// have we published?
|
||||
bool published; |
||||
// pending next completion, or NULL
|
||||
grpc_closure* next_completion; |
||||
// target result address for next completion
|
||||
grpc_resolver_result** target_result; |
||||
} fake_resolver; |
||||
|
||||
static void fake_resolver_destroy(grpc_exec_ctx* exec_ctx, grpc_resolver* gr) { |
||||
fake_resolver* r = (fake_resolver*)gr; |
||||
gpr_mu_destroy(&r->mu); |
||||
gpr_free(r->target_name); |
||||
grpc_lb_addresses_destroy(r->addresses, NULL /* user_data_destroy */); |
||||
gpr_free(r->lb_policy_name); |
||||
gpr_free(r); |
||||
} |
||||
|
||||
static void fake_resolver_shutdown(grpc_exec_ctx* exec_ctx, |
||||
grpc_resolver* resolver) { |
||||
fake_resolver* r = (fake_resolver*)resolver; |
||||
gpr_mu_lock(&r->mu); |
||||
if (r->next_completion != NULL) { |
||||
*r->target_result = NULL; |
||||
grpc_exec_ctx_sched(exec_ctx, r->next_completion, GRPC_ERROR_NONE, NULL); |
||||
r->next_completion = NULL; |
||||
} |
||||
gpr_mu_unlock(&r->mu); |
||||
} |
||||
|
||||
static void fake_resolver_maybe_finish_next_locked(grpc_exec_ctx* exec_ctx, |
||||
fake_resolver* r) { |
||||
if (r->next_completion != NULL && !r->published) { |
||||
r->published = true; |
||||
*r->target_result = grpc_resolver_result_create( |
||||
r->target_name, |
||||
grpc_lb_addresses_copy(r->addresses, NULL /* user_data_copy */), |
||||
r->lb_policy_name, NULL /* lb_policy_args */); |
||||
grpc_exec_ctx_sched(exec_ctx, r->next_completion, GRPC_ERROR_NONE, NULL); |
||||
r->next_completion = NULL; |
||||
} |
||||
} |
||||
|
||||
static void fake_resolver_channel_saw_error(grpc_exec_ctx* exec_ctx, |
||||
grpc_resolver* resolver) { |
||||
fake_resolver* r = (fake_resolver*)resolver; |
||||
gpr_mu_lock(&r->mu); |
||||
r->published = false; |
||||
fake_resolver_maybe_finish_next_locked(exec_ctx, r); |
||||
gpr_mu_unlock(&r->mu); |
||||
} |
||||
|
||||
static void fake_resolver_next(grpc_exec_ctx* exec_ctx, grpc_resolver* resolver, |
||||
grpc_resolver_result** target_result, |
||||
grpc_closure* on_complete) { |
||||
fake_resolver* r = (fake_resolver*)resolver; |
||||
gpr_mu_lock(&r->mu); |
||||
GPR_ASSERT(!r->next_completion); |
||||
r->next_completion = on_complete; |
||||
r->target_result = target_result; |
||||
fake_resolver_maybe_finish_next_locked(exec_ctx, r); |
||||
gpr_mu_unlock(&r->mu); |
||||
} |
||||
|
||||
static const grpc_resolver_vtable fake_resolver_vtable = { |
||||
fake_resolver_destroy, fake_resolver_shutdown, |
||||
fake_resolver_channel_saw_error, fake_resolver_next}; |
||||
|
||||
//
|
||||
// fake_resolver_factory
|
||||
//
|
||||
|
||||
static void fake_resolver_factory_ref(grpc_resolver_factory* factory) {} |
||||
|
||||
static void fake_resolver_factory_unref(grpc_resolver_factory* factory) {} |
||||
|
||||
static void do_nothing(void* ignored) {} |
||||
|
||||
static grpc_resolver* fake_resolver_create(grpc_resolver_factory* factory, |
||||
grpc_resolver_args* args) { |
||||
if (0 != strcmp(args->uri->authority, "")) { |
||||
gpr_log(GPR_ERROR, "authority based uri's not supported by the %s scheme", |
||||
args->uri->scheme); |
||||
return NULL; |
||||
} |
||||
// Get lb_enabled arg. Anything other than "0" is interpreted as true.
|
||||
const char* lb_enabled_qpart = |
||||
grpc_uri_get_query_arg(args->uri, "lb_enabled"); |
||||
const bool lb_enabled = |
||||
lb_enabled_qpart != NULL && strcmp("0", lb_enabled_qpart) != 0; |
||||
// Construct addresses.
|
||||
gpr_slice path_slice = |
||||
gpr_slice_new(args->uri->path, strlen(args->uri->path), do_nothing); |
||||
gpr_slice_buffer path_parts; |
||||
gpr_slice_buffer_init(&path_parts); |
||||
gpr_slice_split(path_slice, ",", &path_parts); |
||||
grpc_lb_addresses* addresses = grpc_lb_addresses_create(path_parts.count); |
||||
bool errors_found = false; |
||||
for (size_t i = 0; i < addresses->num_addresses; i++) { |
||||
grpc_uri ith_uri = *args->uri; |
||||
char* part_str = gpr_dump_slice(path_parts.slices[i], GPR_DUMP_ASCII); |
||||
ith_uri.path = part_str; |
||||
if (!parse_ipv4( |
||||
&ith_uri, |
||||
(struct sockaddr_storage*)(&addresses->addresses[i].address.addr), |
||||
&addresses->addresses[i].address.len)) { |
||||
errors_found = true; |
||||
} |
||||
gpr_free(part_str); |
||||
addresses->addresses[i].is_balancer = lb_enabled; |
||||
if (errors_found) break; |
||||
} |
||||
gpr_slice_buffer_destroy(&path_parts); |
||||
gpr_slice_unref(path_slice); |
||||
if (errors_found) { |
||||
grpc_lb_addresses_destroy(addresses, NULL /* user_data_destroy */); |
||||
return NULL; |
||||
} |
||||
// Instantiate resolver.
|
||||
fake_resolver* r = gpr_malloc(sizeof(fake_resolver)); |
||||
memset(r, 0, sizeof(*r)); |
||||
r->target_name = gpr_strdup(args->uri->path); |
||||
r->addresses = addresses; |
||||
r->lb_policy_name = |
||||
gpr_strdup(grpc_uri_get_query_arg(args->uri, "lb_policy")); |
||||
gpr_mu_init(&r->mu); |
||||
grpc_resolver_init(&r->base, &fake_resolver_vtable); |
||||
return &r->base; |
||||
} |
||||
|
||||
static char* fake_resolver_get_default_authority(grpc_resolver_factory* factory, |
||||
grpc_uri* uri) { |
||||
const char* path = uri->path; |
||||
if (path[0] == '/') ++path; |
||||
return gpr_strdup(path); |
||||
} |
||||
|
||||
static const grpc_resolver_factory_vtable fake_resolver_factory_vtable = { |
||||
fake_resolver_factory_ref, fake_resolver_factory_unref, |
||||
fake_resolver_create, fake_resolver_get_default_authority, "test"}; |
||||
|
||||
static grpc_resolver_factory fake_resolver_factory = { |
||||
&fake_resolver_factory_vtable}; |
||||
|
||||
void grpc_fake_resolver_init(void) { |
||||
grpc_register_resolver_type(&fake_resolver_factory); |
||||
} |
||||
@ -0,0 +1,39 @@ |
||||
//
|
||||
// Copyright 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.
|
||||
//
|
||||
|
||||
#ifndef GRPC_TEST_CORE_END2END_FAKE_RESOLVER_H |
||||
#define GRPC_TEST_CORE_END2END_FAKE_RESOLVER_H |
||||
|
||||
#include "test/core/util/test_config.h" |
||||
|
||||
void grpc_fake_resolver_init(); |
||||
|
||||
#endif /* GRPC_TEST_CORE_END2END_FAKE_RESOLVER_H */ |
||||
@ -0,0 +1,38 @@ |
||||
#!/usr/bin/env bash |
||||
# Copyright 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. |
||||
# |
||||
# Reboots Jenkins worker |
||||
# |
||||
# NOTE: No empty lines should appear in this file before igncr is set! |
||||
set -ex -o igncr || set -ex |
||||
|
||||
# Give 5 seconds to finish the current job, then kill the jenkins slave process |
||||
# to avoid running any other jobs on the worker and restart the worker. |
||||
nohup sh -c 'sleep 5; killall java; sudo reboot' & |
||||
@ -0,0 +1,42 @@ |
||||
#!/usr/bin/env bash |
||||
# Copyright 2015, 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. |
||||
# |
||||
# This script is invoked by Jenkins and triggers a test run, bypassing |
||||
# all args to the test script. |
||||
# |
||||
# Setting up rvm environment BEFORE we set -ex. |
||||
[[ -s /etc/profile.d/rvm.sh ]] && . /etc/profile.d/rvm.sh |
||||
# To prevent cygwin bash complaining about empty lines ending with \r |
||||
# we set the igncr option. The option doesn't exist on Linux, so we fallback |
||||
# to just 'set -ex' there. |
||||
# NOTE: No empty lines should appear in this file before igncr is set! |
||||
set -ex -o igncr || set -ex |
||||
|
||||
python tools/run_tests/run_tests_matrix.py $@ |
||||
@ -0,0 +1,46 @@ |
||||
#!/bin/bash |
||||
# Copyright 2015, 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. |
||||
# |
||||
# Create a workspace in a subdirectory to allow running multiple builds in isolation. |
||||
# WORKSPACE_NAME env variable needs to contain name of the workspace to create. |
||||
# All cmdline args will be passed to run_tests.py script (executed in the |
||||
# newly created workspace) |
||||
set -ex |
||||
|
||||
cd $(dirname $0)/../.. |
||||
|
||||
rm -rf "${WORKSPACE_NAME}" |
||||
# TODO(jtattermusch): clone --recursive fetches the submodules from github. |
||||
# Try avoiding that to save time and network capacity. |
||||
git clone --recursive . "${WORKSPACE_NAME}" |
||||
|
||||
echo "Running run_tests.py in workspace ${WORKSPACE_NAME}" |
||||
python "${WORKSPACE_NAME}/tools/run_tests/run_tests.py" $@ |
||||
|
||||
@ -0,0 +1,282 @@ |
||||
#!/usr/bin/env python2.7 |
||||
# Copyright 2015, 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 test matrix.""" |
||||
|
||||
import argparse |
||||
import jobset |
||||
import multiprocessing |
||||
import os |
||||
import report_utils |
||||
import sys |
||||
|
||||
_ROOT = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../..')) |
||||
os.chdir(_ROOT) |
||||
|
||||
# Set the timeout high to allow enough time for sanitizers and pre-building |
||||
# clang docker. |
||||
_RUNTESTS_TIMEOUT = 2*60*60 |
||||
|
||||
# Number of jobs assigned to each run_tests.py instance |
||||
_INNER_JOBS = 2 |
||||
|
||||
|
||||
def _docker_jobspec(name, runtests_args=[]): |
||||
"""Run a single instance of run_tests.py in a docker container""" |
||||
test_job = jobset.JobSpec( |
||||
cmdline=['python', 'tools/run_tests/run_tests.py', |
||||
'--use_docker', |
||||
'-t', |
||||
'-j', str(_INNER_JOBS), |
||||
'-x', 'report_%s.xml' % name] + runtests_args, |
||||
shortname='run_tests_%s' % name, |
||||
timeout_seconds=_RUNTESTS_TIMEOUT) |
||||
return test_job |
||||
|
||||
|
||||
def _workspace_jobspec(name, runtests_args=[], workspace_name=None): |
||||
"""Run a single instance of run_tests.py in a separate workspace""" |
||||
if not workspace_name: |
||||
workspace_name = 'workspace_%s' % name |
||||
env = {'WORKSPACE_NAME': workspace_name} |
||||
test_job = jobset.JobSpec( |
||||
cmdline=['tools/run_tests/run_tests_in_workspace.sh', |
||||
'-t', |
||||
'-j', str(_INNER_JOBS), |
||||
'-x', '../report_%s.xml' % name] + runtests_args, |
||||
environ=env, |
||||
shortname='run_tests_%s' % name, |
||||
timeout_seconds=_RUNTESTS_TIMEOUT) |
||||
return test_job |
||||
|
||||
|
||||
def _generate_jobs(languages, configs, platforms, |
||||
arch=None, compiler=None, |
||||
labels=[], extra_args=[]): |
||||
result = [] |
||||
for language in languages: |
||||
for platform in platforms: |
||||
for config in configs: |
||||
name = '%s_%s_%s' % (language, platform, config) |
||||
runtests_args = ['-l', language, |
||||
'-c', config] |
||||
if arch or compiler: |
||||
name += '_%s_%s' % (arch, compiler) |
||||
runtests_args += ['--arch', arch, |
||||
'--compiler', compiler] |
||||
|
||||
runtests_args += extra_args |
||||
if platform == 'linux': |
||||
job = _docker_jobspec(name=name, runtests_args=runtests_args) |
||||
else: |
||||
job = _workspace_jobspec(name=name, runtests_args=runtests_args) |
||||
|
||||
job.labels = [platform, config, language] + labels |
||||
result.append(job) |
||||
return result |
||||
|
||||
|
||||
def _create_test_jobs(extra_args=[]): |
||||
test_jobs = [] |
||||
# supported on linux only |
||||
test_jobs += _generate_jobs(languages=['sanity', 'php7'], |
||||
configs=['dbg', 'opt'], |
||||
platforms=['linux'], |
||||
labels=['basictests'], |
||||
extra_args=extra_args) |
||||
|
||||
# supported on all platforms. |
||||
test_jobs += _generate_jobs(languages=['c', 'csharp', 'node', 'python'], |
||||
configs=['dbg', 'opt'], |
||||
platforms=['linux', 'macos', 'windows'], |
||||
labels=['basictests'], |
||||
extra_args=extra_args) |
||||
|
||||
# supported on linux and mac. |
||||
test_jobs += _generate_jobs(languages=['c++', 'ruby', 'php'], |
||||
configs=['dbg', 'opt'], |
||||
platforms=['linux', 'macos'], |
||||
labels=['basictests'], |
||||
extra_args=extra_args) |
||||
|
||||
# supported on mac only. |
||||
test_jobs += _generate_jobs(languages=['objc'], |
||||
configs=['dbg', 'opt'], |
||||
platforms=['macos'], |
||||
labels=['basictests'], |
||||
extra_args=extra_args) |
||||
|
||||
# sanitizers |
||||
test_jobs += _generate_jobs(languages=['c'], |
||||
configs=['msan', 'asan', 'tsan'], |
||||
platforms=['linux'], |
||||
labels=['sanitizers'], |
||||
extra_args=extra_args) |
||||
test_jobs += _generate_jobs(languages=['c++'], |
||||
configs=['asan', 'tsan'], |
||||
platforms=['linux'], |
||||
labels=['sanitizers'], |
||||
extra_args=extra_args) |
||||
return test_jobs |
||||
|
||||
|
||||
def _create_portability_test_jobs(extra_args=[]): |
||||
test_jobs = [] |
||||
# portability C x86 |
||||
test_jobs += _generate_jobs(languages=['c'], |
||||
configs=['dbg'], |
||||
platforms=['linux'], |
||||
arch='x86', |
||||
compiler='default', |
||||
labels=['portability'], |
||||
extra_args=extra_args) |
||||
|
||||
# portability C and C++ on x64 |
||||
for compiler in ['gcc4.4', 'gcc4.6', 'gcc5.3', |
||||
'clang3.5', 'clang3.6', 'clang3.7']: |
||||
test_jobs += _generate_jobs(languages=['c', 'c++'], |
||||
configs=['dbg'], |
||||
platforms=['linux'], |
||||
arch='x64', |
||||
compiler=compiler, |
||||
labels=['portability'], |
||||
extra_args=extra_args) |
||||
|
||||
# portability C on Windows |
||||
for arch in ['x86', 'x64']: |
||||
for compiler in ['vs2013', 'vs2015']: |
||||
test_jobs += _generate_jobs(languages=['c'], |
||||
configs=['dbg'], |
||||
platforms=['windows'], |
||||
arch=arch, |
||||
compiler=compiler, |
||||
labels=['portability'], |
||||
extra_args=extra_args) |
||||
|
||||
test_jobs += _generate_jobs(languages=['python'], |
||||
configs=['dbg'], |
||||
platforms=['linux'], |
||||
arch='default', |
||||
compiler='python3.4', |
||||
labels=['portability'], |
||||
extra_args=extra_args) |
||||
|
||||
test_jobs += _generate_jobs(languages=['csharp'], |
||||
configs=['dbg'], |
||||
platforms=['linux'], |
||||
arch='default', |
||||
compiler='coreclr', |
||||
labels=['portability'], |
||||
extra_args=extra_args) |
||||
return test_jobs |
||||
|
||||
|
||||
def _allowed_labels(): |
||||
"""Returns a list of existing job labels.""" |
||||
all_labels = set() |
||||
for job in _create_test_jobs() + _create_portability_test_jobs(): |
||||
for label in job.labels: |
||||
all_labels.add(label) |
||||
return sorted(all_labels) |
||||
|
||||
|
||||
argp = argparse.ArgumentParser(description='Run a matrix of run_tests.py tests.') |
||||
argp.add_argument('-j', '--jobs', |
||||
default=multiprocessing.cpu_count()/_INNER_JOBS, |
||||
type=int, |
||||
help='Number of concurrent run_tests.py instances.') |
||||
argp.add_argument('-f', '--filter', |
||||
choices=_allowed_labels(), |
||||
nargs='+', |
||||
default=[], |
||||
help='Filter targets to run by label with AND semantics.') |
||||
argp.add_argument('--build_only', |
||||
default=False, |
||||
action='store_const', |
||||
const=True, |
||||
help='Pass --build_only flag to run_tests.py instances.') |
||||
argp.add_argument('--force_default_poller', default=False, action='store_const', const=True, |
||||
help='Pass --force_default_poller to run_tests.py instances.') |
||||
argp.add_argument('--dry_run', |
||||
default=False, |
||||
action='store_const', |
||||
const=True, |
||||
help='Only print what would be run.') |
||||
args = argp.parse_args() |
||||
|
||||
extra_args = [] |
||||
if args.build_only: |
||||
extra_args.append('--build_only') |
||||
if args.force_default_poller: |
||||
extra_args.append('--force_default_poller') |
||||
|
||||
all_jobs = _create_test_jobs(extra_args=extra_args) + _create_portability_test_jobs(extra_args=extra_args) |
||||
|
||||
jobs = [] |
||||
for job in all_jobs: |
||||
if not args.filter or all(filter in job.labels for filter in args.filter): |
||||
jobs.append(job) |
||||
|
||||
if not jobs: |
||||
jobset.message('FAILED', 'No test suites match given criteria.', |
||||
do_newline=True) |
||||
sys.exit(1) |
||||
|
||||
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 or into subworkspaces.') |
||||
|
||||
print |
||||
print 'Will run these tests:' |
||||
for job in jobs: |
||||
if args.dry_run: |
||||
print ' %s: "%s"' % (job.shortname, ' '.join(job.cmdline)) |
||||
else: |
||||
print ' %s' % job.shortname |
||||
print |
||||
|
||||
if args.dry_run: |
||||
print '--dry_run was used, exiting' |
||||
sys.exit(1) |
||||
|
||||
jobset.message('START', 'Running test matrix.', do_newline=True) |
||||
num_failures, resultset = jobset.run(jobs, |
||||
newline_on_success=True, |
||||
travis=True, |
||||
maxjobs=args.jobs) |
||||
report_utils.render_junit_xml_report(resultset, 'report.xml') |
||||
|
||||
if num_failures == 0: |
||||
jobset.message('SUCCESS', 'All run_tests.py instance finished successfully.', |
||||
do_newline=True) |
||||
else: |
||||
jobset.message('FAILED', 'Some run_tests.py instance have failed.', |
||||
do_newline=True) |
||||
sys.exit(1) |
||||
Loading…
Reference in new issue