Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2017 gRPC authors. |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
Vijay Pai | c90a856 | 2018-03-08 21:20:24 -0800 | [diff] [blame] | 19 | #include <grpcpp/impl/codegen/config.h> |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 20 | #include <gtest/gtest.h> |
| 21 | |
Vijay Pai | c90a856 | 2018-03-08 21:20:24 -0800 | [diff] [blame] | 22 | #include <grpcpp/server.h> |
| 23 | #include <grpcpp/server_builder.h> |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 24 | |
Yash Tibrewal | 8cf1470 | 2017-12-06 09:47:54 -0800 | [diff] [blame] | 25 | #include <grpc/grpc.h> |
| 26 | |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 27 | #include "src/proto/grpc/testing/echo.grpc.pb.h" |
| 28 | #include "test/core/util/port.h" |
| 29 | |
| 30 | namespace grpc { |
| 31 | namespace { |
| 32 | |
| 33 | testing::EchoTestService::Service g_service; |
| 34 | |
| 35 | grpc::string MakePort() { |
| 36 | std::ostringstream s; |
| 37 | int p = grpc_pick_unused_port_or_die(); |
| 38 | s << "localhost:" << p; |
| 39 | return s.str(); |
| 40 | } |
| 41 | |
Ara Ayvazyan | bd5dae2 | 2018-03-30 10:59:59 -0700 | [diff] [blame^] | 42 | const grpc::string& GetPort() { |
| 43 | static grpc::string g_port = MakePort(); |
| 44 | return g_port; |
| 45 | } |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 46 | |
| 47 | TEST(ServerBuilderTest, NoOp) { ServerBuilder b; } |
| 48 | |
| 49 | TEST(ServerBuilderTest, CreateServerNoPorts) { |
| 50 | ServerBuilder().RegisterService(&g_service).BuildAndStart()->Shutdown(); |
| 51 | } |
| 52 | |
| 53 | TEST(ServerBuilderTest, CreateServerOnePort) { |
| 54 | ServerBuilder() |
| 55 | .RegisterService(&g_service) |
Ara Ayvazyan | bd5dae2 | 2018-03-30 10:59:59 -0700 | [diff] [blame^] | 56 | .AddListeningPort(GetPort(), InsecureServerCredentials()) |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 57 | .BuildAndStart() |
| 58 | ->Shutdown(); |
| 59 | } |
| 60 | |
| 61 | TEST(ServerBuilderTest, CreateServerRepeatedPort) { |
| 62 | ServerBuilder() |
| 63 | .RegisterService(&g_service) |
Ara Ayvazyan | bd5dae2 | 2018-03-30 10:59:59 -0700 | [diff] [blame^] | 64 | .AddListeningPort(GetPort(), InsecureServerCredentials()) |
| 65 | .AddListeningPort(GetPort(), InsecureServerCredentials()) |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 66 | .BuildAndStart() |
| 67 | ->Shutdown(); |
| 68 | } |
| 69 | |
| 70 | TEST(ServerBuilderTest, CreateServerRepeatedPortWithDisallowedReusePort) { |
| 71 | EXPECT_EQ(ServerBuilder() |
| 72 | .RegisterService(&g_service) |
Ara Ayvazyan | bd5dae2 | 2018-03-30 10:59:59 -0700 | [diff] [blame^] | 73 | .AddListeningPort(GetPort(), InsecureServerCredentials()) |
| 74 | .AddListeningPort(GetPort(), InsecureServerCredentials()) |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 75 | .AddChannelArgument(GRPC_ARG_ALLOW_REUSEPORT, 0) |
| 76 | .BuildAndStart(), |
| 77 | nullptr); |
| 78 | } |
| 79 | |
| 80 | } // namespace |
| 81 | } // namespace grpc |
| 82 | |
| 83 | int main(int argc, char** argv) { |
| 84 | ::testing::InitGoogleTest(&argc, argv); |
Yash Tibrewal | 8cf1470 | 2017-12-06 09:47:54 -0800 | [diff] [blame] | 85 | grpc_init(); |
| 86 | int ret = RUN_ALL_TESTS(); |
| 87 | grpc_shutdown(); |
| 88 | return ret; |
Craig Tiller | 12352b2 | 2017-01-10 15:16:14 -0800 | [diff] [blame] | 89 | } |