From ed0821856533e63bee4f7da54f1a9e6e0b1f12e9 Mon Sep 17 00:00:00 2001 From: Alexander Larsson Date: Wed, 13 May 2009 13:06:58 +0200 Subject: [PATCH] Add tests for local GIOStream GFile ops --- gio/tests/Makefile.am | 4 + gio/tests/readwrite.c | 293 ++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 297 insertions(+) create mode 100644 gio/tests/readwrite.c diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am index eddaaa2..f327443 100644 --- a/gio/tests/Makefile.am +++ b/gio/tests/Makefile.am @@ -19,6 +19,7 @@ progs_ldadd = \ TEST_PROGS += \ memory-input-stream \ memory-output-stream \ + readwrite \ g-file \ g-file-info \ data-input-stream \ @@ -44,6 +45,9 @@ memory_output_stream_LDADD = $(progs_ldadd) g_file_SOURCES = g-file.c g_file_LDADD = $(progs_ldadd) +readwrite_SOURCES = readwrite.c +readwrite_LDADD = $(progs_ldadd) + g_file_info_SOURCES = g-file-info.c g_file_info_LDADD = $(progs_ldadd) diff --git a/gio/tests/readwrite.c b/gio/tests/readwrite.c new file mode 100644 index 0000000..01b3206 --- /dev/null +++ b/gio/tests/readwrite.c @@ -0,0 +1,293 @@ +#include +#include +#include +#include +#include + +static char *original_data = "This is some test data that we can put in a file..."; +static char *new_data = "new data.."; + +static void +verify_pos (GIOStream *iostream, goffset expected_pos) +{ + goffset pos; + + pos = g_seekable_tell (G_SEEKABLE (iostream)); + g_assert_cmpint (pos, ==, expected_pos); + + pos = g_seekable_tell (G_SEEKABLE (g_io_stream_get_input_stream (iostream))); + g_assert_cmpint (pos, ==, expected_pos); + + pos = g_seekable_tell (G_SEEKABLE (g_io_stream_get_output_stream (iostream))); + g_assert_cmpint (pos, ==, expected_pos); +} + +static void +verify_iostream (GFileIOStream *file_iostream) +{ + gboolean res; + GIOStream *iostream; + GError *error; + GInputStream *in; + GOutputStream *out; + char buffer[1024]; + gsize n_bytes; + char *modified_data; + + iostream = G_IO_STREAM (file_iostream); + + verify_pos (iostream, 0); + + in = g_io_stream_get_input_stream (iostream); + out = g_io_stream_get_output_stream (iostream); + + res = g_input_stream_read_all (in, buffer, 20, &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint ((int)n_bytes, ==, 20); + + g_assert (memcmp (buffer, original_data, 20) == 0); + + verify_pos (iostream, 20); + + res = g_seekable_seek (G_SEEKABLE (iostream), + -10, G_SEEK_END, + NULL, NULL); + g_assert (res); + verify_pos (iostream, strlen (original_data) - 10); + + res = g_input_stream_read_all (in, buffer, 20, &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint ((int)n_bytes, ==, 10); + g_assert (memcmp (buffer, original_data + strlen (original_data) - 10, 10) == 0); + + verify_pos (iostream, strlen (original_data)); + + res = g_seekable_seek (G_SEEKABLE (iostream), + 10, G_SEEK_SET, + NULL, NULL); + + verify_pos (iostream, 10); + + res = g_output_stream_write_all (out, new_data, strlen (new_data), + &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint (n_bytes, ==, strlen (new_data)); + + verify_pos (iostream, 10 + strlen (new_data)); + + res = g_seekable_seek (G_SEEKABLE (iostream), + 0, G_SEEK_SET, + NULL, NULL); + g_assert (res); + verify_pos (iostream, 0); + + res = g_input_stream_read_all (in, buffer, strlen (original_data), &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint ((int)n_bytes, ==, strlen (original_data)); + buffer[n_bytes] = 0; + + modified_data = g_strdup (original_data); + memcpy (modified_data + 10, new_data, strlen (new_data)); + g_assert_cmpstr (buffer, ==, modified_data); + + verify_pos (iostream, strlen (original_data)); + + res = g_seekable_seek (G_SEEKABLE (iostream), + 0, G_SEEK_SET, + NULL, NULL); + g_assert (res); + verify_pos (iostream, 0); + + res = g_output_stream_close (out, NULL, NULL); + g_assert (res); + + res = g_input_stream_read_all (in, buffer, 15, &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint ((int)n_bytes, ==, 15); + g_assert (memcmp (buffer, modified_data, 15) == 0); + + error = NULL; + res = g_output_stream_write_all (out, new_data, strlen (new_data), + &n_bytes, NULL, &error); + g_assert (!res); + g_assert (error != NULL); + g_assert (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CLOSED)); + + error = NULL; + res = g_io_stream_close (iostream, NULL, &error); + g_assert (res && error == NULL); + + g_free (modified_data); +} + +static void +test_g_file_open_readwrite (void) +{ + char *tmpfile; + int fd; + gboolean res; + GFileIOStream *file_iostream; + char *path; + GFile *file; + GError *error; + + fd = g_file_open_tmp ("readwrite_XXXXXX", + &tmpfile, NULL); + g_assert (fd != -1); + close (fd); + + res = g_file_set_contents (tmpfile, + original_data, -1, NULL); + g_assert (res); + + path = g_build_filename (g_get_home_dir (), "g-a-nonexisting-file", NULL); + file = g_file_new_for_path (path); + g_free (path); + error = NULL; + file_iostream = g_file_open_readwrite (file, NULL, &error); + g_assert (file_iostream == NULL); + g_assert (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_NOT_FOUND)); + g_error_free (error); + g_object_unref (file); + + file = g_file_new_for_path (tmpfile); + error = NULL; + file_iostream = g_file_open_readwrite (file, NULL, &error); + g_assert (file_iostream != NULL); + + verify_iostream (file_iostream); + + g_object_unref (file_iostream); + + g_unlink (tmpfile); + g_free (tmpfile); +} + +static void +test_g_file_create_readwrite (void) +{ + char *tmpfile; + int fd; + gboolean res; + GFileIOStream *file_iostream; + GOutputStream *out; + GFile *file; + GError *error; + gsize n_bytes; + + fd = g_file_open_tmp ("readwrite_XXXXXX", + &tmpfile, NULL); + g_assert (fd != -1); + close (fd); + + file = g_file_new_for_path (tmpfile); + error = NULL; + file_iostream = g_file_create_readwrite (file, 0, NULL, &error); + g_assert (file_iostream == NULL); + g_assert (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_EXISTS)); + + g_unlink (tmpfile); + file_iostream = g_file_create_readwrite (file, 0, NULL, &error); + g_assert (file_iostream != NULL); + + out = g_io_stream_get_output_stream (G_IO_STREAM (file_iostream)); + res = g_output_stream_write_all (out, original_data, strlen (original_data), + &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint (n_bytes, ==, strlen (original_data)); + + res = g_seekable_seek (G_SEEKABLE (file_iostream), + 0, G_SEEK_SET, + NULL, NULL); + g_assert (res); + + verify_iostream (file_iostream); + + g_object_unref (file_iostream); + + g_unlink (tmpfile); + g_free (tmpfile); +} + +static void +test_g_file_replace_readwrite (void) +{ + char *tmpfile, *backup, *data; + int fd; + gboolean res; + GFileIOStream *file_iostream; + GInputStream *in; + GOutputStream *out; + GFile *file; + GError *error; + char buffer[1024]; + gsize n_bytes; + + fd = g_file_open_tmp ("readwrite_XXXXXX", + &tmpfile, NULL); + g_assert (fd != -1); + close (fd); + + res = g_file_set_contents (tmpfile, + new_data, -1, NULL); + g_assert (res); + + file = g_file_new_for_path (tmpfile); + error = NULL; + file_iostream = g_file_replace_readwrite (file, NULL, + TRUE, 0, NULL, &error); + g_assert (file_iostream != NULL); + + in = g_io_stream_get_input_stream (G_IO_STREAM (file_iostream)); + + /* Ensure its empty */ + res = g_input_stream_read_all (in, buffer, sizeof buffer, &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint ((int)n_bytes, ==, 0); + + out = g_io_stream_get_output_stream (G_IO_STREAM (file_iostream)); + res = g_output_stream_write_all (out, original_data, strlen (original_data), + &n_bytes, NULL, NULL); + g_assert (res); + g_assert_cmpint (n_bytes, ==, strlen (original_data)); + + res = g_seekable_seek (G_SEEKABLE (file_iostream), + 0, G_SEEK_SET, + NULL, NULL); + g_assert (res); + + verify_iostream (file_iostream); + + g_object_unref (file_iostream); + + backup = g_strconcat (tmpfile, "~", NULL); + res = g_file_get_contents (backup, + &data, + NULL, NULL); + g_assert (res); + g_assert_cmpstr (data, ==, new_data); + g_free (data); + g_unlink (backup); + g_free (backup); + + g_unlink (tmpfile); + g_free (tmpfile); +} + + +int +main (int argc, + char *argv[]) +{ + g_type_init (); + g_test_init (&argc, &argv, NULL); + + g_test_add_func ("/readwrite/test_g_file_open_readwrite", + test_g_file_open_readwrite); + g_test_add_func ("/readwrite/test_g_file_create_readwrite", + test_g_file_create_readwrite); + g_test_add_func ("/readwrite/test_g_file_replace_readwrite", + test_g_file_replace_readwrite); + + return g_test_run(); +} -- 2.7.4