diff options
Diffstat (limited to '')
-rw-r--r-- | src/lib/test-ostream-buffer.c | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/src/lib/test-ostream-buffer.c b/src/lib/test-ostream-buffer.c new file mode 100644 index 0000000..4a449f6 --- /dev/null +++ b/src/lib/test-ostream-buffer.c @@ -0,0 +1,108 @@ +/* Copyright (c) 2017-2018 Dovecot authors, see the included COPYING file */ + +#include "test-lib.h" +#include "buffer.h" +#include "str.h" +#include "randgen.h" +#include "istream.h" +#include "ostream.h" + +#define MAX_BUFSIZE 256 + +static void test_ostream_buffer_random_once(void) +{ + buffer_t *buffer; + struct ostream *output; + char buf[MAX_BUFSIZE*4], randbuf[MAX_BUFSIZE]; + unsigned int i, offset, size; + + buffer = buffer_create_dynamic(default_pool, 8); + + memset(buf, 0, sizeof(buf)); + + output = o_stream_create_buffer(buffer); + o_stream_cork(output); + + size = i_rand_minmax(1, MAX_BUFSIZE); + random_fill(randbuf, size); + memcpy(buf, randbuf, size); + test_assert(o_stream_send(output, buf, size) > 0); + + for (i = 0; i < 10; i++) { + offset = i_rand_limit(MAX_BUFSIZE * 3); + size = i_rand_minmax(1, MAX_BUFSIZE); + random_fill(randbuf, size); + memcpy(buf + offset, randbuf, size); + test_assert(o_stream_pwrite(output, randbuf, size, offset) == 0); + if (i_rand_limit(10) == 0) + test_assert(o_stream_flush(output) > 0); + } + + o_stream_uncork(output); + test_assert(o_stream_finish(output) > 0); + + i_assert(buffer->used <= MAX_BUFSIZE*4); + test_assert(memcmp(buf, buffer->data, buffer->used) == 0); + + o_stream_unref(&output); + buffer_free(&buffer); +} + +static void test_ostream_buffer_random(void) +{ + unsigned int i; + + test_begin("ostream buffer pwrite random"); + for (i = 0; i < 100; i++) T_BEGIN { + test_ostream_buffer_random_once(); + } T_END; + test_end(); +} + +static void test_ostream_buffer_size(void) +{ + struct ostream *output; + string_t *str = t_str_new(64); + + test_begin("ostream buffer size/available"); + output = o_stream_create_buffer(str); + test_assert(o_stream_get_buffer_used_size(output) == 0); + test_assert(o_stream_get_buffer_avail_size(output) == SIZE_MAX); + + /* test shrinking sink's max buffer size */ + o_stream_set_max_buffer_size(output, 10); + test_assert(o_stream_get_buffer_used_size(output) == 0); + test_assert(o_stream_get_buffer_avail_size(output) == 10); + + /* partial send */ + const char *partial_input = "01234567890123456789"; + ssize_t ret = o_stream_send_str(output, partial_input); + test_assert(ret == 10); + test_assert(o_stream_get_buffer_used_size(output) == 10); + test_assert(o_stream_get_buffer_avail_size(output) == 0); + + /* increase max buffer size so that it can hold the whole message */ + o_stream_set_max_buffer_size(output, 100); + test_assert(o_stream_get_buffer_used_size(output) == 10); + test_assert(o_stream_get_buffer_avail_size(output) == 90); + + /* send the rest */ + ret += o_stream_send_str(output, partial_input + ret); + test_assert(ret == (ssize_t)strlen(partial_input)); + test_assert(output->offset == str_len(str)); + test_assert(o_stream_get_buffer_used_size(output) == 20); + test_assert(o_stream_get_buffer_avail_size(output) == 80); + + /* check buffered data */ + test_assert(strcmp(str_c(str), partial_input) == 0); + + o_stream_unref(&output); + + test_end(); +} + +void test_ostream_buffer(void) +{ + test_ostream_buffer_random(); + test_ostream_buffer_size(); +} |