1 <%def name="resolver_component_tests(tests)">#!/usr/bin/env python 2 # Copyright 2015 gRPC authors. 3 # 4 # Licensed under the Apache License, Version 2.0 (the "License"); 5 # you may not use this file except in compliance with the License. 6 # You may obtain a copy of the License at 7 # 8 # http://www.apache.org/licenses/LICENSE-2.0 9 # 10 # Unless required by applicable law or agreed to in writing, software 11 # distributed under the License is distributed on an "AS IS" BASIS, 12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 # See the License for the specific language governing permissions and 14 # limitations under the License. 15 16 # This file is auto-generated 17 18 import argparse 19 import sys 20 import subprocess 21 import tempfile 22 import os 23 import time 24 import signal 25 import platform 26 27 28 argp = argparse.ArgumentParser(description='Run c-ares resolver tests') 29 argp.add_argument('--test_bin_path', default=None, type=str, 30 help='Path to gtest test binary to invoke.') 31 argp.add_argument('--dns_server_bin_path', default=None, type=str, 32 help='Path to local DNS server python script.') 33 argp.add_argument('--records_config_path', default=None, type=str, 34 help=('Path to DNS records yaml file that ' 35 'specifies records for the DNS sever. ')) 36 argp.add_argument('--dns_server_port', default=None, type=int, 37 help=('Port that local DNS server is listening on.')) 38 argp.add_argument('--dns_resolver_bin_path', default=None, type=str, 39 help=('Path to the DNS health check utility.')) 40 argp.add_argument('--tcp_connect_bin_path', default=None, type=str, 41 help=('Path to the TCP health check utility.')) 42 args = argp.parse_args() 43 44 def test_runner_log(msg): 45 sys.stderr.write('\n%s: %s\n' % (__file__, msg)) 46 47 def python_args(arg_list): 48 if platform.system() == 'Windows': 49 return [sys.executable] + arg_list 50 return arg_list 51 52 cur_resolver = os.environ.get('GRPC_DNS_RESOLVER') 53 if cur_resolver and cur_resolver != 'ares': 54 test_runner_log(('WARNING: cur resolver set to %s. This set of tests ' 55 'needs to use GRPC_DNS_RESOLVER=ares.')) 56 test_runner_log('Exit 1 without running tests.') 57 sys.exit(1) 58 os.environ.update({'GRPC_DNS_RESOLVER': 'ares'}) 59 os.environ.update({'GRPC_TRACE': 'cares_resolver'}) 60 61 def wait_until_dns_server_is_up(args, 62 dns_server_subprocess, 63 dns_server_subprocess_output): 64 for i in range(0, 30): 65 test_runner_log('Health check: attempt to connect to DNS server over TCP.') 66 tcp_connect_subprocess = subprocess.Popen(python_args([ 67 args.tcp_connect_bin_path, 68 '--server_host', '127.0.0.1', 69 '--server_port', str(args.dns_server_port), 70 '--timeout', str(1)])) 71 tcp_connect_subprocess.communicate() 72 if tcp_connect_subprocess.returncode == 0: 73 test_runner_log(('Health check: attempt to make an A-record ' 74 'query to DNS server.')) 75 dns_resolver_subprocess = subprocess.Popen(python_args([ 76 args.dns_resolver_bin_path, 77 '--qname', 'health-check-local-dns-server-is-alive.resolver-tests.grpctestingexp', 78 '--server_host', '127.0.0.1', 79 '--server_port', str(args.dns_server_port)]), 80 stdout=subprocess.PIPE) 81 dns_resolver_stdout, _ = dns_resolver_subprocess.communicate() 82 if dns_resolver_subprocess.returncode == 0: 83 if '123.123.123.123' in dns_resolver_stdout: 84 test_runner_log(('DNS server is up! ' 85 'Successfully reached it over UDP and TCP.')) 86 return 87 time.sleep(0.1) 88 dns_server_subprocess.kill() 89 dns_server_subprocess.wait() 90 test_runner_log(('Failed to reach DNS server over TCP and/or UDP. ' 91 'Exitting without running tests.')) 92 test_runner_log('======= DNS server stdout ' 93 '(merged stdout and stderr) =============') 94 with open(dns_server_subprocess_output, 'r') as l: 95 test_runner_log(l.read()) 96 test_runner_log('======= end DNS server output=========') 97 sys.exit(1) 98 99 dns_server_subprocess_output = tempfile.mktemp() 100 with open(dns_server_subprocess_output, 'w') as l: 101 dns_server_subprocess = subprocess.Popen(python_args([ 102 args.dns_server_bin_path, 103 '--port', str(args.dns_server_port), 104 '--records_config_path', args.records_config_path]), 105 stdin=subprocess.PIPE, 106 stdout=l, 107 stderr=l) 108 109 def _quit_on_signal(signum, _frame): 110 test_runner_log('Received signal: %d' % signum) 111 dns_server_subprocess.kill() 112 dns_server_subprocess.wait() 113 sys.exit(1) 114 115 signal.signal(signal.SIGINT, _quit_on_signal) 116 signal.signal(signal.SIGTERM, _quit_on_signal) 117 wait_until_dns_server_is_up(args, 118 dns_server_subprocess, 119 dns_server_subprocess_output) 120 num_test_failures = 0 121 122 % for test in tests: 123 test_runner_log('Run test with target: %s' % '${test['test_title']}')\ 124 125 current_test_subprocess = subprocess.Popen([\ 126 127 args.test_bin_path,\ 128 129 % for arg_name_and_value in test['arg_names_and_values']: 130 \ 131 '--${arg_name_and_value[0]}', '${arg_name_and_value[1]}',\ 132 133 \ 134 % endfor 135 '--local_dns_server_address', '127.0.0.1:%d' % args.dns_server_port])\ 136 137 current_test_subprocess.communicate()\ 138 139 if current_test_subprocess.returncode != 0:\ 140 141 num_test_failures += 1 142 143 % endfor 144 test_runner_log('now kill DNS server') 145 dns_server_subprocess.kill() 146 dns_server_subprocess.wait() 147 test_runner_log('%d tests failed.' % num_test_failures) 148 sys.exit(num_test_failures)</%def> 149