[glib] Add some buffered output tests

[Matthias Clasen <matthiasc src gnome org>]

commit aec652677bed2d82201b0234e04e95f164e20770
Author: Matthias Clasen <mclasen redhat com>
Date:   Mon Jul 5 11:12:49 2010 -0400

    Add some buffered output tests

 gio/tests/Makefile.am              |    4 ++
 gio/tests/buffered-output-stream.c |   86 ++++++++++++++++++++++++++++++++++++
 2 files changed, 90 insertions(+), 0 deletions(-)
---
diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am
index 988fc45..8886f1b 100644
--- a/gio/tests/Makefile.am
+++ b/gio/tests/Makefile.am
@@ -32,6 +32,7 @@ TEST_PROGS +=	 		\
 	data-output-stream 	\
 	g-icon			\
 	buffered-input-stream	\
+	buffered-output-stream	\
 	sleepy-stream		\
 	filter-streams		\
 	simple-async-result	\
@@ -130,6 +131,9 @@ g_icon_LDADD	= $(progs_ldadd)
 buffered_input_stream_SOURCES	= buffered-input-stream.c
 buffered_input_stream_LDADD	= $(progs_ldadd)
 
+buffered_output_stream_SOURCES	= buffered-output-stream.c
+buffered_output_stream_LDADD	= $(progs_ldadd)
+
 live_g_file_SOURCES	  = live-g-file.c
 live_g_file_LDADD	  = $(progs_ldadd)
 
diff --git a/gio/tests/buffered-output-stream.c b/gio/tests/buffered-output-stream.c
new file mode 100644
index 0000000..e3e3c6f
--- /dev/null
+++ b/gio/tests/buffered-output-stream.c
@@ -0,0 +1,86 @@
+#include <gio/gio.h>
+
+static void
+test_write (void)
+{
+  GOutputStream *base;
+  GOutputStream *out;
+  GError *error;
+  const gchar buffer[] = "abcdefghijklmnopqrstuvwxyz";
+
+  base = g_memory_output_stream_new (g_malloc0 (20), 20, g_realloc, g_free);
+  out = g_buffered_output_stream_new (base);
+
+  g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 4096);
+  g_assert (!g_buffered_output_stream_get_auto_grow (G_BUFFERED_OUTPUT_STREAM (out)));
+  g_buffered_output_stream_set_buffer_size (G_BUFFERED_OUTPUT_STREAM (out), 16);
+  g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 16);
+
+  error = NULL;
+  g_assert_cmpint (g_output_stream_write (out, buffer, 10, NULL, &error), ==, 10);
+  g_assert_no_error (error);
+
+  g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
+
+  g_assert_cmpint (g_output_stream_write (out, buffer + 10, 10, NULL, &error), ==, 6);
+  g_assert_no_error (error);
+
+  g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
+  g_assert (g_output_stream_flush (out, NULL, &error));
+  g_assert_no_error (error);
+  g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 16);
+
+  g_assert_cmpstr (g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (base)), ==, "abcdefghijklmnop");
+
+  g_object_unref (out);
+  g_object_unref (base);
+}
+
+static void
+test_grow (void)
+{
+  GOutputStream *base;
+  GOutputStream *out;
+  GError *error;
+  const gchar buffer[] = "abcdefghijklmnopqrstuvwxyz";
+
+  base = g_memory_output_stream_new (g_malloc0 (30), 30, g_realloc, g_free);
+  out = g_buffered_output_stream_new (base);
+
+  g_buffered_output_stream_set_buffer_size (G_BUFFERED_OUTPUT_STREAM (out), 16);
+  g_buffered_output_stream_set_auto_grow (G_BUFFERED_OUTPUT_STREAM (out), TRUE);
+
+  error = NULL;
+  g_assert_cmpint (g_output_stream_write (out, buffer, 10, NULL, &error), ==, 10);
+  g_assert_no_error (error);
+
+  g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), ==, 16);
+  g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
+
+  g_assert_cmpint (g_output_stream_write (out, buffer + 10, 10, NULL, &error), ==, 10);
+  g_assert_no_error (error);
+
+  g_assert_cmpint (g_buffered_output_stream_get_buffer_size (G_BUFFERED_OUTPUT_STREAM (out)), >=, 20);
+  g_assert_cmpint (g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (base)), ==, 0);
+
+  g_assert (g_output_stream_flush (out, NULL, &error));
+  g_assert_no_error (error);
+
+  g_assert_cmpstr (g_memory_output_stream_get_data (G_MEMORY_OUTPUT_STREAM (base)), ==, "abcdefghijklmnopqrst");
+
+  g_object_unref (out);
+  g_object_unref (base);
+}
+
+int
+main (int argc, char *argv[])
+{
+  g_type_init ();
+
+  g_test_init (&argc, &argv, NULL);
+
+  g_test_add_func ("/buffered-output-stream/write", test_write);
+  g_test_add_func ("/buffered-output-stream/grow", test_grow);
+
+  return g_test_run ();
+}