--- /dev/null
+/* GStreamer
+ * Copyright (C) 2013 Collabora Ltd.
+ * Author: Sebastian Dröge <sebastian.droege@collabora.co.uk>
+ *
+ * gstcontext.c: Unit test for GstContext
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, 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
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#include <gst/check/gstcheck.h>
+
+GST_START_TEST (test_basic)
+{
+ GstContext *c1, *c2;
+ GstStructure *s1, *s2;
+
+ c1 = gst_context_new ();
+ fail_unless (c1 != NULL);
+ fail_unless (GST_IS_CONTEXT (c1));
+ s1 = (GstStructure *) gst_context_get_structure (c1);
+ fail_unless (s1 != NULL);
+ gst_structure_set (s1, "foobar", G_TYPE_INT, 1, NULL);
+
+ c2 = gst_context_copy (c1);
+ fail_unless (c2 != NULL);
+ fail_unless (GST_IS_CONTEXT (c2));
+ s2 = (GstStructure *) gst_context_get_structure (c2);
+ fail_unless (s2 != NULL);
+ fail_unless (gst_structure_is_equal (s1, s2));
+
+ gst_context_unref (c1);
+ gst_context_unref (c2);
+}
+
+GST_END_TEST;
+
+typedef struct
+{
+ GstElement parent;
+ gboolean set_before_ready;
+ gboolean set_from_need_context;
+ gboolean create_self;
+} GstContextElement;
+
+typedef struct
+{
+ GstElementClass parent_class;
+} GstContextElementClass;
+
+GType gst_context_element_get_type (void);
+
+G_DEFINE_TYPE (GstContextElement, gst_context_element, GST_TYPE_ELEMENT);
+
+static void
+gst_context_element_set_context (GstElement * element, GstContext * context)
+{
+ GST_ELEMENT_CLASS (gst_context_element_parent_class)->set_context (element,
+ context);
+}
+
+static GstStateChangeReturn
+gst_context_element_change_state (GstElement * element,
+ GstStateChange transition)
+{
+ GstContextElement *celement = (GstContextElement *) element;
+
+ if (transition == GST_STATE_CHANGE_NULL_TO_READY) {
+ GstContext *context;
+ GstStructure *s;
+ GstMessage *msg;
+ gboolean have_foobar = FALSE;
+
+ context = gst_element_get_context (element);
+ if (context) {
+ s = gst_context_get_structure (context);
+ if (gst_structure_has_field (s, "foobar"))
+ have_foobar = TRUE;
+ gst_context_unref (context);
+ }
+
+ if (celement->set_before_ready && !have_foobar)
+ return GST_STATE_CHANGE_FAILURE;
+ else if (celement->set_before_ready)
+ return
+ GST_ELEMENT_CLASS (gst_context_element_parent_class)->change_state
+ (element, transition);
+
+ if (celement->set_from_need_context && have_foobar)
+ return GST_STATE_CHANGE_FAILURE;
+
+ if (!have_foobar) {
+ /* Here we would first query downstream for a context but we have no pads */
+ msg = gst_message_new_need_context (GST_OBJECT (element));
+ gst_message_add_context_type (msg, "foobar");
+ gst_element_post_message (element, msg);
+
+ context = gst_element_get_context (element);
+ if (context) {
+ s = gst_context_get_structure (context);
+ if (gst_structure_has_field (s, "foobar"))
+ have_foobar = TRUE;
+ gst_context_unref (context);
+ }
+ }
+
+ if (celement->set_from_need_context && !have_foobar)
+ return GST_STATE_CHANGE_FAILURE;
+ else if (celement->set_from_need_context)
+ return
+ GST_ELEMENT_CLASS (gst_context_element_parent_class)->change_state
+ (element, transition);
+
+ if (celement->create_self && have_foobar)
+ return GST_STATE_CHANGE_FAILURE;
+
+ if (!have_foobar) {
+ context = gst_element_get_context (element);
+ if (context)
+ context = gst_context_make_writable (context);
+ else
+ context = gst_context_new ();
+ s = gst_context_get_structure (context);
+ gst_structure_set (s, "foobar", G_TYPE_INT, 123, NULL);
+ gst_element_set_context (element, context);
+ msg =
+ gst_message_new_have_context (GST_OBJECT (element),
+ gst_context_ref (context));
+ gst_element_post_message (element, msg);
+ gst_context_unref (context);
+ }
+ return
+ GST_ELEMENT_CLASS (gst_context_element_parent_class)->change_state
+ (element, transition);
+ }
+
+ return
+ GST_ELEMENT_CLASS (gst_context_element_parent_class)->change_state
+ (element, transition);
+}
+
+static void
+gst_context_element_class_init (GstContextElementClass * klass)
+{
+ GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
+
+ gstelement_class->set_context = gst_context_element_set_context;
+ gstelement_class->change_state = gst_context_element_change_state;
+}
+
+static void
+gst_context_element_init (GstContextElement * self)
+{
+
+}
+
+GST_START_TEST (test_element_set_before_ready)
+{
+ GstBus *bus;
+ GstElement *element;
+ GstContext *context, *context2;
+ GstStructure *s, *s2;
+ GstMessage *msg;
+
+ element = g_object_new (gst_context_element_get_type (), NULL);
+ bus = gst_bus_new ();
+ gst_element_set_bus (element, bus);
+
+ ((GstContextElement *) element)->set_before_ready = TRUE;
+
+ fail_if (gst_element_set_state (element,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS);
+ fail_if (gst_bus_pop (bus) != NULL);
+
+ context = gst_context_new ();
+ s = gst_context_get_structure (context);
+ gst_structure_set (s, "foobar", G_TYPE_INT, 123, NULL);
+ gst_element_set_context (element, context);
+ fail_unless (gst_element_set_state (element,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS);
+ fail_unless ((msg =
+ gst_bus_pop_filtered (bus, GST_MESSAGE_STATE_CHANGED)) != NULL);
+ gst_message_unref (msg);
+ fail_if (gst_bus_pop (bus) != NULL);
+
+ context2 = gst_element_get_context (element);
+ fail_unless (GST_IS_CONTEXT (context2));
+ s2 = gst_context_get_structure (context2);
+ fail_unless (gst_structure_is_equal (s, s2));
+
+ gst_context_unref (context);
+ gst_context_unref (context2);
+
+ gst_element_set_bus (element, NULL);
+ fail_unless (gst_element_set_state (element,
+ GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS);
+
+ gst_object_unref (bus);
+ gst_object_unref (element);
+}
+
+GST_END_TEST;
+
+static GstBusSyncReply
+sync_handler (GstBus * bus, GstMessage * message, gpointer user_data)
+{
+ if (GST_MESSAGE_TYPE (message) == GST_MESSAGE_NEED_CONTEXT) {
+ guint n;
+ const gchar *type;
+ GstElement *element = GST_ELEMENT (GST_MESSAGE_SRC (message));
+ GstContext *context;
+ GstStructure *s;
+
+ n = gst_message_get_n_context_types (message);
+ fail_unless (n == 1);
+ fail_unless (gst_message_parse_nth_context_type (message, 0, &type));
+ fail_unless_equals_string (type, "foobar");
+ context = gst_element_get_context (element);
+ if (context)
+ context = gst_context_make_writable (context);
+ else
+ context = gst_context_new ();
+ s = gst_context_get_structure (context);
+ gst_structure_set (s, "foobar", G_TYPE_INT, 123, NULL);
+ gst_element_set_context (element, context);
+ gst_context_unref (context);
+ }
+
+ return GST_BUS_PASS;
+}
+
+GST_START_TEST (test_element_set_from_need_context)
+{
+ GstBus *bus;
+ GstElement *element;
+ GstContext *context;
+ GstStructure *s;
+ GstMessage *msg;
+
+ element = g_object_new (gst_context_element_get_type (), NULL);
+ bus = gst_bus_new ();
+ gst_bus_set_sync_handler (bus, sync_handler, NULL, NULL);
+ gst_element_set_bus (element, bus);
+
+ ((GstContextElement *) element)->set_from_need_context = TRUE;
+
+ fail_unless (gst_element_set_state (element,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS);
+ fail_unless ((msg =
+ gst_bus_pop_filtered (bus, GST_MESSAGE_NEED_CONTEXT)) != NULL);
+ gst_message_unref (msg);
+ fail_unless ((msg =
+ gst_bus_pop_filtered (bus, GST_MESSAGE_STATE_CHANGED)) != NULL);
+ gst_message_unref (msg);
+ fail_if (gst_bus_pop (bus) != NULL);
+
+ context = gst_element_get_context (element);
+ fail_unless (GST_IS_CONTEXT (context));
+ s = gst_context_get_structure (context);
+ fail_unless (gst_structure_has_field (s, "foobar"));
+
+ gst_context_unref (context);
+
+ gst_element_set_bus (element, NULL);
+ fail_unless (gst_element_set_state (element,
+ GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS);
+
+ gst_object_unref (bus);
+ gst_object_unref (element);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_element_create_self)
+{
+ GstBus *bus;
+ GstElement *element;
+ GstContext *context;
+ GstStructure *s;
+ GstMessage *msg;
+
+ element = g_object_new (gst_context_element_get_type (), NULL);
+ bus = gst_bus_new ();
+ gst_element_set_bus (element, bus);
+
+ ((GstContextElement *) element)->create_self = TRUE;
+
+ fail_unless (gst_element_set_state (element,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS);
+ fail_unless ((msg =
+ gst_bus_pop_filtered (bus, GST_MESSAGE_NEED_CONTEXT)) != NULL);
+ gst_message_unref (msg);
+ fail_unless ((msg =
+ gst_bus_pop_filtered (bus, GST_MESSAGE_HAVE_CONTEXT)) != NULL);
+ gst_message_parse_have_context (msg, &context);
+ fail_unless (GST_IS_CONTEXT (context));
+ s = gst_context_get_structure (context);
+ fail_unless (gst_structure_has_field (s, "foobar"));
+ gst_context_unref (context);
+ gst_message_unref (msg);
+ fail_unless ((msg =
+ gst_bus_pop_filtered (bus, GST_MESSAGE_STATE_CHANGED)) != NULL);
+ gst_message_unref (msg);
+ fail_if (gst_bus_pop (bus) != NULL);
+
+ context = gst_element_get_context (element);
+ fail_unless (GST_IS_CONTEXT (context));
+ s = gst_context_get_structure (context);
+ fail_unless (gst_structure_has_field (s, "foobar"));
+
+ gst_context_unref (context);
+
+ gst_element_set_bus (element, NULL);
+ fail_unless (gst_element_set_state (element,
+ GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS);
+
+ gst_object_unref (bus);
+ gst_object_unref (element);
+}
+
+GST_END_TEST;
+
+static Suite *
+gst_context_suite (void)
+{
+ Suite *s = suite_create ("GstContext");
+ TCase *tc_chain = tcase_create ("context tests");
+
+ tcase_set_timeout (tc_chain, 0);
+
+ suite_add_tcase (s, tc_chain);
+ tcase_add_test (tc_chain, test_basic);
+ tcase_add_test (tc_chain, test_element_set_before_ready);
+ tcase_add_test (tc_chain, test_element_set_from_need_context);
+ tcase_add_test (tc_chain, test_element_create_self);
+
+ return s;
+}
+
+GST_CHECK_MAIN (gst_context);