mirror of
https://gitlab.gnome.org/GNOME/glib.git
synced 2024-12-27 07:56:14 +01:00
995a2eb50b
Signed-off-by: Simon McVittie <simon.mcvittie@collabora.co.uk> Bug: https://bugzilla.gnome.org/show_bug.cgi?id=666115 Reviewed-by: Matthias Clasen <mclasen@redhat.com>
504 lines
14 KiB
C
504 lines
14 KiB
C
/* GLib testing framework examples and tests
|
|
* Copyright (C) 2008 Red Hat, Inc.
|
|
* Authors: Tomas Bzatek <tbzatek@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>
|
|
|
|
#define MAX_LINES 0xFFF
|
|
#define MAX_BYTES 0x10000
|
|
|
|
static void
|
|
test_basic (void)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
gint val;
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
|
|
g_object_get (stream, "byte-order", &val, NULL);
|
|
g_assert_cmpint (val, ==, G_DATA_STREAM_BYTE_ORDER_BIG_ENDIAN);
|
|
g_object_set (stream, "byte-order", G_DATA_STREAM_BYTE_ORDER_LITTLE_ENDIAN, NULL);
|
|
g_assert_cmpint (g_data_input_stream_get_byte_order (G_DATA_INPUT_STREAM (stream)), ==, G_DATA_STREAM_BYTE_ORDER_LITTLE_ENDIAN);
|
|
|
|
g_object_get (stream, "newline-type", &val, NULL);
|
|
g_assert_cmpint (val, ==, G_DATA_STREAM_NEWLINE_TYPE_LF);
|
|
g_object_set (stream, "newline-type", G_DATA_STREAM_NEWLINE_TYPE_CR_LF, NULL);
|
|
g_assert_cmpint (g_data_input_stream_get_newline_type (G_DATA_INPUT_STREAM (stream)), ==, G_DATA_STREAM_NEWLINE_TYPE_CR_LF);
|
|
|
|
g_object_unref (stream);
|
|
g_object_unref (base_stream);
|
|
}
|
|
|
|
static void
|
|
test_seek_to_start (GInputStream *stream)
|
|
{
|
|
GError *error = NULL;
|
|
gboolean res = g_seekable_seek (G_SEEKABLE (stream), 0, G_SEEK_SET, NULL, &error);
|
|
g_assert_cmpint (res, ==, TRUE);
|
|
g_assert_no_error (error);
|
|
}
|
|
|
|
static void
|
|
test_read_lines (GDataStreamNewlineType newline_type)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
GError *error = NULL;
|
|
char *data;
|
|
int line;
|
|
const char* lines[MAX_LINES];
|
|
const char* endl[4] = {"\n", "\r", "\r\n", "\n"};
|
|
|
|
/* prepare data */
|
|
int i;
|
|
for (i = 0; i < MAX_LINES; i++)
|
|
lines[i] = "some_text";
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
g_assert (base_stream != NULL);
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
g_assert(stream != NULL);
|
|
|
|
/* Byte order testing */
|
|
g_data_input_stream_set_byte_order (G_DATA_INPUT_STREAM (stream), G_DATA_STREAM_BYTE_ORDER_BIG_ENDIAN);
|
|
g_assert_cmpint (g_data_input_stream_get_byte_order (G_DATA_INPUT_STREAM (stream)), ==, G_DATA_STREAM_BYTE_ORDER_BIG_ENDIAN);
|
|
g_data_input_stream_set_byte_order (G_DATA_INPUT_STREAM (stream), G_DATA_STREAM_BYTE_ORDER_LITTLE_ENDIAN);
|
|
g_assert_cmpint (g_data_input_stream_get_byte_order (G_DATA_INPUT_STREAM (stream)), ==, G_DATA_STREAM_BYTE_ORDER_LITTLE_ENDIAN);
|
|
|
|
/* Line ends testing */
|
|
g_data_input_stream_set_newline_type (G_DATA_INPUT_STREAM (stream), newline_type);
|
|
g_assert_cmpint (g_data_input_stream_get_newline_type (G_DATA_INPUT_STREAM (stream)), ==, newline_type);
|
|
|
|
|
|
/* Add sample data */
|
|
for (i = 0; i < MAX_LINES; i++)
|
|
g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (base_stream),
|
|
g_strconcat (lines[i], endl[newline_type], NULL), -1, g_free);
|
|
|
|
/* Seek to the start */
|
|
test_seek_to_start (base_stream);
|
|
|
|
/* Test read line */
|
|
error = NULL;
|
|
data = (char*)1;
|
|
line = 0;
|
|
while (data)
|
|
{
|
|
gsize length = -1;
|
|
data = g_data_input_stream_read_line (G_DATA_INPUT_STREAM (stream), &length, NULL, &error);
|
|
if (data)
|
|
{
|
|
g_assert_cmpstr (data, ==, lines[line]);
|
|
g_free (data);
|
|
g_assert_no_error (error);
|
|
line++;
|
|
}
|
|
if (error)
|
|
g_error_free (error);
|
|
}
|
|
g_assert_cmpint (line, ==, MAX_LINES);
|
|
|
|
|
|
g_object_unref (base_stream);
|
|
g_object_unref (stream);
|
|
}
|
|
|
|
static void
|
|
test_read_lines_LF (void)
|
|
{
|
|
test_read_lines (G_DATA_STREAM_NEWLINE_TYPE_LF);
|
|
}
|
|
|
|
static void
|
|
test_read_lines_CR (void)
|
|
{
|
|
test_read_lines (G_DATA_STREAM_NEWLINE_TYPE_CR);
|
|
}
|
|
|
|
static void
|
|
test_read_lines_CR_LF (void)
|
|
{
|
|
test_read_lines (G_DATA_STREAM_NEWLINE_TYPE_CR_LF);
|
|
}
|
|
|
|
static void
|
|
test_read_lines_any (void)
|
|
{
|
|
test_read_lines (G_DATA_STREAM_NEWLINE_TYPE_ANY);
|
|
}
|
|
|
|
static void
|
|
test_read_lines_LF_valid_utf8 (void)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
GError *error = NULL;
|
|
char *line;
|
|
guint n_lines = 0;
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
|
|
g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (base_stream),
|
|
"foo\nthis is valid UTF-8 ☺!\nbar\n", -1, NULL);
|
|
|
|
/* Test read line */
|
|
error = NULL;
|
|
while (TRUE)
|
|
{
|
|
gsize length = -1;
|
|
line = g_data_input_stream_read_line_utf8 (G_DATA_INPUT_STREAM (stream), &length, NULL, &error);
|
|
g_assert_no_error (error);
|
|
if (line == NULL)
|
|
break;
|
|
n_lines++;
|
|
g_free (line);
|
|
}
|
|
g_assert_cmpint (n_lines, ==, 3);
|
|
|
|
g_object_unref (base_stream);
|
|
g_object_unref (stream);
|
|
}
|
|
|
|
static void
|
|
test_read_lines_LF_invalid_utf8 (void)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
GError *error = NULL;
|
|
char *line;
|
|
guint n_lines = 0;
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
|
|
g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (base_stream),
|
|
"foo\nthis is not valid UTF-8 \xE5 =(\nbar\n", -1, NULL);
|
|
|
|
/* Test read line */
|
|
error = NULL;
|
|
while (TRUE)
|
|
{
|
|
gsize length = -1;
|
|
line = g_data_input_stream_read_line_utf8 (G_DATA_INPUT_STREAM (stream), &length, NULL, &error);
|
|
if (n_lines == 0)
|
|
g_assert_no_error (error);
|
|
else
|
|
{
|
|
g_assert (error != NULL);
|
|
g_clear_error (&error);
|
|
g_free (line);
|
|
break;
|
|
}
|
|
n_lines++;
|
|
g_free (line);
|
|
}
|
|
g_assert_cmpint (n_lines, ==, 1);
|
|
|
|
g_object_unref (base_stream);
|
|
g_object_unref (stream);
|
|
}
|
|
|
|
static void
|
|
test_read_until (void)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
GError *error = NULL;
|
|
char *data;
|
|
int line;
|
|
int i;
|
|
|
|
#define REPEATS 10 /* number of rounds */
|
|
#define DATA_STRING " part1 # part2 $ part3 % part4 ^"
|
|
#define DATA_PART_LEN 7 /* number of characters between separators */
|
|
#define DATA_SEP "#$%^"
|
|
#define DATA_SEP_LEN 4
|
|
const int DATA_PARTS_NUM = DATA_SEP_LEN * REPEATS;
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
|
|
for (i = 0; i < REPEATS; i++)
|
|
g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (base_stream), DATA_STRING, -1, NULL);
|
|
|
|
/* Test stop characters */
|
|
error = NULL;
|
|
data = (char*)1;
|
|
line = 0;
|
|
while (data)
|
|
{
|
|
gsize length = -1;
|
|
data = g_data_input_stream_read_until (G_DATA_INPUT_STREAM (stream), DATA_SEP, &length, NULL, &error);
|
|
if (data)
|
|
{
|
|
g_assert_cmpint (strlen (data), ==, DATA_PART_LEN);
|
|
g_free (data);
|
|
g_assert_no_error (error);
|
|
line++;
|
|
}
|
|
}
|
|
g_assert_no_error (error);
|
|
g_assert_cmpint (line, ==, DATA_PARTS_NUM);
|
|
|
|
g_object_unref (base_stream);
|
|
g_object_unref (stream);
|
|
}
|
|
|
|
static void
|
|
test_read_upto (void)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
GError *error = NULL;
|
|
char *data;
|
|
int line;
|
|
int i;
|
|
guchar stop_char;
|
|
|
|
#undef REPEATS
|
|
#undef DATA_STRING
|
|
#undef DATA_PART_LEN
|
|
#undef DATA_SEP
|
|
#undef DATA_SEP_LEN
|
|
#define REPEATS 10 /* number of rounds */
|
|
#define DATA_STRING " part1 # part2 $ part3 \0 part4 ^"
|
|
#define DATA_PART_LEN 7 /* number of characters between separators */
|
|
#define DATA_SEP "#$\0^"
|
|
#define DATA_SEP_LEN 4
|
|
const int DATA_PARTS_NUM = DATA_SEP_LEN * REPEATS;
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
|
|
for (i = 0; i < REPEATS; i++)
|
|
g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (base_stream), DATA_STRING, 32, NULL);
|
|
|
|
/* Test stop characters */
|
|
error = NULL;
|
|
data = (char*)1;
|
|
line = 0;
|
|
while (data)
|
|
{
|
|
gsize length = -1;
|
|
data = g_data_input_stream_read_upto (G_DATA_INPUT_STREAM (stream), DATA_SEP, DATA_SEP_LEN, &length, NULL, &error);
|
|
if (data)
|
|
{
|
|
g_assert_cmpint (strlen (data), ==, DATA_PART_LEN);
|
|
g_assert_no_error (error);
|
|
line++;
|
|
|
|
stop_char = g_data_input_stream_read_byte (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
g_assert (memchr (DATA_SEP, stop_char, DATA_SEP_LEN) != NULL);
|
|
g_assert_no_error (error);
|
|
}
|
|
g_free (data);
|
|
}
|
|
g_assert_no_error (error);
|
|
g_assert_cmpint (line, ==, DATA_PARTS_NUM);
|
|
|
|
g_object_unref (base_stream);
|
|
g_object_unref (stream);
|
|
}
|
|
enum TestDataType {
|
|
TEST_DATA_BYTE = 0,
|
|
TEST_DATA_INT16,
|
|
TEST_DATA_UINT16,
|
|
TEST_DATA_INT32,
|
|
TEST_DATA_UINT32,
|
|
TEST_DATA_INT64,
|
|
TEST_DATA_UINT64
|
|
};
|
|
|
|
#define TEST_DATA_RETYPE_BUFF(a, t, v) \
|
|
(a == TEST_DATA_BYTE ? (t) *(guchar*)v : \
|
|
(a == TEST_DATA_INT16 ? (t) *(gint16*)v : \
|
|
(a == TEST_DATA_UINT16 ? (t) *(guint16*)v : \
|
|
(a == TEST_DATA_INT32 ? (t) *(gint32*)v : \
|
|
(a == TEST_DATA_UINT32 ? (t) *(guint32*)v : \
|
|
(a == TEST_DATA_INT64 ? (t) *(gint64*)v : \
|
|
(t) *(guint64*)v ))))))
|
|
|
|
|
|
static void
|
|
test_data_array (GInputStream *stream, GInputStream *base_stream,
|
|
gpointer buffer, int len,
|
|
enum TestDataType data_type, GDataStreamByteOrder byte_order)
|
|
{
|
|
GError *error = NULL;
|
|
int pos = 0;
|
|
int data_size = 1;
|
|
gint64 data;
|
|
GDataStreamByteOrder native;
|
|
gboolean swap;
|
|
|
|
/* Seek to start */
|
|
test_seek_to_start (base_stream);
|
|
|
|
/* Set correct data size */
|
|
switch (data_type)
|
|
{
|
|
case TEST_DATA_BYTE:
|
|
data_size = 1;
|
|
break;
|
|
case TEST_DATA_INT16:
|
|
case TEST_DATA_UINT16:
|
|
data_size = 2;
|
|
break;
|
|
case TEST_DATA_INT32:
|
|
case TEST_DATA_UINT32:
|
|
data_size = 4;
|
|
break;
|
|
case TEST_DATA_INT64:
|
|
case TEST_DATA_UINT64:
|
|
data_size = 8;
|
|
break;
|
|
default:
|
|
g_assert_not_reached ();
|
|
break;
|
|
}
|
|
|
|
/* Set flag to swap bytes if needed */
|
|
native = (G_BYTE_ORDER == G_BIG_ENDIAN) ? G_DATA_STREAM_BYTE_ORDER_BIG_ENDIAN : G_DATA_STREAM_BYTE_ORDER_LITTLE_ENDIAN;
|
|
swap = (byte_order != G_DATA_STREAM_BYTE_ORDER_HOST_ENDIAN) && (byte_order != native);
|
|
|
|
data = 1;
|
|
while (data != 0)
|
|
{
|
|
switch (data_type)
|
|
{
|
|
case TEST_DATA_BYTE:
|
|
data = g_data_input_stream_read_byte (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
break;
|
|
case TEST_DATA_INT16:
|
|
data = g_data_input_stream_read_int16 (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
if (swap)
|
|
data = (gint16)GUINT16_SWAP_LE_BE((gint16)data);
|
|
break;
|
|
case TEST_DATA_UINT16:
|
|
data = g_data_input_stream_read_uint16 (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
if (swap)
|
|
data = (guint16)GUINT16_SWAP_LE_BE((guint16)data);
|
|
break;
|
|
case TEST_DATA_INT32:
|
|
data = g_data_input_stream_read_int32 (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
if (swap)
|
|
data = (gint32)GUINT32_SWAP_LE_BE((gint32)data);
|
|
break;
|
|
case TEST_DATA_UINT32:
|
|
data = g_data_input_stream_read_uint32 (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
if (swap)
|
|
data = (guint32)GUINT32_SWAP_LE_BE((guint32)data);
|
|
break;
|
|
case TEST_DATA_INT64:
|
|
data = g_data_input_stream_read_int64 (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
if (swap)
|
|
data = (gint64)GUINT64_SWAP_LE_BE((gint64)data);
|
|
break;
|
|
case TEST_DATA_UINT64:
|
|
data = g_data_input_stream_read_uint64 (G_DATA_INPUT_STREAM (stream), NULL, &error);
|
|
if (swap)
|
|
data = (guint64)GUINT64_SWAP_LE_BE((guint64)data);
|
|
break;
|
|
default:
|
|
g_assert_not_reached ();
|
|
break;
|
|
}
|
|
if (!error)
|
|
g_assert_cmpint (data, ==, TEST_DATA_RETYPE_BUFF(data_type, gint64, ((guchar*)buffer + pos)));
|
|
|
|
pos += data_size;
|
|
}
|
|
if (pos < len + 1)
|
|
g_assert_no_error (error);
|
|
if (error)
|
|
g_error_free (error);
|
|
g_assert_cmpint (pos - data_size, ==, len);
|
|
}
|
|
|
|
static void
|
|
test_read_int (void)
|
|
{
|
|
GInputStream *stream;
|
|
GInputStream *base_stream;
|
|
GRand *randomizer;
|
|
int i;
|
|
gpointer buffer;
|
|
|
|
randomizer = g_rand_new ();
|
|
buffer = g_malloc0 (MAX_BYTES);
|
|
|
|
/* Fill in some random data */
|
|
for (i = 0; i < MAX_BYTES; i++)
|
|
{
|
|
guchar x = 0;
|
|
while (! x)
|
|
x = (guchar)g_rand_int (randomizer);
|
|
*(guchar*)((guchar*)buffer + sizeof(guchar) * i) = x;
|
|
}
|
|
|
|
base_stream = g_memory_input_stream_new ();
|
|
stream = G_INPUT_STREAM (g_data_input_stream_new (base_stream));
|
|
g_memory_input_stream_add_data (G_MEMORY_INPUT_STREAM (base_stream), buffer, MAX_BYTES, NULL);
|
|
|
|
|
|
for (i = 0; i < 3; i++)
|
|
{
|
|
int j;
|
|
g_data_input_stream_set_byte_order (G_DATA_INPUT_STREAM (stream), i);
|
|
|
|
for (j = 0; j <= TEST_DATA_UINT64; j++)
|
|
test_data_array (stream, base_stream, buffer, MAX_BYTES, j, i);
|
|
}
|
|
|
|
g_object_unref (base_stream);
|
|
g_object_unref (stream);
|
|
g_rand_free (randomizer);
|
|
g_free (buffer);
|
|
}
|
|
|
|
|
|
int
|
|
main (int argc,
|
|
char *argv[])
|
|
{
|
|
g_type_init ();
|
|
g_test_init (&argc, &argv, NULL);
|
|
|
|
g_test_add_func ("/data-input-stream/basic", test_basic);
|
|
g_test_add_func ("/data-input-stream/read-lines-LF", test_read_lines_LF);
|
|
g_test_add_func ("/data-input-stream/read-lines-LF-valid-utf8", test_read_lines_LF_valid_utf8);
|
|
g_test_add_func ("/data-input-stream/read-lines-LF-invalid-utf8", test_read_lines_LF_invalid_utf8);
|
|
g_test_add_func ("/data-input-stream/read-lines-CR", test_read_lines_CR);
|
|
g_test_add_func ("/data-input-stream/read-lines-CR-LF", test_read_lines_CR_LF);
|
|
g_test_add_func ("/data-input-stream/read-lines-any", test_read_lines_any);
|
|
g_test_add_func ("/data-input-stream/read-until", test_read_until);
|
|
g_test_add_func ("/data-input-stream/read-upto", test_read_upto);
|
|
g_test_add_func ("/data-input-stream/read-int", test_read_int);
|
|
|
|
return g_test_run();
|
|
}
|