buffered-input-stream.c 13.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
/* GLib testing framework examples and tests
 * Copyright (C) 2008 Red Hat, Inc.
 * Authors: Matthias Clasen <mclasen@redhat.com>
 *
 * 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>

28 29 30 31 32 33 34 35 36 37 38
static void
test_peek (void)
{
  GInputStream *base;
  GInputStream *in;
  gssize npeek;
  char *buffer;

  base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
  in = g_buffered_input_stream_new_sized (base, 64);

Matthias Clasen's avatar
Matthias Clasen committed
39 40 41 42
  g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), 5, NULL, NULL);
  g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 5);
  g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), -1, NULL, NULL);
  g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, strlen ("abcdefjhijk"));
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61

  buffer = g_new0 (char, 64);
  npeek = g_buffered_input_stream_peek (G_BUFFERED_INPUT_STREAM (in), buffer, 2, 3);
  g_assert_cmpint (npeek, ==, 3);
  g_assert_cmpstr ("cde", ==, buffer);
  g_free (buffer);

  buffer = g_new0 (char, 64);
  npeek = g_buffered_input_stream_peek (G_BUFFERED_INPUT_STREAM (in), buffer, 9, 5);
  g_assert_cmpint (npeek, ==, 2);
  g_assert_cmpstr ("jk", ==, buffer);
  g_free (buffer);

  buffer = g_new0 (char, 64);
  npeek = g_buffered_input_stream_peek (G_BUFFERED_INPUT_STREAM (in), buffer, 75, 3);
  g_assert_cmpint (npeek, ==, 0);
  g_free (buffer);

  g_object_unref (in);
62
  g_object_unref (base);
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82
}

static void
test_peek_buffer (void)
{
  GInputStream *base;
  GInputStream *in;
  gssize nfill;
  gsize bufsize;
  char *buffer;

  base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
  in = g_buffered_input_stream_new (base);

  nfill = g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), strlen ("abcdefghijk"), NULL, NULL);
  buffer = (char *) g_buffered_input_stream_peek_buffer (G_BUFFERED_INPUT_STREAM (in), &bufsize);
  g_assert_cmpint (nfill, ==, bufsize);
  g_assert (0 == strncmp ("abcdefghijk", buffer, bufsize));

  g_object_unref (in);
83
  g_object_unref (base);
84 85 86 87 88 89 90 91 92 93 94
}

static void
test_set_buffer_size (void)
{
  GInputStream *base;
  GInputStream *in;
  gsize size, bufsize;

  base = g_memory_input_stream_new_from_data ("abcdefghijk", -1, NULL);
  in = g_buffered_input_stream_new (base);
95 96
  size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
  g_assert_cmpint (size, ==, 4096);
97 98 99 100 101 102 103 104 105 106 107

  g_buffered_input_stream_set_buffer_size (G_BUFFERED_INPUT_STREAM (in), 64);
  size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
  g_assert_cmpint (size, ==, 64);

  /* size cannot shrink below current content len */
  g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), strlen ("abcdefghijk"), NULL, NULL);
  g_buffered_input_stream_peek_buffer (G_BUFFERED_INPUT_STREAM (in), &bufsize);
  g_buffered_input_stream_set_buffer_size (G_BUFFERED_INPUT_STREAM (in), 2);
  size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
  g_assert_cmpint (size, ==, bufsize);
108 109
  g_object_get (in, "buffer-size", &size, NULL);
  g_assert_cmpint (size, ==, bufsize);
110 111

  g_object_unref (in);
112 113 114 115 116 117 118

  in = g_buffered_input_stream_new_sized (base, 64);
  size = g_buffered_input_stream_get_buffer_size (G_BUFFERED_INPUT_STREAM (in));
  g_assert_cmpint (size, ==, 64);

  g_object_unref (in);
  g_object_unref (base);
119 120
}

121 122 123 124 125
static void
test_read_byte (void)
{
  GInputStream *base;
  GInputStream *in;
Matthias Clasen's avatar
Matthias Clasen committed
126
  GError *error;
127 128 129

  g_test_bug ("562393");

Matthias Clasen's avatar
Matthias Clasen committed
130
  base = g_memory_input_stream_new_from_data ("abcdefgh", -1, NULL);
131 132
  in = g_buffered_input_stream_new (base);

Matthias Clasen's avatar
Matthias Clasen committed
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
  error = NULL;
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'a');
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'b');
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'c');
  g_assert_no_error (error);

  g_assert_cmpint (g_input_stream_skip (in, 3, NULL, &error), ==, 3);
  g_assert_no_error (error);

  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'g');
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'h');
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, -1);
  g_assert_no_error (error);

  g_assert (g_input_stream_close (in, NULL, &error));
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, -1);
  g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CLOSED);
