// // // Copyright 2021 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 #include #include "absl/strings/str_cat.h" #include #include #include #include "src/proto/grpc/reflection/v1alpha/reflection.grpc.pb.h" #include "test/core/test_util/port.h" #include "test/core/test_util/test_config.h" namespace grpc { namespace testing { class AdminServicesTest : public ::testing::Test { public: void SetUp() override { std::string address = absl::StrCat("localhost:", grpc_pick_unused_port_or_die()); // Create admin server grpc::reflection::InitProtoReflectionServerBuilderPlugin(); ServerBuilder builder; builder.AddListeningPort(address, InsecureServerCredentials()); grpc::AddAdminServices(&builder); server_ = builder.BuildAndStart(); // Create channel auto reflection_stub = reflection::v1alpha::ServerReflection::NewStub( CreateChannel(address, InsecureChannelCredentials())); stream_ = reflection_stub->ServerReflectionInfo(&reflection_ctx_); } std::vector GetServiceList() { std::vector services; reflection::v1alpha::ServerReflectionRequest request; reflection::v1alpha::ServerReflectionResponse response; request.set_list_services(""); stream_->Write(request); stream_->Read(&response); for (auto& service : response.list_services_response().service()) { services.push_back(service.name()); } return services; } private: std::unique_ptr server_; ClientContext reflection_ctx_; std::shared_ptr< ClientReaderWriter> stream_; }; TEST_F(AdminServicesTest, ValidateRegisteredServices) { // Using Contains here, because the server builder might register other // services in certain environments. EXPECT_THAT( GetServiceList(), ::testing::AllOf( ::testing::Contains("grpc.channelz.v1.Channelz"), ::testing::Contains("grpc.reflection.v1alpha.ServerReflection"))); #if defined(GRPC_NO_XDS) || defined(DISABLED_XDS_PROTO_IN_CC) EXPECT_THAT(GetServiceList(), ::testing::Not(::testing::Contains( "envoy.service.status.v3.ClientStatusDiscoveryService"))); #else EXPECT_THAT(GetServiceList(), ::testing::Contains( "envoy.service.status.v3.ClientStatusDiscoveryService")); #endif // GRPC_NO_XDS or DISABLED_XDS_PROTO_IN_CC } } // namespace testing } // namespace grpc int main(int argc, char** argv) { grpc::testing::TestEnvironment env(&argc, argv); ::testing::InitGoogleTest(&argc, argv); int ret = RUN_ALL_TESTS(); return ret; }