diff options
Diffstat (limited to 'test/core/surface/byte_buffer_reader_test.c')
-rw-r--r-- | test/core/surface/byte_buffer_reader_test.c | 207 |
1 files changed, 109 insertions, 98 deletions
diff --git a/test/core/surface/byte_buffer_reader_test.c b/test/core/surface/byte_buffer_reader_test.c index 560e0ac7b2..dd56ba86e4 100644 --- a/test/core/surface/byte_buffer_reader_test.c +++ b/test/core/surface/byte_buffer_reader_test.c @@ -48,72 +48,79 @@ #define LOG_TEST(x) gpr_log(GPR_INFO, "%s", x) -static void test_read_one_slice(void) { +static void +test_read_one_slice (void) +{ gpr_slice slice; grpc_byte_buffer *buffer; grpc_byte_buffer_reader reader; gpr_slice first_slice, second_slice; int first_code, second_code; - LOG_TEST("test_read_one_slice"); - slice = gpr_slice_from_copied_string("test"); - buffer = grpc_raw_byte_buffer_create(&slice, 1); - gpr_slice_unref(slice); - grpc_byte_buffer_reader_init(&reader, buffer); - first_code = grpc_byte_buffer_reader_next(&reader, &first_slice); - GPR_ASSERT(first_code != 0); - GPR_ASSERT(memcmp(GPR_SLICE_START_PTR(first_slice), "test", 4) == 0); - gpr_slice_unref(first_slice); - second_code = grpc_byte_buffer_reader_next(&reader, &second_slice); - GPR_ASSERT(second_code == 0); - grpc_byte_buffer_destroy(buffer); + LOG_TEST ("test_read_one_slice"); + slice = gpr_slice_from_copied_string ("test"); + buffer = grpc_raw_byte_buffer_create (&slice, 1); + gpr_slice_unref (slice); + grpc_byte_buffer_reader_init (&reader, buffer); + first_code = grpc_byte_buffer_reader_next (&reader, &first_slice); + GPR_ASSERT (first_code != 0); + GPR_ASSERT (memcmp (GPR_SLICE_START_PTR (first_slice), "test", 4) == 0); + gpr_slice_unref (first_slice); + second_code = grpc_byte_buffer_reader_next (&reader, &second_slice); + GPR_ASSERT (second_code == 0); + grpc_byte_buffer_destroy (buffer); } -static void test_read_one_slice_malloc(void) { +static void +test_read_one_slice_malloc (void) +{ gpr_slice slice; grpc_byte_buffer *buffer; grpc_byte_buffer_reader reader; gpr_slice first_slice, second_slice; int first_code, second_code; - LOG_TEST("test_read_one_slice_malloc"); - slice = gpr_slice_malloc(4); - memcpy(GPR_SLICE_START_PTR(slice), "test", 4); - buffer = grpc_raw_byte_buffer_create(&slice, 1); - gpr_slice_unref(slice); - grpc_byte_buffer_reader_init(&reader, buffer); - first_code = grpc_byte_buffer_reader_next(&reader, &first_slice); - GPR_ASSERT(first_code != 0); - GPR_ASSERT(memcmp(GPR_SLICE_START_PTR(first_slice), "test", 4) == 0); - gpr_slice_unref(first_slice); - second_code = grpc_byte_buffer_reader_next(&reader, &second_slice); - GPR_ASSERT(second_code == 0); - grpc_byte_buffer_destroy(buffer); + LOG_TEST ("test_read_one_slice_malloc"); + slice = gpr_slice_malloc (4); + memcpy (GPR_SLICE_START_PTR (slice), "test", 4); + buffer = grpc_raw_byte_buffer_create (&slice, 1); + gpr_slice_unref (slice); + grpc_byte_buffer_reader_init (&reader, buffer); + first_code = grpc_byte_buffer_reader_next (&reader, &first_slice); + GPR_ASSERT (first_code != 0); + GPR_ASSERT (memcmp (GPR_SLICE_START_PTR (first_slice), "test", 4) == 0); + gpr_slice_unref (first_slice); + second_code = grpc_byte_buffer_reader_next (&reader, &second_slice); + GPR_ASSERT (second_code == 0); + grpc_byte_buffer_destroy (buffer); } -static void test_read_none_compressed_slice(void) { +static void +test_read_none_compressed_slice (void) +{ gpr_slice slice; grpc_byte_buffer *buffer; grpc_byte_buffer_reader reader; gpr_slice first_slice, second_slice; int first_code, second_code; - LOG_TEST("test_read_none_compressed_slice"); - slice = gpr_slice_from_copied_string("test"); - buffer = grpc_raw_byte_buffer_create(&slice, 1); - gpr_slice_unref(slice); - grpc_byte_buffer_reader_init(&reader, buffer); - first_code = grpc_byte_buffer_reader_next(&reader, &first_slice); - GPR_ASSERT(first_code != 0); - GPR_ASSERT(memcmp(GPR_SLICE_START_PTR(first_slice), "test", 4) == 0); - gpr_slice_unref(first_slice); - second_code = grpc_byte_buffer_reader_next(&reader, &second_slice); - GPR_ASSERT(second_code == 0); - grpc_byte_buffer_destroy(buffer); + LOG_TEST ("test_read_none_compressed_slice"); + slice = gpr_slice_from_copied_string ("test"); + buffer = grpc_raw_byte_buffer_create (&slice, 1); + gpr_slice_unref (slice); + grpc_byte_buffer_reader_init (&reader, buffer); + first_code = grpc_byte_buffer_reader_next (&reader, &first_slice); + GPR_ASSERT (first_code != 0); + GPR_ASSERT (memcmp (GPR_SLICE_START_PTR (first_slice), "test", 4) == 0); + gpr_slice_unref (first_slice); + second_code = grpc_byte_buffer_reader_next (&reader, &second_slice); + GPR_ASSERT (second_code == 0); + grpc_byte_buffer_destroy (buffer); } -static void read_compressed_slice(grpc_compression_algorithm algorithm, - size_t input_size) { +static void +read_compressed_slice (grpc_compression_algorithm algorithm, size_t input_size) +{ gpr_slice input_slice; gpr_slice_buffer sliceb_in; gpr_slice_buffer sliceb_out; @@ -122,76 +129,80 @@ static void read_compressed_slice(grpc_compression_algorithm algorithm, gpr_slice read_slice; size_t read_count = 0; - gpr_slice_buffer_init(&sliceb_in); - gpr_slice_buffer_init(&sliceb_out); - - input_slice = gpr_slice_malloc(input_size); - memset(GPR_SLICE_START_PTR(input_slice), 'a', input_size); - gpr_slice_buffer_add(&sliceb_in, input_slice); /* takes ownership */ - GPR_ASSERT(grpc_msg_compress(algorithm, &sliceb_in, &sliceb_out)); - - buffer = grpc_raw_compressed_byte_buffer_create(sliceb_out.slices, - sliceb_out.count, algorithm); - grpc_byte_buffer_reader_init(&reader, buffer); - - while (grpc_byte_buffer_reader_next(&reader, &read_slice)) { - GPR_ASSERT(memcmp(GPR_SLICE_START_PTR(read_slice), - GPR_SLICE_START_PTR(input_slice) + read_count, - GPR_SLICE_LENGTH(read_slice)) == 0); - read_count += GPR_SLICE_LENGTH(read_slice); - gpr_slice_unref(read_slice); - } - GPR_ASSERT(read_count == input_size); - grpc_byte_buffer_reader_destroy(&reader); - grpc_byte_buffer_destroy(buffer); - gpr_slice_buffer_destroy(&sliceb_out); - gpr_slice_buffer_destroy(&sliceb_in); + gpr_slice_buffer_init (&sliceb_in); + gpr_slice_buffer_init (&sliceb_out); + + input_slice = gpr_slice_malloc (input_size); + memset (GPR_SLICE_START_PTR (input_slice), 'a', input_size); + gpr_slice_buffer_add (&sliceb_in, input_slice); /* takes ownership */ + GPR_ASSERT (grpc_msg_compress (algorithm, &sliceb_in, &sliceb_out)); + + buffer = grpc_raw_compressed_byte_buffer_create (sliceb_out.slices, sliceb_out.count, algorithm); + grpc_byte_buffer_reader_init (&reader, buffer); + + while (grpc_byte_buffer_reader_next (&reader, &read_slice)) + { + GPR_ASSERT (memcmp (GPR_SLICE_START_PTR (read_slice), GPR_SLICE_START_PTR (input_slice) + read_count, GPR_SLICE_LENGTH (read_slice)) == 0); + read_count += GPR_SLICE_LENGTH (read_slice); + gpr_slice_unref (read_slice); + } + GPR_ASSERT (read_count == input_size); + grpc_byte_buffer_reader_destroy (&reader); + grpc_byte_buffer_destroy (buffer); + gpr_slice_buffer_destroy (&sliceb_out); + gpr_slice_buffer_destroy (&sliceb_in); } -static void test_read_gzip_compressed_slice(void) { +static void +test_read_gzip_compressed_slice (void) +{ const size_t INPUT_SIZE = 2048; - LOG_TEST("test_read_gzip_compressed_slice"); - read_compressed_slice(GRPC_COMPRESS_GZIP, INPUT_SIZE); + LOG_TEST ("test_read_gzip_compressed_slice"); + read_compressed_slice (GRPC_COMPRESS_GZIP, INPUT_SIZE); } -static void test_read_deflate_compressed_slice(void) { +static void +test_read_deflate_compressed_slice (void) +{ const size_t INPUT_SIZE = 2048; - LOG_TEST("test_read_deflate_compressed_slice"); - read_compressed_slice(GRPC_COMPRESS_DEFLATE, INPUT_SIZE); + LOG_TEST ("test_read_deflate_compressed_slice"); + read_compressed_slice (GRPC_COMPRESS_DEFLATE, INPUT_SIZE); } -static void test_byte_buffer_from_reader(void) { +static void +test_byte_buffer_from_reader (void) +{ gpr_slice slice; grpc_byte_buffer *buffer, *buffer_from_reader; grpc_byte_buffer_reader reader; - LOG_TEST("test_byte_buffer_from_reader"); - slice = gpr_slice_malloc(4); - memcpy(GPR_SLICE_START_PTR(slice), "test", 4); - buffer = grpc_raw_byte_buffer_create(&slice, 1); - gpr_slice_unref(slice); - grpc_byte_buffer_reader_init(&reader, buffer); - - buffer_from_reader = grpc_raw_byte_buffer_from_reader(&reader); - GPR_ASSERT(buffer->type == buffer_from_reader->type); - GPR_ASSERT(buffer_from_reader->data.raw.compression == GRPC_COMPRESS_NONE); - GPR_ASSERT(buffer_from_reader->data.raw.slice_buffer.count == 1); - GPR_ASSERT(memcmp(GPR_SLICE_START_PTR( - buffer_from_reader->data.raw.slice_buffer.slices[0]), - "test", 4) == 0); - - grpc_byte_buffer_destroy(buffer); - grpc_byte_buffer_destroy(buffer_from_reader); + LOG_TEST ("test_byte_buffer_from_reader"); + slice = gpr_slice_malloc (4); + memcpy (GPR_SLICE_START_PTR (slice), "test", 4); + buffer = grpc_raw_byte_buffer_create (&slice, 1); + gpr_slice_unref (slice); + grpc_byte_buffer_reader_init (&reader, buffer); + + buffer_from_reader = grpc_raw_byte_buffer_from_reader (&reader); + GPR_ASSERT (buffer->type == buffer_from_reader->type); + GPR_ASSERT (buffer_from_reader->data.raw.compression == GRPC_COMPRESS_NONE); + GPR_ASSERT (buffer_from_reader->data.raw.slice_buffer.count == 1); + GPR_ASSERT (memcmp (GPR_SLICE_START_PTR (buffer_from_reader->data.raw.slice_buffer.slices[0]), "test", 4) == 0); + + grpc_byte_buffer_destroy (buffer); + grpc_byte_buffer_destroy (buffer_from_reader); } -int main(int argc, char **argv) { - grpc_test_init(argc, argv); - test_read_one_slice(); - test_read_one_slice_malloc(); - test_read_none_compressed_slice(); - test_read_gzip_compressed_slice(); - test_read_deflate_compressed_slice(); - test_byte_buffer_from_reader(); +int +main (int argc, char **argv) +{ + grpc_test_init (argc, argv); + test_read_one_slice (); + test_read_one_slice_malloc (); + test_read_none_compressed_slice (); + test_read_gzip_compressed_slice (); + test_read_deflate_compressed_slice (); + test_byte_buffer_from_reader (); return 0; } |