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