diff options
Diffstat (limited to 'test/core/channel')
-rw-r--r-- | test/core/channel/channel_args_test.c | 125 | ||||
-rw-r--r-- | test/core/channel/channel_stack_test.c | 145 |
2 files changed, 126 insertions, 144 deletions
diff --git a/test/core/channel/channel_args_test.c b/test/core/channel/channel_args_test.c index 207196ebdb..0b74dee41e 100644 --- a/test/core/channel/channel_args_test.c +++ b/test/core/channel/channel_args_test.c @@ -40,9 +40,7 @@ #include "test/core/util/test_config.h" -static void -test_create (void) -{ +static void test_create(void) { grpc_arg arg_int; grpc_arg arg_string; grpc_arg to_add[2]; @@ -58,94 +56,87 @@ test_create (void) to_add[0] = arg_int; to_add[1] = arg_string; - ch_args = grpc_channel_args_copy_and_add (NULL, to_add, 2); + ch_args = grpc_channel_args_copy_and_add(NULL, to_add, 2); - GPR_ASSERT (ch_args->num_args == 2); - GPR_ASSERT (strcmp (ch_args->args[0].key, arg_int.key) == 0); - GPR_ASSERT (ch_args->args[0].type == arg_int.type); - GPR_ASSERT (ch_args->args[0].value.integer == arg_int.value.integer); + GPR_ASSERT(ch_args->num_args == 2); + GPR_ASSERT(strcmp(ch_args->args[0].key, arg_int.key) == 0); + GPR_ASSERT(ch_args->args[0].type == arg_int.type); + GPR_ASSERT(ch_args->args[0].value.integer == arg_int.value.integer); - GPR_ASSERT (strcmp (ch_args->args[1].key, arg_string.key) == 0); - GPR_ASSERT (ch_args->args[1].type == arg_string.type); - GPR_ASSERT (strcmp (ch_args->args[1].value.string, arg_string.value.string) == 0); + GPR_ASSERT(strcmp(ch_args->args[1].key, arg_string.key) == 0); + GPR_ASSERT(ch_args->args[1].type == arg_string.type); + GPR_ASSERT(strcmp(ch_args->args[1].value.string, arg_string.value.string) == + 0); - grpc_channel_args_destroy (ch_args); + grpc_channel_args_destroy(ch_args); } -static void -test_set_compression_algorithm (void) -{ +static void test_set_compression_algorithm(void) { grpc_channel_args *ch_args; - ch_args = grpc_channel_args_set_compression_algorithm (NULL, GRPC_COMPRESS_GZIP); - GPR_ASSERT (ch_args->num_args == 1); - GPR_ASSERT (strcmp (ch_args->args[0].key, GRPC_COMPRESSION_ALGORITHM_ARG) == 0); - GPR_ASSERT (ch_args->args[0].type == GRPC_ARG_INTEGER); + ch_args = + grpc_channel_args_set_compression_algorithm(NULL, GRPC_COMPRESS_GZIP); + GPR_ASSERT(ch_args->num_args == 1); + GPR_ASSERT(strcmp(ch_args->args[0].key, GRPC_COMPRESSION_ALGORITHM_ARG) == 0); + GPR_ASSERT(ch_args->args[0].type == GRPC_ARG_INTEGER); - grpc_channel_args_destroy (ch_args); + grpc_channel_args_destroy(ch_args); } -static void -test_compression_algorithm_states (void) -{ +static void test_compression_algorithm_states(void) { grpc_channel_args *ch_args, *ch_args_wo_gzip, *ch_args_wo_gzip_deflate; unsigned states_bitset; size_t i; - ch_args = grpc_channel_args_copy_and_add (NULL, NULL, 0); + ch_args = grpc_channel_args_copy_and_add(NULL, NULL, 0); /* by default, all enabled */ - states_bitset = (unsigned) grpc_channel_args_compression_algorithm_get_states (ch_args); + states_bitset = + (unsigned)grpc_channel_args_compression_algorithm_get_states(ch_args); - for (i = 0; i < GRPC_COMPRESS_ALGORITHMS_COUNT; i++) - { - GPR_ASSERT (GPR_BITGET (states_bitset, i)); - } + for (i = 0; i < GRPC_COMPRESS_ALGORITHMS_COUNT; i++) { + GPR_ASSERT(GPR_BITGET(states_bitset, i)); + } /* disable gzip and deflate */ - ch_args_wo_gzip = grpc_channel_args_compression_algorithm_set_state (&ch_args, GRPC_COMPRESS_GZIP, 0); - GPR_ASSERT (ch_args == ch_args_wo_gzip); - ch_args_wo_gzip_deflate = grpc_channel_args_compression_algorithm_set_state (&ch_args_wo_gzip, GRPC_COMPRESS_DEFLATE, 0); - GPR_ASSERT (ch_args_wo_gzip == ch_args_wo_gzip_deflate); - - states_bitset = (unsigned) grpc_channel_args_compression_algorithm_get_states (ch_args_wo_gzip_deflate); - for (i = 0; i < GRPC_COMPRESS_ALGORITHMS_COUNT; i++) - { - if (i == GRPC_COMPRESS_GZIP || i == GRPC_COMPRESS_DEFLATE) - { - GPR_ASSERT (GPR_BITGET (states_bitset, i) == 0); - } - else - { - GPR_ASSERT (GPR_BITGET (states_bitset, i) != 0); - } + ch_args_wo_gzip = grpc_channel_args_compression_algorithm_set_state( + &ch_args, GRPC_COMPRESS_GZIP, 0); + GPR_ASSERT(ch_args == ch_args_wo_gzip); + ch_args_wo_gzip_deflate = grpc_channel_args_compression_algorithm_set_state( + &ch_args_wo_gzip, GRPC_COMPRESS_DEFLATE, 0); + GPR_ASSERT(ch_args_wo_gzip == ch_args_wo_gzip_deflate); + + states_bitset = (unsigned)grpc_channel_args_compression_algorithm_get_states( + ch_args_wo_gzip_deflate); + for (i = 0; i < GRPC_COMPRESS_ALGORITHMS_COUNT; i++) { + if (i == GRPC_COMPRESS_GZIP || i == GRPC_COMPRESS_DEFLATE) { + GPR_ASSERT(GPR_BITGET(states_bitset, i) == 0); + } else { + GPR_ASSERT(GPR_BITGET(states_bitset, i) != 0); } + } /* re-enabled gzip only */ - ch_args_wo_gzip = grpc_channel_args_compression_algorithm_set_state (&ch_args_wo_gzip_deflate, GRPC_COMPRESS_GZIP, 1); - GPR_ASSERT (ch_args_wo_gzip == ch_args_wo_gzip_deflate); - - states_bitset = (unsigned) grpc_channel_args_compression_algorithm_get_states (ch_args_wo_gzip); - for (i = 0; i < GRPC_COMPRESS_ALGORITHMS_COUNT; i++) - { - if (i == GRPC_COMPRESS_DEFLATE) - { - GPR_ASSERT (GPR_BITGET (states_bitset, i) == 0); - } - else - { - GPR_ASSERT (GPR_BITGET (states_bitset, i) != 0); - } + ch_args_wo_gzip = grpc_channel_args_compression_algorithm_set_state( + &ch_args_wo_gzip_deflate, GRPC_COMPRESS_GZIP, 1); + GPR_ASSERT(ch_args_wo_gzip == ch_args_wo_gzip_deflate); + + states_bitset = (unsigned)grpc_channel_args_compression_algorithm_get_states( + ch_args_wo_gzip); + for (i = 0; i < GRPC_COMPRESS_ALGORITHMS_COUNT; i++) { + if (i == GRPC_COMPRESS_DEFLATE) { + GPR_ASSERT(GPR_BITGET(states_bitset, i) == 0); + } else { + GPR_ASSERT(GPR_BITGET(states_bitset, i) != 0); } + } - grpc_channel_args_destroy (ch_args); + grpc_channel_args_destroy(ch_args); } -int -main (int argc, char **argv) -{ - grpc_test_init (argc, argv); - test_create (); - test_set_compression_algorithm (); - test_compression_algorithm_states (); +int main(int argc, char **argv) { + grpc_test_init(argc, argv); + test_create(); + test_set_compression_algorithm(); + test_compression_algorithm_states(); return 0; } diff --git a/test/core/channel/channel_stack_test.c b/test/core/channel/channel_stack_test.c index 24d5093b60..178fe2aa87 100644 --- a/test/core/channel/channel_stack_test.c +++ b/test/core/channel/channel_stack_test.c @@ -41,63 +41,55 @@ #include "test/core/util/test_config.h" -static void -channel_init_func (grpc_exec_ctx * exec_ctx, grpc_channel_element * elem, grpc_channel * master, const grpc_channel_args * args, grpc_mdctx * metadata_context, int is_first, int is_last) -{ - GPR_ASSERT (args->num_args == 1); - GPR_ASSERT (args->args[0].type == GRPC_ARG_INTEGER); - GPR_ASSERT (0 == strcmp (args->args[0].key, "test_key")); - GPR_ASSERT (args->args[0].value.integer == 42); - GPR_ASSERT (is_first); - GPR_ASSERT (is_last); - *(int *) (elem->channel_data) = 0; +static void channel_init_func(grpc_exec_ctx *exec_ctx, + grpc_channel_element *elem, grpc_channel *master, + const grpc_channel_args *args, + grpc_mdctx *metadata_context, int is_first, + int is_last) { + GPR_ASSERT(args->num_args == 1); + GPR_ASSERT(args->args[0].type == GRPC_ARG_INTEGER); + GPR_ASSERT(0 == strcmp(args->args[0].key, "test_key")); + GPR_ASSERT(args->args[0].value.integer == 42); + GPR_ASSERT(is_first); + GPR_ASSERT(is_last); + *(int *)(elem->channel_data) = 0; } -static void -call_init_func (grpc_exec_ctx * exec_ctx, grpc_call_element * elem, const void *server_transport_data, grpc_transport_stream_op * initial_op) -{ - ++*(int *) (elem->channel_data); - *(int *) (elem->call_data) = 0; +static void call_init_func(grpc_exec_ctx *exec_ctx, grpc_call_element *elem, + const void *server_transport_data, + grpc_transport_stream_op *initial_op) { + ++*(int *)(elem->channel_data); + *(int *)(elem->call_data) = 0; } -static void -channel_destroy_func (grpc_exec_ctx * exec_ctx, grpc_channel_element * elem) -{ -} +static void channel_destroy_func(grpc_exec_ctx *exec_ctx, + grpc_channel_element *elem) {} -static void -call_destroy_func (grpc_exec_ctx * exec_ctx, grpc_call_element * elem) -{ - ++*(int *) (elem->channel_data); +static void call_destroy_func(grpc_exec_ctx *exec_ctx, + grpc_call_element *elem) { + ++*(int *)(elem->channel_data); } -static void -call_func (grpc_exec_ctx * exec_ctx, grpc_call_element * elem, grpc_transport_stream_op * op) -{ - ++*(int *) (elem->call_data); +static void call_func(grpc_exec_ctx *exec_ctx, grpc_call_element *elem, + grpc_transport_stream_op *op) { + ++*(int *)(elem->call_data); } -static void -channel_func (grpc_exec_ctx * exec_ctx, grpc_channel_element * elem, grpc_transport_op * op) -{ - ++*(int *) (elem->channel_data); +static void channel_func(grpc_exec_ctx *exec_ctx, grpc_channel_element *elem, + grpc_transport_op *op) { + ++*(int *)(elem->channel_data); } -static char * -get_peer (grpc_exec_ctx * exec_ctx, grpc_call_element * elem) -{ - return gpr_strdup ("peer"); +static char *get_peer(grpc_exec_ctx *exec_ctx, grpc_call_element *elem) { + return gpr_strdup("peer"); } -static void -test_create_channel_stack (void) -{ - const grpc_channel_filter filter = { call_func, channel_func, - sizeof (int), call_init_func, - call_destroy_func, sizeof (int), - channel_init_func, channel_destroy_func, - get_peer, "some_test_filter" - }; +static void test_create_channel_stack(void) { + const grpc_channel_filter filter = {call_func, channel_func, + sizeof(int), call_init_func, + call_destroy_func, sizeof(int), + channel_init_func, channel_destroy_func, + get_peer, "some_test_filter"}; const grpc_channel_filter *filters = &filter; grpc_channel_stack *channel_stack; grpc_call_stack *call_stack; @@ -110,7 +102,7 @@ test_create_channel_stack (void) int *call_data; grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; - metadata_context = grpc_mdctx_create (); + metadata_context = grpc_mdctx_create(); arg.type = GRPC_ARG_INTEGER; arg.key = "test_key"; @@ -119,39 +111,38 @@ test_create_channel_stack (void) chan_args.num_args = 1; chan_args.args = &arg; - channel_stack = gpr_malloc (grpc_channel_stack_size (&filters, 1)); - grpc_channel_stack_init (&exec_ctx, &filters, 1, NULL, &chan_args, metadata_context, channel_stack); - GPR_ASSERT (channel_stack->count == 1); - channel_elem = grpc_channel_stack_element (channel_stack, 0); - channel_data = (int *) channel_elem->channel_data; - GPR_ASSERT (*channel_data == 0); - - call_stack = gpr_malloc (channel_stack->call_stack_size); - grpc_call_stack_init (&exec_ctx, channel_stack, NULL, NULL, call_stack); - GPR_ASSERT (call_stack->count == 1); - call_elem = grpc_call_stack_element (call_stack, 0); - GPR_ASSERT (call_elem->filter == channel_elem->filter); - GPR_ASSERT (call_elem->channel_data == channel_elem->channel_data); - call_data = (int *) call_elem->call_data; - GPR_ASSERT (*call_data == 0); - GPR_ASSERT (*channel_data == 1); - - grpc_call_stack_destroy (&exec_ctx, call_stack); - gpr_free (call_stack); - GPR_ASSERT (*channel_data == 2); - - grpc_channel_stack_destroy (&exec_ctx, channel_stack); - gpr_free (channel_stack); - - grpc_mdctx_unref (metadata_context); - - GPR_ASSERT (grpc_closure_list_empty (closure_list)); + channel_stack = gpr_malloc(grpc_channel_stack_size(&filters, 1)); + grpc_channel_stack_init(&exec_ctx, &filters, 1, NULL, &chan_args, + metadata_context, channel_stack); + GPR_ASSERT(channel_stack->count == 1); + channel_elem = grpc_channel_stack_element(channel_stack, 0); + channel_data = (int *)channel_elem->channel_data; + GPR_ASSERT(*channel_data == 0); + + call_stack = gpr_malloc(channel_stack->call_stack_size); + grpc_call_stack_init(&exec_ctx, channel_stack, NULL, NULL, call_stack); + GPR_ASSERT(call_stack->count == 1); + call_elem = grpc_call_stack_element(call_stack, 0); + GPR_ASSERT(call_elem->filter == channel_elem->filter); + GPR_ASSERT(call_elem->channel_data == channel_elem->channel_data); + call_data = (int *)call_elem->call_data; + GPR_ASSERT(*call_data == 0); + GPR_ASSERT(*channel_data == 1); + + grpc_call_stack_destroy(&exec_ctx, call_stack); + gpr_free(call_stack); + GPR_ASSERT(*channel_data == 2); + + grpc_channel_stack_destroy(&exec_ctx, channel_stack); + gpr_free(channel_stack); + + grpc_mdctx_unref(metadata_context); + + GPR_ASSERT(grpc_closure_list_empty(closure_list)); } -int -main (int argc, char **argv) -{ - grpc_test_init (argc, argv); - test_create_channel_stack (); +int main(int argc, char **argv) { + grpc_test_init(argc, argv); + test_create_channel_stack(); return 0; } |