| /* GLib testing framework examples and tests |
| * Copyright (C) 2007 Imendio AB |
| * Authors: Tim Janik |
| * |
| * This work is provided "as is"; redistribution and modification |
| * in whole or in part, in any medium, physical or electronic is |
| * permitted without restriction. |
| * |
| * This work is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. |
| * |
| * In no event shall the authors or contributors be liable for any |
| * direct, indirect, incidental, special, exemplary, or consequential |
| * damages (including, but not limited to, procurement of substitute |
| * goods or services; loss of use, data, or profits; or business |
| * interruption) however caused and on any theory of liability, whether |
| * in contract, strict liability, or tort (including negligence or |
| * otherwise) arising in any way out of the use of this software, even |
| * if advised of the possibility of such damage. |
| */ |
| |
| #include <glib/glib.h> |
| #include <gio/gio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| static void |
| test_read_chunks (void) |
| { |
| const char *data1 = "abcdefghijklmnopqrstuvwxyz"; |
| const char *data2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; |
| const char *result = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"; |
| char buffer[128]; |
| gsize bytes_read, pos, len, chunk_size; |
| GError *error = NULL; |
| GInputStream *stream; |
| gboolean res; |
| |
| 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); |
| len = strlen (data1) + strlen (data2); |
| |
| for (chunk_size = 1; chunk_size < len - 1; chunk_size++) |
| { |
| pos = 0; |
| while (pos < len) |
| { |
| bytes_read = g_input_stream_read (stream, buffer, chunk_size, NULL, &error); |
| g_assert_no_error (error); |
| g_assert_cmpint (bytes_read, ==, MIN (chunk_size, len - pos)); |
| g_assert (strncmp (buffer, result + pos, bytes_read) == 0); |
| |
| pos += bytes_read; |
| } |
| |
| g_assert_cmpint (pos, ==, len); |
| res = g_seekable_seek (G_SEEKABLE (stream), 0, G_SEEK_SET, NULL, &error); |
| g_assert_cmpint (res, ==, TRUE); |
| g_assert_no_error (error); |
| } |
| |
| g_object_unref (stream); |
| } |
| |
| GMainLoop *loop; |
| |
| static void |
| async_read_chunk (GObject *object, |
| GAsyncResult *result, |
| gpointer user_data) |
| { |
| gsize *bytes_read = user_data; |
| GError *error = NULL; |
| |
| *bytes_read = g_input_stream_read_finish (G_INPUT_STREAM (object), |
| result, &error); |
| g_assert_no_error (error); |
| |
| g_main_loop_quit (loop); |
| } |
| |
| static void |
| async_skipped_chunk (GObject *object, |
| GAsyncResult *result, |
| gpointer user_data) |
| { |
| gsize *bytes_skipped = user_data; |
| GError *error = NULL; |
| |
| *bytes_skipped = g_input_stream_skip_finish (G_INPUT_STREAM (object), |
| result, &error); |
| g_assert_no_error (error); |
| |
| g_main_loop_quit (loop); |
| } |
| |
| static void |
| test_async (void) |
| { |
| const char *data1 = "abcdefghijklmnopqrstuvwxyz"; |
| const char *data2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; |
| const char *result = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"; |
| char buffer[128]; |
| gsize bytes_read, bytes_skipped; |
| gsize pos, len, chunk_size; |
| GError *error = NULL; |
| GInputStream *stream; |
| gboolean res; |
| |
| loop = g_main_loop_new (NULL, FALSE); |
| |
| 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); |
| len = strlen (data1) + strlen (data2); |
| |
| for (chunk_size = 1; chunk_size < len - 1; chunk_size++) |
| { |
| pos = 0; |
| while (pos < len) |
| { |
| g_input_stream_read_async (stream, buffer, chunk_size, |
| G_PRIORITY_DEFAULT, NULL, |
| async_read_chunk, &bytes_read); |
| g_main_loop_run (loop); |
| |
| g_assert_cmpint (bytes_read, ==, MIN (chunk_size, len - pos)); |
| g_assert (strncmp (buffer, result + pos, bytes_read) == 0); |
| |
| pos += bytes_read; |
| } |
| |
| g_assert_cmpint (pos, ==, len); |
| res = g_seekable_seek (G_SEEKABLE (stream), 0, G_SEEK_SET, NULL, &error); |
| g_assert_cmpint (res, ==, TRUE); |
| g_assert_no_error (error); |
| |
| pos = 0; |
| while (pos + chunk_size + 1 < len) |
| { |
| g_input_stream_skip_async (stream, chunk_size, |
| G_PRIORITY_DEFAULT, NULL, |
| async_skipped_chunk, &bytes_skipped); |
| g_main_loop_run (loop); |
| |
| g_assert_cmpint (bytes_skipped, ==, MIN (chunk_size, len - pos)); |
| |
| pos += bytes_skipped; |
| } |
| |
| g_input_stream_read_async (stream, buffer, len - pos, |
| G_PRIORITY_DEFAULT, NULL, |
| async_read_chunk, &bytes_read); |
| g_main_loop_run (loop); |
| |
| g_assert_cmpint (bytes_read, ==, len - pos); |
| g_assert (strncmp (buffer, result + pos, bytes_read) == 0); |
| |
| res = g_seekable_seek (G_SEEKABLE (stream), 0, G_SEEK_SET, NULL, &error); |
| g_assert_cmpint (res, ==, TRUE); |
| g_assert_no_error (error); |
| } |
| |
| g_object_unref (stream); |
| g_main_loop_unref (loop); |
| } |
| |
| 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_error_free (error); |
| |
| g_object_unref (stream); |
| } |
| |
| static void |
| test_read_bytes (void) |
| { |
| const char *data1 = "abcdefghijklmnopqrstuvwxyz"; |
| const char *data2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; |
| GInputStream *stream; |
| GError *error = NULL; |
| GBytes *bytes; |
| gsize size; |
| gconstpointer data; |
| |
| 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); |
| |
| bytes = g_input_stream_read_bytes (stream, 26, NULL, &error); |
| g_assert_no_error (error); |
| |
| data = g_bytes_get_data (bytes, &size); |
| g_assert_cmpint (size, ==, 26); |
| g_assert (strncmp (data, data1, 26) == 0); |
| |
| g_bytes_unref (bytes); |
| g_object_unref (stream); |
| } |
| |
| static void |
| test_from_bytes (void) |
| { |
| gchar data[4096], buffer[4096]; |
| GBytes *bytes; |
| GError *error = NULL; |
| GInputStream *stream; |
| gint i; |
| |
| for (i = 0; i < 4096; i++) |
| data[i] = 1 + i % 255; |
| |
| bytes = g_bytes_new_static (data, 4096); |
| stream = g_memory_input_stream_new_from_bytes (bytes); |
| g_assert (g_input_stream_read (stream, buffer, 2048, NULL, &error) == 2048); |
| g_assert_no_error (error); |
| g_assert (strncmp (data, buffer, 2048) == 0); |
| |
| g_object_unref (stream); |
| g_bytes_unref (bytes); |
| } |
| |
| int |
| main (int argc, |
| char *argv[]) |
| { |
| 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/async", test_async); |
| g_test_add_func ("/memory-input-stream/seek", test_seek); |
| g_test_add_func ("/memory-input-stream/truncate", test_truncate); |
| g_test_add_func ("/memory-input-stream/read-bytes", test_read_bytes); |
| g_test_add_func ("/memory-input-stream/from-bytes", test_from_bytes); |
| |
| return g_test_run(); |
| } |