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.
94 lines
3.3 KiB
94 lines
3.3 KiB
<%def name="resolver_component_tests(tests)">#!/bin/bash |
|
# 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 |
|
|
|
set -ex |
|
|
|
# all command args required in this set order |
|
FLAGS_test_bin_path=$(echo "$1" | grep '\--test_bin_path=' | cut -d "=" -f 2) |
|
FLAGS_dns_server_bin_path=$(echo "$2" | grep '\--dns_server_bin_path=' | cut -d "=" -f 2) |
|
FLAGS_records_config_path=$(echo "$3" | grep '\--records_config_path=' | cut -d "=" -f 2) |
|
FLAGS_dns_server_port=$(echo "$4" | grep '\--dns_server_port=' | cut -d "=" -f 2) |
|
FLAGS_dns_resolver_bin_path=$(echo "$5" | grep '\--dns_resolver_bin_path=' | cut -d "=" -f 2) |
|
FLAGS_tcp_connect_bin_path=$(echo "$6" | grep '\--tcp_connect_bin_path=' | cut -d "=" -f 2) |
|
|
|
for cmd_arg in "$FLAGS_test_bin_path" "$FLAGS_dns_server_bin_path" "$FLAGS_records_config_path" "$FLAGS_dns_server_port" "$FLAGS_dns_resolver_bin_path" "$FLAGS_tcp_connect_bin_path"; do |
|
if [[ "$cmd_arg" == "" ]]; then |
|
echo "Missing a CMD arg" && exit 1 |
|
fi |
|
done |
|
|
|
if [[ "$GRPC_DNS_RESOLVER" != "" && "$GRPC_DNS_RESOLVER" != ares ]]; then |
|
echo "This test only works under GRPC_DNS_RESOLVER=ares. Have GRPC_DNS_RESOLVER=$GRPC_DNS_RESOLVER" && exit 1 |
|
fi |
|
export GRPC_DNS_RESOLVER=ares |
|
|
|
"$FLAGS_dns_server_bin_path" --records_config_path="$FLAGS_records_config_path" --port="$FLAGS_dns_server_port" > /dev/null 2>&1 & |
|
DNS_SERVER_PID=$! |
|
echo "Local DNS server started. PID: $DNS_SERVER_PID" |
|
|
|
# Health check local DNS server TCP and UDP ports |
|
for ((i=0;i<30;i++)); |
|
do |
|
echo "Retry health-check local DNS server by attempting a DNS query and TCP handshake" |
|
RETRY=0 |
|
$FLAGS_dns_resolver_bin_path -s 127.0.0.1 -p "$FLAGS_dns_server_port" -n health-check-local-dns-server-is-alive.resolver-tests.grpctestingexp. -t 1 | grep '123.123.123.123' || RETRY=1 |
|
$FLAGS_tcp_connect_bin_path -s 127.0.0.1 -p "$FLAGS_dns_server_port" -t 1 || RETRY=1 |
|
if [[ "$RETRY" == 0 ]]; then |
|
break |
|
fi; |
|
sleep 0.1 |
|
done |
|
|
|
if [[ $RETRY == 1 ]]; then |
|
echo "FAILED TO START LOCAL DNS SERVER" |
|
kill -SIGTERM "$DNS_SERVER_PID" |
|
wait |
|
exit 1 |
|
fi |
|
|
|
function terminate_all { |
|
echo "Received signal. Terminating $! and $DNS_SERVER_PID" |
|
kill -SIGTERM "$!" || true |
|
kill -SIGTERM "$DNS_SERVER_PID" || true |
|
wait |
|
exit 1 |
|
} |
|
|
|
trap terminate_all SIGTERM SIGINT |
|
|
|
EXIT_CODE=0 |
|
# TODO: this test should check for GCE residency and skip tests using _grpclb._tcp.* SRV records once GCE residency checks are made |
|
# in the resolver. |
|
|
|
% for test in tests: |
|
$FLAGS_test_bin_path \\ |
|
|
|
--target_name='${test['target_name']}' \\ |
|
|
|
--expected_addrs='${test['expected_addrs']}' \\ |
|
|
|
--expected_chosen_service_config='${test['expected_chosen_service_config']}' \\ |
|
|
|
--expected_lb_policy='${test['expected_lb_policy']}' \\ |
|
|
|
--local_dns_server_address="127.0.0.1:$FLAGS_dns_server_port" & |
|
wait "$!" || EXIT_CODE=1 |
|
|
|
% endfor |
|
kill -SIGTERM "$DNS_SERVER_PID" || true |
|
wait |
|
exit $EXIT_CODE</%def>
|
|
|