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
  47   base = g_memory_output_stream_new (g_malloc0 (30), 30, g_realloc, g_free);
  48   out = g_buffered_output_stream_new (base);
  49
  50   g_buffered_output_stream_set_buffer_size (G_BUFFERED_OUTPUT_STREAM (out), 16);
  51   g_buffered_output_stream_set_auto_grow (G_BUFFERED_OUTPUT_STREAM (out), TRUE);
  52
  53   error = NULL;
  54   g_assert_cmpint (g_output_stream_write (out, buffer, 10, NULL, &error), ==, 10);
  55   g_assert_no_error (error);
  56
  57   g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 16);
  58   g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
  59
  60   g_assert_cmpint (g_output_stream_write (out, buffer + 10, 10, NULL, &error), ==, 10);
  61   g_assert_no_error (error);
  62
  63   g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), >=, 20);
  64   g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
  65
  66   g_assert (g_output_stream_flush (out, NULL, &error));
  67   g_assert_no_error (error);
  68
  69   g_assert_cmpstr (g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (base)), ==, "abcdefghijklmnopqrst");
  70
  71   g_object_unref (out);
  72   g_object_unref (base);
  73 }
  74
  75 int
  76 main (int argc, char *argv[])
  77 {
  78   g_type_init ();
  79
  80   g_test_init (&argc, &argv, NULL);
  81
  82   g_test_add_func ("/buffered-output-stream/write", test_write);
  83   g_test_add_func ("/buffered-output-stream/grow", test_grow);
  84
  85   return g_test_run ();
  86 }