[glib] Improve coverage of memory stream tests

[Matthias Clasen <matthiasc src gnome org>]

commit 9794a648a9af25a9d409ce6aef97ac95f94c5cfb
Author: Matthias Clasen <mclasen redhat com>
Date:   Thu Jul 29 02:19:46 2010 -0400

    Improve coverage of memory stream tests

 gio/tests/memory-input-stream.c  |   63 ++++++++++++++++++++++++++++++++++++++
 gio/tests/memory-output-stream.c |   29 +++++++++++++++++-
 2 files changed, 91 insertions(+), 1 deletions(-)
---
diff --git a/gio/tests/memory-input-stream.c b/gio/tests/memory-input-stream.c
index 6ffee86..3bb021c 100644
--- a/gio/tests/memory-input-stream.c
+++ b/gio/tests/memory-input-stream.c
@@ -65,6 +65,67 @@ test_read_chunks (void)
     }
 }
 
+static void
+test_seek (void)
+{
+  const char *data1 = "abcdefghijklmnopqrstuvwxyz";
+  const char *data2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
+  GInputStream *stream;
+  GError *error;
+  char buffer[10];
+
+  stream = g_memory_input_stream_new ();
+
+  g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (stream),
+                                  data1, -1, NULL);
+  g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (stream),
+                                  data2, -1, NULL);
+
+  g_assert (G_IS_SEEKABLE (stream));
+  g_assert (g_seekable_can_seek (G_SEEKABLE (stream)));
+
+  error = NULL;
+  g_assert (g_seekable_seek (G_SEEKABLE (stream), 26, G_SEEK_SET, NULL, &error));
+  g_assert_no_error (error);
+  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (stream)), ==, 26);
+
+  g_assert (g_input_stream_read (stream, buffer, 1, NULL, &error) == 1);
+  g_assert_no_error (error);
+
+  g_assert (buffer[0] == 'A');
+
+  g_assert (!g_seekable_seek (G_SEEKABLE (stream), 26, G_SEEK_CUR, NULL, &error));
+  g_assert_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT);
+  g_error_free (error);
+
+  g_object_unref (stream);
+}
+
+static void
+test_truncate (void)
+{
+  const char *data1 = "abcdefghijklmnopqrstuvwxyz";
+  const char *data2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
+  GInputStream *stream;
+  GError *error;
+
+  stream = g_memory_input_stream_new ();
+
+  g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (stream),
+                                  data1, -1, NULL);
+  g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (stream),
+                                  data2, -1, NULL);
+
+  g_assert (G_IS_SEEKABLE (stream));
+  g_assert (!g_seekable_can_truncate (G_SEEKABLE (stream)));
+
+  error = NULL;
+  g_assert (!g_seekable_truncate (G_SEEKABLE (stream), 26, NULL, &error));
+  g_assert_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED);
+
+  g_object_unref (stream);
+}
+
 int
 main (int   argc,
       char *argv[])
@@ -73,6 +134,8 @@ main (int   argc,
   g_test_init (&argc, &argv, NULL);
 
   g_test_add_func ("/memory-input-stream/read-chunks", test_read_chunks);
+  g_test_add_func ("/memory-input-stream/seek", test_seek);
+  g_test_add_func ("/memory-input-stream/truncate", test_truncate);
 
   return g_test_run();
 }
diff --git a/gio/tests/memory-output-stream.c b/gio/tests/memory-output-stream.c
index 44b5624..210ad50 100644
--- a/gio/tests/memory-output-stream.c
+++ b/gio/tests/memory-output-stream.c
@@ -35,6 +35,7 @@ test_truncate (void)
   g_test_bug ("540423");
 
   mo = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
+  g_assert (g_seekable_can_truncate (G_SEEKABLE (mo)));
   o = g_data_output_stream_new (mo);
   for (i = 0; i < 1000; i++)
     {
@@ -54,6 +55,29 @@ test_truncate (void)
 }
 
 static void
+test_seek (void)
+{
+  GOutputStream *mo;
+  GError *error;
+
+  mo = g_memory_output_stream_new (g_new (gchar, 100), 100, g_realloc, g_free);
+
+  g_assert (G_IS_SEEKABLE (mo));
+  g_assert (g_seekable_can_seek (G_SEEKABLE (mo)));
+
+  error = NULL;
+  g_assert (g_seekable_seek (G_SEEKABLE (mo), 26, G_SEEK_SET, NULL, &error));
+  g_assert_no_error (error);
+  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (mo)), ==, 26);
+
+  g_assert (!g_seekable_seek (G_SEEKABLE (mo), 200, G_SEEK_CUR, NULL, &error));
+  g_assert_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT);
+  g_error_free (error);
+
+  g_object_unref (mo);
+}
+
+static void
 test_data_size (void)
 {
   GOutputStream *mo;
@@ -80,7 +104,9 @@ test_data_size (void)
   
   pos = g_memory_output_stream_get_data_size (G_MEMORY_OUTPUT_STREAM (mo));
   g_assert_cmpint (pos, ==, 1);
-  
+
+  g_assert_cmpint (g_memory_output_stream_get_size (G_MEMORY_OUTPUT_STREAM (mo)), ==, 16);
+
   g_object_unref (o);
   g_object_unref (mo);
 }
@@ -130,6 +156,7 @@ main (int   argc,
   g_test_bug_base ("http://bugzilla.gnome.org/";);
 
   g_test_add_func ("/memory-output-stream/truncate", test_truncate);
+  g_test_add_func ("/memory-output-stream/seek", test_seek);
   g_test_add_func ("/memory-output-stream/get-data-size", test_data_size);
   g_test_add_func ("/memory-output-stream/properties", test_properties);