Make changes to fix test failures

pull/18371/head
Karthik Ravi Shankar 6 years ago
parent e27ddbc87e
commit b25f6da3f0
  1. 1
      include/grpcpp/server.h
  2. 2
      src/cpp/server/health/default_health_check_service.h
  3. 6
      src/cpp/server/health/health_check_service.cc
  4. 2
      src/cpp/server/server_cc.cc
  5. 8
      test/cpp/end2end/client_lb_end2end_test.cc
  6. 26
      test/cpp/end2end/health_service_end2end_test.cc

@ -28,6 +28,7 @@
#include <grpc/compression.h> #include <grpc/compression.h>
#include <grpc/support/atm.h> #include <grpc/support/atm.h>
#include <grpcpp/completion_queue.h> #include <grpcpp/completion_queue.h>
#include <grpcpp/health_check_service_interface.h>
#include <grpcpp/impl/call.h> #include <grpcpp/impl/call.h>
#include <grpcpp/impl/codegen/client_interceptor.h> #include <grpcpp/impl/codegen/client_interceptor.h>
#include <grpcpp/impl/codegen/grpc_library.h> #include <grpcpp/impl/codegen/grpc_library.h>

@ -37,7 +37,7 @@ namespace grpc {
// Default implementation of HealthCheckServiceInterface. Server will create and // Default implementation of HealthCheckServiceInterface. Server will create and
// own it. // own it.
class DefaultHealthCheckService final : public HealthCheckServiceInterface { class DefaultHealthCheckService final : public grpc_impl::HealthCheckServiceInterface {
public: public:
enum ServingStatus { NOT_FOUND, SERVING, NOT_SERVING }; enum ServingStatus { NOT_FOUND, SERVING, NOT_SERVING };

@ -16,9 +16,9 @@
* *
*/ */
#include <grpcpp/health_check_service_interface.h> #include <grpcpp/health_check_service_interface_impl.h>
namespace grpc { namespace grpc_impl {
namespace { namespace {
bool g_grpc_default_health_check_service_enabled = false; bool g_grpc_default_health_check_service_enabled = false;
} // namespace } // namespace
@ -31,4 +31,4 @@ void EnableDefaultHealthCheckService(bool enable) {
g_grpc_default_health_check_service_enabled = enable; g_grpc_default_health_check_service_enabled = enable;
} }
} // namespace grpc } // namespace grpc_impl

@ -802,7 +802,7 @@ Server::Server(
if (channel_args.args[i].value.pointer.p == nullptr) { if (channel_args.args[i].value.pointer.p == nullptr) {
health_check_service_disabled_ = true; health_check_service_disabled_ = true;
} else { } else {
health_check_service_.reset(static_cast<HealthCheckServiceInterface*>( health_check_service_.reset(static_cast<grpc_impl::HealthCheckServiceInterface*>(
channel_args.args[i].value.pointer.p)); channel_args.args[i].value.pointer.p));
} }
break; break;

@ -1242,7 +1242,7 @@ TEST_F(ClientLbEnd2endTest,
} }
TEST_F(ClientLbEnd2endTest, RoundRobinWithHealthChecking) { TEST_F(ClientLbEnd2endTest, RoundRobinWithHealthChecking) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
// Start servers. // Start servers.
const int kNumServers = 3; const int kNumServers = 3;
StartServers(kNumServers); StartServers(kNumServers);
@ -1311,11 +1311,11 @@ TEST_F(ClientLbEnd2endTest, RoundRobinWithHealthChecking) {
EXPECT_TRUE(WaitForChannelNotReady(channel.get())); EXPECT_TRUE(WaitForChannelNotReady(channel.get()));
CheckRpcSendFailure(stub); CheckRpcSendFailure(stub);
// Clean up. // Clean up.
EnableDefaultHealthCheckService(false); grpc_impl::EnableDefaultHealthCheckService(false);
} }
TEST_F(ClientLbEnd2endTest, RoundRobinWithHealthCheckingInhibitPerChannel) { TEST_F(ClientLbEnd2endTest, RoundRobinWithHealthCheckingInhibitPerChannel) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
// Start server. // Start server.
const int kNumServers = 1; const int kNumServers = 1;
StartServers(kNumServers); StartServers(kNumServers);
@ -1341,7 +1341,7 @@ TEST_F(ClientLbEnd2endTest, RoundRobinWithHealthCheckingInhibitPerChannel) {
EXPECT_TRUE(WaitForChannelReady(channel2.get(), 1)); EXPECT_TRUE(WaitForChannelReady(channel2.get(), 1));
CheckRpcSendOk(stub2, DEBUG_LOCATION); CheckRpcSendOk(stub2, DEBUG_LOCATION);
// Clean up. // Clean up.
EnableDefaultHealthCheckService(false); grpc_impl::EnableDefaultHealthCheckService(false);
} }
class ClientLbInterceptTrailingMetadataTest : public ClientLbEnd2endTest { class ClientLbInterceptTrailingMetadataTest : public ClientLbEnd2endTest {

@ -285,8 +285,8 @@ class HealthServiceEnd2endTest : public ::testing::Test {
}; };
TEST_F(HealthServiceEnd2endTest, DefaultHealthServiceDisabled) { TEST_F(HealthServiceEnd2endTest, DefaultHealthServiceDisabled) {
EnableDefaultHealthCheckService(false); grpc_impl::EnableDefaultHealthCheckService(false);
EXPECT_FALSE(DefaultHealthCheckServiceEnabled()); EXPECT_FALSE(grpc_impl::DefaultHealthCheckServiceEnabled());
SetUpServer(true, false, false, nullptr); SetUpServer(true, false, false, nullptr);
HealthCheckServiceInterface* default_service = HealthCheckServiceInterface* default_service =
server_->GetHealthCheckService(); server_->GetHealthCheckService();
@ -298,8 +298,8 @@ TEST_F(HealthServiceEnd2endTest, DefaultHealthServiceDisabled) {
} }
TEST_F(HealthServiceEnd2endTest, DefaultHealthService) { TEST_F(HealthServiceEnd2endTest, DefaultHealthService) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
EXPECT_TRUE(DefaultHealthCheckServiceEnabled()); EXPECT_TRUE(grpc_impl::DefaultHealthCheckServiceEnabled());
SetUpServer(true, false, false, nullptr); SetUpServer(true, false, false, nullptr);
VerifyHealthCheckService(); VerifyHealthCheckService();
VerifyHealthCheckServiceStreaming(); VerifyHealthCheckServiceStreaming();
@ -311,19 +311,19 @@ TEST_F(HealthServiceEnd2endTest, DefaultHealthService) {
} }
TEST_F(HealthServiceEnd2endTest, DefaultHealthServiceShutdown) { TEST_F(HealthServiceEnd2endTest, DefaultHealthServiceShutdown) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
EXPECT_TRUE(DefaultHealthCheckServiceEnabled()); EXPECT_TRUE(grpc_impl::DefaultHealthCheckServiceEnabled());
SetUpServer(true, false, false, nullptr); SetUpServer(true, false, false, nullptr);
VerifyHealthCheckServiceShutdown(); VerifyHealthCheckServiceShutdown();
} }
// Provide an empty service to disable the default service. // Provide an empty service to disable the default service.
TEST_F(HealthServiceEnd2endTest, ExplicitlyDisableViaOverride) { TEST_F(HealthServiceEnd2endTest, ExplicitlyDisableViaOverride) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
EXPECT_TRUE(DefaultHealthCheckServiceEnabled()); EXPECT_TRUE(grpc_impl::DefaultHealthCheckServiceEnabled());
std::unique_ptr<HealthCheckServiceInterface> empty_service; std::unique_ptr<HealthCheckServiceInterface> empty_service;
SetUpServer(true, false, true, std::move(empty_service)); SetUpServer(true, false, true, std::move(empty_service));
HealthCheckServiceInterface* service = server_->GetHealthCheckService(); grpc_impl::HealthCheckServiceInterface* service = server_->GetHealthCheckService();
EXPECT_TRUE(service == nullptr); EXPECT_TRUE(service == nullptr);
ResetStubs(); ResetStubs();
@ -333,8 +333,8 @@ TEST_F(HealthServiceEnd2endTest, ExplicitlyDisableViaOverride) {
// Provide an explicit override of health checking service interface. // Provide an explicit override of health checking service interface.
TEST_F(HealthServiceEnd2endTest, ExplicitlyOverride) { TEST_F(HealthServiceEnd2endTest, ExplicitlyOverride) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
EXPECT_TRUE(DefaultHealthCheckServiceEnabled()); EXPECT_TRUE(grpc_impl::DefaultHealthCheckServiceEnabled());
std::unique_ptr<HealthCheckServiceInterface> override_service( std::unique_ptr<HealthCheckServiceInterface> override_service(
new CustomHealthCheckService(&health_check_service_impl_)); new CustomHealthCheckService(&health_check_service_impl_));
HealthCheckServiceInterface* underlying_service = override_service.get(); HealthCheckServiceInterface* underlying_service = override_service.get();
@ -349,8 +349,8 @@ TEST_F(HealthServiceEnd2endTest, ExplicitlyOverride) {
} }
TEST_F(HealthServiceEnd2endTest, ExplicitlyHealthServiceShutdown) { TEST_F(HealthServiceEnd2endTest, ExplicitlyHealthServiceShutdown) {
EnableDefaultHealthCheckService(true); grpc_impl::EnableDefaultHealthCheckService(true);
EXPECT_TRUE(DefaultHealthCheckServiceEnabled()); EXPECT_TRUE(grpc_impl::DefaultHealthCheckServiceEnabled());
std::unique_ptr<HealthCheckServiceInterface> override_service( std::unique_ptr<HealthCheckServiceInterface> override_service(
new CustomHealthCheckService(&health_check_service_impl_)); new CustomHealthCheckService(&health_check_service_impl_));
HealthCheckServiceInterface* underlying_service = override_service.get(); HealthCheckServiceInterface* underlying_service = override_service.get();

Loading…
Cancel
Save