mirror of
https://gitlab.gnome.org/GNOME/glib.git
synced 2025-08-25 18:39:00 +02:00
Make GBufferedOutputStream implement GSeekable
https://bugzilla.gnome.org/show_bug.cgi?id=673034
This commit is contained in:
Maciej Piechotka
committed by
Alexander Larsson
Alexander Larsson
parent
90739baec0
commit
43895e3089
@@ -111,6 +111,199 @@ test_close (void)
|
||||
g_object_unref (base);
|
||||
}
|
||||
|
||||
static void
|
||||
test_seek (void)
|
||||
{
|
||||
GMemoryOutputStream *base;
|
||||
GOutputStream *out;
|
||||
GSeekable *seekable;
|
||||
GError *error;
|
||||
gsize bytes_written;
|
||||
gboolean ret;
|
||||
const gchar buffer[] = "abcdefghijklmnopqrstuvwxyz";
|
||||
|
||||
base = G_MEMORY_OUTPUT_STREAM (g_memory_output_stream_new (g_malloc0 (30), 30, g_realloc, g_free));
|
||||
out = g_buffered_output_stream_new_sized (G_OUTPUT_STREAM (base), 8);
|
||||
seekable = G_SEEKABLE (out);
|
||||
error = NULL;
|
||||
|
||||
/* Write data */
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 0);
|
||||
ret = g_output_stream_write_all (out, buffer, 4, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert_cmpint (bytes_written, ==, 4);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 4);
|
||||
g_assert_cmpint (g_memory_output_stream_get_data_size (base), ==, 0);
|
||||
|
||||
/* Forward relative seek */
|
||||
ret = g_seekable_seek (seekable, 2, G_SEEK_CUR, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 6);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
|
||||
g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
|
||||
g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
|
||||
ret = g_output_stream_write_all (out, buffer, 2, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (bytes_written, ==, 2);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 8);
|
||||
|
||||
/* Backward relative seek */
|
||||
ret = g_seekable_seek (seekable, -4, G_SEEK_CUR, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 4);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
|
||||
g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
|
||||
g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
|
||||
ret = g_output_stream_write_all (out, buffer, 2, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (bytes_written, ==, 2);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 6);
|
||||
|
||||
/* From start */
|
||||
ret = g_seekable_seek (seekable, 2, G_SEEK_SET, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 2);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
|
||||
g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
|
||||
g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[4]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[5]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
|
||||
ret = g_output_stream_write_all (out, buffer, 2, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (bytes_written, ==, 2);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 4);
|
||||
|
||||
/* From end */
|
||||
ret = g_seekable_seek (seekable, 6 - 30, G_SEEK_END, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 6);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[4]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[5]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
|
||||
ret = g_output_stream_write_all (out, buffer + 2, 2, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (bytes_written, ==, 2);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 8);
|
||||
|
||||
/* Check flush */
|
||||
ret = g_output_stream_flush (out, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (ret);
|
||||
g_assert_cmpint (g_seekable_tell (G_SEEKABLE (out)), ==, 8);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[0]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[1]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[2]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[3]);
|
||||
g_assert_cmpint ('a', ==, ((gchar *)g_memory_output_stream_get_data (base))[4]);
|
||||
g_assert_cmpint ('b', ==, ((gchar *)g_memory_output_stream_get_data (base))[5]);
|
||||
g_assert_cmpint ('c', ==, ((gchar *)g_memory_output_stream_get_data (base))[6]);
|
||||
g_assert_cmpint ('d', ==, ((gchar *)g_memory_output_stream_get_data (base))[7]);
|
||||
|
||||
g_object_unref (out);
|
||||
g_object_unref (base);
|
||||
}
|
||||
|
||||
static void
|
||||
test_truncate(void)
|
||||
{
|
||||
GMemoryOutputStream *base_stream;
|
||||
GOutputStream *stream;
|
||||
GSeekable *seekable;
|
||||
GError *error;
|
||||
gsize bytes_written;
|
||||
guchar *stream_data;
|
||||
gsize len;
|
||||
gboolean res;
|
||||
|
||||
len = 8;
|
||||
|
||||
/* Create objects */
|
||||
stream_data = g_malloc0 (len);
|
||||
base_stream = G_MEMORY_OUTPUT_STREAM (g_memory_output_stream_new (stream_data, len, g_realloc, g_free));
|
||||
stream = g_buffered_output_stream_new_sized (G_OUTPUT_STREAM (base_stream), 8);
|
||||
seekable = G_SEEKABLE (stream);
|
||||
|
||||
g_assert (g_seekable_can_truncate (seekable));
|
||||
|
||||
/* Write */
|
||||
g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, len);
|
||||
g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 0);
|
||||
|
||||
error = NULL;
|
||||
res = g_output_stream_write_all (stream, "ab", 2, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (res);
|
||||
res = g_output_stream_write_all (stream, "cd", 2, &bytes_written, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (res);
|
||||
|
||||
res = g_output_stream_flush (stream, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (res);
|
||||
|
||||
g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, len);
|
||||
g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 4);
|
||||
g_assert_cmpint (stream_data[0], ==, 'a');
|
||||
g_assert_cmpint (stream_data[1], ==, 'b');
|
||||
g_assert_cmpint (stream_data[2], ==, 'c');
|
||||
g_assert_cmpint (stream_data[3], ==, 'd');
|
||||
|
||||
/* Truncate at position */
|
||||
res = g_seekable_truncate (seekable, 4, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (res);
|
||||
g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, 4);
|
||||
g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 4);
|
||||
g_assert_cmpint (stream_data[0], ==, 'a');
|
||||
g_assert_cmpint (stream_data[1], ==, 'b');
|
||||
g_assert_cmpint (stream_data[2], ==, 'c');
|
||||
g_assert_cmpint (stream_data[3], ==, 'd');
|
||||
|
||||
/* Truncate beyond position */
|
||||
res = g_seekable_truncate (seekable, 6, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (res);
|
||||
g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, 6);
|
||||
g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 4);
|
||||
g_assert_cmpint (stream_data[0], ==, 'a');
|
||||
g_assert_cmpint (stream_data[1], ==, 'b');
|
||||
g_assert_cmpint (stream_data[2], ==, 'c');
|
||||
g_assert_cmpint (stream_data[3], ==, 'd');
|
||||
|
||||
/* Truncate before position */
|
||||
res = g_seekable_truncate (seekable, 2, NULL, &error);
|
||||
g_assert_no_error (error);
|
||||
g_assert (res);
|
||||
g_assert_cmpint (g_memory_output_stream_get_size (base_stream), ==, 2);
|
||||
g_assert_cmpint (g_memory_output_stream_get_data_size (base_stream), ==, 2);
|
||||
g_assert_cmpint (stream_data[0], ==, 'a');
|
||||
g_assert_cmpint (stream_data[1], ==, 'b');
|
||||
|
||||
g_object_unref (stream);
|
||||
g_object_unref (base_stream);
|
||||
}
|
||||
|
||||
int
|
||||
main (int argc, char *argv[])
|
||||
{
|
||||
@@ -120,6 +313,8 @@ main (int argc, char *argv[])
|
||||
|
||||
g_test_add_func ("/buffered-output-stream/write", test_write);
|
||||
g_test_add_func ("/buffered-output-stream/grow", test_grow);
|
||||
g_test_add_func ("/buffered-output-stream/seek", test_seek);
|
||||
g_test_add_func ("/buffered-output-stream/truncate", test_truncate);
|
||||
g_test_add_func ("/filter-output-stream/close", test_close);
|
||||
|
||||
return g_test_run ();
|
||||
|
||||
Reference in New Issue
Block a user