1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/* Copyright (c) 2013-2018 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "buffer.h"
#include "str.h"
#include "istream.h"
#include "ostream.h"
#include "ostream-dot.h"
#include "test-common.h"
struct dot_test {
const char *input;
const char *output;
};
static void test_ostream_dot_one(const struct dot_test *test)
{
struct istream *test_input;
struct ostream *output, *test_output;
buffer_t *output_data;
const unsigned char *data;
size_t size;
ssize_t ret;
test_input = test_istream_create(test->input);
output_data = t_buffer_create(1024);
test_output = o_stream_create_buffer(output_data);
output = o_stream_create_dot(test_output, FALSE);
while ((ret = i_stream_read(test_input)) > 0 || ret == -2) {
data = i_stream_get_data(test_input, &size);
ret = o_stream_send(output, data, size);
test_assert(ret >= 0);
if (ret <= 0)
break;
i_stream_skip(test_input, ret);
}
test_assert(test_input->eof);
test_assert(o_stream_finish(output) > 0);
test_assert(output->offset == strlen(test->input));
test_assert(test_output->offset == strlen(test->output));
o_stream_unref(&output);
o_stream_unref(&test_output);
test_assert(strcmp(str_c(output_data), test->output) == 0);
i_stream_unref(&test_input);
}
static void test_ostream_dot(void)
{
static struct dot_test tests[] = {
{ "foo\r\n.\r\n", "foo\r\n..\r\n.\r\n" },
{ "foo\n.\n", "foo\r\n..\r\n.\r\n" },
{ ".foo\r\n.\r\nfoo\r\n", "..foo\r\n..\r\nfoo\r\n.\r\n" },
{ ".foo\n.\nfoo\n", "..foo\r\n..\r\nfoo\r\n.\r\n" },
{ "\r\n", "\r\n.\r\n" },
{ "\n", "\r\n.\r\n" },
{ "", "\r\n.\r\n" },
};
unsigned int i;
for (i = 0; i < N_ELEMENTS(tests); i++) {
test_begin(t_strdup_printf("dot ostream[%d]:", i));
test_ostream_dot_one(&tests[i]);
test_end();
}
}
static void test_ostream_dot_parent_almost_full(void)
{
buffer_t *output_data;
struct ostream *test_output, *output;
ssize_t ret;
test_begin("dot ostream parent almost full");
output_data = t_buffer_create(1024);
test_output = test_ostream_create_nonblocking(output_data, 1);
test_ostream_set_max_output_size(test_output, 1);
output = o_stream_create_dot(test_output, FALSE);
ret = o_stream_send(output, "a", 1);
test_assert(ret == 0);
ret = o_stream_send(output, "bc", 2);
test_assert(ret == 0);
o_stream_unref(&output);
o_stream_unref(&test_output);
test_end();
}
int main(void)
{
static void (*const test_functions[])(void) = {
test_ostream_dot,
test_ostream_dot_parent_almost_full,
NULL
};
return test_run(test_functions);
}
|