155
  g_error_free (error);
156

157
  g_object_unref (in);
158 159 160 161
  g_object_unref (base);
}

static void
Matthias Clasen's avatar
Matthias Clasen committed
162
test_read (void)
163 164 165 166
{
  GInputStream *base;
  GInputStream *in;
  gchar buffer[20];
Matthias Clasen's avatar
Matthias Clasen committed
167
  GError *error;
168

Matthias Clasen's avatar
Matthias Clasen committed
169
  base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ", -1, NULL);
170 171 172 173
  in = g_buffered_input_stream_new_sized (base, 8);

  g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 0);

Matthias Clasen's avatar
Matthias Clasen committed
174 175 176
  error = NULL;
  g_assert_cmpint (g_buffered_input_stream_fill (G_BUFFERED_INPUT_STREAM (in), 8, NULL, &error), ==, 8);
  g_assert_no_error (error);
177 178 179 180

  g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 8);

  memset (buffer, 0, 20);
Matthias Clasen's avatar
Matthias Clasen committed
181
  g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 16);
182
  g_assert_cmpstr (buffer, ==, "abcdefghijklmnop");
Matthias Clasen's avatar
Matthias Clasen committed
183
  g_assert_no_error (error);
184 185 186 187

  g_assert_cmpint (g_buffered_input_stream_get_available (G_BUFFERED_INPUT_STREAM (in)), ==, 0);

  memset (buffer, 0, 20);
Matthias Clasen's avatar
Matthias Clasen committed
188
  g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 16);
189
  g_assert_cmpstr (buffer, ==, "qrstuvwxyzABCDEF");
Matthias Clasen's avatar
Matthias Clasen committed
190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
  g_assert_no_error (error);

  memset (buffer, 0, 20);
  g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 16);
  g_assert_cmpstr (buffer, ==, "GHIJKLMNOPQRSTUV");
  g_assert_no_error (error);

  memset (buffer, 0, 20);
  g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 4);
  g_assert_cmpstr (buffer, ==, "WXYZ");
  g_assert_no_error (error);

  memset (buffer, 0, 20);
  g_assert_cmpint (g_input_stream_read (in, &buffer, 16, NULL, &error), ==, 0);
  g_assert_no_error (error);
205 206 207

  g_object_unref (in);
  g_object_unref (base);
208
}
209

Matthias Clasen's avatar
Matthias Clasen committed
210 211 212 213 214
static void
test_skip (void)
{
  GInputStream *base;
  GInputStream *in;
Matthias Clasen's avatar
Matthias Clasen committed
215
  GError *error;
Matthias Clasen's avatar
Matthias Clasen committed
216

Matthias Clasen's avatar
Matthias Clasen committed
217
  base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
Matthias Clasen's avatar
Matthias Clasen committed
218 219
  in = g_buffered_input_stream_new_sized (base, 5);

Matthias Clasen's avatar
Matthias Clasen committed
220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246
  error = NULL;
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'a');
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'b');
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'c');
  g_assert_no_error (error);

  g_assert_cmpint (g_input_stream_skip (in, 7, NULL, &error), ==, 7);
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'k');
  g_assert_no_error (error);

  g_assert_cmpint (g_input_stream_skip (in, 10, NULL, &error), ==, 10);
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'v');
  g_assert_no_error (error);

  g_assert_cmpint (g_input_stream_skip (in, 20, NULL, &error), ==, 20);
  g_assert_no_error (error);
  g_assert_cmpint (g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error), ==, 'Q');
  g_assert_no_error (error);

  g_assert_cmpint (g_input_stream_skip (in, 10, NULL, &error), ==, 8);
  g_assert_no_error (error);
  g_assert_cmpint (g_input_stream_skip (in, 10, NULL, &error), ==, 0);
  g_assert_no_error (error);
Matthias Clasen's avatar
Matthias Clasen committed
247 248

  g_object_unref (in);
249
  g_object_unref (base);
Matthias Clasen's avatar
Matthias Clasen committed
250 251
}

