[testing] add test client for use in XDS federation tests (#32020)
[testing] add test client for use in XDS federation testspull/32037/head
apolcyn
2 years ago
committed by
GitHub
parent
3a3f7ebc25
commit
c952d37814
5 changed files with 179 additions and 29 deletions
@ -0,0 +1,122 @@ |
||||
//
|
||||
//
|
||||
// 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.
|
||||
//
|
||||
//
|
||||
|
||||
#include <memory> |
||||
#include <thread> |
||||
|
||||
#include "absl/flags/flag.h" |
||||
#include "absl/strings/str_split.h" |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
|
||||
#include "src/core/lib/gpr/string.h" |
||||
#include "test/core/util/test_config.h" |
||||
#include "test/cpp/interop/client_helper.h" |
||||
#include "test/cpp/interop/interop_client.h" |
||||
#include "test/cpp/util/create_test_channel.h" |
||||
#include "test/cpp/util/test_config.h" |
||||
|
||||
ABSL_FLAG(std::string, server_uris, "", |
||||
"Comma-separated list of sever URIs to make RPCs to"); |
||||
ABSL_FLAG(std::string, credentials_types, "", |
||||
"Comma-separated list of credentials, each entry is used for the " |
||||
"server of the corresponding index in server_uris. Supported values: " |
||||
"compute_engine_channel_creds, INSECURE_CREDENTIALS"); |
||||
ABSL_FLAG(int32_t, soak_iterations, 10, |
||||
"The number of iterations to use for the two soak tests: rpc_soak " |
||||
"and channel_soak"); |
||||
ABSL_FLAG(int32_t, soak_max_failures, 0, |
||||
"The number of iterations in soak tests that are allowed to fail " |
||||
"(either due to non-OK status code or exceeding the per-iteration " |
||||
"max acceptable latency)."); |
||||
ABSL_FLAG(int32_t, soak_per_iteration_max_acceptable_latency_ms, 1000, |
||||
"The number of milliseconds a single iteration in the two soak tests " |
||||
"(rpc_soak and channel_soak) should take."); |
||||
ABSL_FLAG( |
||||
int32_t, soak_overall_timeout_seconds, 10, |
||||
"The overall number of seconds after which a soak test should stop and " |
||||
"fail, if the desired number of iterations have not yet completed."); |
||||
ABSL_FLAG(int32_t, soak_min_time_ms_between_rpcs, 0, |
||||
"The minimum time in milliseconds between consecutive RPCs in a soak " |
||||
"test (rpc_soak or channel_soak), useful for limiting QPS"); |
||||
ABSL_FLAG(std::string, test_case, "rpc_soak", |
||||
"Configure different test cases. Valid options are: " |
||||
"rpc_soak: sends --soak_iterations large_unary RPCs; " |
||||
"channel_soak: sends --soak_iterations RPCs, rebuilding the channel " |
||||
"each time"); |
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc::testing::TestEnvironment env(&argc, argv); |
||||
grpc::testing::InitTest(&argc, &argv, true); |
||||
gpr_log(GPR_INFO, "Testing these cases: %s", |
||||
absl::GetFlag(FLAGS_test_case).c_str()); |
||||
std::string test_case = absl::GetFlag(FLAGS_test_case); |
||||
// validate flags
|
||||
std::vector<std::string> uris = |
||||
absl::StrSplit(absl::GetFlag(FLAGS_server_uris), ','); |
||||
std::vector<std::string> creds = |
||||
absl::StrSplit(absl::GetFlag(FLAGS_credentials_types), ','); |
||||
if (uris.size() != creds.size()) { |
||||
gpr_log(GPR_ERROR, |
||||
"Number of entries in --server_uris %ld != number of entries in " |
||||
"--credentials_types %ld", |
||||
uris.size(), creds.size()); |
||||
GPR_ASSERT(0); |
||||
} |
||||
if (uris.empty()) { |
||||
gpr_log(GPR_ERROR, "--server_uris has zero entries"); |
||||
GPR_ASSERT(0); |
||||
} |
||||
// construct and start clients
|
||||
std::vector<std::thread> threads; |
||||
for (size_t i = 0; i < uris.size(); i++) { |
||||
threads.push_back(std::thread([uris, creds, i, test_case]() { |
||||
auto channel_creation_func = [uris, creds, i]() { |
||||
return grpc::CreateTestChannel(uris[i], creds[i], |
||||
nullptr /* call creds */); |
||||
}; |
||||
grpc::testing::InteropClient client(channel_creation_func, true, false); |
||||
if (test_case == "rpc_soak") { |
||||
client.DoRpcSoakTest( |
||||
uris[i], absl::GetFlag(FLAGS_soak_iterations), |
||||
absl::GetFlag(FLAGS_soak_max_failures), |
||||
absl::GetFlag(FLAGS_soak_per_iteration_max_acceptable_latency_ms), |
||||
absl::GetFlag(FLAGS_soak_min_time_ms_between_rpcs), |
||||
absl::GetFlag(FLAGS_soak_overall_timeout_seconds)); |
||||
} else if (test_case == "channel_soak") { |
||||
client.DoChannelSoakTest( |
||||
uris[i], absl::GetFlag(FLAGS_soak_iterations), |
||||
absl::GetFlag(FLAGS_soak_max_failures), |
||||
absl::GetFlag(FLAGS_soak_per_iteration_max_acceptable_latency_ms), |
||||
absl::GetFlag(FLAGS_soak_min_time_ms_between_rpcs), |
||||
absl::GetFlag(FLAGS_soak_overall_timeout_seconds)); |
||||
} else { |
||||
gpr_log(GPR_ERROR, |
||||
"Invalid test case, must be either rpc_soak or channel_soak"); |
||||
GPR_ASSERT(0); |
||||
} |
||||
})); |
||||
} |
||||
for (auto& thd : threads) { |
||||
thd.join(); |
||||
} |
||||
gpr_log(GPR_INFO, "All clients done!"); |
||||
return 0; |
||||
} |
||||
Loading…
Reference in new issue