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 int
 115 main (int argc, char *argv[])
 116 {
 117   g_type_init ();
 118
 119   g_test_init (&argc, &argv, NULL);
 120
 121   g_test_add_func ("/buffered-output-stream/write", test_write);
 122   g_test_add_func ("/buffered-output-stream/grow", test_grow);
 123   g_test_add_func ("/filter-output-stream/close", test_close);
 124
 125   return g_test_run ();
 126 }