Matthias Clasen's avatar
Matthias Clasen committed
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
static void
test_close (void)
{
  GInputStream *base;
  GInputStream *in;
  GError *error;

  base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
  in = g_buffered_input_stream_new (base);

  g_assert (g_filter_input_stream_get_close_base_stream (G_FILTER_INPUT_STREAM (in)));

  error = NULL;
  g_assert (g_input_stream_close (in, NULL, &error));
  g_assert_no_error (error);
  g_assert (g_input_stream_is_closed (base));

  g_object_unref (in);
  g_object_unref (base);

  base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
  in = g_buffered_input_stream_new (base);

  g_filter_input_stream_set_close_base_stream (G_FILTER_INPUT_STREAM (in), FALSE);

  error = NULL;
  g_assert (g_input_stream_close (in, NULL, &error));
  g_assert_no_error (error);
  g_assert (!g_input_stream_is_closed (base));

  g_object_unref (in);
  g_object_unref (base);
}

286 287 288 289 290 291 292 293 294 295 296 297 298 299
static void
test_seek (void)
{
  GInputStream *base;
  GInputStream *in;
  GError *error;
  gint byte;
  gboolean ret;

  base = g_memory_input_stream_new_from_data ("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVXYZ", -1, NULL);
  in = g_buffered_input_stream_new_sized (base, 4);
  error = NULL;

  /* Seek by read */
Dan Winship's avatar
Dan Winship committed
300
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 0);
301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'a');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 1);

  /* Seek forward (in buffer) */
  ret = g_seekable_seek (G_SEEKABLE (in), 1, G_SEEK_CUR, NULL, &error);
  g_assert_no_error (error);
  g_assert (ret);
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 2);
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'c');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 3);

  /* Seek backward (in buffer) */
  ret = g_seekable_seek (G_SEEKABLE (in), -2, G_SEEK_CUR, NULL, &error);
  g_assert_no_error (error);
  g_assert (ret);
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 1);
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'b');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 2);

  /* Seek forward (outside buffer) */
  ret = g_seekable_seek (G_SEEKABLE (in), 6, G_SEEK_CUR, NULL, &error);
  g_assert_no_error (error);
  g_assert (ret);
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 8);
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'i');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 9);

  /* Seek backward (outside buffer) */
  ret = g_seekable_seek (G_SEEKABLE (in), -6, G_SEEK_CUR, NULL, &error);
  g_assert_no_error (error);
  g_assert (ret);
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 3);
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'd');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 4);

  /* Seek from beginning */
  ret = g_seekable_seek (G_SEEKABLE (in), 8, G_SEEK_SET, NULL, &error);
  g_assert_no_error (error);
  g_assert (ret);
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 8);
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'i');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 9);

  /* Seek from end */
  ret = g_seekable_seek (G_SEEKABLE (in), -1, G_SEEK_END, NULL, &error);
  g_assert_no_error (error);
  g_assert (ret);
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 50);
  byte = g_buffered_input_stream_read_byte (G_BUFFERED_INPUT_STREAM (in), NULL, &error);
  g_assert_no_error (error);
  g_assert_cmpint (byte, ==, 'Z');
  g_assert_cmpint (g_seekable_tell (G_SEEKABLE (in)), ==, 51);

  /* Cleanup */
  g_object_unref (in);
  g_object_unref (base);
}

371 372 373 374 375 376 377
int
main (int   argc,
      char *argv[])
{
  g_test_init (&argc, &argv, NULL);
  g_test_bug_base ("http://bugzilla.gnome.org/");

378 379 380
  g_test_add_func ("/buffered-input-stream/peek", test_peek);
  g_test_add_func ("/buffered-input-stream/peek-buffer", test_peek_buffer);
  g_test_add_func ("/buffered-input-stream/set-buffer-size", test_set_buffer_size);
381
  g_test_add_func ("/buffered-input-stream/read-byte", test_read_byte);
Matthias Clasen's avatar
Matthias Clasen committed
382
  g_test_add_func ("/buffered-input-stream/read", test_read);
Matthias Clasen's avatar
Matthias Clasen committed
383
  g_test_add_func ("/buffered-input-stream/skip", test_skip);
384
  g_test_add_func ("/buffered-input-stream/seek", test_seek);
Matthias Clasen's avatar
Matthias Clasen committed
385
  g_test_add_func ("/filter-input-stream/close", test_close);
386 387 388

  return g_test_run();
}