unit: Add util function for unit tests

1 files changed, 109 insertions, 0 deletions

diff --git a/unit/utils.c b/unit/utils.c
new file mode 100644
index 00000000..bd424392
--- /dev/null
+++ b/unit/utils.c
@@ -0,0 +1,109 @@
+/*
+ *
+ *  Connection Manager
+ *
+ *  Copyright (C) 2011  BWM CarIT GmbH. All rights reserved.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License version 2 as
+ *  published by the Free Software Foundation.
+ *
+ *  This program 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 General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdlib.h>
+
+#include "gdbus/gdbus.h"
+
+#include "test-connman.h"
+
+#define ENABLE_WRAPPER 1
+
+gboolean util_quit_loop(gpointer data)
+{
+	struct test_fix *fix = data;
+
+	g_main_loop_quit(fix->main_loop);
+
+	return FALSE;
+}
+
+guint util_idle_call(struct test_fix *fix, GSourceFunc func,
+			GDestroyNotify notify)
+{
+	GSource *source;
+	guint id;
+
+	source = g_idle_source_new();
+	g_source_set_callback(source, func, fix, notify);
+	id = g_source_attach(source, g_main_loop_get_context(fix->main_loop));
+	g_source_unref(source);
+
+	return id;
+}
+
+guint util_call(struct test_fix *fix, GSourceFunc func,
+		GDestroyNotify notify)
+{
+	GSource *source;
+	guint id;
+
+	source = g_timeout_source_new(0);
+	g_source_set_callback(source, func, fix, notify);
+	id = g_source_attach(source, g_main_loop_get_context(fix->main_loop));
+	g_source_unref(source);
+
+	return id;
+}
+
+void util_setup(struct test_fix *fix, gconstpointer data)
+{
+	fix->main_loop = g_main_loop_new(NULL, FALSE);
+	fix->main_connection = g_dbus_setup_private(DBUS_BUS_SYSTEM,
+							NULL, NULL);
+}
+
+void util_teardown(struct test_fix *fix, gconstpointer data)
+{
+	dbus_connection_close(fix->main_connection);
+	dbus_connection_unref(fix->main_connection);
+
+	g_main_loop_unref(fix->main_loop);
+}
+
+static void util_wrapper(struct test_fix *fix, gconstpointer data)
+{
+	GSourceFunc func = data;
+#if ENABLE_WRAPPER
+	if (g_test_trap_fork(60 * 1000 * 1000, 0) == TRUE) {
+		util_call(fix, func, NULL);
+		g_main_loop_run(fix->main_loop);
+		exit(0);
+	}
+
+	g_test_trap_assert_passed();
+#else
+	util_call(fix, func, NULL);
+	g_main_loop_run(fix->main_loop);
+#endif
+}
+
+void util_test_add(const char *test_name, GSourceFunc test_func,
+			util_test_setup_cb setup_cb,
+			util_test_teardown_cb teardown_cb)
+{
+	g_test_add(test_name, struct test_fix, test_func,
+		setup_cb, util_wrapper, teardown_cb);
+}