1 //
2 //
3 // Copyright 2015 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/lib/channel/channel_stack.h"
20
21 #include <grpc/support/alloc.h>
22
23 #include "absl/status/status.h"
24 #include "absl/types/optional.h"
25 #include "gtest/gtest.h"
26 #include "src/core/config/core_configuration.h"
27 #include "src/core/lib/channel/channel_args.h"
28 #include "src/core/lib/channel/channel_args_preconditioning.h"
29 #include "src/core/lib/iomgr/exec_ctx.h"
30 #include "src/core/util/status_helper.h"
31 #include "test/core/test_util/test_config.h"
32
channel_init_func(grpc_channel_element * elem,grpc_channel_element_args * args)33 static grpc_error_handle channel_init_func(grpc_channel_element* elem,
34 grpc_channel_element_args* args) {
35 int test_value = args->channel_args.GetInt("test_key").value_or(-1);
36 EXPECT_EQ(test_value, 42);
37 auto* ee = args->channel_args
38 .GetObject<grpc_event_engine::experimental::EventEngine>();
39 EXPECT_NE(ee, nullptr);
40 EXPECT_TRUE(args->is_first);
41 EXPECT_TRUE(args->is_last);
42 *static_cast<int*>(elem->channel_data) = 0;
43 return absl::OkStatus();
44 }
45
call_init_func(grpc_call_element * elem,const grpc_call_element_args *)46 static grpc_error_handle call_init_func(
47 grpc_call_element* elem, const grpc_call_element_args* /*args*/) {
48 ++*static_cast<int*>(elem->channel_data);
49 *static_cast<int*>(elem->call_data) = 0;
50 return absl::OkStatus();
51 }
52
channel_destroy_func(grpc_channel_element *)53 static void channel_destroy_func(grpc_channel_element* /*elem*/) {}
54
call_destroy_func(grpc_call_element * elem,const grpc_call_final_info *,grpc_closure *)55 static void call_destroy_func(grpc_call_element* elem,
56 const grpc_call_final_info* /*final_info*/,
57 grpc_closure* /*ignored*/) {
58 ++*static_cast<int*>(elem->channel_data);
59 }
60
call_func(grpc_call_element * elem,grpc_transport_stream_op_batch *)61 static void call_func(grpc_call_element* elem,
62 grpc_transport_stream_op_batch* /*op*/) {
63 ++*static_cast<int*>(elem->call_data);
64 }
65
channel_func(grpc_channel_element * elem,grpc_transport_op *)66 static void channel_func(grpc_channel_element* elem,
67 grpc_transport_op* /*op*/) {
68 ++*static_cast<int*>(elem->channel_data);
69 }
70
free_channel(void * arg,grpc_error_handle)71 static void free_channel(void* arg, grpc_error_handle /*error*/) {
72 grpc_channel_stack_destroy(static_cast<grpc_channel_stack*>(arg));
73 gpr_free(arg);
74 }
75
free_call(void * arg,grpc_error_handle)76 static void free_call(void* arg, grpc_error_handle /*error*/) {
77 grpc_call_stack_destroy(static_cast<grpc_call_stack*>(arg), nullptr, nullptr);
78 gpr_free(arg);
79 }
80
TEST(ChannelStackTest,CreateChannelStack)81 TEST(ChannelStackTest, CreateChannelStack) {
82 const grpc_channel_filter filter = {
83 call_func,
84 channel_func,
85 sizeof(int),
86 call_init_func,
87 grpc_call_stack_ignore_set_pollset_or_pollset_set,
88 call_destroy_func,
89 sizeof(int),
90 channel_init_func,
91 grpc_channel_stack_no_post_init,
92 channel_destroy_func,
93 grpc_channel_next_get_info,
94 GRPC_UNIQUE_TYPE_NAME_HERE("some_test_filter")};
95 const grpc_channel_filter* filters = &filter;
96 grpc_channel_stack* channel_stack;
97 grpc_call_stack* call_stack;
98 grpc_channel_element* channel_elem;
99 grpc_call_element* call_elem;
100 int* channel_data;
101 int* call_data;
102 grpc_core::ExecCtx exec_ctx;
103 grpc_slice path = grpc_slice_from_static_string("/service/method");
104
105 channel_stack = static_cast<grpc_channel_stack*>(
106 gpr_malloc(grpc_channel_stack_size(&filters, 1)));
107 auto channel_args = grpc_core::CoreConfiguration::Get()
108 .channel_args_preconditioning()
109 .PreconditionChannelArgs(nullptr)
110 .Set("test_key", 42);
111 ASSERT_TRUE(GRPC_LOG_IF_ERROR(
112 "grpc_channel_stack_init",
113 grpc_channel_stack_init(1, free_channel, channel_stack, &filters, 1,
114 channel_args, "test", channel_stack)));
115 EXPECT_EQ(channel_stack->count, 1);
116 channel_elem = grpc_channel_stack_element(channel_stack, 0);
117 channel_data = static_cast<int*>(channel_elem->channel_data);
118 EXPECT_EQ(*channel_data, 0);
119
120 call_stack =
121 static_cast<grpc_call_stack*>(gpr_malloc(channel_stack->call_stack_size));
122 const grpc_call_element_args args = {
123 call_stack, // call_stack
124 nullptr, // server_transport_data
125 path, // path
126 gpr_get_cycle_counter(), // start_time
127 grpc_core::Timestamp::InfFuture(), // deadline
128 nullptr, // arena
129 nullptr, // call_combiner
130 };
131 grpc_error_handle error =
132 grpc_call_stack_init(channel_stack, 1, free_call, call_stack, &args);
133 ASSERT_TRUE(error.ok()) << grpc_core::StatusToString(error);
134 EXPECT_EQ(call_stack->count, 1);
135 call_elem = grpc_call_stack_element(call_stack, 0);
136 EXPECT_EQ(call_elem->filter, channel_elem->filter);
137 EXPECT_EQ(call_elem->channel_data, channel_elem->channel_data);
138 call_data = static_cast<int*>(call_elem->call_data);
139 EXPECT_EQ(*call_data, 0);
140 EXPECT_EQ(*channel_data, 1);
141
142 GRPC_CALL_STACK_UNREF(call_stack, "done");
143 grpc_core::ExecCtx::Get()->Flush();
144 EXPECT_EQ(*channel_data, 2);
145
146 GRPC_CHANNEL_STACK_UNREF(channel_stack, "done");
147
148 grpc_slice_unref(path);
149 }
150
main(int argc,char ** argv)151 int main(int argc, char** argv) {
152 grpc::testing::TestEnvironment env(&argc, argv);
153 ::testing::InitGoogleTest(&argc, argv);
154 grpc::testing::TestGrpcScope grpc_scope;
155 return RUN_ALL_TESTS();
156 }
157