gio/tests/buffered-input-stream.c

   1 /* GLib testing framework examples and tests
   2  * Copyright (C) 2008 Red Hat, Inc.
   3  * Authors: Matthias Clasen <mclasen@redhat.com>
   4  *
   5  * This work is provided "as is"; redistribution and modification
   6  * in whole or in part, in any medium, physical or electronic is
   7  * permitted without restriction.
   8  *
   9  * This work is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  12  *
  13  * In no event shall the authors or contributors be liable for any
  14  * direct, indirect, incidental, special, exemplary, or consequential
  15  * damages (including, but not limited to, procurement of substitute
  16  * goods or services; loss of use, data, or profits; or business
  17  * interruption) however caused and on any theory of liability, whether
  18  * in contract, strict liability, or tort (including negligence or
  19  * otherwise) arising in any way out of the use of this software, even
  20  * if advised of the possibility of such damage.
  21  */
  22
  23 #include <glib/glib.h>
  24 #include <gio/gio.h>
  25 #include <stdlib.h>
  26 #include <string.h>
  27
  28 static void
  29 test_peek (void)
  30 {
  31   GInputStream *base;
  32   GInputStream *in;
  33   gssize npeek;
  34   char *buffer;
  35
  36   base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
  37   in = g_buffered_input_stream_new_sized (base, 64);
  38
  39   g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), 5, NULL, NULL);
  40   g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 5);
  41   g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), -1, NULL, NULL);
  42   g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, strlen ("abcdefjhijk"));
  43
  44   buffer = g_new0 (char, 64);
  45   npeek = g_buffered_input_stream_peek (G_BUFFERED_INPUT_STREAM (in), buffer, 2, 3);
  46   g_assert_cmpint (npeek, ==, 3);
  47   g_assert_cmpstr ("cde", ==, buffer);
  48   g_free (buffer);
  49
  50   buffer = g_new0 (char, 64);
  51   npeek = g_buffered_input_stream_peek (G_BUFFERED_INPUT_STREAM (in), buffer, 9, 5);
  52   g_assert_cmpint (npeek, ==, 2);
  53   g_assert_cmpstr ("jk", ==, buffer);
  54   g_free (buffer);
  55
  56   buffer = g_new0 (char, 64);
  57   npeek = g_buffered_input_stream_peek (G_BUFFERED_INPUT_STREAM (in), buffer, 75, 3);
  58   g_assert_cmpint (npeek, ==, 0);
  59   g_free (buffer);
  60
  61   g_object_unref (in);
  62   g_object_unref (base);
  63 }
  64
  65 static void
  66 test_peek_buffer (void)
  67 {
  68   GInputStream *base;
  69   GInputStream *in;
  70   gssize nfill;
  71   gsize bufsize;
  72   char *buffer;
  73
  74   base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
  75   in = g_buffered_input_stream_new (base);
  76
  77   nfill = g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), strlen ("abcdefghijk"), NULL, NULL);
  78   buffer = (char *) g_buffered_input_stream_peek_buffer (G_BUFFERED_INPUT_STREAM (in), &bufsize);
  79   g_assert_cmpint (nfill, ==, bufsize);
  80   g_assert (0 == strncmp ("abcdefghijk", buffer, bufsize));
  81
  82   g_object_unref (in);
  83   g_object_unref (base);
  84 }
  85
  86 static void
  87 test_set_buffer_size (void)
  88 {
  89   GInputStream *base;
  90   GInputStream *in;
  91   gsize size, bufsize;
  92
  93   base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
  94   in = g_buffered_input_stream_new (base);
  95   size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
  96   g_assert_cmpint (size, ==, 4096);
  97
  98   g_buffered_input_stream_set_buffer_size (G_BUFFERED_INPUT_STREAM (in), 64);
  99   size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
 100   g_assert_cmpint (size, ==, 64);
 101
 102   /* size cannot shrink below current content len */
 103   g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), strlen ("abcdefghijk"), NULL, NULL);
 104   g_buffered_input_stream_peek_buffer (G_BUFFERED_INPUT_STREAM (in), &bufsize);
 105   g_buffered_input_stream_set_buffer_size (G_BUFFERED_INPUT_STREAM (in), 2);
 106   size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
 107   g_assert_cmpint (size, ==, bufsize);
 108   g_object_get (in, "buffer-size", &size, NULL);
 109   g_assert_cmpint (size, ==, bufsize);
 110
 111   g_object_unref (in);
 112
 113   in = g_buffered_input_stream_new_sized (base, 64);
 114   size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
 115   g_assert_cmpint (size, ==, 64);
 116
 117   g_object_unref (in);
 118   g_object_unref (base);
 119 }
 120
 121 static void
 122 test_read_byte (void)
 123 {
 124   GInputStream *base;
 125   GInputStream *in;
 126   GError *error;
 127
 128   g_test_bug ("562393");
 129
 130   base = g_memory_input_stream_new_from_data ("abcdefgh", -1, NULL);
 131   in = g_buffered_input_stream_new (base);
 132
 133   error = NULL;
 134   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'a');
 135   g_assert_no_error (error);
 136   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'b');
 137   g_assert_no_error (error);
 138   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'c');
 139   g_assert_no_error (error);
 140
 141   g_assert_cmpint (g_input_stream_skip (in, 3, NULL, &error), ==, 3);
 142   g_assert_no_error (error);
 143
 144   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'g');
 145   g_assert_no_error (error);
 146   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'h');
 147   g_assert_no_error (error);
 148   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, -1);
 149   g_assert_no_error (error);
 150
 151   g_assert (g_input_stream_close (in, NULL, &error));
 152   g_assert_no_error (error);
 153   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, -1);
 154   g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CLOSED);
 155   g_error_free (error);
 156
 157   g_object_unref (in);
 158   g_object_unref (base);
 159 }
 160
 161 static void
 162 test_read (void)
 163 {
 164   GInputStream *base;
 165   GInputStream *in;
 166   gchar buffer[20];
 167   GError *error;
 168
 169   base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ", -1, NULL);
 170   in = g_buffered_input_stream_new_sized (base, 8);
 171
 172   g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 0);
 173
 174   error = NULL;
 175   g_assert_cmpint (g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), 8, NULL, &error), ==, 8);
 176   g_assert_no_error (error);
 177
 178   g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 8);
 179
 180   memset (buffer, 0, 20);
 181   g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 16);
 182   g_assert_cmpstr (buffer, ==, "abcdefghijklmnop");
 183   g_assert_no_error (error);
 184
 185   g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 0);
 186
 187   memset (buffer, 0, 20);
 188   g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 16);
 189   g_assert_cmpstr (buffer, ==, "qrstuvwxyzABCDEF");
 190   g_assert_no_error (error);
 191
 192   memset (buffer, 0, 20);
 193   g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 16);
 194   g_assert_cmpstr (buffer, ==, "GHIJKLMNOPQRSTUV");
 195   g_assert_no_error (error);
 196
 197   memset (buffer, 0, 20);
 198   g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 4);
 199   g_assert_cmpstr (buffer, ==, "WXYZ");
 200   g_assert_no_error (error);
 201
 202   memset (buffer, 0, 20);
 203   g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 0);
 204   g_assert_no_error (error);
 205
 206   g_object_unref (in);
 207   g_object_unref (base);
 208 }
 209
 210 static void
 211 test_skip (void)
 212 {
 213   GInputStream *base;
 214   GInputStream *in;
 215   GError *error;
 216
 217   base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
 218   in = g_buffered_input_stream_new_sized (base, 5);
 219
 220   error = NULL;
 221   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'a');
 222   g_assert_no_error (error);
 223   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'b');
 224   g_assert_no_error (error);
 225   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'c');
 226   g_assert_no_error (error);
 227
 228   g_assert_cmpint (g_input_stream_skip (in, 7, NULL, &error), ==, 7);
 229   g_assert_no_error (error);
 230   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'k');
 231   g_assert_no_error (error);
 232
 233   g_assert_cmpint (g_input_stream_skip (in, 10, NULL, &error), ==, 10);
 234   g_assert_no_error (error);
 235   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'v');
 236   g_assert_no_error (error);
 237
 238   g_assert_cmpint (g_input_stream_skip (in, 20, NULL, &error), ==, 20);
 239   g_assert_no_error (error);
 240   g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'Q');
 241   g_assert_no_error (error);
 242
 243   g_assert_cmpint (g_input_stream_skip (in, 10, NULL, &error), ==, 8);
 244   g_assert_no_error (error);
 245   g_assert_cmpint (g_input_stream_skip (in, 10, NULL, &error), ==, 0);
 246   g_assert_no_error (error);
 247
 248   g_object_unref (in);
 249   g_object_unref (base);
 250 }
 251
 252 static void
 253 test_close (void)
 254 {
 255   GInputStream *base;
 256   GInputStream *in;
 257   GError *error;
 258
 259   base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
 260   in = g_buffered_input_stream_new (base);
 261
 262   g_assert (g_filter_input_stream_get_close_base_stream (G_FILTER_INPUT_STREAM (in)));
 263
 264   error = NULL;
 265   g_assert (g_input_stream_close (in, NULL, &error));
 266   g_assert_no_error (error);
 267   g_assert (g_input_stream_is_closed (base));
 268
 269   g_object_unref (in);
 270   g_object_unref (base);
 271
 272   base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
 273   in = g_buffered_input_stream_new (base);
 274
 275   g_filter_input_stream_set_close_base_stream (G_FILTER_INPUT_STREAM (in), FALSE);
 276
 277   error = NULL;
 278   g_assert (g_input_stream_close (in, NULL, &error));
 279   g_assert_no_error (error);
 280   g_assert (!g_input_stream_is_closed (base));
 281
 282   g_object_unref (in);
 283   g_object_unref (base);
 284 }
 285
 286 static void
 287 test_seek (void)
 288 {
 289   GInputStream *base;
 290   GInputStream *in;
 291   GError *error;
 292   gint byte;
 293   gboolean ret;
 294
 295   base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
 296   in = g_buffered_input_stream_new_sized (base, 4);
 297   error = NULL;
 298
 299   /* Seek by read */
 300   g_assert_cmpstr (g_seekable_tell (G_SEEKABLE (in)), ==, 0);
 301   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 302   g_assert_no_error (error);
 303   g_assert_cmpint (byte, ==, 'a');
 304   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 1);
 305
 306   /* Seek forward (in buffer) */
 307   ret = g_seekable_seek (G_SEEKABLE (in), 1, G_SEEK_CUR, NULL, &error);
 308   g_assert_no_error (error);
 309   g_assert (ret);
 310   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 2);
 311   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 312   g_assert_no_error (error);
 313   g_assert_cmpint (byte, ==, 'c');
 314   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 3);
 315
 316   /* Seek backward (in buffer) */
 317   ret = g_seekable_seek (G_SEEKABLE (in), -2, G_SEEK_CUR, NULL, &error);
 318   g_assert_no_error (error);
 319   g_assert (ret);
 320   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 1);
 321   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 322   g_assert_no_error (error);
 323   g_assert_cmpint (byte, ==, 'b');
 324   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 2);
 325
 326   /* Seek forward (outside buffer) */
 327   ret = g_seekable_seek (G_SEEKABLE (in), 6, G_SEEK_CUR, NULL, &error);
 328   g_assert_no_error (error);
 329   g_assert (ret);
 330   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 8);
 331   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 332   g_assert_no_error (error);
 333   g_assert_cmpint (byte, ==, 'i');
 334   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 9);
 335
 336   /* Seek backward (outside buffer) */
 337   ret = g_seekable_seek (G_SEEKABLE (in), -6, G_SEEK_CUR, NULL, &error);
 338   g_assert_no_error (error);
 339   g_assert (ret);
 340   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 3);
 341   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 342   g_assert_no_error (error);
 343   g_assert_cmpint (byte, ==, 'd');
 344   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 4);
 345
 346   /* Seek from beginning */
 347   ret = g_seekable_seek (G_SEEKABLE (in), 8, G_SEEK_SET, NULL, &error);
 348   g_assert_no_error (error);
 349   g_assert (ret);
 350   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 8);
 351   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 352   g_assert_no_error (error);
 353   g_assert_cmpint (byte, ==, 'i');
 354   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 9);
 355
 356   /* Seek from end */
 357   ret = g_seekable_seek (G_SEEKABLE (in), -1, G_SEEK_END, NULL, &error);
 358   g_assert_no_error (error);
 359   g_assert (ret);
 360   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 50);
 361   byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
 362   g_assert_no_error (error);
 363   g_assert_cmpint (byte, ==, 'Z');
 364   g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 51);
 365
 366   /* Cleanup */
 367   g_object_unref (in);
 368   g_object_unref (base);
 369 }
 370
 371 int
 372 main (int   argc,
 373       char *argv[])
 374 {
 375   g_type_init ();
 376   g_test_init (&argc, &argv, NULL);
 377   g_test_bug_base ("http://bugzilla.gnome.org/");
 378
 379   g_test_add_func ("/buffered-input-stream/peek", test_peek);
 380   g_test_add_func ("/buffered-input-stream/peek-buffer", test_peek_buffer);
 381   g_test_add_func ("/buffered-input-stream/set-buffer-size", test_set_buffer_size);
 382   g_test_add_func ("/buffered-input-stream/read-byte", test_read_byte);
 383   g_test_add_func ("/buffered-input-stream/read", test_read);
 384   g_test_add_func ("/buffered-input-stream/skip", test_skip);
 385   g_test_add_func ("/buffered-input-stream/seek", test_seek);
 386   g_test_add_func ("/filter-input-stream/close", test_close);
 387
 388   return g_test_run();
 389 }