context: use context on buffers instead of caps
[platform/upstream/gstreamer.git] / tests / check / elements / tee.c
index e48bc40..b07dba6 100644 (file)
@@ -3,6 +3,8 @@
  * unit test for tee
  *
  * Copyright (C) <2007> Wim Taymans <wim dot taymans at gmail dot com>
+ * Copyright (C) <2008> Ole André Vadla Ravnås <ole.andre.ravnas@tandberg.com>
+ * Copyright (C) <2008> Christian Berentsen <christian.berentsen@tandberg.com>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Library General Public
@@ -114,7 +116,7 @@ GST_START_TEST (test_stress)
 {
   GstElement *pipeline;
   GstElement *tee;
-  gchar *desc;
+  const gchar *desc;
   GstBus *bus;
   GstMessage *msg;
   gint i;
@@ -164,15 +166,427 @@ GST_START_TEST (test_stress)
 
 GST_END_TEST;
 
+typedef struct
+{
+  GstElement *tee;
+  GstCaps *caps;
+  GstPad *start_srcpad;
+  GstPad *tee_sinkpad;
+  GstPad *tee_srcpad;
+  GstPad *final_sinkpad;
+  GThread *app_thread;
+  gint countdown;
+  gboolean app_thread_prepped;
+  gboolean bufferalloc_blocked;
+} BufferAllocHarness;
+
+static void
+buffer_alloc_harness_setup (BufferAllocHarness * h, gint countdown)
+{
+  h->tee = gst_check_setup_element ("tee");
+  fail_if (h->tee == NULL);
+
+  h->countdown = countdown;
+
+  fail_unless_equals_int (gst_element_set_state (h->tee, GST_STATE_PLAYING),
+      TRUE);
+
+  h->caps = gst_caps_new_simple ("video/x-raw-yuv", NULL);
+
+  h->start_srcpad = gst_pad_new ("src", GST_PAD_SRC);
+  fail_if (h->start_srcpad == NULL);
+  fail_unless (gst_pad_set_caps (h->start_srcpad, h->caps) == TRUE);
+  fail_unless (gst_pad_set_active (h->start_srcpad, TRUE) == TRUE);
+
+  h->tee_sinkpad = gst_element_get_static_pad (h->tee, "sink");
+  fail_if (h->tee_sinkpad == NULL);
+
+  h->tee_srcpad = gst_element_get_request_pad (h->tee, "src%d");
+  fail_if (h->tee_srcpad == NULL);
+
+  h->final_sinkpad = gst_pad_new ("sink", GST_PAD_SINK);
+  fail_if (h->final_sinkpad == NULL);
+  fail_unless (gst_pad_set_caps (h->final_sinkpad, h->caps) == TRUE);
+  fail_unless (gst_pad_set_active (h->final_sinkpad, TRUE) == TRUE);
+  g_object_set_qdata (G_OBJECT (h->final_sinkpad),
+      g_quark_from_static_string ("buffer-alloc-harness"), h);
+
+  fail_unless_equals_int (gst_pad_link (h->start_srcpad, h->tee_sinkpad),
+      GST_PAD_LINK_OK);
+  fail_unless_equals_int (gst_pad_link (h->tee_srcpad, h->final_sinkpad),
+      GST_PAD_LINK_OK);
+}
+
+static void
+buffer_alloc_harness_teardown (BufferAllocHarness * h)
+{
+  if (h->app_thread)
+    g_thread_join (h->app_thread);
+
+  gst_pad_set_active (h->final_sinkpad, FALSE);
+  gst_object_unref (h->final_sinkpad);
+  gst_object_unref (h->tee_srcpad);
+  gst_object_unref (h->tee_sinkpad);
+  gst_pad_set_active (h->start_srcpad, FALSE);
+  gst_object_unref (h->start_srcpad);
+  gst_caps_unref (h->caps);
+  gst_check_teardown_element (h->tee);
+}
+
+#if 0
+static gpointer
+app_thread_func (gpointer data)
+{
+  BufferAllocHarness *h = data;
+
+  /* Signal that we are about to call release_request_pad(). */
+  g_mutex_lock (check_mutex);
+  h->app_thread_prepped = TRUE;
+  g_cond_signal (check_cond);
+  g_mutex_unlock (check_mutex);
+
+  /* Simulate that the app releases the pad while the streaming thread is in
+   * buffer_alloc below. */
+  gst_element_release_request_pad (h->tee, h->tee_srcpad);
+
+  /* Signal the bufferalloc function below if it's still waiting. */
+  g_mutex_lock (check_mutex);
+  h->bufferalloc_blocked = FALSE;
+  g_cond_signal (check_cond);
+  g_mutex_unlock (check_mutex);
+
+  return NULL;
+}
+#endif
+
+#if 0
+static GstFlowReturn
+final_sinkpad_bufferalloc (GstPad * pad, guint64 offset, guint size,
+    GstCaps * caps, GstBuffer ** buf)
+{
+  BufferAllocHarness *h;
+  GTimeVal deadline;
+
+  h = g_object_get_qdata (G_OBJECT (pad),
+      g_quark_from_static_string ("buffer-alloc-harness"));
+  g_assert (h != NULL);
+
+  if (--(h->countdown) == 0) {
+    /* Time to make the app release the pad. */
+    h->app_thread_prepped = FALSE;
+    h->bufferalloc_blocked = TRUE;
+
+    h->app_thread = g_thread_create (app_thread_func, h, TRUE, NULL);
+    fail_if (h->app_thread == NULL);
+
+    /* Wait for the app thread to get ready to call release_request_pad(). */
+    g_mutex_lock (check_mutex);
+    while (!h->app_thread_prepped)
+      g_cond_wait (check_cond, check_mutex);
+    g_mutex_unlock (check_mutex);
+
+    /* Now wait for it to do that within a second, to avoid deadlocking
+     * in the event of future changes to the locking semantics. */
+    g_mutex_lock (check_mutex);
+    g_get_current_time (&deadline);
+    deadline.tv_sec += 1;
+    while (h->bufferalloc_blocked) {
+      if (!g_cond_timed_wait (check_cond, check_mutex, &deadline))
+        break;
+    }
+    g_mutex_unlock (check_mutex);
+  }
+
+  *buf = gst_buffer_new_and_alloc (size);
+  gst_buffer_set_caps (*buf, caps);
+
+  return GST_FLOW_OK;
+}
+#endif
+
+/* Simulate an app releasing the pad while the first alloc_buffer() is in
+ * progress. */
+GST_START_TEST (test_release_while_buffer_alloc)
+{
+  BufferAllocHarness h;
+
+  buffer_alloc_harness_setup (&h, 1);
+
+  buffer_alloc_harness_teardown (&h);
+}
+
+GST_END_TEST;
+
+/* Simulate an app releasing the pad while the second alloc_buffer() is in
+ * progress. */
+GST_START_TEST (test_release_while_second_buffer_alloc)
+{
+  BufferAllocHarness h;
+
+  buffer_alloc_harness_setup (&h, 2);
+
+  buffer_alloc_harness_teardown (&h);
+}
+
+GST_END_TEST;
+
+/* Check the internal pads of tee */
+GST_START_TEST (test_internal_links)
+{
+  GstElement *tee;
+  GstPad *sinkpad, *srcpad1, *srcpad2;
+  GstIterator *it;
+  GstIteratorResult res;
+  gpointer val1, val2;
+
+  tee = gst_check_setup_element ("tee");
+
+  sinkpad = gst_element_get_static_pad (tee, "sink");
+  fail_unless (sinkpad != NULL);
+  it = gst_pad_iterate_internal_links (sinkpad);
+  fail_unless (it != NULL);
+
+  /* iterator should not return anything */
+  val1 = NULL;
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_DONE);
+  fail_unless (val1 == NULL);
+
+  srcpad1 = gst_element_get_request_pad (tee, "src%d");
+  fail_unless (srcpad1 != NULL);
+
+  /* iterator should resync */
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_RESYNC);
+  fail_unless (val1 == NULL);
+  gst_iterator_resync (it);
+
+  /* we should get something now */
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_OK);
+  fail_unless (GST_PAD_CAST (val1) == srcpad1);
+
+  gst_object_unref (val1);
+
+  val1 = NULL;
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_DONE);
+  fail_unless (val1 == NULL);
+
+  srcpad2 = gst_element_get_request_pad (tee, "src%d");
+  fail_unless (srcpad2 != NULL);
+
+  /* iterator should resync */
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_RESYNC);
+  fail_unless (val1 == NULL);
+  gst_iterator_resync (it);
+
+  /* we should get one of the 2 pads now */
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_OK);
+  fail_unless (GST_PAD_CAST (val1) == srcpad1
+      || GST_PAD_CAST (val1) == srcpad2);
+
+  /* and the other */
+  res = gst_iterator_next (it, &val2);
+  fail_unless (res == GST_ITERATOR_OK);
+  fail_unless (GST_PAD_CAST (val2) == srcpad1
+      || GST_PAD_CAST (val2) == srcpad2);
+  fail_unless (val1 != val2);
+  gst_object_unref (val1);
+  gst_object_unref (val2);
+
+  val1 = NULL;
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_DONE);
+  fail_unless (val1 == NULL);
+
+  gst_iterator_free (it);
+
+  /* get an iterator for the other direction */
+  it = gst_pad_iterate_internal_links (srcpad1);
+  fail_unless (it != NULL);
+
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_OK);
+  fail_unless (GST_PAD_CAST (val1) == sinkpad);
+  gst_object_unref (val1);
+
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_DONE);
+  gst_iterator_free (it);
+
+  it = gst_pad_iterate_internal_links (srcpad2);
+  fail_unless (it != NULL);
+
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_OK);
+  fail_unless (GST_PAD_CAST (val1) == sinkpad);
+  gst_object_unref (val1);
+
+  res = gst_iterator_next (it, &val1);
+  fail_unless (res == GST_ITERATOR_DONE);
+
+  gst_iterator_free (it);
+  gst_object_unref (srcpad1);
+  gst_object_unref (srcpad2);
+  gst_object_unref (sinkpad);
+  gst_object_unref (tee);
+}
+
+GST_END_TEST;
+
+static GstFlowReturn
+_fake_chain (GstPad * pad, GstBuffer * buffer)
+{
+  gst_buffer_unref (buffer);
+  return GST_FLOW_OK;
+}
+
+static GstFlowReturn
+_fake_chain_error (GstPad * pad, GstBuffer * buffer)
+{
+  gst_buffer_unref (buffer);
+  return GST_FLOW_ERROR;
+}
+
+GST_START_TEST (test_flow_aggregation)
+{
+  GstPad *mysrc, *mysink1, *mysink2;
+  GstPad *teesink, *teesrc1, *teesrc2;
+  GstElement *tee;
+  GstBuffer *buffer;
+  GstCaps *caps;
+
+  caps = gst_caps_new_simple ("test/test", NULL);
+
+  tee = gst_element_factory_make ("tee", NULL);
+  fail_unless (tee != NULL);
+  teesink = gst_element_get_static_pad (tee, "sink");
+  fail_unless (teesink != NULL);
+  teesrc1 = gst_element_get_request_pad (tee, "src%d");
+  fail_unless (teesrc1 != NULL);
+  teesrc2 = gst_element_get_request_pad (tee, "src%d");
+  fail_unless (teesrc2 != NULL);
+
+  mysink1 = gst_pad_new ("mysink1", GST_PAD_SINK);
+  gst_pad_set_caps (mysink1, caps);
+  mysink2 = gst_pad_new ("mysink2", GST_PAD_SINK);
+  gst_pad_set_caps (mysink2, caps);
+  mysrc = gst_pad_new ("mysrc", GST_PAD_SRC);
+  gst_pad_set_caps (mysrc, caps);
+
+  gst_pad_set_chain_function (mysink1, _fake_chain);
+  gst_pad_set_active (mysink1, TRUE);
+  gst_pad_set_chain_function (mysink2, _fake_chain);
+  gst_pad_set_active (mysink2, TRUE);
+
+  fail_unless (gst_pad_link (mysrc, teesink) == GST_PAD_LINK_OK);
+  fail_unless (gst_pad_link (teesrc1, mysink1) == GST_PAD_LINK_OK);
+  fail_unless (gst_pad_link (teesrc2, mysink2) == GST_PAD_LINK_OK);
+
+  fail_unless (gst_element_set_state (tee,
+          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS);
+
+  buffer = gst_buffer_new ();
+  //gst_buffer_set_caps (buffer, caps);
+
+  /* First check if everything works in normal state */
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_OK);
+
+  /* One pad being in wrong state must result in wrong state */
+  gst_pad_set_active (mysink2, FALSE);
+  fail_unless (gst_pad_push (mysrc,
+          gst_buffer_ref (buffer)) == GST_FLOW_WRONG_STATE);
+
+  gst_pad_set_active (mysink1, FALSE);
+  gst_pad_set_active (mysink2, TRUE);
+  fail_unless (gst_pad_push (mysrc,
+          gst_buffer_ref (buffer)) == GST_FLOW_WRONG_STATE);
+
+  gst_pad_set_active (mysink2, FALSE);
+  fail_unless (gst_pad_push (mysrc,
+          gst_buffer_ref (buffer)) == GST_FLOW_WRONG_STATE);
+
+  /* Test if everything still works in normal state */
+  gst_pad_set_active (mysink1, TRUE);
+  gst_pad_set_active (mysink2, TRUE);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_OK);
+
+  /* One unlinked pad must return OK, two unlinked pads must return NOT_LINKED */
+  fail_unless (gst_pad_unlink (teesrc1, mysink1) == TRUE);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_OK);
+
+  fail_unless (gst_pad_link (teesrc1, mysink1) == GST_PAD_LINK_OK);
+  fail_unless (gst_pad_unlink (teesrc2, mysink2) == TRUE);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_OK);
+
+  fail_unless (gst_pad_unlink (teesrc1, mysink1) == TRUE);
+  fail_unless (gst_pad_push (mysrc,
+          gst_buffer_ref (buffer)) == GST_FLOW_NOT_LINKED);
+
+  /* Test if everything still works in normal state */
+  fail_unless (gst_pad_link (teesrc1, mysink1) == GST_PAD_LINK_OK);
+  fail_unless (gst_pad_link (teesrc2, mysink2) == GST_PAD_LINK_OK);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_OK);
+
+  /* One pad returning ERROR should result in ERROR */
+  gst_pad_set_chain_function (mysink1, _fake_chain_error);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_ERROR);
+
+  gst_pad_set_chain_function (mysink1, _fake_chain);
+  gst_pad_set_chain_function (mysink2, _fake_chain_error);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_ERROR);
+
+  gst_pad_set_chain_function (mysink1, _fake_chain_error);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_ERROR);
+
+  /* And now everything still needs to work */
+  gst_pad_set_chain_function (mysink1, _fake_chain);
+  gst_pad_set_chain_function (mysink2, _fake_chain);
+  fail_unless (gst_pad_push (mysrc, gst_buffer_ref (buffer)) == GST_FLOW_OK);
+
+  fail_unless (gst_element_set_state (tee,
+          GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS);
+
+  fail_unless (gst_pad_unlink (mysrc, teesink) == TRUE);
+  fail_unless (gst_pad_unlink (teesrc1, mysink1) == TRUE);
+  fail_unless (gst_pad_unlink (teesrc2, mysink2) == TRUE);
+
+
+  gst_object_unref (teesink);
+  gst_object_unref (teesrc1);
+  gst_object_unref (teesrc2);
+  gst_element_release_request_pad (tee, teesrc1);
+  gst_element_release_request_pad (tee, teesrc2);
+  gst_object_unref (tee);
+
+  gst_object_unref (mysink1);
+  gst_object_unref (mysink2);
+  gst_object_unref (mysrc);
+  gst_caps_unref (caps);
+  gst_buffer_unref (buffer);
+}
+
+GST_END_TEST;
+
 static Suite *
 tee_suite (void)
 {
   Suite *s = suite_create ("tee");
   TCase *tc_chain = tcase_create ("general");
 
+  /* Set the timeout to a much larger time - 3 minutes */
+  tcase_set_timeout (tc_chain, 180);
+
   suite_add_tcase (s, tc_chain);
   tcase_add_test (tc_chain, test_num_buffers);
   tcase_add_test (tc_chain, test_stress);
+  tcase_add_test (tc_chain, test_release_while_buffer_alloc);
+  tcase_add_test (tc_chain, test_release_while_second_buffer_alloc);
+  tcase_add_test (tc_chain, test_internal_links);
+  tcase_add_test (tc_chain, test_flow_aggregation);
 
   return s;
 }