|
|
|
<%def name="resolver_component_tests(tests)">#!/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_DNS_RESOLVER': 'ares'})
|
|
|
|
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
|
|
|
|
|
|
|
|
% for test in tests:
|
|
|
|
test_runner_log('Run test with target: %s' % '${test['test_title']}')\
|
|
|
|
|
|
|
|
current_test_subprocess = subprocess.Popen([\
|
|
|
|
|
|
|
|
args.test_bin_path,\
|
|
|
|
|
|
|
|
% for arg_name_and_value in test['arg_names_and_values']:
|
|
|
|
\
|
|
|
|
'--${arg_name_and_value[0]}', '${arg_name_and_value[1]}',\
|
|
|
|
|
|
|
|
\
|
|
|
|
% endfor
|
|
|
|
'--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
|
|
|
|
|
|
|
|
% endfor
|
|
|
|
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)</%def>
|