pollcore!wip/gmaincontext

2 files changed, 187 insertions, 0 deletions

diff --git a/glib/tests/Makefile.am b/glib/tests/Makefile.am
index daf4229f7..1de080ce9 100644
--- a/glib/tests/Makefile.am
+++ b/glib/tests/Makefile.am
@@ -12,6 +12,8 @@ spawn_multithreaded_LDFLAGS = $(patsubst -lgcov,,$(LDFLAGS))
 
 # -----------------------------------------------------------------------------
 
+noinst_PROGRAMS = pollcore
+
 dist_test_data = \
 	keyfiletest.ini			\
 	pages.ini			\
@@ -71,6 +73,7 @@ test_programs = \
 	option-context			\
 	option-argv0			\
 	pattern				\
+	pollcore			\
 	private				\
 	protocol			\
 	queue				\
diff --git a/glib/tests/pollcore.c b/glib/tests/pollcore.c
new file mode 100644
index 000000000..20b4ab3f5
--- /dev/null
+++ b/glib/tests/pollcore.c
@@ -0,0 +1,184 @@
+/*
+ * Copyright © 2014 Canonical Limited
+ *
+ * This library 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 of the licence, or (at your option) any later version.
+ *
+ * This library 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 this library; if not, see <http://www.gnu.org/licenses/>.
+ *
+ * Author: Ryan Lortie <desrt@desrt.ca>
+ */
+
+#include "config.h"
+
+#include <glib.h>
+
+#include "gpollcore.h"
+
+#include <sys/poll.h>
+#include <unistd.h>
+
+static void
+update_free (gpointer data)
+{
+  GPollUpdate *update = data;
+
+  g_slice_free (GPollUpdate, update);
+}
+
+static void
+add_update (GHashTable *table,
+            gint        fd,
+            guint       old_events,
+            guint       new_events,
+            gpointer    user_data)
+{
+  GPollUpdate *update;
+
+  update = g_hash_table_lookup (table, GINT_TO_POINTER (fd));
+
+  if (!update)
+    {
+      update = g_slice_new (GPollUpdate);
+      update->old_events = old_events;
+      update->user_data = user_data;
+
+      g_hash_table_insert (table, GINT_TO_POINTER (fd), update);
+    }
+  else
+    {
+      g_assert_cmpint (update->new_events, ==, old_events);
+      g_assert (update->user_data == user_data);
+    }
+
+  update->new_events = new_events;
+
+  /* XXX: what to do if only user_data changed? */
+  if (update->new_events == update->old_events)
+    g_hash_table_remove (table, GINT_TO_POINTER (fd));
+}
+
+static gboolean
+is_ready (gint fd)
+{
+  struct pollfd pfd;
+
+  pfd.fd = fd;
+  pfd.events = POLLIN;
+
+  return poll (&pfd, 1, 0);
+}
+
+static gpointer
+kick_core (gpointer user_data)
+{
+  GPollCore *core = user_data;
+
+  g_usleep (G_TIME_SPAN_SECOND * 0.1);
+  g_poll_core_set_ready_time (core, 0);
+  return NULL;
+}
+
+static void
+test_pollcore (void)
+{
+  GPollEvent events[10];
+  GHashTable *updates;
+  GPollCore core;
+  GMutex lock;
+  gint pipes[2];
+  gint64 time;
+  gchar b;
+  gint fd;
+  gint r;
+
+  g_mutex_init (&lock);
+  g_mutex_lock (&lock);
+
+  updates = g_hash_table_new_full (NULL, NULL, NULL, update_free);
+
+  pipe (pipes);
+
+  g_poll_core_init (&core);
+
+  fd = g_poll_core_get_unix_fd (&core);
+
+  r = g_poll_core_update_and_collect (&core, NULL, NULL, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 0);
+  g_assert (!is_ready (fd));
+
+  add_update (updates, pipes[1], 0, G_IO_IN, test_pollcore);
+  add_update (updates, pipes[1], G_IO_IN, G_IO_OUT, test_pollcore);
+  r = g_poll_core_update_and_collect (&core, updates, NULL, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 1);
+  g_hash_table_remove_all (updates);
+  g_assert (is_ready (fd));
+
+  add_update (updates, pipes[0], 0, G_IO_IN, test_pollcore);
+  r = g_poll_core_update_and_collect (&core, updates, NULL, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 1);
+  g_hash_table_remove_all (updates);
+  g_assert (is_ready (fd));
+
+  write (pipes[1], "x", 1);
+
+  r = g_poll_core_update_and_collect (&core, updates, NULL, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 2);
+  g_assert (is_ready (fd));
+
+  read (pipes[0], &b, 1);
+
+  time = g_get_monotonic_time () + G_TIME_SPAN_SECOND * 0.1;
+
+  add_update (updates, pipes[1], G_IO_OUT, 0, NULL);
+  r = g_poll_core_update_and_collect (&core, updates, &time, events, G_N_ELEMENTS (events));
+  g_hash_table_remove_all (updates);
+  g_assert_cmpint (r, ==, 0);
+  g_assert (!is_ready (fd));
+
+  g_poll_core_wait (&core, &lock);
+  g_assert (is_ready (fd));
+
+  r = g_poll_core_update_and_collect (&core, NULL, NULL, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 1);
+  g_assert (is_ready (fd));
+
+  time = -1;
+  r = g_poll_core_update_and_collect (&core, updates, &time, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 0);
+  g_assert (!is_ready (fd));
+
+  g_thread_unref (g_thread_new ("kicker", kick_core, &core));
+
+  g_poll_core_wait (&core, &lock);
+  r = g_poll_core_update_and_collect (&core, NULL, NULL, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 1);
+  g_assert (is_ready (fd));
+
+  r = g_poll_core_update_and_collect (&core, NULL, &time, events, G_N_ELEMENTS (events));
+  g_assert_cmpint (r, ==, 0);
+  g_assert (!is_ready (fd));
+
+  g_poll_core_clear (&core);
+
+  g_mutex_unlock (&lock);
+  g_mutex_clear (&lock);
+}
+
+int
+main (int argc, char **argv)
+{
+  g_test_init (&argc, &argv, NULL);
+
+  g_test_add_func ("/glib/pollcore", test_pollcore);
+
+  return g_test_run ();
+}