tests: Add some basic tests for GDataBuffer

2 files changed, 129 insertions, 0 deletions

diff --git a/gdata/tests/Makefile.am b/gdata/tests/Makefile.am
index a3ca6264..7ee767ea 100644
--- a/gdata/tests/Makefile.am
+++ b/gdata/tests/Makefile.am
@@ -47,6 +47,7 @@ LDADD = \
 	$(NULL)
 
 test_programs = \
+	buffer \
 	general \
 	calendar \
 	contacts \
diff --git a/gdata/tests/buffer.c b/gdata/tests/buffer.c
new file mode 100644
index 00000000..02012d6c
--- /dev/null
+++ b/gdata/tests/buffer.c
@@ -0,0 +1,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 ();
+}