src/lib/test-istream-concat.c

	test-istream-concat.c revision 32142dc3b19a1dff23ef351596703f1a43ea51a0
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose/* Copyright (c) 2009-2013 Dovecot authors, see the included COPYING file */
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#include "test-lib.h"
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#include "istream-private.h"
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#include "istream-concat.h"
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#include <stdlib.h>
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#include <fcntl.h>
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#include <unistd.h>
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#define TEST_MAX_ISTREAM_COUNT 10
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#define TEST_MAX_ISTREAM_SIZE 1024
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#define TEST_MAX_BUFFER_SIZE 128
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bosestatic void test_istream_concat_one(unsigned int buffer_size)
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose{
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    static const char *input_string = "xyz";
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#define STREAM_COUNT 5
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose#define STREAM_BYTES 3
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    struct istream *streams[STREAM_COUNT+1];
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    struct istream *input;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    const unsigned char *data;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    size_t size;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    unsigned int i, j;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    for (i = 0; i < STREAM_COUNT; i++) {
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        streams[i] = test_istream_create(input_string);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        test_istream_set_allow_eof(streams[i], TRUE);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        test_istream_set_size(streams[i], 0);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    }
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    streams[i] = NULL;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    input = i_stream_create_concat(streams);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    for (i = 0; i/STREAM_BYTES < STREAM_COUNT; i++) {
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        test_istream_set_size(streams[i/STREAM_BYTES], (i%STREAM_BYTES) + 1);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        test_assert(i_stream_read(input) == 1);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        if (i < buffer_size) {
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose            data = i_stream_get_data(input, &size);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose            test_assert(size == i+1);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        } else {
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose            i_stream_skip(input, 1);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose            data = i_stream_get_data(input, &size);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose            test_assert(size == buffer_size);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        }
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        for (j = 0; j < size; j++) {
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose            test_assert((char)data[j] == input_string[(input->v_offset + j) % STREAM_BYTES]);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose        }
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    }
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    i_stream_unref(&input);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose}
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bosestatic void test_istream_concat_random(void)
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose{
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    struct istream **streams, *input, *input1, *input2;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    const unsigned char *data;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    unsigned char *w_data;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    size_t size = 0;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    unsigned int i, j, offset, stream_count, data_len;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    srand(3);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    stream_count = (rand() % TEST_MAX_ISTREAM_COUNT) + 2;
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    streams = t_new(struct istream *, stream_count + 1);
8eb981dd8bc85aee7a913c6f0096ad47f3382339Sumit Bose    for (i = 0, offset = 0; i < stream_count; i++) {
        data_len = rand() % TEST_MAX_ISTREAM_SIZE + 1;
        w_data = t_malloc(data_len);
        for (j = 0; j < data_len; j++)
            w_data[j] = offset++;
        streams[i] = test_istream_create_data(w_data, data_len);
        test_istream_set_allow_eof(streams[i], TRUE);
    }
    streams[i] = NULL;
    i_assert(offset > 0);

    input = i_stream_create_concat(streams);
    i_stream_set_max_buffer_size(input, TEST_MAX_BUFFER_SIZE);
    input1 = i_stream_create_limit(input, (uoff_t)-1);
    input2 = i_stream_create_limit(input, (uoff_t)-1);
    i_stream_unref(&input);

    for (i = 0; i < 1000; i++) {
        input = i % 2 == 0 ? input1 : input2;
        if (rand() % 3 == 0) {
            i_stream_seek(input, rand() % offset);
        } else {
            ssize_t ret = i_stream_read(input);
            size = i_stream_get_data_size(input);
            if (ret == -2) {
                test_assert(size >= TEST_MAX_BUFFER_SIZE);
            } else if (input->v_offset + size != offset) {
                test_assert(ret > 0);
                test_assert(input->v_offset + ret <= offset);
                i_stream_skip(input, rand() % ret);

                data = i_stream_get_data(input, &size);
                for (j = 0; j < size; j++) {
                    test_assert(data[j] == (input->v_offset + j) % 256);
                }
            }
        }
    }
    for (i = 0; i < stream_count; i++)
        i_stream_unref(&streams[i]);
    i_stream_unref(&input1);
    i_stream_unref(&input2);
}

void test_istream_concat(void)
{
    unsigned int i;

    test_begin("istream concat");
    for (i = 1; i < STREAM_BYTES*STREAM_COUNT; i++) {
        test_istream_concat_one(i);
    }
    test_end();

    test_begin("istream concat random");
    for (i = 0; i < 100; i++) T_BEGIN {
        test_istream_concat_random();
    } T_END;
    test_end();
}