1 //
2 // Copyright 2022 gRPC authors.
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16
17 #include <memory>
18
19 #include "absl/types/optional.h"
20 #include "gtest/gtest.h"
21
22 #include <grpc/impl/channel_arg_names.h>
23 #include <grpc/support/port_platform.h>
24
25 #include "src/core/client_channel/client_channel_filter.h"
26 #include "src/core/client_channel/subchannel_pool_interface.h"
27 #include "src/core/lib/channel/channel_args.h"
28 #include "src/core/resolver/endpoint_addresses.h"
29 #include "test/core/util/test_config.h"
30
31 namespace grpc_core {
32 namespace testing {
33 namespace {
34
TEST(MakeSubchannelArgs,UsesChannelDefaultAuthorityByDefault)35 TEST(MakeSubchannelArgs, UsesChannelDefaultAuthorityByDefault) {
36 ChannelArgs args = ClientChannelFilter::MakeSubchannelArgs(
37 ChannelArgs(), ChannelArgs(), nullptr, "foo.example.com");
38 EXPECT_EQ(args.GetString(GRPC_ARG_DEFAULT_AUTHORITY), "foo.example.com");
39 }
40
TEST(MakeSubchannelArgs,DefaultAuthorityFromChannelArgs)41 TEST(MakeSubchannelArgs, DefaultAuthorityFromChannelArgs) {
42 ChannelArgs args = ClientChannelFilter::MakeSubchannelArgs(
43 ChannelArgs().Set(GRPC_ARG_DEFAULT_AUTHORITY, "bar.example.com"),
44 ChannelArgs(), nullptr, "foo.example.com");
45 EXPECT_EQ(args.GetString(GRPC_ARG_DEFAULT_AUTHORITY), "bar.example.com");
46 }
47
TEST(MakeSubchannelArgs,DefaultAuthorityFromResolver)48 TEST(MakeSubchannelArgs, DefaultAuthorityFromResolver) {
49 ChannelArgs args = ClientChannelFilter::MakeSubchannelArgs(
50 ChannelArgs(),
51 ChannelArgs().Set(GRPC_ARG_DEFAULT_AUTHORITY, "bar.example.com"), nullptr,
52 "foo.example.com");
53 EXPECT_EQ(args.GetString(GRPC_ARG_DEFAULT_AUTHORITY), "bar.example.com");
54 }
55
TEST(MakeSubchannelArgs,DefaultAuthorityFromChannelArgsOverridesValueFromResolver)56 TEST(MakeSubchannelArgs,
57 DefaultAuthorityFromChannelArgsOverridesValueFromResolver) {
58 ChannelArgs args = ClientChannelFilter::MakeSubchannelArgs(
59 ChannelArgs().Set(GRPC_ARG_DEFAULT_AUTHORITY, "bar.example.com"),
60 ChannelArgs().Set(GRPC_ARG_DEFAULT_AUTHORITY, "baz.example.com"), nullptr,
61 "foo.example.com");
62 EXPECT_EQ(args.GetString(GRPC_ARG_DEFAULT_AUTHORITY), "bar.example.com");
63 }
64
TEST(MakeSubchannelArgs,ArgsFromChannelTrumpPerAddressArgs)65 TEST(MakeSubchannelArgs, ArgsFromChannelTrumpPerAddressArgs) {
66 ChannelArgs args = ClientChannelFilter::MakeSubchannelArgs(
67 ChannelArgs().Set("foo", 1), ChannelArgs().Set("foo", 2), nullptr,
68 "foo.example.com");
69 EXPECT_EQ(args.GetInt("foo"), 1);
70 }
71
TEST(MakeSubchannelArgs,StripsOutNoSubchannelArgs)72 TEST(MakeSubchannelArgs, StripsOutNoSubchannelArgs) {
73 ChannelArgs args = ClientChannelFilter::MakeSubchannelArgs(
74 ChannelArgs().Set(GRPC_ARG_NO_SUBCHANNEL_PREFIX "foo", 1),
75 ChannelArgs().Set(GRPC_ARG_NO_SUBCHANNEL_PREFIX "bar", 1), nullptr,
76 "foo.example.com");
77 EXPECT_EQ(args.GetString(GRPC_ARG_NO_SUBCHANNEL_PREFIX "foo"), absl::nullopt);
78 EXPECT_EQ(args.GetString(GRPC_ARG_NO_SUBCHANNEL_PREFIX "bar"), absl::nullopt);
79 }
80
81 } // namespace
82 } // namespace testing
83 } // namespace grpc_core
84
main(int argc,char ** argv)85 int main(int argc, char** argv) {
86 ::testing::InitGoogleTest(&argc, argv);
87 grpc::testing::TestEnvironment env(&argc, argv);
88 auto result = RUN_ALL_TESTS();
89 return result;
90 }
91