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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#include <test.h>
#include <alloc.h>
#include <writer.h>
static Writer *global_w;
static bool global_w_closed;
/* Override libc's fwrite(). */
static size_t CFENGINE_TEST_fwrite(const void *ptr, size_t size, size_t nmemb,
ARG_UNUSED FILE *stream)
{
for (int i = 0; i < nmemb; ++i)
{
char *b = xstrndup(ptr + i * size, size);
WriterWrite(global_w, b);
free(b);
}
return nmemb * size;
}
/* Override libc's fclose(). */
static int CFENGINE_TEST_fclose(ARG_UNUSED FILE *stream)
{
global_w_closed = true;
return 0;
}
#include <writer.c> /* MUST be included after the overrides. */
void test_empty_file_buffer(void)
{
global_w = StringWriter();
global_w_closed = false;
Writer *w = FileWriter(NULL);
assert_int_equal(StringWriterLength(global_w), 0);
assert_string_equal(StringWriterData(global_w), "");
WriterClose(w);
WriterClose(global_w);
assert_int_equal(global_w_closed, true);
}
void test_write_empty_file_buffer(void)
{
global_w = StringWriter();
Writer *w = FileWriter(NULL);
WriterWrite(w, "");
assert_int_equal(StringWriterLength(global_w), 0);
assert_string_equal(StringWriterData(global_w), "");
WriterClose(w);
WriterClose(global_w);
assert_int_equal(global_w_closed, true);
}
void test_write_file_buffer(void)
{
global_w = StringWriter();
Writer *w = FileWriter(NULL);
WriterWrite(w, "123");
assert_int_equal(StringWriterLength(global_w), 3);
assert_string_equal(StringWriterData(global_w), "123");
WriterClose(w);
WriterClose(global_w);
assert_int_equal(global_w_closed, true);
}
void test_multiwrite_file_buffer(void)
{
global_w = StringWriter();
Writer *w = FileWriter(NULL);
WriterWrite(w, "123");
WriterWrite(w, "456");
assert_int_equal(StringWriterLength(global_w), 6);
assert_string_equal(StringWriterData(global_w), "123456");
WriterClose(w);
WriterClose(global_w);
assert_int_equal(global_w_closed, true);
}
int main()
{
PRINT_TEST_BANNER();
const UnitTest tests[] =
{
unit_test(test_empty_file_buffer),
unit_test(test_write_empty_file_buffer),
unit_test(test_write_file_buffer),
unit_test(test_multiwrite_file_buffer),
};
return run_tests(tests);
}
/* STUB */
void __ProgrammingError(ARG_UNUSED const char *file,
ARG_UNUSED int lineno,
ARG_UNUSED const char *format, ...)
{
fail();
exit(42);
}
void FatalError(ARG_UNUSED char *s, ...)
{
fail();
exit(42);
}
|
