1 files changed, 83 insertions, 0 deletions
diff --git a/test/endless/test-custom-container.c b/test/endless/test-custom-container.c
new file mode 100644
index 0000000..42cf99f
--- /dev/null
+++ b/test/endless/test-custom-container.c
@@ -0,0 +1,83 @@
+/* Copyright 2014 Endless Mobile, Inc. */
+
+#include <gtk/gtk.h>
+#include <endless/endless.h>
+
+#include "run-tests.h"
+
+typedef struct
+{
+  GtkContainer *container;
+  GtkWidget *child1;
+  GtkWidget *child2;
+  GtkWidget *child3;
+} CustomContainerFixture;
+
+#define ADD_CUSTOM_CONTAINER_TEST(path, test_func) \
+  g_test_add ((path), CustomContainerFixture, NULL, \
+              custom_container_fixture_setup, \
+              (test_func), \
+              custom_container_fixture_teardown)
+
+
+static void
+custom_container_fixture_setup (CustomContainerFixture *fixture,
+                                gconstpointer           unused G_GNUC_UNUSED)
+{
+  // We acquire the widget ref so they don't automatically get destroyed after
+  // being removed from the container.
+  fixture->child1 = g_object_ref_sink (gtk_label_new ("1"));
+  fixture->child2 = g_object_ref_sink (gtk_label_new ("2"));
+  fixture->child3 = g_object_ref_sink (gtk_label_new ("3"));
+  fixture->container = GTK_CONTAINER (eos_custom_container_new ());
+}
+
+static void
+custom_container_fixture_teardown (CustomContainerFixture *fixture,
+                                   gconstpointer           unused G_GNUC_UNUSED)
+{
+  gtk_widget_destroy (fixture->child1);
+  gtk_widget_destroy (fixture->child2);
+  gtk_widget_destroy (fixture->child3);
+  gtk_widget_destroy ((GtkWidget *) fixture->container);
+  g_object_unref (fixture->child1);
+  g_object_unref (fixture->child2);
+  g_object_unref (fixture->child3);
+}
+
+static void
+test_custom_container_add (CustomContainerFixture *fixture,
+                           gconstpointer           unused G_GNUC_UNUSED)
+{
+  gtk_container_add (fixture->container, fixture->child1);
+  gtk_container_add (fixture->container, fixture->child2);
+  gtk_container_add (fixture->container, fixture->child3);
+
+  g_assert (gtk_widget_get_parent (fixture->child1) == GTK_WIDGET (fixture->container));
+  GList *children = gtk_container_get_children (fixture->container);
+  g_assert (g_list_length (children) == 3);
+  g_assert (g_list_find (children, fixture->child1) != NULL);
+  g_assert (g_list_find (children, fixture->child2) != NULL);
+  g_assert (g_list_find (children, fixture->child3) != NULL);
+}
+
+static void
+test_custom_container_remove (CustomContainerFixture *fixture,
+                              gconstpointer           unused G_GNUC_UNUSED)
+{
+  gtk_container_add (fixture->container, fixture->child1);
+  gtk_container_add (fixture->container, fixture->child2);
+  gtk_container_add (fixture->container, fixture->child3);
+  gtk_container_remove (fixture->container, fixture->child2);
+
+  g_assert (gtk_widget_get_parent (fixture->child2) != GTK_WIDGET (fixture->container));
+  GList *children = gtk_container_get_children (fixture->container);
+  g_assert (g_list_find (children, fixture->child2) == NULL);
+}
+
+void
+add_custom_container_tests (void)
+{
+  ADD_CUSTOM_CONTAINER_TEST ("/custom-container/add", test_custom_container_add);
+  ADD_CUSTOM_CONTAINER_TEST ("/custom-container/remove", test_custom_container_remove);
+}