Alexander Polcyn
7 years ago
parent
2d894a8c3b
commit
ac8b72f75b
11 changed files with 327 additions and 5 deletions
@ -0,0 +1,116 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2017 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 <grpcpp/impl/codegen/config.h> |
||||||
|
#include <gtest/gtest.h> |
||||||
|
|
||||||
|
#include <grpcpp/server.h> |
||||||
|
#include <grpcpp/server_builder.h> |
||||||
|
|
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include "src/core/lib/iomgr/socket_mutator.h" |
||||||
|
#include "src/proto/grpc/testing/echo.grpc.pb.h" |
||||||
|
#include "test/core/util/port.h" |
||||||
|
|
||||||
|
/* This test does a sanity check that grpc_socket_mutator's
|
||||||
|
* are used by servers. It's meant to protect code and end-to-end |
||||||
|
* tests that rely on this functionality but which live outside |
||||||
|
* of the grpc github repo. */ |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace { |
||||||
|
|
||||||
|
bool mock_socket_mutator_mutate_fd(int, grpc_socket_mutator*); |
||||||
|
int mock_socket_mutator_compare(grpc_socket_mutator*, grpc_socket_mutator*); |
||||||
|
void mock_socket_mutator_destroy(grpc_socket_mutator*); |
||||||
|
|
||||||
|
const grpc_socket_mutator_vtable mock_socket_mutator_vtable = { |
||||||
|
mock_socket_mutator_mutate_fd, |
||||||
|
mock_socket_mutator_compare, |
||||||
|
mock_socket_mutator_destroy, |
||||||
|
}; |
||||||
|
|
||||||
|
class MockSocketMutator : public grpc_socket_mutator { |
||||||
|
public: |
||||||
|
MockSocketMutator() : mutate_fd_call_count_(0) { |
||||||
|
grpc_socket_mutator_init(this, &mock_socket_mutator_vtable); |
||||||
|
} |
||||||
|
int mutate_fd_call_count_; |
||||||
|
}; |
||||||
|
|
||||||
|
bool mock_socket_mutator_mutate_fd(int fd, grpc_socket_mutator* m) { |
||||||
|
MockSocketMutator* s = reinterpret_cast<MockSocketMutator*>(m); |
||||||
|
s->mutate_fd_call_count_++; |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
int mock_socket_mutator_compare(grpc_socket_mutator* a, |
||||||
|
grpc_socket_mutator* b) { |
||||||
|
return (uintptr_t)a - (uintptr_t)b; |
||||||
|
} |
||||||
|
|
||||||
|
void mock_socket_mutator_destroy(grpc_socket_mutator* m) { |
||||||
|
MockSocketMutator* s = reinterpret_cast<MockSocketMutator*>(m); |
||||||
|
delete s; |
||||||
|
} |
||||||
|
|
||||||
|
class MockSocketMutatorServerBuilderOption : public grpc::ServerBuilderOption { |
||||||
|
public: |
||||||
|
MockSocketMutatorServerBuilderOption(MockSocketMutator* mock_socket_mutator) |
||||||
|
: mock_socket_mutator_(mock_socket_mutator) {} |
||||||
|
|
||||||
|
void UpdateArguments(ChannelArguments* args) override { |
||||||
|
args->SetSocketMutator(mock_socket_mutator_); |
||||||
|
} |
||||||
|
|
||||||
|
void UpdatePlugins( |
||||||
|
std::vector<std::unique_ptr<ServerBuilderPlugin>>*) override{}; |
||||||
|
|
||||||
|
MockSocketMutator* mock_socket_mutator_; |
||||||
|
}; |
||||||
|
|
||||||
|
TEST(ServerBuilderWithSocketMutatorTest, CreateServerWithSocketMutator) { |
||||||
|
auto address = "localhost:" + std::to_string(grpc_pick_unused_port_or_die()); |
||||||
|
auto mock_socket_mutator = new MockSocketMutator(); |
||||||
|
std::unique_ptr<grpc::ServerBuilderOption> mock_socket_mutator_builder_option( |
||||||
|
new MockSocketMutatorServerBuilderOption(mock_socket_mutator)); |
||||||
|
testing::EchoTestService::Service echo_service; |
||||||
|
EXPECT_EQ(mock_socket_mutator->mutate_fd_call_count_, 0); |
||||||
|
ServerBuilder builder; |
||||||
|
builder.RegisterService(&echo_service); |
||||||
|
builder.AddListeningPort(address, InsecureServerCredentials()); |
||||||
|
builder.SetOption(std::move(mock_socket_mutator_builder_option)); |
||||||
|
std::unique_ptr<grpc::Server> server(builder.BuildAndStart()); |
||||||
|
EXPECT_NE(server, nullptr); |
||||||
|
// Only assert that the socket mutator was used.
|
||||||
|
EXPECT_GE(mock_socket_mutator->mutate_fd_call_count_, 1); |
||||||
|
server->Shutdown(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
::testing::InitGoogleTest(&argc, argv); |
||||||
|
grpc_init(); |
||||||
|
int ret = RUN_ALL_TESTS(); |
||||||
|
grpc_shutdown(); |
||||||
|
return ret; |
||||||
|
} |
||||||
Loading…
Reference in new issue