mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
293 lines
15 KiB
293 lines
15 KiB
#!/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. |
|
|
|
# This file is auto-generated |
|
|
|
import argparse |
|
import sys |
|
import subprocess |
|
import tempfile |
|
import os |
|
import time |
|
import signal |
|
import platform |
|
|
|
|
|
argp = argparse.ArgumentParser(description='Run c-ares resolver tests') |
|
argp.add_argument('--test_bin_path', default=None, type=str, |
|
help='Path to gtest test binary to invoke.') |
|
argp.add_argument('--dns_server_bin_path', default=None, type=str, |
|
help='Path to local DNS server python script.') |
|
argp.add_argument('--records_config_path', default=None, type=str, |
|
help=('Path to DNS records yaml file that ' |
|
'specifies records for the DNS sever. ')) |
|
argp.add_argument('--dns_server_port', default=None, type=int, |
|
help=('Port that local DNS server is listening on.')) |
|
argp.add_argument('--dns_resolver_bin_path', default=None, type=str, |
|
help=('Path to the DNS health check utility.')) |
|
argp.add_argument('--tcp_connect_bin_path', default=None, type=str, |
|
help=('Path to the TCP health check utility.')) |
|
args = argp.parse_args() |
|
|
|
def test_runner_log(msg): |
|
sys.stderr.write('\n%s: %s\n' % (__file__, msg)) |
|
|
|
def python_args(arg_list): |
|
if platform.system() == 'Windows': |
|
return [sys.executable] + arg_list |
|
return arg_list |
|
|
|
cur_resolver = os.environ.get('GRPC_DNS_RESOLVER') |
|
if cur_resolver and cur_resolver != 'ares': |
|
test_runner_log(('WARNING: cur resolver set to %s. This set of tests ' |
|
'needs to use GRPC_DNS_RESOLVER=ares.')) |
|
test_runner_log('Exit 1 without running tests.') |
|
sys.exit(1) |
|
os.environ.update({'GRPC_TRACE': 'cares_resolver'}) |
|
|
|
def wait_until_dns_server_is_up(args, |
|
dns_server_subprocess, |
|
dns_server_subprocess_output): |
|
for i in range(0, 30): |
|
test_runner_log('Health check: attempt to connect to DNS server over TCP.') |
|
tcp_connect_subprocess = subprocess.Popen(python_args([ |
|
args.tcp_connect_bin_path, |
|
'--server_host', '127.0.0.1', |
|
'--server_port', str(args.dns_server_port), |
|
'--timeout', str(1)])) |
|
tcp_connect_subprocess.communicate() |
|
if tcp_connect_subprocess.returncode == 0: |
|
test_runner_log(('Health check: attempt to make an A-record ' |
|
'query to DNS server.')) |
|
dns_resolver_subprocess = subprocess.Popen(python_args([ |
|
args.dns_resolver_bin_path, |
|
'--qname', 'health-check-local-dns-server-is-alive.resolver-tests.grpctestingexp', |
|
'--server_host', '127.0.0.1', |
|
'--server_port', str(args.dns_server_port)]), |
|
stdout=subprocess.PIPE) |
|
dns_resolver_stdout, _ = dns_resolver_subprocess.communicate() |
|
if dns_resolver_subprocess.returncode == 0: |
|
if '123.123.123.123' in dns_resolver_stdout: |
|
test_runner_log(('DNS server is up! ' |
|
'Successfully reached it over UDP and TCP.')) |
|
return |
|
time.sleep(0.1) |
|
dns_server_subprocess.kill() |
|
dns_server_subprocess.wait() |
|
test_runner_log(('Failed to reach DNS server over TCP and/or UDP. ' |
|
'Exitting without running tests.')) |
|
test_runner_log('======= DNS server stdout ' |
|
'(merged stdout and stderr) =============') |
|
with open(dns_server_subprocess_output, 'r') as l: |
|
test_runner_log(l.read()) |
|
test_runner_log('======= end DNS server output=========') |
|
sys.exit(1) |
|
|
|
dns_server_subprocess_output = tempfile.mktemp() |
|
with open(dns_server_subprocess_output, 'w') as l: |
|
dns_server_subprocess = subprocess.Popen(python_args([ |
|
args.dns_server_bin_path, |
|
'--port', str(args.dns_server_port), |
|
'--records_config_path', args.records_config_path]), |
|
stdin=subprocess.PIPE, |
|
stdout=l, |
|
stderr=l) |
|
|
|
def _quit_on_signal(signum, _frame): |
|
test_runner_log('Received signal: %d' % signum) |
|
dns_server_subprocess.kill() |
|
dns_server_subprocess.wait() |
|
sys.exit(1) |
|
|
|
signal.signal(signal.SIGINT, _quit_on_signal) |
|
signal.signal(signal.SIGTERM, _quit_on_signal) |
|
wait_until_dns_server_is_up(args, |
|
dns_server_subprocess, |
|
dns_server_subprocess_output) |
|
num_test_failures = 0 |
|
|
|
test_runner_log('Run test with target: %s' % 'no-srv-ipv4-single-target.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'no-srv-ipv4-single-target.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '5.5.5.5:443,False', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv4-single-target.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv4-single-target.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:1234,True', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv4-multi-target.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv4-multi-target.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.5:1234,True;1.2.3.6:1234,True;1.2.3.7:1234,True', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv6-single-target.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv6-single-target.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '[2607:f8b0:400a:801::1001]:1234,True', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv6-multi-target.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv6-multi-target.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '[2607:f8b0:400a:801::1002]:1234,True;[2607:f8b0:400a:801::1003]:1234,True;[2607:f8b0:400a:801::1004]:1234,True', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv4-simple-service-config.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv4-simple-service-config.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:1234,True', |
|
'--expected_chosen_service_config', '{"loadBalancingPolicy":"round_robin","methodConfig":[{"name":[{"method":"Foo","service":"SimpleService","waitForReady":true}]}]}', |
|
'--expected_lb_policy', 'round_robin', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'ipv4-no-srv-simple-service-config.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'ipv4-no-srv-simple-service-config.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '{"loadBalancingPolicy":"round_robin","methodConfig":[{"name":[{"method":"Foo","service":"NoSrvSimpleService","waitForReady":true}]}]}', |
|
'--expected_lb_policy', 'round_robin', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'ipv4-no-config-for-cpp.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'ipv4-no-config-for-cpp.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'ipv4-cpp-config-has-zero-percentage.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'ipv4-cpp-config-has-zero-percentage.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'ipv4-second-language-is-cpp.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'ipv4-second-language-is-cpp.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '{"loadBalancingPolicy":"round_robin","methodConfig":[{"name":[{"method":"Foo","service":"CppService","waitForReady":true}]}]}', |
|
'--expected_lb_policy', 'round_robin', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'ipv4-config-with-percentages.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'ipv4-config-with-percentages.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '{"loadBalancingPolicy":"round_robin","methodConfig":[{"name":[{"method":"Foo","service":"AlwaysPickedService","waitForReady":true}]}]}', |
|
'--expected_lb_policy', 'round_robin', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv4-target-has-backend-and-balancer.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv4-target-has-backend-and-balancer.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:1234,True;1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'srv-ipv6-target-has-backend-and-balancer.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'srv-ipv6-target-has-backend-and-balancer.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '[2607:f8b0:400a:801::1002]:1234,True;[2607:f8b0:400a:801::1002]:443,False', |
|
'--expected_chosen_service_config', '', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('Run test with target: %s' % 'ipv4-config-causing-fallback-to-tcp.resolver-tests-version-4.grpctestingexp.') |
|
current_test_subprocess = subprocess.Popen([ |
|
args.test_bin_path, |
|
'--target_name', 'ipv4-config-causing-fallback-to-tcp.resolver-tests-version-4.grpctestingexp.', |
|
'--expected_addrs', '1.2.3.4:443,False', |
|
'--expected_chosen_service_config', '{"loadBalancingPolicy":"round_robin","methodConfig":[{"name":[{"method":"Foo","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooTwo","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooThree","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooFour","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooFive","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooSix","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooSeven","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooEight","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooNine","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooTen","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooEleven","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooTwelve","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooTwelve","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooTwelve","service":"SimpleService","waitForReady":true}]},{"name":[{"method":"FooTwelve","service":"SimpleService","waitForReady":true}]}]}', |
|
'--expected_lb_policy', '', |
|
'--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port]) |
|
current_test_subprocess.communicate() |
|
if current_test_subprocess.returncode != 0: |
|
num_test_failures += 1 |
|
|
|
test_runner_log('now kill DNS server') |
|
dns_server_subprocess.kill() |
|
dns_server_subprocess.wait() |
|
test_runner_log('%d tests failed.' % num_test_failures) |
|
sys.exit(num_test_failures)
|
|
|