channel_stack_test.cc
Go to the documentation of this file.
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 
20 
21 #include <string.h>
22 
23 #include <grpc/grpc.h>
24 #include <grpc/support/alloc.h>
25 #include <grpc/support/log.h>
27 
30 
33  GPR_ASSERT(args->channel_args->num_args == 1);
34  GPR_ASSERT(args->channel_args->args[0].type == GRPC_ARG_INTEGER);
35  GPR_ASSERT(0 == strcmp(args->channel_args->args[0].key, "test_key"));
36  GPR_ASSERT(args->channel_args->args[0].value.integer == 42);
37  GPR_ASSERT(args->is_first);
38  GPR_ASSERT(args->is_last);
39  *static_cast<int*>(elem->channel_data) = 0;
40  return GRPC_ERROR_NONE;
41 }
42 
44  grpc_call_element* elem, const grpc_call_element_args* /*args*/) {
45  ++*static_cast<int*>(elem->channel_data);
46  *static_cast<int*>(elem->call_data) = 0;
47  return GRPC_ERROR_NONE;
48 }
49 
50 static void channel_destroy_func(grpc_channel_element* /*elem*/) {}
51 
53  const grpc_call_final_info* /*final_info*/,
54  grpc_closure* /*ignored*/) {
55  ++*static_cast<int*>(elem->channel_data);
56 }
57 
60  ++*static_cast<int*>(elem->call_data);
61 }
62 
64  grpc_transport_op* /*op*/) {
65  ++*static_cast<int*>(elem->channel_data);
66 }
67 
68 static void free_channel(void* arg, grpc_error_handle /*error*/) {
70  gpr_free(arg);
71 }
72 
73 static void free_call(void* arg, grpc_error_handle /*error*/) {
74  grpc_call_stack_destroy(static_cast<grpc_call_stack*>(arg), nullptr, nullptr);
75  gpr_free(arg);
76 }
77 
78 static void test_create_channel_stack(void) {
79  const grpc_channel_filter filter = {
80  call_func,
81  nullptr,
83  sizeof(int),
87  sizeof(int),
92  "some_test_filter"};
93  const grpc_channel_filter* filters = &filter;
94  grpc_channel_stack* channel_stack;
95  grpc_call_stack* call_stack;
96  grpc_channel_element* channel_elem;
97  grpc_call_element* call_elem;
98  grpc_arg arg;
99  grpc_channel_args chan_args;
100  int* channel_data;
101  int* call_data;
103  grpc_slice path = grpc_slice_from_static_string("/service/method");
104 
106  arg.key = const_cast<char*>("test_key");
107  arg.value.integer = 42;
108 
109  chan_args.num_args = 1;
110  chan_args.args = &arg;
111 
112  channel_stack = static_cast<grpc_channel_stack*>(
113  gpr_malloc(grpc_channel_stack_size(&filters, 1)));
115  "grpc_channel_stack_init",
116  grpc_channel_stack_init(1, free_channel, channel_stack, &filters, 1,
117  &chan_args, "test", channel_stack)));
118  GPR_ASSERT(channel_stack->count == 1);
119  channel_elem = grpc_channel_stack_element(channel_stack, 0);
120  channel_data = static_cast<int*>(channel_elem->channel_data);
121  GPR_ASSERT(*channel_data == 0);
122 
123  call_stack =
124  static_cast<grpc_call_stack*>(gpr_malloc(channel_stack->call_stack_size));
125  const grpc_call_element_args args = {
126  call_stack, /* call_stack */
127  nullptr, /* server_transport_data */
128  nullptr, /* context */
129  path, /* path */
130  gpr_get_cycle_counter(), /* start_time */
131  grpc_core::Timestamp::InfFuture(), /* deadline */
132  nullptr, /* arena */
133  nullptr, /* call_combiner */
134  };
136  grpc_call_stack_init(channel_stack, 1, free_call, call_stack, &args);
138  GPR_ASSERT(call_stack->count == 1);
139  call_elem = grpc_call_stack_element(call_stack, 0);
140  GPR_ASSERT(call_elem->filter == channel_elem->filter);
141  GPR_ASSERT(call_elem->channel_data == channel_elem->channel_data);
142  call_data = static_cast<int*>(call_elem->call_data);
143  GPR_ASSERT(*call_data == 0);
144  GPR_ASSERT(*channel_data == 1);
145 
146  GRPC_CALL_STACK_UNREF(call_stack, "done");
148  GPR_ASSERT(*channel_data == 2);
149 
150  GRPC_CHANNEL_STACK_UNREF(channel_stack, "done");
151 
153 }
154 
155 int main(int argc, char** argv) {
156  grpc::testing::TestEnvironment env(&argc, argv);
157  grpc_init();
159  grpc_shutdown();
160  return 0;
161 }
grpc_call_stack::count
size_t count
Definition: channel_stack.h:239
grpc_arg
Definition: grpc_types.h:103
GRPC_ERROR_NONE
#define GRPC_ERROR_NONE
Definition: error.h:234
channel_init_func
static grpc_error_handle channel_init_func(grpc_channel_element *elem, grpc_channel_element_args *args)
Definition: channel_stack_test.cc:31
log.h
channel_func
static void channel_func(grpc_channel_element *elem, grpc_transport_op *)
Definition: channel_stack_test.cc:63
generate.env
env
Definition: generate.py:37
GRPC_ARG_INTEGER
@ GRPC_ARG_INTEGER
Definition: grpc_types.h:81
grpc_channel_stack
Definition: channel_stack.h:202
grpc_channel_stack_element
grpc_channel_element * grpc_channel_stack_element(grpc_channel_stack *channel_stack, size_t index)
Definition: channel_stack.cc:78
call_data
struct connected_channel_call_data call_data
string.h
GRPC_CHANNEL_STACK_UNREF
#define GRPC_CHANNEL_STACK_UNREF(channel_stack, reason)
Definition: channel_stack.h:299
gpr_free
GPRAPI void gpr_free(void *ptr)
Definition: alloc.cc:51
grpc_channel_element
Definition: channel_stack.h:186
elem
Timer elem
Definition: event_engine/iomgr_event_engine/timer_heap_test.cc:109
error
grpc_error_handle error
Definition: retry_filter.cc:499
arg::value
void * value
Definition: cmdline.cc:44
gpr_malloc
GPRAPI void * gpr_malloc(size_t size)
Definition: alloc.cc:29
grpc_call_stack_init
grpc_error_handle grpc_call_stack_init(grpc_channel_stack *channel_stack, int initial_refs, grpc_iomgr_cb_func destroy, void *destroy_arg, const grpc_call_element_args *elem_args)
Definition: channel_stack.cc:180
GRPC_LOG_IF_ERROR
#define GRPC_LOG_IF_ERROR(what, error)
Definition: error.h:398
check_documentation.path
path
Definition: check_documentation.py:57
grpc_call_stack_ignore_set_pollset_or_pollset_set
void grpc_call_stack_ignore_set_pollset_or_pollset_set(grpc_call_element *, grpc_polling_entity *)
Definition: channel_stack.cc:233
grpc_call_stack_element
grpc_call_element * grpc_call_stack_element(grpc_call_stack *call_stack, size_t index)
Definition: channel_stack.cc:100
grpc_channel_stack_destroy
void grpc_channel_stack_destroy(grpc_channel_stack *stack)
Definition: channel_stack.cc:166
grpc_call_element::call_data
void * call_data
Definition: channel_stack.h:197
grpc_call_element
Definition: channel_stack.h:194
grpc_channel_args
Definition: grpc_types.h:132
grpc_transport_op
Definition: transport.h:452
grpc_channel_stack::call_stack_size
size_t call_stack_size
Definition: channel_stack.h:208
string_util.h
GRPC_CALL_STACK_UNREF
#define GRPC_CALL_STACK_UNREF(call_stack, reason)
Definition: channel_stack.h:295
arg::type
argtype type
Definition: cmdline.cc:43
asyncio_get_stats.args
args
Definition: asyncio_get_stats.py:40
xds_interop_client.int
int
Definition: xds_interop_client.py:113
GPR_ASSERT
#define GPR_ASSERT(x)
Definition: include/grpc/impl/codegen/log.h:94
grpc_core::ExecCtx::Flush
bool Flush()
Definition: exec_ctx.cc:69
grpc_channel_element::channel_data
void * channel_data
Definition: channel_stack.h:188
free_call
static void free_call(void *arg, grpc_error_handle)
Definition: channel_stack_test.cc:73
grpc_channel_stack_no_post_init
void grpc_channel_stack_no_post_init(grpc_channel_stack *, grpc_channel_element *)
Definition: channel_stack.cc:282
channel_stack.h
grpc_channel_element::filter
const grpc_channel_filter * filter
Definition: channel_stack.h:187
grpc_call_stack
Definition: channel_stack.h:233
grpc_call_element::filter
const grpc_channel_filter * filter
Definition: channel_stack.h:195
grpc.h
grpc_channel_next_get_info
void grpc_channel_next_get_info(grpc_channel_element *elem, const grpc_channel_info *channel_info)
Definition: channel_stack.cc:258
free_channel
static void free_channel(void *arg, grpc_error_handle)
Definition: channel_stack_test.cc:68
grpc_channel_args::num_args
size_t num_args
Definition: grpc_types.h:133
arg
Definition: cmdline.cc:40
grpc_slice_from_static_string
GPRAPI grpc_slice grpc_slice_from_static_string(const char *source)
Definition: slice/slice.cc:89
grpc_slice
Definition: include/grpc/impl/codegen/slice.h:65
grpc_call_stack_destroy
void grpc_call_stack_destroy(grpc_call_stack *stack, const grpc_call_final_info *final_info, grpc_closure *then_schedule_closure)
Definition: channel_stack.cc:236
grpc_call_element_args
Definition: channel_stack.h:80
slice_internal.h
main
int main(int argc, char **argv)
Definition: channel_stack_test.cc:155
grpc_core::ExecCtx
Definition: exec_ctx.h:97
grpc_channel_filter
Definition: channel_stack.h:111
call_destroy_func
static void call_destroy_func(grpc_call_element *elem, const grpc_call_final_info *, grpc_closure *)
Definition: channel_stack_test.cc:52
test_config.h
grpc_channel_stack_size
size_t grpc_channel_stack_size(const grpc_channel_filter **filters, size_t filter_count)
Definition: channel_stack.cc:51
call_func
static void call_func(grpc_call_element *elem, grpc_transport_stream_op_batch *)
Definition: channel_stack_test.cc:58
grpc_channel_stack::count
size_t count
Definition: channel_stack.h:205
exec_ctx
grpc_core::ExecCtx exec_ctx
Definition: end2end_binder_transport_test.cc:75
channel_data
struct connected_channel_channel_data channel_data
alloc.h
grpc::testing::TestEnvironment
Definition: test/core/util/test_config.h:54
grpc_channel_stack_init
grpc_error_handle grpc_channel_stack_init(int initial_refs, grpc_iomgr_cb_func destroy, void *destroy_arg, const grpc_channel_filter **filters, size_t filter_count, const grpc_channel_args *channel_args, const char *name, grpc_channel_stack *stack)
Definition: channel_stack.cc:105
arg
struct arg arg
test_create_channel_stack
static void test_create_channel_stack(void)
Definition: channel_stack_test.cc:78
grpc_channel_element_args
Definition: channel_stack.h:74
grpc_core::Timestamp::InfFuture
static constexpr Timestamp InfFuture()
Definition: src/core/lib/gprpp/time.h:79
call_init_func
static grpc_error_handle call_init_func(grpc_call_element *elem, const grpc_call_element_args *)
Definition: channel_stack_test.cc:43
grpc_call_final_info
Definition: channel_stack.h:95
grpc_init
GRPCAPI void grpc_init(void)
Definition: init.cc:146
grpc_error
Definition: error_internal.h:42
grpc_transport_stream_op_batch
Definition: transport.h:284
grpc_channel_args::args
grpc_arg * args
Definition: grpc_types.h:134
grpc_closure
Definition: closure.h:56
grpc_call_element::channel_data
void * channel_data
Definition: channel_stack.h:196
grpc_shutdown
GRPCAPI void grpc_shutdown(void)
Definition: init.cc:209
grpc_core::ExecCtx::Get
static ExecCtx * Get()
Definition: exec_ctx.h:205
grpc_slice_unref_internal
void grpc_slice_unref_internal(const grpc_slice &slice)
Definition: slice_refcount.h:39
GRPC_ERROR_IS_NONE
#define GRPC_ERROR_IS_NONE(err)
Definition: error.h:241
channel_destroy_func
static void channel_destroy_func(grpc_channel_element *)
Definition: channel_stack_test.cc:50


grpc
Author(s):
autogenerated on Thu Mar 13 2025 02:58:44