test-istream-chain.c revision bac6427712674cd4b6146ef51f90c14eb9603db6
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen/* Copyright (c) 2016 Dovecot authors, see the included COPYING file */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen#include "test-lib.h"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen#include "istream-private.h"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen#include "istream-chain.h"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainenstatic void test_istream_chain_basic(void)
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen{
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream *input, *test_input, *test_input2;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream_chain *chain;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen const unsigned char *data;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen size_t size;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_begin("istream chain");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input = test_istream_create("stream1");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input2 = test_istream_create("STREAM2");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen input = i_stream_create_chain(&chain);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* no input */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* stream1 input */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 7);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen data = i_stream_get_data(input, &size);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 7 && memcmp(data, "stream1", 7) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen data = i_stream_get_data(input, &size);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 7 && memcmp(data, "stream1", 7) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* STREAM2 input */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input2);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 7);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen data = i_stream_get_data(input, &size);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 14 && memcmp(data, "stream1STREAM2", 14) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen data = i_stream_get_data(input, &size);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 14 && memcmp(data, "stream1STREAM2", 14) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* EOF */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append_eof(chain);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == -1 &&
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen input->eof && input->stream_errno == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen data = i_stream_get_data(input, &size);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 14 && memcmp(data, "stream1STREAM2", 14) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_unref(&input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_is_eof(test_input));
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_is_eof(test_input2));
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_unref(&test_input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_unref(&test_input2);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_end();
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen}
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainenstatic void test_istream_chain_early_end(void)
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen{
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream *input, *test_input;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream_chain *chain;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_begin("istream chain early end");
ba1a5db879b08d2fefcb42160af67853cdfe7687Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input = test_istream_create("string");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_istream_set_size(test_input, 3);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_istream_set_allow_eof(test_input, FALSE);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen input = i_stream_create_chain(&chain);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 3);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_istream_set_size(test_input, 5);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 2);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* with current implementation we could skip less than 5 and have
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen v_offset<5, but I don't think that can work in all situations.
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen the normal case is anyway that we'll read everything up until some
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen point and skip over all the data up to there. */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_skip(input, 5);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_unref(&input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(test_input->v_offset == 5);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_unref(&test_input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_end();
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen}
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainenstatic void test_istream_chain_accumulate(void)
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen{
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream *input;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream *test_input, *test_input2, *test_input3, *test_input4,
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen *test_input5;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen struct istream_chain *chain;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen const unsigned char *data;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen size_t size;
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_begin("istream chain");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input = test_istream_create("aaaaaaaaaaaaaaaaaaaa");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input2 = test_istream_create("bbbbbbbbbbbbbbbbbbbbbbbbb");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input3 = test_istream_create("cccccccccccccccccccccccccccccc");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input4 = test_istream_create("ddddddddddddddddddddddddd");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_input5 = test_istream_create("eeeeeeeeeeeeeeeeeeee");
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen input = i_stream_create_chain(&chain);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* no input */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read(input) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* first stream */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read_data(input, &data, &size, 0) == 1);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 20);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(memcmp(data, "aaaaaaaaaaaaaaaaaaaa", 20) == 0);
ba1a5db879b08d2fefcb42160af67853cdfe7687Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* partially skip */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_skip(input, 12);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* second stream */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input2);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read_data(input, &data, &size, 10) == 1);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 33);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(memcmp(data, "aaaaaaaa"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen "bbbbbbbbbbbbbbbbbbbbbbbbb", 33) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* partially skip */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_skip(input, 12);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* third stream */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input3);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read_data(input, &data, &size, 25) == 1);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 51);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(memcmp(data, "bbbbbbbbbbbbbbbbbbbbb"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen "cccccccccccccccccccccccccccccc", 51) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* partially skip */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_skip(input, 12);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* forth stream */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input4);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read_data(input, &data, &size, 40) == 1);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(size == 64);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(memcmp(data, "bbbbbbbbb"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen "cccccccccccccccccccccccccccccc"
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen "ddddddddddddddddddddddddd", 64) == 0);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* partially skip */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_skip(input, 6);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen /* fifth stream */
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen i_stream_chain_append(chain, test_input5);
6509aa7306649042e828fef9e2f280f4535ea137Timo Sirainen test_assert(i_stream_read_data(input, &data, &size, 60) == 1);
test_assert(size == 78);
test_assert(memcmp(data, "bbb"
"cccccccccccccccccccccccccccccc"
"ddddddddddddddddddddddddd"
"eeeeeeeeeeeeeeeeeeee", 78) == 0);
/* EOF */
i_stream_chain_append_eof(chain);
test_assert(i_stream_read(input) == -1);
test_assert(input->eof && input->stream_errno == 0);
test_assert(i_stream_read_data(input, &data, &size, 78) == -1);
test_assert(size == 78);
test_assert(memcmp(data, "bbb"
"cccccccccccccccccccccccccccccc"
"ddddddddddddddddddddddddd"
"eeeeeeeeeeeeeeeeeeee", 78) == 0);
/* skip rest */
i_stream_skip(input, 78);
test_assert(i_stream_read(input) == -1);
test_assert(input->eof && input->stream_errno == 0);
data = i_stream_get_data(input, &size);
test_assert(size == 0);
i_stream_unref(&input);
i_stream_unref(&test_input);
i_stream_unref(&test_input2);
i_stream_unref(&test_input3);
test_end();
}
void test_istream_chain(void)
{
test_istream_chain_basic();
test_istream_chain_early_end();
test_istream_chain_accumulate();
}