test/cpp/naming/resolver_component_tests_runner_invoker.cc

   1 /*
   2  *
   3  * Copyright 2017 gRPC authors.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 #include <signal.h>
  20 #include <string.h>
  21 #include <unistd.h>
  22
  23 #include <string>
  24 #include <thread>
  25 #include <vector>
  26
  27 #include "absl/flags/flag.h"
  28
  29 #include <grpc/grpc.h>
  30 #include <grpc/support/alloc.h>
  31 #include <grpc/support/log.h>
  32 #include <grpc/support/string_util.h>
  33
  34 #ifdef __FreeBSD__
  35 #include <sys/wait.h>
  36 #endif
  37
  38 #include "src/core/lib/gpr/env.h"
  39 #include "test/core/util/port.h"
  40 #include "test/core/util/test_config.h"
  41 #include "test/cpp/util/subprocess.h"
  42 #include "test/cpp/util/test_config.h"
  43
  44 ABSL_FLAG(
  45     bool, running_under_bazel, false,
  46     "True if this test is running under bazel. "
  47     "False indicates that this test is running under run_tests.py. "
  48     "Child process test binaries are located differently based on this flag. ");
  49
  50 ABSL_FLAG(std::string, test_bin_name, "",
  51           "Name, without the preceding path, of the test binary");
  52
  53 ABSL_FLAG(std::string, grpc_test_directory_relative_to_test_srcdir,
  54           "/com_github_grpc_grpc",
  55           "This flag only applies if runner_under_bazel is true. This "
  56           "flag is ignored if runner_under_bazel is false. "
  57           "Directory of the <repo-root>/test directory relative to bazel's "
  58           "TEST_SRCDIR environment variable");
  59
  60 ABSL_FLAG(std::string, extra_args, "",
  61           "Comma-separated list of opaque command args to plumb through to "
  62           "the binary pointed at by --test_bin_name");
  63
  64 using grpc::SubProcess;
  65
  66 namespace grpc {
  67
  68 namespace testing {
  69
  70 void InvokeResolverComponentTestsRunner(
  71     std::string test_runner_bin_path, const std::string& test_bin_path,
  72     const std::string& dns_server_bin_path,
  73     const std::string& records_config_path,
  74     const std::string& dns_resolver_bin_path,
  75     const std::string& tcp_connect_bin_path) {
  76   int dns_server_port = grpc_pick_unused_port_or_die();
  77
  78   SubProcess* test_driver = new SubProcess(
  79       {std::move(test_runner_bin_path), "--test_bin_path=" + test_bin_path,
  80        "--dns_server_bin_path=" + dns_server_bin_path,
  81        "--records_config_path=" + records_config_path,
  82        "--dns_server_port=" + std::to_string(dns_server_port),
  83        "--dns_resolver_bin_path=" + dns_resolver_bin_path,
  84        "--tcp_connect_bin_path=" + tcp_connect_bin_path,
  85        "--extra_args=" + absl::GetFlag(FLAGS_extra_args)});
  86   gpr_mu test_driver_mu;
  87   gpr_mu_init(&test_driver_mu);
  88   gpr_cv test_driver_cv;
  89   gpr_cv_init(&test_driver_cv);
  90   int test_driver_done = 0;
  91   int status = test_driver->Join();
  92   if (WIFEXITED(status)) {
  93     if (WEXITSTATUS(status)) {
  94       gpr_log(GPR_INFO,
  95               "Resolver component test test-runner exited with code %d",
  96               WEXITSTATUS(status));
  97       abort();
  98     }
  99   } else if (WIFSIGNALED(status)) {
 100     gpr_log(GPR_INFO,
 101             "Resolver component test test-runner ended from signal %d",
 102             WTERMSIG(status));
 103     abort();
 104   } else {
 105     gpr_log(GPR_INFO,
 106             "Resolver component test test-runner ended with unknown status %d",
 107             status);
 108     abort();
 109   }
 110   gpr_mu_lock(&test_driver_mu);
 111   test_driver_done = 1;
 112   gpr_cv_signal(&test_driver_cv);
 113   gpr_mu_unlock(&test_driver_mu);
 114   delete test_driver;
 115   gpr_mu_destroy(&test_driver_mu);
 116   gpr_cv_destroy(&test_driver_cv);
 117 }
 118
 119 }  // namespace testing
 120
 121 }  // namespace grpc
 122
 123 int main(int argc, char** argv) {
 124   grpc::testing::TestEnvironment env(argc, argv);
 125   grpc::testing::InitTest(&argc, &argv, true);
 126   grpc_init();
 127   GPR_ASSERT(!absl::GetFlag(FLAGS_test_bin_name).empty());
 128   std::string my_bin = argv[0];
 129   if (absl::GetFlag(FLAGS_running_under_bazel)) {
 130     GPR_ASSERT(!absl::GetFlag(FLAGS_grpc_test_directory_relative_to_test_srcdir)
 131                     .empty());
 132     // Use bazel's TEST_SRCDIR environment variable to locate the "test data"
 133     // binaries.
 134     char* test_srcdir = gpr_getenv("TEST_SRCDIR");
 135     std::string const bin_dir =
 136         test_srcdir +
 137         absl::GetFlag(FLAGS_grpc_test_directory_relative_to_test_srcdir) +
 138         std::string("/test/cpp/naming");
 139     // Invoke bazel's executeable links to the .sh and .py scripts (don't use
 140     // the .sh and .py suffixes) to make
 141     // sure that we're using bazel's test environment.
 142     grpc::testing::InvokeResolverComponentTestsRunner(
 143         bin_dir + "/resolver_component_tests_runner",
 144         bin_dir + "/" + absl::GetFlag(FLAGS_test_bin_name),
 145         bin_dir + "/utils/dns_server",
 146         bin_dir + "/resolver_test_record_groups.yaml",
 147         bin_dir + "/utils/dns_resolver", bin_dir + "/utils/tcp_connect");
 148     gpr_free(test_srcdir);
 149   } else {
 150     // Get the current binary's directory relative to repo root to invoke the
 151     // correct build config (asan/tsan/dbg, etc.).
 152     std::string const bin_dir = my_bin.substr(0, my_bin.rfind('/'));
 153     // Invoke the .sh and .py scripts directly where they are in source code.
 154     grpc::testing::InvokeResolverComponentTestsRunner(
 155         "test/cpp/naming/resolver_component_tests_runner.py",
 156         bin_dir + "/" + absl::GetFlag(FLAGS_test_bin_name),
 157         "test/cpp/naming/utils/dns_server.py",
 158         "test/cpp/naming/resolver_test_record_groups.yaml",
 159         "test/cpp/naming/utils/dns_resolver.py",
 160         "test/cpp/naming/utils/tcp_connect.py");
 161   }
 162   grpc_shutdown();
 163   return 0;
 164 }