1 #include <gio/gio.h>
2
3 static void
4 test_write (void)
5 {
6 GOutputStream *base;
7 GOutputStream *out;
8 GError *error;
9 const gchar buffer[] = "abcdefghijklmnopqrstuvwxyz";
10
11 base = g_memory_output_stream_new (g_malloc0 (20), 20, NULL, g_free);
12 out = g_buffered_output_stream_new (base);
13
14 g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 4096);
15 g_assert (!g_buffered_output_stream_get_auto_grow (G_BUFFERED_OUTPUT_STREAM (out)));
16 g_object_set (out, "auto-grow", TRUE, NULL);
17 g_assert (g_buffered_output_stream_get_auto_grow (G_BUFFERED_OUTPUT_STREAM (out)));
18 g_object_set (out, "auto-grow", FALSE, NULL);
19
20 g_buffered_output_stream_set_buffer_size (G_BUFFERED_OUTPUT_STREAM (out), 16);
21 g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 16);
22
23 error = NULL;
24 g_assert_cmpint (g_output_stream_write (out, buffer, 10, NULL, &error), ==, 10);
25 g_assert_no_error (error);
26
27 g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
28
29 g_assert_cmpint (g_output_stream_write (out, buffer + 10, 10, NULL, &error), ==, 6);
30 g_assert_no_error (error);
31
32 g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
33 g_assert (g_output_stream_flush (out, NULL, &error));
34 g_assert_no_error (error);
35 g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 16);
36
37 g_assert_cmpstr (g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (base)), ==, "abcdefghijklmnop");
38
39 g_object_unref (out);
40 g_object_unref (base);
41 }
42
43 static void
44 test_grow (void)
45 {
46 GOutputStream *base;
47 GOutputStream *out;
48 GError *error;
49 const gchar buffer[] = "abcdefghijklmnopqrstuvwxyz";
50 gint size;
51 gboolean grow;
52
53 base = g_memory_output_stream_new (g_malloc0 (30), 30, g_realloc, g_free);
54 out = g_buffered_output_stream_new_sized (base, 16);
55
56 g_buffered_output_stream_set_auto_grow (G_BUFFERED_OUTPUT_STREAM (out), TRUE);
57
58 g_object_get (out, "buffer-size", &size, "auto-grow", &grow, NULL);
59 g_assert_cmpint (size, ==, 16);
60 g_assert (grow);
61
62 g_assert (g_seekable_can_seek (G_SEEKABLE (out)));
63
64 error = NULL;
65 g_assert_cmpint (g_output_stream_write (out, buffer, 10, NULL, &error), ==, 10);
66 g_assert_no_error (error);
67
68 g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 16);
69 g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
70
71 g_assert_cmpint (g_output_stream_write (out, buffer + 10, 10, NULL, &error), ==, 10);
72 g_assert_no_error (error);
73
74 g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), >=, 20);
75 g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
76
77 g_assert (g_output_stream_flush (out, NULL, &error));
78 g_assert_no_error (error);
79
80 g_assert_cmpstr (g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (base)), ==, "abcdefghijklmnopqrst");
81
82 g_object_unref (out);
83 g_object_unref (base);
84 }
85
86 static void
87 test_close (void)
88 {
89 GOutputStream *base;
90 GOutputStream *out;
91 GError *error;
92
93 base = g_memory_output_stream_new (g_malloc0 (30), 30, g_realloc, g_free);
94 out = g_buffered_output_stream_new (base);
95
96 g_assert (g_filter_output_stream_get_close_base_stream (G_FILTER_OUTPUT_STREAM (out)));
97
98 error = NULL;
99 g_assert (g_output_stream_close (out, NULL, &error));
100 g_assert_no_error (error);
101 g_assert (g_output_stream_is_closed (base));
102
103 g_object_unref (out);
104 g_object_unref (base);
105
106 base = g_memory_output_stream_new (g_malloc0 (30), 30, g_realloc, g_free);
107 out = g_buffered_output_stream_new (base);
108
109 g_filter_output_stream_set_close_base_stream (G_FILTER_OUTPUT_STREAM (out), FALSE);
110
111 error = NULL;
112 g_assert (g_output_stream_close (out, NULL, &error));
113 g_assert_no_error (error);
114 g_assert (!g_output_stream_is_closed (base));
115
116 g_object_unref (out);
117 g_object_unref (base);
118 }
119
120 static void
121 test_seek (void)
122 {
123 GMemoryOutputStream *base;
124 GOutputStream *out;
125 GSeekable *seekable;
126 GError *error;
127 gsize bytes_written;
128 gboolean ret;
129 const gchar buffer[] = "abcdefghijklmnopqrstuvwxyz";
130
131 base = G_MEMORY_OUTPUT_STREAM (g_memory_output_stream_new (g_malloc0 (30), 30, NULL, g_free));
132 out = g_buffered_output_stream_new_sized (G_OUTPUT_STREAM (base), 8);
133 seekable = G_SEEKABLE (out);
134 error = NULL;
135
136 /* Write data */
137 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 0);
138 ret = g_output_stream_write_all (out, buffer, 4, &bytes_written, NULL, &error);
139 g_assert_no_error (error);
140 g_assert_cmpint (bytes_written, ==, 4);
141 g_assert (ret);
142 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 4);
143 g_assert_cmpint (g_memory_output_stream_get_data_size (base), ==, 0);
144
145 /* Forward relative seek */
146 ret = g_seekable_seek (seekable, 2, G_SEEK_CUR, NULL, &error);
147 g_assert_no_error (error);
148 g_assert (ret);
149 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 6);
150 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
151 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
152 g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
153 g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
154 ret = g_output_stream_write_all (out, buffer, 2, &bytes_written, NULL, &error);
155 g_assert_no_error (error);
156 g_assert (ret);
157 g_assert_cmpint (bytes_written, ==, 2);
158 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 8);
159
160 /* Backward relative seek */
161 ret = g_seekable_seek (seekable, -4, G_SEEK_CUR, NULL, &error);
162 g_assert_no_error (error);
163 g_assert (ret);
164 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 4);
165 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
166 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
167 g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
168 g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
169 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
170 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
171 ret = g_output_stream_write_all (out, buffer, 2, &bytes_written, NULL, &error);
172 g_assert_no_error (error);
173 g_assert (ret);
174 g_assert_cmpint (bytes_written, ==, 2);
175 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 6);
176
177 /* From start */
178 ret = g_seekable_seek (seekable, 2, G_SEEK_SET, NULL, &error);
179 g_assert_no_error (error);
180 g_assert (ret);
181 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 2);
182 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
183 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
184 g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
185 g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
186 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[4]);
187 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[5]);
188 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
189 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
190 ret = g_output_stream_write_all (out, buffer, 2, &bytes_written, NULL, &error);
191 g_assert_no_error (error);
192 g_assert (ret);
193 g_assert_cmpint (bytes_written, ==, 2);
194 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 4);
195
196 /* From end */
197 ret = g_seekable_seek (seekable, 6 - 30, G_SEEK_END, NULL, &error);
198 g_assert_no_error (error);
199 g_assert (ret);
200 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 6);
201 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
202 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
203 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
204 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
205 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[4]);
206 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[5]);
207 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
208 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
209 ret = g_output_stream_write_all (out, buffer + 2, 2, &bytes_written, NULL, &error);
210 g_assert_no_error (error);
211 g_assert (ret);
212 g_assert_cmpint (bytes_written, ==, 2);
213 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 8);
214
215 /* Check flush */
216 ret = g_output_stream_flush (out, NULL, &error);
217 g_assert_no_error (error);
218 g_assert (ret);
219 g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 8);
220 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
221 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
222 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
223 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
224 g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[4]);
225 g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[5]);
226 g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
227 g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
228
229 g_object_unref (out);
230 g_object_unref (base);
231 }
232
233 static void
234 test_truncate(void)
235 {
236 GMemoryOutputStream *base_stream;
237 GOutputStream *stream;
238 GSeekable *seekable;
239 GError *error;
240 gsize bytes_written;
241 guchar *stream_data;
242 gsize len;
243 gboolean res;
244
245 len = 8;
246
247 /* Create objects */
248 stream_data = g_malloc0 (len);
249 base_stream = G_MEMORY_OUTPUT_STREAM (g_memory_output_stream_new (stream_data, len, g_realloc, g_free));
250 stream = g_buffered_output_stream_new_sized (G_OUTPUT_STREAM (base_stream), 8);
251 seekable = G_SEEKABLE (stream);
252
253 g_assert (g_seekable_can_truncate (seekable));
254
255 /* Write */
256 g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, len);
257 g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 0);
258
259 error = NULL;
260 res = g_output_stream_write_all (stream, "ab", 2, &bytes_written, NULL, &error);
261 g_assert_no_error (error);
262 g_assert (res);
263 res = g_output_stream_write_all (stream, "cd", 2, &bytes_written, NULL, &error);
264 g_assert_no_error (error);
265 g_assert (res);
266
267 res = g_output_stream_flush (stream, NULL, &error);
268 g_assert_no_error (error);
269 g_assert (res);
270
271 g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, len);
272 g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 4);
273 stream_data = g_memory_output_stream_get_data (base_stream);
274 g_assert_cmpint (stream_data[0], ==, 'a');
275 g_assert_cmpint (stream_data[1], ==, 'b');
276 g_assert_cmpint (stream_data[2], ==, 'c');
277 g_assert_cmpint (stream_data[3], ==, 'd');
278
279 /* Truncate at position */
280 res = g_seekable_truncate (seekable, 4, NULL, &error);
281 g_assert_no_error (error);
282 g_assert (res);
283 g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, 4);
284 g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 4);
285 stream_data = g_memory_output_stream_get_data (base_stream);
286 g_assert_cmpint (stream_data[0], ==, 'a');
287 g_assert_cmpint (stream_data[1], ==, 'b');
288 g_assert_cmpint (stream_data[2], ==, 'c');
289 g_assert_cmpint (stream_data[3], ==, 'd');
290
291 /* Truncate beyond position */
292 res = g_seekable_truncate (seekable, 6, NULL, &error);
293 g_assert_no_error (error);
294 g_assert (res);
295 g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, 6);
296 g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 6);
297 stream_data = g_memory_output_stream_get_data (base_stream);
298 g_assert_cmpint (stream_data[0], ==, 'a');
299 g_assert_cmpint (stream_data[1], ==, 'b');
300 g_assert_cmpint (stream_data[2], ==, 'c');
301 g_assert_cmpint (stream_data[3], ==, 'd');
302
303 /* Truncate before position */
304 res = g_seekable_truncate (seekable, 2, NULL, &error);
305 g_assert_no_error (error);
306 g_assert (res);
307 g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, 2);
308 g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 2);
309 stream_data = g_memory_output_stream_get_data (base_stream);
310 g_assert_cmpint (stream_data[0], ==, 'a');
311 g_assert_cmpint (stream_data[1], ==, 'b');
312
313 g_object_unref (stream);
314 g_object_unref (base_stream);
315 }
316
317 int
318 main (int argc, char *argv[])
319 {
320 g_test_init (&argc, &argv, NULL);
321
322 g_test_add_func ("/buffered-output-stream/write", test_write);
323 g_test_add_func ("/buffered-output-stream/grow", test_grow);
324 g_test_add_func ("/buffered-output-stream/seek", test_seek);
325 g_test_add_func ("/buffered-output-stream/truncate", test_truncate);
326 g_test_add_func ("/filter-output-stream/close", test_close);
327
328 return g_test_run ();
329 }