gio/tests/buffered-output-stream.c

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