summaryrefslogtreecommitdiff
path: root/gdata/tests/buffer.c
blob: 02012d6cef75aab19d3abedaf06c25bc8e1645ce (plain)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
/*
 * GData Client
 * Copyright (C) Philip Withnall 2016 <philip@tecnocode.co.uk>
 *
 * GData Client is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * GData Client 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.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with GData Client.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "config.h"

#include <glib.h>
#include <unistd.h>

#include "gdata.h"
#include "common.h"

/* gdata-buffer.h is private, so just include the C file for easy testing. */
#include "gdata-buffer.c"


typedef struct {
	gpointer unused;
} Fixture;


static void
set_up (Fixture *f, gconstpointer user_data)
{
	/* Abort if we end up blocking. */
	alarm (30);
}

static void
tear_down (Fixture *f, gconstpointer user_data)
{
	/* Reset the alarm. */
	alarm (0);
}

static void
test_buffer_construction (Fixture *f, gconstpointer user_data)
{
	GDataBuffer *buffer = NULL;  /* owned */

	buffer = gdata_buffer_new ();
	gdata_buffer_free (buffer);
}

static void
test_buffer_instant_eof (Fixture *f, gconstpointer user_data)
{
	GDataBuffer *buffer = NULL;  /* owned */
	gboolean reached_eof = FALSE;
	guint8 buf[1];

	buffer = gdata_buffer_new ();

	g_assert_false (gdata_buffer_push_data (buffer, NULL, 0));
	g_assert_cmpuint (gdata_buffer_pop_data (buffer, buf, sizeof (buf),
	                                         &reached_eof, NULL), ==, 0);
	g_assert_true (reached_eof);

	gdata_buffer_free (buffer);
}

static void
test_buffer_basic (Fixture *f, gconstpointer user_data)
{
	GDataBuffer *buffer = NULL;  /* owned */
	gboolean reached_eof = FALSE;
	guint8 buf[100];
	guint8 buf2[100];
	gsize i;

	buffer = gdata_buffer_new ();

	for (i = 0; i < sizeof (buf); i++)
		buf[i] = i;

	g_assert_true (gdata_buffer_push_data (buffer, buf, sizeof (buf)));
	g_assert_false (gdata_buffer_push_data (buffer, NULL, 0));

	g_assert_cmpuint (gdata_buffer_pop_data (buffer, buf2,
	                                         sizeof (buf2) / 2,
	                                         &reached_eof, NULL), ==,
	                                         sizeof (buf2) / 2);
	g_assert_false (reached_eof);
	g_assert_cmpuint (gdata_buffer_pop_data (buffer,
	                                         buf2 + sizeof (buf2) / 2,
	                                         sizeof (buf2) / 2,
	                                         &reached_eof, NULL), ==,
	                                         sizeof (buf2) / 2);
	g_assert_true (reached_eof);

	for (i = 0; i < sizeof (buf); i++)
		g_assert_cmpuint (buf[i], ==, buf2[i]);

	gdata_buffer_free (buffer);
}

int
main (int argc, char *argv[])
{
	gdata_test_init (argc, argv);

	/* Only print out headers, since we're sending a lot of data. */
	g_setenv ("LIBGDATA_DEBUG", "2" /* GDATA_LOG_HEADERS */, TRUE);

	g_test_add ("/buffer/construction", Fixture, NULL,
	            set_up, test_buffer_construction, tear_down);
	g_test_add ("/buffer/instant-eof", Fixture, NULL,
	            set_up, test_buffer_instant_eof, tear_down);
	g_test_add ("/buffer/basic", Fixture, NULL,
	            set_up, test_buffer_basic, tear_down);

	return g_test_run ();
}