1 //
2 //
3 // Copyright 2021 gRPC authors.
4 //
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
8 //
9 // http://www.apache.org/licenses/LICENSE-2.0
10 //
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.
16 //
17 //
18
19 #include "src/core/server/xds_channel_stack_modifier.h"
20
21 #include <grpc/grpc.h>
22
23 #include <algorithm>
24 #include <string>
25
26 #include "gtest/gtest.h"
27 #include "src/core/config/core_configuration.h"
28 #include "src/core/lib/channel/channel_args.h"
29 #include "src/core/lib/channel/channel_stack.h"
30 #include "src/core/lib/channel/channel_stack_builder_impl.h"
31 #include "src/core/lib/iomgr/endpoint.h"
32 #include "src/core/lib/iomgr/exec_ctx.h"
33 #include "src/core/lib/iomgr/iomgr_fwd.h"
34 #include "src/core/lib/surface/channel_init.h"
35 #include "src/core/lib/surface/channel_stack_type.h"
36 #include "src/core/lib/transport/transport.h"
37 #include "test/core/test_util/test_config.h"
38
39 namespace grpc_core {
40 namespace testing {
41 namespace {
42
43 // Test that XdsChannelStackModifier can be safely copied to channel args
44 // and destroyed
TEST(XdsChannelStackModifierTest,CopyChannelArgs)45 TEST(XdsChannelStackModifierTest, CopyChannelArgs) {
46 grpc_init();
47 auto channel_stack_modifier = MakeRefCounted<XdsChannelStackModifier>(
48 std::vector<const grpc_channel_filter*>{});
49 grpc_arg arg = channel_stack_modifier->MakeChannelArg();
50 grpc_channel_args* args = grpc_channel_args_copy_and_add(nullptr, &arg, 1);
51 EXPECT_EQ(channel_stack_modifier,
52 XdsChannelStackModifier::GetFromChannelArgs(*args));
53 grpc_channel_args_destroy(args);
54 grpc_shutdown();
55 }
56
57 // Test compare on channel args with the same XdsChannelStackModifier
TEST(XdsChannelStackModifierTest,ChannelArgsCompare)58 TEST(XdsChannelStackModifierTest, ChannelArgsCompare) {
59 grpc_init();
60 auto channel_stack_modifier = MakeRefCounted<XdsChannelStackModifier>(
61 std::vector<const grpc_channel_filter*>{});
62 grpc_arg arg = channel_stack_modifier->MakeChannelArg();
63 grpc_channel_args* args = grpc_channel_args_copy_and_add(nullptr, &arg, 1);
64 grpc_channel_args* new_args = grpc_channel_args_copy(args);
65 EXPECT_EQ(XdsChannelStackModifier::GetFromChannelArgs(*new_args),
66 XdsChannelStackModifier::GetFromChannelArgs(*args));
67 grpc_channel_args_destroy(args);
68 grpc_channel_args_destroy(new_args);
69 grpc_shutdown();
70 }
71
72 const UniqueTypeName kTestFilter1 = GRPC_UNIQUE_TYPE_NAME_HERE("test_filter_1");
73 const UniqueTypeName kTestFilter2 = GRPC_UNIQUE_TYPE_NAME_HERE("test_filter_2");
74
75 namespace {
76 class FakeTransport final : public Transport {
77 public:
filter_stack_transport()78 FilterStackTransport* filter_stack_transport() override { return nullptr; }
client_transport()79 ClientTransport* client_transport() override { return nullptr; }
server_transport()80 ServerTransport* server_transport() override { return nullptr; }
81
GetTransportName() const82 absl::string_view GetTransportName() const override { return "fake"; }
SetPollset(grpc_stream *,grpc_pollset *)83 void SetPollset(grpc_stream*, grpc_pollset*) override {}
SetPollsetSet(grpc_stream *,grpc_pollset_set *)84 void SetPollsetSet(grpc_stream*, grpc_pollset_set*) override {}
PerformOp(grpc_transport_op *)85 void PerformOp(grpc_transport_op*) override {}
Orphan()86 void Orphan() override {}
87 };
88 } // namespace
89
90 // Test filters insertion
TEST(XdsChannelStackModifierTest,XdsHttpFiltersInsertion)91 TEST(XdsChannelStackModifierTest, XdsHttpFiltersInsertion) {
92 CoreConfiguration::Reset();
93 grpc_init();
94 // Add 2 test filters to XdsChannelStackModifier
95 const grpc_channel_filter test_filter_1 = {
96 nullptr, nullptr, 0, nullptr, nullptr, nullptr,
97 0, nullptr, nullptr, nullptr, nullptr, kTestFilter1};
98 const grpc_channel_filter test_filter_2 = {
99 nullptr, nullptr, 0, nullptr, nullptr, nullptr,
100 0, nullptr, nullptr, nullptr, nullptr, kTestFilter2};
101 auto channel_stack_modifier = MakeRefCounted<XdsChannelStackModifier>(
102 std::vector<const grpc_channel_filter*>{&test_filter_1, &test_filter_2});
103 grpc_arg arg = channel_stack_modifier->MakeChannelArg();
104 FakeTransport fake_transport;
105 // Create a phony ChannelStackBuilder object
106 grpc_channel_args* args = grpc_channel_args_copy_and_add(nullptr, &arg, 1);
107 ChannelStackBuilderImpl builder(
108 "test", GRPC_SERVER_CHANNEL,
109 ChannelArgs::FromC(args).SetObject<Transport>(&fake_transport));
110 grpc_channel_args_destroy(args);
111 // Construct channel stack and verify that the test filters were successfully
112 // added
113 {
114 ExecCtx exec_ctx;
115 ASSERT_TRUE(CoreConfiguration::Get().channel_init().CreateStack(&builder));
116 }
117 std::vector<std::string> filters;
118 for (const auto& entry : *builder.mutable_stack()) {
119 filters.push_back(std::string(entry->name.name()));
120 }
121 filters.resize(3);
122 EXPECT_EQ(filters, std::vector<std::string>(
123 {"server", std::string(kTestFilter1.name()),
124 std::string(kTestFilter2.name())}));
125 grpc_shutdown();
126 }
127
128 } // namespace
129 } // namespace testing
130 } // namespace grpc_core
131
main(int argc,char ** argv)132 int main(int argc, char** argv) {
133 ::testing::InitGoogleTest(&argc, argv);
134 grpc::testing::TestEnvironment env(&argc, argv);
135 int ret = RUN_ALL_TESTS();
136 return ret;
137 }
138