--- /dev/null
+/* GStreamer
+ *
+ * some unit tests for GstBaseSrc
+ *
+ * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>
+ *
+ * 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., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+#include <gst/gst.h>
+#include <gst/check/gstcheck.h>
+#include <gst/base/gstbasesrc.h>
+
+static void
+eos_event_counter (GstObject * pad, GstEvent * event, guint * p_num_eos)
+{
+ fail_unless (event != NULL);
+ fail_unless (GST_IS_EVENT (event));
+
+ if (GST_EVENT_TYPE (event) == GST_EVENT_EOS)
+ *p_num_eos += 1;
+}
+
+/* basesrc_eos_events_push_live_op:
+ * - make sure source does send an EOS event when operating in push
+ * mode and being set to READY explicitly (like one might with
+ * live sources)
+ */
+GST_START_TEST (basesrc_eos_events_push_live_op)
+{
+ GstStateChangeReturn state_ret;
+ GstElement *src, *sink, *pipe;
+ GstMessage *msg;
+ GstBus *bus;
+ GstPad *srcpad;
+ guint probe, num_eos = 0;
+
+ pipe = gst_pipeline_new ("pipeline");
+ sink = gst_element_factory_make ("fakesink", "sink");
+ src = gst_element_factory_make ("fakesrc", "src");
+
+ g_assert (pipe != NULL);
+ g_assert (sink != NULL);
+ g_assert (src != NULL);
+
+ fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
+ fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
+
+ fail_unless (gst_element_link (src, sink) == TRUE);
+
+ g_object_set (sink, "can-activate-push", TRUE, NULL);
+ g_object_set (sink, "can-activate-pull", FALSE, NULL);
+
+ g_object_set (src, "can-activate-push", TRUE, NULL);
+ g_object_set (src, "can-activate-pull", FALSE, NULL);
+
+ /* set up event probe to count EOS events */
+ srcpad = gst_element_get_pad (src, "src");
+ fail_unless (srcpad != NULL);
+
+ probe = gst_pad_add_event_probe (srcpad,
+ G_CALLBACK (eos_event_counter), &num_eos);
+
+ bus = gst_element_get_bus (pipe);
+
+ gst_element_set_state (pipe, GST_STATE_PLAYING);
+ state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
+ fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
+
+ /* wait a second, then do controlled shutdown */
+ g_usleep (GST_USECOND * 1);
+
+ /* shut down source only (should send EOS event) ... */
+ gst_element_set_state (src, GST_STATE_NULL);
+ state_ret = gst_element_get_state (src, NULL, NULL, -1);
+ fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
+
+ fail_unless (gst_element_set_locked_state (src, TRUE) == TRUE);
+
+ /* ... and wait for the EOS message from the sink */
+ msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
+ fail_unless (msg != NULL);
+ fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
+ fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
+
+ /* should be exactly one EOS event */
+ fail_unless (num_eos == 1);
+
+ gst_element_set_state (pipe, GST_STATE_NULL);
+ gst_element_get_state (pipe, NULL, NULL, -1);
+
+ /* make sure source hasn't sent a second one when going PAUSED => READY */
+ fail_unless (num_eos == 1);
+
+ gst_pad_remove_event_probe (srcpad, probe);
+ gst_object_unref (srcpad);
+ gst_message_unref (msg);
+ gst_object_unref (bus);
+ gst_object_unref (pipe);
+}
+
+GST_END_TEST;
+
+/* basesrc_eos_events_push:
+ * - make sure source only sends one EOS when operating in push-mode,
+ * reaching the max number of buffers, and is then shut down.
+ */
+GST_START_TEST (basesrc_eos_events_push)
+{
+ GstStateChangeReturn state_ret;
+ GstElement *src, *sink, *pipe;
+ GstMessage *msg;
+ GstBus *bus;
+ GstPad *srcpad;
+ guint probe, num_eos = 0;
+
+ pipe = gst_pipeline_new ("pipeline");
+ sink = gst_element_factory_make ("fakesink", "sink");
+ src = gst_element_factory_make ("fakesrc", "src");
+
+ g_assert (pipe != NULL);
+ g_assert (sink != NULL);
+ g_assert (src != NULL);
+
+ fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
+ fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
+
+ fail_unless (gst_element_link (src, sink) == TRUE);
+
+ g_object_set (sink, "can-activate-push", TRUE, NULL);
+ g_object_set (sink, "can-activate-pull", FALSE, NULL);
+
+ g_object_set (src, "can-activate-push", TRUE, NULL);
+ g_object_set (src, "can-activate-pull", FALSE, NULL);
+ g_object_set (src, "num-buffers", 8, NULL);
+
+ /* set up event probe to count EOS events */
+ srcpad = gst_element_get_pad (src, "src");
+ fail_unless (srcpad != NULL);
+
+ probe = gst_pad_add_event_probe (srcpad,
+ G_CALLBACK (eos_event_counter), &num_eos);
+
+ bus = gst_element_get_bus (pipe);
+
+ gst_element_set_state (pipe, GST_STATE_PLAYING);
+ state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
+ fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
+
+ msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
+ fail_unless (msg != NULL);
+ fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
+ fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
+
+ /* should be exactly one EOS event */
+ fail_unless (num_eos == 1);
+
+ gst_element_set_state (pipe, GST_STATE_NULL);
+ gst_element_get_state (pipe, NULL, NULL, -1);
+
+ /* make sure source hasn't sent a second one when going PAUSED => READY */
+ fail_unless (num_eos == 1);
+
+ gst_pad_remove_event_probe (srcpad, probe);
+ gst_object_unref (srcpad);
+ gst_message_unref (msg);
+ gst_object_unref (bus);
+ gst_object_unref (pipe);
+}
+
+GST_END_TEST;
+
+/* basesrc_eos_events_pull_live_op:
+ * - make sure source doesn't send an EOS event when operating in
+ * pull mode and being set to READY explicitly (like one might with
+ * live sources)
+ */
+GST_START_TEST (basesrc_eos_events_pull_live_op)
+{
+ GstStateChangeReturn state_ret;
+ GstElement *src, *sink, *pipe;
+ GstPad *srcpad;
+ guint probe, num_eos = 0;
+
+ pipe = gst_pipeline_new ("pipeline");
+ sink = gst_element_factory_make ("fakesink", "sink");
+ src = gst_element_factory_make ("fakesrc", "src");
+
+ g_assert (pipe != NULL);
+ g_assert (sink != NULL);
+ g_assert (src != NULL);
+
+ fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
+ fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
+
+ fail_unless (gst_element_link (src, sink) == TRUE);
+
+ g_object_set (sink, "can-activate-push", FALSE, NULL);
+ g_object_set (sink, "can-activate-pull", TRUE, NULL);
+
+ g_object_set (src, "can-activate-push", FALSE, NULL);
+ g_object_set (src, "can-activate-pull", TRUE, NULL);
+
+ /* set up event probe to count EOS events */
+ srcpad = gst_element_get_pad (src, "src");
+ fail_unless (srcpad != NULL);
+
+ probe = gst_pad_add_event_probe (srcpad,
+ G_CALLBACK (eos_event_counter), &num_eos);
+
+ gst_element_set_state (pipe, GST_STATE_PLAYING);
+ state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
+ fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
+
+ /* wait a second, then do controlled shutdown */
+ g_usleep (GST_USECOND * 1);
+
+ /* shut down source only ... */
+ gst_element_set_state (src, GST_STATE_NULL);
+ state_ret = gst_element_get_state (src, NULL, NULL, -1);
+ fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
+
+ fail_unless (gst_element_set_locked_state (src, TRUE) == TRUE);
+
+ /* source shouldn't have sent any EOS event in pull mode */
+ fail_unless (num_eos == 0);
+
+ gst_element_set_state (pipe, GST_STATE_NULL);
+ gst_element_get_state (pipe, NULL, NULL, -1);
+
+ /* make sure source hasn't sent an EOS when going PAUSED => READY either */
+ fail_unless (num_eos == 0);
+
+ gst_pad_remove_event_probe (srcpad, probe);
+ gst_object_unref (srcpad);
+ gst_object_unref (pipe);
+}
+
+GST_END_TEST;
+
+/* basesrc_eos_events_pull:
+ * - makes sure source doesn't send EOS event when reaching the max.
+ * number of buffers configured in pull-mode
+ * - make sure source doesn't send EOS event either when being shut down
+ * (PAUSED => READY state change) after EOSing in pull mode
+ */
+GST_START_TEST (basesrc_eos_events_pull)
+{
+ GstStateChangeReturn state_ret;
+ GstElement *src, *sink, *pipe;
+ GstMessage *msg;
+ GstBus *bus;
+ GstPad *srcpad;
+ guint probe, num_eos = 0;
+
+ pipe = gst_pipeline_new ("pipeline");
+ sink = gst_element_factory_make ("fakesink", "sink");
+ src = gst_element_factory_make ("fakesrc", "src");
+
+ g_assert (pipe != NULL);
+ g_assert (sink != NULL);
+ g_assert (src != NULL);
+
+ fail_unless (gst_bin_add (GST_BIN (pipe), src) == TRUE);
+ fail_unless (gst_bin_add (GST_BIN (pipe), sink) == TRUE);
+
+ fail_unless (gst_element_link (src, sink) == TRUE);
+
+ g_object_set (sink, "can-activate-push", FALSE, NULL);
+ g_object_set (sink, "can-activate-pull", TRUE, NULL);
+
+ g_object_set (src, "can-activate-push", FALSE, NULL);
+ g_object_set (src, "can-activate-pull", TRUE, NULL);
+ g_object_set (src, "num-buffers", 8, NULL);
+
+ /* set up event probe to count EOS events */
+ srcpad = gst_element_get_pad (src, "src");
+ fail_unless (srcpad != NULL);
+
+ probe = gst_pad_add_event_probe (srcpad,
+ G_CALLBACK (eos_event_counter), &num_eos);
+
+ bus = gst_element_get_bus (pipe);
+
+ gst_element_set_state (pipe, GST_STATE_PLAYING);
+ state_ret = gst_element_get_state (pipe, NULL, NULL, -1);
+ fail_unless (state_ret == GST_STATE_CHANGE_SUCCESS);
+
+ msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
+ fail_unless (msg != NULL);
+ fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
+ fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
+
+ /* source shouldn't have sent any EOS event in pull mode */
+ fail_unless (num_eos == 0);
+
+ gst_element_set_state (pipe, GST_STATE_NULL);
+ gst_element_get_state (pipe, NULL, NULL, -1);
+
+ /* make sure source hasn't sent an EOS when going PAUSED => READY either */
+ fail_unless (num_eos == 0);
+
+ gst_pad_remove_event_probe (srcpad, probe);
+ gst_object_unref (srcpad);
+ gst_message_unref (msg);
+ gst_object_unref (bus);
+ gst_object_unref (pipe);
+}
+
+GST_END_TEST;
+
+
+Suite *
+gst_basesrc_suite (void)
+{
+ Suite *s = suite_create ("GstBaseSrc");
+ TCase *tc = tcase_create ("general");
+
+ suite_add_tcase (s, tc);
+ tcase_add_test (tc, basesrc_eos_events_pull);
+ tcase_add_test (tc, basesrc_eos_events_push);
+ tcase_add_test (tc, basesrc_eos_events_push_live_op);
+ tcase_add_test (tc, basesrc_eos_events_pull_live_op);
+
+ return s;
+}
+
+int
+main (int argc, char **argv)
+{
+ int nf;
+
+ Suite *s = gst_basesrc_suite ();
+ SRunner *sr = srunner_create (s);
+
+ gst_check_init (&argc, &argv);
+
+ srunner_run_all (sr, CK_NORMAL);
+ nf = srunner_ntests_failed (sr);
+ srunner_free (sr);
+
+ return nf;
+}