3
* Copyright 2015, Google Inc.
6
* Redistribution and use in source and binary forms, with or without
7
* modification, are permitted provided that the following conditions are
10
* * Redistributions of source code must retain the above copyright
11
* notice, this list of conditions and the following disclaimer.
12
* * Redistributions in binary form must reproduce the above
13
* copyright notice, this list of conditions and the following disclaimer
14
* in the documentation and/or other materials provided with the
16
* * Neither the name of Google Inc. nor the names of its
17
* contributors may be used to endorse or promote products derived from
18
* this software without specific prior written permission.
20
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34
#include "src/core/channel/channel_stack.h"
38
#include <grpc/support/alloc.h>
39
#include <grpc/support/log.h>
40
#include "test/core/util/test_config.h"
42
static void channel_init_func(grpc_channel_element *elem, grpc_channel *master,
43
const grpc_channel_args *args,
44
grpc_mdctx *metadata_context, int is_first,
46
GPR_ASSERT(args->num_args == 1);
47
GPR_ASSERT(args->args[0].type == GRPC_ARG_INTEGER);
48
GPR_ASSERT(0 == strcmp(args->args[0].key, "test_key"));
49
GPR_ASSERT(args->args[0].value.integer == 42);
52
*(int *)(elem->channel_data) = 0;
55
static void call_init_func(grpc_call_element *elem,
56
const void *server_transport_data,
57
grpc_transport_stream_op *initial_op) {
58
++*(int *)(elem->channel_data);
59
*(int *)(elem->call_data) = 0;
62
static void channel_destroy_func(grpc_channel_element *elem) {}
64
static void call_destroy_func(grpc_call_element *elem) {
65
++*(int *)(elem->channel_data);
68
static void call_func(grpc_call_element *elem, grpc_transport_stream_op *op) {
69
++*(int *)(elem->call_data);
72
static void channel_func(grpc_channel_element *elem, grpc_transport_op *op) {
73
++*(int *)(elem->channel_data);
76
static void test_create_channel_stack(void) {
77
const grpc_channel_filter filter = {
78
call_func, channel_func, sizeof(int),
79
call_init_func, call_destroy_func, sizeof(int),
80
channel_init_func, channel_destroy_func, "some_test_filter"};
81
const grpc_channel_filter *filters = &filter;
82
grpc_channel_stack *channel_stack;
83
grpc_call_stack *call_stack;
84
grpc_channel_element *channel_elem;
85
grpc_call_element *call_elem;
87
grpc_channel_args chan_args;
88
grpc_mdctx *metadata_context;
92
metadata_context = grpc_mdctx_create();
94
arg.type = GRPC_ARG_INTEGER;
96
arg.value.integer = 42;
98
chan_args.num_args = 1;
99
chan_args.args = &arg;
101
channel_stack = gpr_malloc(grpc_channel_stack_size(&filters, 1));
102
grpc_channel_stack_init(&filters, 1, NULL, &chan_args, metadata_context,
104
GPR_ASSERT(channel_stack->count == 1);
105
channel_elem = grpc_channel_stack_element(channel_stack, 0);
106
channel_data = (int *)channel_elem->channel_data;
107
GPR_ASSERT(*channel_data == 0);
109
call_stack = gpr_malloc(channel_stack->call_stack_size);
110
grpc_call_stack_init(channel_stack, NULL, NULL, call_stack);
111
GPR_ASSERT(call_stack->count == 1);
112
call_elem = grpc_call_stack_element(call_stack, 0);
113
GPR_ASSERT(call_elem->filter == channel_elem->filter);
114
GPR_ASSERT(call_elem->channel_data == channel_elem->channel_data);
115
call_data = (int *)call_elem->call_data;
116
GPR_ASSERT(*call_data == 0);
117
GPR_ASSERT(*channel_data == 1);
119
grpc_call_stack_destroy(call_stack);
120
gpr_free(call_stack);
121
GPR_ASSERT(*channel_data == 2);
123
grpc_channel_stack_destroy(channel_stack);
124
gpr_free(channel_stack);
126
grpc_mdctx_unref(metadata_context);
129
int main(int argc, char **argv) {
130
grpc_test_init(argc, argv);
131
test_create_channel_stack();