mirror of https://github.com/grpc/grpc.git
Delete dns resolver connectivity test (#28994)
apolcyn
3 years ago
committed by
GitHub
parent
33135abbee
commit
bbe49dc16b
5 changed files with 0 additions and 370 deletions
@ -1,214 +0,0 @@ |
||||
/*
|
||||
* |
||||
* 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 <string.h> |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/support/alloc.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h" |
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/config/core_configuration.h" |
||||
#include "src/core/lib/iomgr/resolve_address.h" |
||||
#include "src/core/lib/iomgr/resolve_address_impl.h" |
||||
#include "src/core/lib/iomgr/timer.h" |
||||
#include "src/core/lib/iomgr/work_serializer.h" |
||||
#include "src/core/lib/resolver/resolver.h" |
||||
#include "src/core/lib/resolver/resolver_registry.h" |
||||
#include "src/core/lib/resolver/server_address.h" |
||||
#include "test/core/util/test_config.h" |
||||
|
||||
static gpr_mu g_mu; |
||||
static bool g_fail_resolution = true; |
||||
static std::shared_ptr<grpc_core::WorkSerializer>* g_work_serializer; |
||||
|
||||
namespace { |
||||
|
||||
class TestDNSResolver : public grpc_core::DNSResolver { |
||||
public: |
||||
class TestDNSRequest : public grpc_core::DNSResolver::Request { |
||||
public: |
||||
explicit TestDNSRequest( |
||||
std::function<void(absl::StatusOr<std::vector<grpc_resolved_address>>)> |
||||
on_done) |
||||
: on_done_(std::move(on_done)) {} |
||||
|
||||
void Start() override { |
||||
gpr_mu_lock(&g_mu); |
||||
if (g_fail_resolution) { |
||||
g_fail_resolution = false; |
||||
gpr_mu_unlock(&g_mu); |
||||
new grpc_core::DNSCallbackExecCtxScheduler( |
||||
std::move(on_done_), absl::UnknownError("Forced Failure")); |
||||
} else { |
||||
gpr_mu_unlock(&g_mu); |
||||
std::vector<grpc_resolved_address> addrs; |
||||
grpc_resolved_address phony_resolved_address; |
||||
memset(&phony_resolved_address, 0, sizeof(phony_resolved_address)); |
||||
addrs.push_back(phony_resolved_address); |
||||
new grpc_core::DNSCallbackExecCtxScheduler(std::move(on_done_), addrs); |
||||
} |
||||
} |
||||
|
||||
void Orphan() override { Unref(); } |
||||
|
||||
private: |
||||
std::function<void(absl::StatusOr<std::vector<grpc_resolved_address>>)> |
||||
on_done_; |
||||
}; |
||||
|
||||
grpc_core::OrphanablePtr<grpc_core::DNSResolver::Request> ResolveName( |
||||
absl::string_view name, absl::string_view /* default_port */, |
||||
grpc_pollset_set* /* interested_parties */, |
||||
std::function<void(absl::StatusOr<std::vector<grpc_resolved_address>>)> |
||||
on_done) override { |
||||
GPR_ASSERT("test" == name); |
||||
return grpc_core::MakeOrphanable<TestDNSRequest>(std::move(on_done)); |
||||
} |
||||
|
||||
absl::StatusOr<std::vector<grpc_resolved_address>> ResolveNameBlocking( |
||||
absl::string_view /* name */, |
||||
absl::string_view /* default_port */) override { |
||||
GPR_ASSERT(0); |
||||
} |
||||
}; |
||||
|
||||
} // namespace
|
||||
|
||||
static grpc_ares_request* my_dns_lookup_ares( |
||||
const char* /*dns_server*/, const char* addr, const char* /*default_port*/, |
||||
grpc_pollset_set* /*interested_parties*/, grpc_closure* on_done, |
||||
std::unique_ptr<grpc_core::ServerAddressList>* addresses, |
||||
std::unique_ptr<grpc_core::ServerAddressList>* /*balancer_addresses*/, |
||||
char** /*service_config_json*/, int /*query_timeout_ms*/) { // NOLINT
|
||||
gpr_mu_lock(&g_mu); |
||||
GPR_ASSERT(0 == strcmp("test", addr)); |
||||
grpc_error_handle error = GRPC_ERROR_NONE; |
||||
if (g_fail_resolution) { |
||||
g_fail_resolution = false; |
||||
gpr_mu_unlock(&g_mu); |
||||
error = GRPC_ERROR_CREATE_FROM_STATIC_STRING("Forced Failure"); |
||||
} else { |
||||
gpr_mu_unlock(&g_mu); |
||||
*addresses = absl::make_unique<grpc_core::ServerAddressList>(); |
||||
grpc_resolved_address phony_resolved_address; |
||||
memset(&phony_resolved_address, 0, sizeof(phony_resolved_address)); |
||||
phony_resolved_address.len = 123; |
||||
(*addresses)->emplace_back(phony_resolved_address, nullptr); |
||||
} |
||||
grpc_core::ExecCtx::Run(DEBUG_LOCATION, on_done, error); |
||||
return nullptr; |
||||
} |
||||
|
||||
static void my_cancel_ares_request(grpc_ares_request* request) { |
||||
GPR_ASSERT(request == nullptr); |
||||
} |
||||
|
||||
static grpc_core::OrphanablePtr<grpc_core::Resolver> create_resolver( |
||||
const char* name, |
||||
std::unique_ptr<grpc_core::Resolver::ResultHandler> result_handler) { |
||||
grpc_core::ResolverFactory* factory = grpc_core::CoreConfiguration::Get() |
||||
.resolver_registry() |
||||
.LookupResolverFactory("dns"); |
||||
absl::StatusOr<grpc_core::URI> uri = grpc_core::URI::Parse(name); |
||||
if (!uri.ok()) { |
||||
gpr_log(GPR_ERROR, "%s", uri.status().ToString().c_str()); |
||||
GPR_ASSERT(uri.ok()); |
||||
} |
||||
grpc_core::ResolverArgs args; |
||||
args.uri = std::move(*uri); |
||||
args.work_serializer = *g_work_serializer; |
||||
args.result_handler = std::move(result_handler); |
||||
grpc_core::OrphanablePtr<grpc_core::Resolver> resolver = |
||||
factory->CreateResolver(std::move(args)); |
||||
return resolver; |
||||
} |
||||
|
||||
class ResultHandler : public grpc_core::Resolver::ResultHandler { |
||||
public: |
||||
struct ResolverOutput { |
||||
grpc_core::Resolver::Result result; |
||||
gpr_event ev; |
||||
|
||||
ResolverOutput() { gpr_event_init(&ev); } |
||||
}; |
||||
|
||||
void SetOutput(ResolverOutput* output) { |
||||
gpr_atm_rel_store(&output_, reinterpret_cast<gpr_atm>(output)); |
||||
} |
||||
|
||||
void ReportResult(grpc_core::Resolver::Result result) override { |
||||
ResolverOutput* output = |
||||
reinterpret_cast<ResolverOutput*>(gpr_atm_acq_load(&output_)); |
||||
GPR_ASSERT(output != nullptr); |
||||
output->result = std::move(result); |
||||
gpr_event_set(&output->ev, reinterpret_cast<void*>(1)); |
||||
} |
||||
|
||||
private: |
||||
gpr_atm output_ = 0; // ResolverOutput*
|
||||
}; |
||||
|
||||
// interleave waiting for an event with a timer check
|
||||
static bool wait_loop(int deadline_seconds, gpr_event* ev) { |
||||
while (deadline_seconds) { |
||||
gpr_log(GPR_DEBUG, "Test: waiting for %d more seconds", deadline_seconds); |
||||
if (gpr_event_wait(ev, grpc_timeout_seconds_to_deadline(1))) return true; |
||||
deadline_seconds--; |
||||
|
||||
grpc_core::ExecCtx exec_ctx; |
||||
grpc_timer_check(nullptr); |
||||
} |
||||
return false; |
||||
} |
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc::testing::TestEnvironment env(argc, argv); |
||||
|
||||
grpc_init(); |
||||
gpr_mu_init(&g_mu); |
||||
auto work_serializer = std::make_shared<grpc_core::WorkSerializer>(); |
||||
g_work_serializer = &work_serializer; |
||||
grpc_core::SetDNSResolver(new TestDNSResolver()); |
||||
grpc_dns_lookup_ares = my_dns_lookup_ares; |
||||
grpc_cancel_ares_request = my_cancel_ares_request; |
||||
|
||||
{ |
||||
grpc_core::ExecCtx exec_ctx; |
||||
ResultHandler* result_handler = new ResultHandler(); |
||||
grpc_core::OrphanablePtr<grpc_core::Resolver> resolver = create_resolver( |
||||
"dns:test", |
||||
std::unique_ptr<grpc_core::Resolver::ResultHandler>(result_handler)); |
||||
ResultHandler::ResolverOutput output1; |
||||
result_handler->SetOutput(&output1); |
||||
resolver->StartLocked(); |
||||
grpc_core::ExecCtx::Get()->Flush(); |
||||
GPR_ASSERT(wait_loop(5, &output1.ev)); |
||||
GPR_ASSERT(!output1.result.addresses.ok()); |
||||
|
||||
ResultHandler::ResolverOutput output2; |
||||
result_handler->SetOutput(&output2); |
||||
grpc_core::ExecCtx::Get()->Flush(); |
||||
GPR_ASSERT(wait_loop(30, &output2.ev)); |
||||
GPR_ASSERT(output2.result.addresses.ok()); |
||||
GPR_ASSERT(!output2.result.addresses->empty()); |
||||
} |
||||
|
||||
grpc_shutdown(); |
||||
gpr_mu_destroy(&g_mu); |
||||
} |
||||
Loading…
Reference in new issue