<%def name="resolver_component_tests(tests)">#!/usr/bin/env python3 # 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 os import platform import signal import subprocess import sys import tempfile import time 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.')) argp.add_argument('--extra_args', default='', type=str, help=('Comma-separate list of command args to ' 'plumb through to --test_bin_path')) 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' and arg_list[0].endswith('.py'): 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,cares_address_sorting'}) experiments = os.environ.get('GRPC_EXPERIMENTS') if experiments is not None and 'event_engine_dns' in experiments: os.environ.update({'GRPC_TRACE': 'event_engine_client_channel_resolver,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(str.encode('ascii')) if dns_resolver_subprocess.returncode == 0: if '123.123.123.123'.encode('ascii') 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 ] + args.extra_args.split(','))\ 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)