2 * aggregator.c - GstAggregator testsuite
3 * Copyright (C) 2006 Alessandro Decina <alessandro.d@gmail.com>
4 * Copyright (C) 2014 Mathieu Duponchelle <mathieu.duponchelle@oencreed.com>
5 * Copyright (C) 2014 Thibault Saunier <tsaunier@opencreed.com>
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Library General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Library General Public License for more details.
17 * You should have received a copy of the GNU Library General Public
18 * License along with this library; if not, write to the
19 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
20 * Boston, MA 02110-1301, USA.
27 #include <gst/check/gstcheck.h>
28 #include <gst/base/gstaggregator.h>
30 /* dummy aggregator based element */
32 #define GST_TYPE_TEST_AGGREGATOR (gst_test_aggregator_get_type ())
33 #define GST_TEST_AGGREGATOR(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_TEST_AGGREGATOR, GstTestAggregator))
34 #define GST_TEST_AGGREGATOR_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_TEST_AGGREGATOR, GstTestAggregatorClass))
35 #define GST_TEST_AGGREGATOR_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_TEST_AGGREGATOR, GstTestAggregatorClass))
37 #define fail_error_message(msg) \
40 gst_message_parse_error(msg, &error, NULL); \
41 fail_unless(FALSE, "Error Message from %s : %s", \
42 GST_OBJECT_NAME (GST_MESSAGE_SRC(msg)), error->message); \
43 g_error_free (error); \
46 typedef struct _GstTestAggregator GstTestAggregator;
47 typedef struct _GstTestAggregatorClass GstTestAggregatorClass;
49 static GType gst_test_aggregator_get_type (void);
51 #define BUFFER_DURATION 100000000 /* 10 frames per second */
52 #define TEST_GAP_PTS 0
53 #define TEST_GAP_DURATION (5 * GST_SECOND)
55 struct _GstTestAggregator
60 gboolean gap_expected;
63 struct _GstTestAggregatorClass
65 GstAggregatorClass parent_class;
69 gst_test_aggregator_aggregate (GstAggregator * aggregator, gboolean timeout)
72 gboolean all_eos = TRUE;
73 GstTestAggregator *testagg;
76 gboolean done_iterating = FALSE;
78 testagg = GST_TEST_AGGREGATOR (aggregator);
80 iter = gst_element_iterate_sink_pads (GST_ELEMENT (testagg));
81 while (!done_iterating) {
82 GValue value = { 0, };
83 GstAggregatorPad *pad;
85 switch (gst_iterator_next (iter, &value)) {
87 pad = g_value_get_object (&value);
89 if (gst_aggregator_pad_is_eos (pad) == FALSE)
92 if (testagg->gap_expected == TRUE) {
93 buf = gst_aggregator_pad_get_buffer (pad);
95 fail_unless (GST_BUFFER_PTS (buf) == TEST_GAP_PTS);
96 fail_unless (GST_BUFFER_DURATION (buf) == TEST_GAP_DURATION);
97 fail_unless (GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_GAP));
98 fail_unless (GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DROPPABLE));
99 gst_buffer_unref (buf);
100 testagg->gap_expected = FALSE;
103 gst_aggregator_pad_drop_buffer (pad);
105 g_value_reset (&value);
107 case GST_ITERATOR_RESYNC:
108 gst_iterator_resync (iter);
110 case GST_ITERATOR_ERROR:
111 GST_WARNING_OBJECT (testagg, "Sinkpads iteration error");
112 done_iterating = TRUE;
114 case GST_ITERATOR_DONE:
115 done_iterating = TRUE;
119 gst_iterator_free (iter);
121 if (all_eos == TRUE) {
122 GST_INFO_OBJECT (testagg, "no data available, must be EOS");
123 gst_pad_push_event (aggregator->srcpad, gst_event_new_eos ());
127 buf = gst_buffer_new ();
128 GST_BUFFER_TIMESTAMP (buf) = testagg->timestamp;
129 GST_BUFFER_DURATION (buf) = BUFFER_DURATION;
130 testagg->timestamp += BUFFER_DURATION;
132 gst_aggregator_finish_buffer (aggregator, buf);
134 /* We just check finish_frame return FLOW_OK */
138 #define gst_test_aggregator_parent_class parent_class
139 G_DEFINE_TYPE (GstTestAggregator, gst_test_aggregator, GST_TYPE_AGGREGATOR);
142 gst_test_aggregator_class_init (GstTestAggregatorClass * klass)
144 GstElementClass *gstelement_class = (GstElementClass *) klass;
145 GstAggregatorClass *base_aggregator_class = (GstAggregatorClass *) klass;
147 static GstStaticPadTemplate _src_template =
148 GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
149 GST_STATIC_CAPS_ANY);
151 static GstStaticPadTemplate _sink_template =
152 GST_STATIC_PAD_TEMPLATE ("sink_%u", GST_PAD_SINK, GST_PAD_REQUEST,
153 GST_STATIC_CAPS_ANY);
155 gst_element_class_add_static_pad_template (gstelement_class, &_src_template);
157 gst_element_class_add_static_pad_template (gstelement_class, &_sink_template);
159 gst_element_class_set_static_metadata (gstelement_class, "Aggregator",
160 "Testing", "Combine N buffers", "Stefan Sauer <ensonic@users.sf.net>");
162 base_aggregator_class->aggregate =
163 GST_DEBUG_FUNCPTR (gst_test_aggregator_aggregate);
167 gst_test_aggregator_init (GstTestAggregator * self)
169 GstAggregator *agg = GST_AGGREGATOR (self);
170 gst_segment_init (&agg->segment, GST_FORMAT_TIME);
172 self->gap_expected = FALSE;
176 gst_test_aggregator_plugin_init (GstPlugin * plugin)
178 return gst_element_register (plugin, "testaggregator", GST_RANK_NONE,
179 GST_TYPE_TEST_AGGREGATOR);
183 gst_test_aggregator_plugin_register (void)
185 return gst_plugin_register_static (GST_VERSION_MAJOR,
189 gst_test_aggregator_plugin_init,
190 VERSION, GST_LICENSE, PACKAGE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);
197 GstElement *aggregator;
198 GstPad *sinkpad, *srcpad;
199 GstFlowReturn expected_result;
201 /* ------------------
202 * ----------- --------|-- |
203 * | srcpad | -- | sinkpad | aggregator |
204 * ----------- --------|-- |
206 * This is for 1 Chain, we can have several
213 GstPad *srcpad, /* srcpad of the GstAggregator */
214 *sinkpad; /* fake sinkpad to which GstAggregator.srcpad is linked */
216 GstElement *aggregator;
218 /* -----------------|
219 * | ---------- -----------
220 * | aggregator | srcpad | -- | sinkpad |
221 * | ---------- -----------
225 gint flush_start_events, flush_stop_events;
228 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
231 GST_STATIC_CAPS_ANY);
233 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
236 GST_STATIC_CAPS_ANY);
239 start_flow (ChainData * chain_data)
244 gst_pad_push_event (chain_data->srcpad, gst_event_new_stream_start ("test"));
246 caps = gst_caps_new_empty_simple ("foo/x-bar");
247 gst_pad_push_event (chain_data->srcpad, gst_event_new_caps (caps));
248 gst_caps_unref (caps);
250 gst_segment_init (&segment, GST_FORMAT_TIME);
251 gst_pad_push_event (chain_data->srcpad, gst_event_new_segment (&segment));
255 push_buffer (gpointer user_data)
258 ChainData *chain_data = (ChainData *) user_data;
260 start_flow (chain_data);
262 GST_DEBUG ("Pushing buffer on pad: %s:%s",
263 GST_DEBUG_PAD_NAME (chain_data->sinkpad));
264 flow = gst_pad_push (chain_data->srcpad, chain_data->buffer);
265 fail_unless (flow == chain_data->expected_result,
266 "got flow %s instead of %s on %s:%s", gst_flow_get_name (flow),
267 gst_flow_get_name (chain_data->expected_result),
268 GST_DEBUG_PAD_NAME (chain_data->sinkpad));
269 chain_data->buffer = NULL;
275 push_event (gpointer user_data)
277 ChainData *chain_data = (ChainData *) user_data;
278 GstTestAggregator *aggregator = (GstTestAggregator *) chain_data->aggregator;
279 GstEventType event_type;
281 start_flow (chain_data);
283 GST_INFO_OBJECT (chain_data->srcpad, "Pushing event: %"
284 GST_PTR_FORMAT, chain_data->event);
286 event_type = GST_EVENT_TYPE (chain_data->event);
287 switch (event_type) {
289 aggregator->gap_expected = TRUE;
295 fail_unless (gst_pad_push_event (chain_data->srcpad,
296 chain_data->event) == TRUE);
303 _aggregate_timeout (GMainLoop * ml)
305 g_main_loop_quit (ml);
307 fail_unless ("No buffer found on aggregator.srcpad -> TIMEOUT" == NULL);
313 _quit (GMainLoop * ml)
315 GST_DEBUG ("QUITING ML");
316 g_main_loop_quit (ml);
318 return G_SOURCE_REMOVE;
321 static GstPadProbeReturn
322 _aggregated_cb (GstPad * pad, GstPadProbeInfo * info, GMainLoop * ml)
324 GST_DEBUG ("Should quit ML");
325 g_idle_add ((GSourceFunc) _quit, ml);
327 return GST_PAD_PROBE_REMOVE;
330 static GstPadProbeReturn
331 downstream_probe_cb (GstPad * pad, GstPadProbeInfo * info, TestData * test)
333 GST_DEBUG ("PROBING ");
334 if (info->type & GST_PAD_PROBE_TYPE_EVENT_FLUSH) {
335 if (GST_EVENT_TYPE (GST_PAD_PROBE_INFO_EVENT (info)) ==
336 GST_EVENT_FLUSH_START) {
338 g_atomic_int_inc (&test->flush_start_events);
339 GST_DEBUG ("==========> FLUSH: %i", test->flush_start_events);
340 } else if (GST_EVENT_TYPE (GST_PAD_PROBE_INFO_EVENT (info)) ==
341 GST_EVENT_FLUSH_STOP)
342 g_atomic_int_inc (&test->flush_stop_events);
344 return GST_PAD_PROBE_OK;
349 * Not thread safe, will create a new ChainData which contains
350 * an activated src pad linked to a requested sink pad of @agg, and
351 * a newly allocated buffer ready to be pushed. Caller needs to
352 * clear with _chain_data_clear after.
355 _chain_data_init (ChainData * data, GstElement * agg)
357 static gint num_src_pads = 0;
358 gchar *pad_name = g_strdup_printf ("src%d", num_src_pads);
362 data->srcpad = gst_pad_new_from_static_template (&srctemplate, pad_name);
364 gst_pad_set_active (data->srcpad, TRUE);
365 data->aggregator = agg;
366 data->buffer = gst_buffer_new ();
367 data->sinkpad = gst_element_get_request_pad (agg, "sink_%u");
368 fail_unless (GST_IS_PAD (data->sinkpad));
369 fail_unless (gst_pad_link (data->srcpad, data->sinkpad) == GST_PAD_LINK_OK);
373 _chain_data_clear (ChainData * data)
376 gst_buffer_unref (data->buffer);
378 gst_object_unref (data->srcpad);
380 gst_object_unref (data->sinkpad);
384 _test_chain (GstPad * pad, GstObject * object, GstBuffer * buffer)
386 /* accept any buffers */
387 gst_buffer_unref (buffer);
392 _test_data_init (TestData * test, gboolean needs_flushing)
394 test->aggregator = gst_element_factory_make ("testaggregator", NULL);
395 gst_element_set_state (test->aggregator, GST_STATE_PLAYING);
396 test->ml = g_main_loop_new (NULL, TRUE);
397 test->srcpad = GST_AGGREGATOR (test->aggregator)->srcpad;
399 GST_DEBUG ("Srcpad: %p", test->srcpad);
401 if (needs_flushing) {
402 static gint num_sink_pads = 0;
403 gchar *pad_name = g_strdup_printf ("sink%d", num_sink_pads);
406 test->sinkpad = gst_pad_new_from_static_template (&sinktemplate, pad_name);
407 gst_pad_set_chain_function (test->sinkpad, _test_chain);
408 gst_pad_set_active (test->sinkpad, TRUE);
410 fail_unless (gst_pad_link (test->srcpad, test->sinkpad) == GST_PAD_LINK_OK);
411 gst_pad_add_probe (test->srcpad, GST_PAD_PROBE_TYPE_EVENT_FLUSH,
412 (GstPadProbeCallback) downstream_probe_cb, test, NULL);
414 gst_pad_add_probe (test->srcpad, GST_PAD_PROBE_TYPE_BUFFER,
415 (GstPadProbeCallback) _aggregated_cb, test->ml, NULL);
420 g_timeout_add (1000, (GSourceFunc) _aggregate_timeout, test->ml);
424 _test_data_clear (TestData * test)
426 gst_element_set_state (test->aggregator, GST_STATE_NULL);
427 gst_object_unref (test->aggregator);
430 gst_object_unref (test->sinkpad);
432 g_main_loop_unref (test->ml);
435 GST_START_TEST (test_aggregate)
437 GThread *thread1, *thread2;
439 ChainData data1 = { 0, };
440 ChainData data2 = { 0, };
441 TestData test = { 0, };
443 _test_data_init (&test, FALSE);
444 _chain_data_init (&data1, test.aggregator);
445 _chain_data_init (&data2, test.aggregator);
447 thread1 = g_thread_try_new ("gst-check", push_buffer, &data1, NULL);
448 thread2 = g_thread_try_new ("gst-check", push_buffer, &data2, NULL);
450 g_main_loop_run (test.ml);
451 g_source_remove (test.timeout_id);
454 /* these will return immediately as when the data is popped the threads are
455 * unlocked and will terminate */
456 g_thread_join (thread1);
457 g_thread_join (thread2);
459 _chain_data_clear (&data1);
460 _chain_data_clear (&data2);
461 _test_data_clear (&test);
466 GST_START_TEST (test_aggregate_eos)
468 GThread *thread1, *thread2;
470 ChainData data1 = { 0, };
471 ChainData data2 = { 0, };
472 TestData test = { 0, };
474 _test_data_init (&test, FALSE);
475 _chain_data_init (&data1, test.aggregator);
476 _chain_data_init (&data2, test.aggregator);
478 data2.event = gst_event_new_eos ();
480 thread1 = g_thread_try_new ("gst-check", push_buffer, &data1, NULL);
481 thread2 = g_thread_try_new ("gst-check", push_event, &data2, NULL);
483 g_main_loop_run (test.ml);
484 g_source_remove (test.timeout_id);
486 /* these will return immediately as when the data is popped the threads are
487 * unlocked and will terminate */
488 g_thread_join (thread1);
489 g_thread_join (thread2);
491 _chain_data_clear (&data1);
492 _chain_data_clear (&data2);
493 _test_data_clear (&test);
498 GST_START_TEST (test_aggregate_gap)
502 ChainData data = { 0, };
503 TestData test = { 0, };
505 _test_data_init (&test, FALSE);
506 _chain_data_init (&data, test.aggregator);
508 data.event = gst_event_new_gap (TEST_GAP_PTS, TEST_GAP_DURATION);
510 thread = g_thread_try_new ("gst-check", push_event, &data, NULL);
512 g_main_loop_run (test.ml);
513 g_source_remove (test.timeout_id);
515 /* these will return immediately as when the data is popped the threads are
516 * unlocked and will terminate */
517 g_thread_join (thread);
519 _chain_data_clear (&data);
520 _test_data_clear (&test);
525 #define NUM_BUFFERS 3
527 handoff (GstElement * fakesink, GstBuffer * buf, GstPad * pad, guint * count)
530 GST_DEBUG ("HANDOFF: %i", *count);
533 /* Test a linear pipeline using aggregator */
534 GST_START_TEST (test_linear_pipeline)
538 GstElement *pipeline, *src, *agg, *sink;
542 pipeline = gst_pipeline_new ("pipeline");
543 src = gst_check_setup_element ("fakesrc");
544 g_object_set (src, "num-buffers", NUM_BUFFERS, "sizetype", 2, "sizemax", 4,
546 agg = gst_check_setup_element ("testaggregator");
547 sink = gst_check_setup_element ("fakesink");
548 g_object_set (sink, "signal-handoffs", TRUE, NULL);
549 g_signal_connect (sink, "handoff", (GCallback) handoff, &count);
551 fail_unless (gst_bin_add (GST_BIN (pipeline), src));
552 fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
553 fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
554 fail_unless (gst_element_link (src, agg));
555 fail_unless (gst_element_link (agg, sink));
557 bus = gst_element_get_bus (pipeline);
558 fail_if (bus == NULL);
559 gst_element_set_state (pipeline, GST_STATE_PLAYING);
561 msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
562 fail_if (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_EOS);
563 gst_message_unref (msg);
565 fail_unless_equals_int (count, NUM_BUFFERS);
567 gst_element_set_state (pipeline, GST_STATE_NULL);
568 gst_object_unref (bus);
569 gst_object_unref (pipeline);
574 GST_START_TEST (test_two_src_pipeline)
578 GstElement *pipeline, *src, *src1, *agg, *sink;
582 pipeline = gst_pipeline_new ("pipeline");
583 src = gst_element_factory_make ("fakesrc", NULL);
584 g_object_set (src, "num-buffers", NUM_BUFFERS, "sizetype", 2, "sizemax", 4,
587 src1 = gst_element_factory_make ("fakesrc", NULL);
588 g_object_set (src1, "num-buffers", NUM_BUFFERS + 1, "sizetype", 2, "sizemax",
591 agg = gst_check_setup_element ("testaggregator");
592 sink = gst_check_setup_element ("fakesink");
593 g_object_set (sink, "signal-handoffs", TRUE, NULL);
594 g_signal_connect (sink, "handoff", (GCallback) handoff, &count);
596 fail_unless (gst_bin_add (GST_BIN (pipeline), src));
597 fail_unless (gst_bin_add (GST_BIN (pipeline), src1));
598 fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
599 fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
600 fail_unless (gst_element_link (src, agg));
601 fail_unless (gst_element_link (src1, agg));
602 fail_unless (gst_element_link (agg, sink));
604 bus = gst_element_get_bus (pipeline);
605 fail_if (bus == NULL);
606 gst_element_set_state (pipeline, GST_STATE_PLAYING);
608 msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
609 fail_if (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_EOS);
610 gst_message_unref (msg);
612 fail_unless_equals_int (count, NUM_BUFFERS + 1);
614 gst_element_set_state (pipeline, GST_STATE_NULL);
615 gst_object_unref (bus);
616 gst_object_unref (pipeline);
621 static GstPadProbeReturn
622 _drop_buffer_probe_cb (GstPad * pad, GstPadProbeInfo * info, gpointer user_data)
626 if (GST_IS_BUFFER (info->data)) {
627 wait = GPOINTER_TO_INT (user_data);
629 g_usleep (wait / 1000);
630 return GST_PAD_PROBE_DROP;
633 return GST_PAD_PROBE_PASS;
636 #define TIMEOUT_NUM_BUFFERS 20
638 _test_timeout (gint buffer_wait)
642 GstElement *pipeline, *src, *src1, *agg, *sink;
647 pipeline = gst_pipeline_new ("pipeline");
648 src = gst_element_factory_make ("fakesrc", NULL);
649 g_object_set (src, "num-buffers", TIMEOUT_NUM_BUFFERS, "sizetype", 2,
650 "sizemax", 4, "is-live", TRUE, "datarate", 4000, NULL);
652 src1 = gst_element_factory_make ("fakesrc", NULL);
653 g_object_set (src1, "num-buffers", TIMEOUT_NUM_BUFFERS, "sizetype", 2,
654 "sizemax", 4, "is-live", TRUE, "datarate", 4000, NULL);
656 agg = gst_check_setup_element ("testaggregator");
657 g_object_set (agg, "latency", GST_USECOND, NULL);
658 sink = gst_check_setup_element ("fakesink");
659 g_object_set (sink, "signal-handoffs", TRUE, NULL);
660 g_signal_connect (sink, "handoff", (GCallback) handoff, &count);
662 fail_unless (gst_bin_add (GST_BIN (pipeline), src));
663 fail_unless (gst_bin_add (GST_BIN (pipeline), src1));
664 fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
665 fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
667 src1pad = gst_element_get_static_pad (src1, "src");
668 fail_if (src1pad == NULL);
669 gst_pad_add_probe (src1pad, GST_PAD_PROBE_TYPE_BLOCK_DOWNSTREAM,
670 (GstPadProbeCallback) _drop_buffer_probe_cb,
671 GINT_TO_POINTER (buffer_wait), NULL);
673 fail_unless (gst_element_link (src, agg));
674 fail_unless (gst_element_link (src1, agg));
675 fail_unless (gst_element_link (agg, sink));
677 bus = gst_element_get_bus (pipeline);
678 fail_if (bus == NULL);
679 gst_element_set_state (pipeline, GST_STATE_PLAYING);
681 msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
682 fail_if (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_EOS);
683 gst_message_unref (msg);
685 /* cannot rely on the exact number of buffers as the timeout may produce
686 * more buffers with the unsynchronized _aggregate() implementation in
688 fail_if (count < TIMEOUT_NUM_BUFFERS);
690 gst_element_set_state (pipeline, GST_STATE_NULL);
691 gst_object_unref (src1pad);
692 gst_object_unref (bus);
693 gst_object_unref (pipeline);
696 GST_START_TEST (test_timeout_pipeline)
703 GST_START_TEST (test_timeout_pipeline_with_wait)
705 _test_timeout (1000000 /* 1 ms */ );
710 GST_START_TEST (test_flushing_seek)
713 GThread *thread1, *thread2;
715 ChainData data1 = { 0, };
716 ChainData data2 = { 0, };
717 TestData test = { 0, };
719 _test_data_init (&test, TRUE);
721 /* Queue a buffer in agg:sink_1. Then do a flushing seek and check that the
722 * new flushing seek logic is triggered. On the first FLUSH_START call the
723 * buffers queued in collectpads should get flushed. Only one FLUSH_START and
724 * one FLUSH_STOP should be forwarded downstream.
726 _chain_data_init (&data1, test.aggregator);
727 _chain_data_init (&data2, test.aggregator);
728 GST_BUFFER_TIMESTAMP (data2.buffer) = 0;
730 gst_segment_init (&GST_AGGREGATOR (test.aggregator)->segment,
733 /* now do a successful flushing seek */
734 event = gst_event_new_seek (1, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH,
735 GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET, 10 * GST_SECOND);
736 fail_unless (gst_pad_send_event (test.srcpad, event));
738 /* flushing starts once one of the upstream elements sends the first
740 fail_unless_equals_int (test.flush_start_events, 0);
741 fail_unless_equals_int (test.flush_stop_events, 0);
743 /* flush ogg:sink_0. This flushs collectpads, calls ::flush() and sends
744 * FLUSH_START downstream */
745 GST_DEBUG ("Flushing: %s:%s", GST_DEBUG_PAD_NAME (data2.sinkpad));
746 fail_unless (gst_pad_push_event (data2.srcpad, gst_event_new_flush_start ()));
748 /* expect this buffer to be flushed */
749 data2.expected_result = GST_FLOW_FLUSHING;
750 thread2 = g_thread_try_new ("gst-check", push_buffer, &data2, NULL);
752 fail_unless (gst_pad_push_event (data1.srcpad, gst_event_new_flush_start ()));
753 fail_unless_equals_int (test.flush_start_events, 1);
754 fail_unless_equals_int (test.flush_stop_events, 0);
756 /* the first FLUSH_STOP is not forwarded downstream */
757 fail_unless (gst_pad_push_event (data1.srcpad,
758 gst_event_new_flush_stop (TRUE)));
759 fail_unless_equals_int (test.flush_start_events, 1);
760 fail_unless_equals_int (test.flush_stop_events, 0);
762 /* at this point even the other pad agg:sink_1 should be flushing so thread2
763 * should have stopped */
764 g_thread_join (thread2);
766 /* push a buffer on agg:sink_0 to trigger one collect after flushing to verify
767 * that flushing completes once all the pads have been flushed */
768 thread1 = g_thread_try_new ("gst-check", push_buffer, &data1, NULL);
770 /* flush agg:sink_1 as well. This completes the flushing seek so a FLUSH_STOP is
772 gst_pad_push_event (data2.srcpad, gst_event_new_flush_stop (TRUE));
774 /* and the last FLUSH_STOP is forwarded downstream */
775 fail_unless_equals_int (test.flush_stop_events, 1);
777 /* Check collected */
778 gst_pad_add_probe (test.srcpad, GST_PAD_PROBE_TYPE_BUFFER,
779 (GstPadProbeCallback) _aggregated_cb, test.ml, NULL);
781 data2.event = gst_event_new_eos ();
782 thread2 = g_thread_try_new ("gst-check", push_event, &data2, NULL);
784 g_main_loop_run (test.ml);
785 g_source_remove (test.timeout_id);
787 fail_unless_equals_int (test.flush_stop_events, 1);
789 /* these will return immediately as at this point the threads have been
790 * unlocked and are finished */
791 g_thread_join (thread1);
792 g_thread_join (thread2);
794 _chain_data_clear (&data1);
795 _chain_data_clear (&data2);
796 _test_data_clear (&test);
803 infinite_seek (guint num_srcs, guint num_seeks, gboolean is_live)
807 GstElement *pipeline, *src, *agg, *sink;
810 gboolean seek_res, carry_on = TRUE;
812 gst_init (NULL, NULL);
814 pipeline = gst_pipeline_new ("pipeline");
816 agg = gst_check_setup_element ("testaggregator");
817 sink = gst_check_setup_element ("fakesink");
820 g_object_set (agg, "latency", GST_MSECOND, NULL);
822 fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
823 fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
824 fail_unless (gst_element_link (agg, sink));
826 for (i = 0; i < num_srcs; i++) {
827 src = gst_element_factory_make ("fakesrc", NULL);
828 g_object_set (src, "sizetype", 2, "sizemax", 4,
829 "format", GST_FORMAT_TIME, "datarate", 1000, NULL);
831 g_object_set (src, "is-live", TRUE, NULL);
832 fail_unless (gst_bin_add (GST_BIN (pipeline), src));
833 fail_unless (gst_element_link (src, agg));
836 bus = gst_element_get_bus (pipeline);
837 fail_if (bus == NULL);
838 gst_element_set_state (pipeline, GST_STATE_PLAYING);
839 while (count < num_seeks && carry_on) {
840 message = gst_bus_poll (bus, GST_MESSAGE_ANY, GST_SECOND / 10);
842 switch (GST_MESSAGE_TYPE (message)) {
843 case GST_MESSAGE_EOS:
845 /* we should check if we really finished here */
846 GST_WARNING ("Got an EOS");
850 case GST_MESSAGE_STATE_CHANGED:
854 if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipeline)) {
855 gst_message_parse_state_changed (message, NULL, &new, NULL);
857 if (new != GST_STATE_PLAYING)
860 GST_INFO ("Seeking (num: %i)", count);
862 gst_element_seek_simple (sink, GST_FORMAT_TIME,
863 GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, 0);
864 GST_INFO ("seek result is : %d", seek_res);
865 fail_unless (seek_res != 0);
871 case GST_MESSAGE_ERROR:
872 GST_ERROR ("Error on the bus: %" GST_PTR_FORMAT, message);
874 fail_error_message (message);
879 gst_message_unref (message);
883 gst_element_set_state (pipeline, GST_STATE_NULL);
884 gst_object_unref (bus);
885 gst_object_unref (pipeline);
888 GST_START_TEST (test_infinite_seek)
890 infinite_seek (2, 500, FALSE);
895 GST_START_TEST (test_infinite_seek_50_src)
897 infinite_seek (50, 100, FALSE);
902 GST_START_TEST (test_infinite_seek_50_src_live)
904 infinite_seek (50, 100, TRUE);
911 GstElement *agg, *src, *pipeline;
916 static GstPadProbeReturn
917 pad_probe_cb (GstPad * pad, GstPadProbeInfo * info, RemoveElementData * data)
921 GST_INFO_OBJECT (pad, "Removing pad");
923 peer = gst_pad_get_peer (pad);
924 gst_pad_unlink (pad, peer);
925 gst_element_release_request_pad (data->agg, peer);
926 fail_unless (gst_bin_remove (GST_BIN (data->pipeline), data->src));
927 gst_object_unref (peer);
929 g_mutex_lock (data->lock);
930 g_cond_broadcast (data->cond);
931 g_mutex_unlock (data->lock);
933 return GST_PAD_PROBE_OK;
936 GST_START_TEST (test_add_remove)
938 /* Used to notify that we removed the pad from */
945 gboolean carry_on = TRUE;
946 guint num_iterations = 100;
949 GstElement *pipeline, *src, *src1 = NULL, *agg, *sink;
953 gst_init (NULL, NULL);
954 g_mutex_init (&lock);
957 pipeline = gst_pipeline_new ("pipeline");
959 agg = gst_check_setup_element ("testaggregator");
960 sink = gst_check_setup_element ("fakesink");
962 fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
963 fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
964 fail_unless (gst_element_link (agg, sink));
966 bus = gst_element_get_bus (pipeline);
967 while (count < num_iterations) {
969 src = gst_element_factory_make ("fakesrc", NULL);
970 g_object_set (src, "num-buffers", 100000, "sizetype", 2, "sizemax", 4,
971 "format", GST_FORMAT_TIME, "datarate", 1000, NULL);
972 gst_element_set_locked_state (src, TRUE);
973 fail_unless (gst_bin_add (GST_BIN (pipeline), src));
974 fail_unless (gst_element_link (src, agg));
975 gst_element_set_locked_state (src, FALSE);
976 fail_unless (gst_element_sync_state_with_parent (src));
979 gst_element_set_state (pipeline, GST_STATE_PLAYING);
981 /* Now make sure the seek happend */
984 message = gst_bus_timed_pop (bus, -1);
985 switch (GST_MESSAGE_TYPE (message)) {
986 case GST_MESSAGE_EOS:
988 /* we should check if we really finished here */
989 GST_WARNING ("Got an EOS");
993 case GST_MESSAGE_STATE_CHANGED:
995 if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipeline)) {
996 gst_message_parse_state_changed (message, NULL, &state, NULL);
998 if (state == GST_STATE_PLAYING) {
999 RemoveElementData data;
1003 GST_DEBUG ("First run, not removing any element yet");
1008 data.src = gst_object_ref (src1);
1012 data.pipeline = pipeline;
1013 pad = gst_element_get_static_pad (data.src, "src");
1015 g_mutex_lock (&lock);
1016 gst_pad_add_probe (pad, GST_PAD_PROBE_TYPE_BLOCK_DOWNSTREAM,
1017 (GstPadProbeCallback) pad_probe_cb, &data, NULL);
1018 GST_INFO ("Waiting for %" GST_PTR_FORMAT " %s", pad,
1019 gst_element_state_get_name (GST_STATE (data.src)));
1020 g_cond_wait (&cond, &lock);
1021 g_mutex_unlock (&lock);
1022 gst_object_unref (pad);
1024 /* We can not set state from the streaming thread so we
1025 * need to make sure that the source has been removed
1026 * before setting its state to NULL */
1027 gst_element_set_state (data.src, GST_STATE_NULL);
1029 gst_object_unref (data.src);
1035 case GST_MESSAGE_ERROR:
1037 GST_ERROR ("Error on the bus: %" GST_PTR_FORMAT, message);
1039 fail_error_message (message);
1046 gst_message_unref (message);
1049 GST_INFO ("Seeking");
1050 fail_unless (gst_element_seek_simple (pipeline, GST_FORMAT_TIME,
1051 GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, 0));
1056 gst_element_set_state (pipeline, GST_STATE_NULL);
1057 gst_object_unref (bus);
1058 gst_object_unref (pipeline);
1059 g_mutex_clear (&lock);
1060 g_cond_clear (&cond);
1065 GST_START_TEST (test_change_state_intensive)
1068 GstMessage *message;
1069 GstElement *pipeline, *src, *agg, *sink;
1071 gint i, state_i = 0, num_srcs = 3;
1072 gboolean carry_on = TRUE, ready = FALSE;
1073 GstStateChangeReturn state_return;
1074 GstState wanted_state, wanted_states[] = {
1075 GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_READY,
1076 GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_READY,
1077 GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_READY,
1078 GST_STATE_PAUSED, GST_STATE_READY, GST_STATE_PAUSED, GST_STATE_READY,
1079 GST_STATE_PAUSED, GST_STATE_READY, GST_STATE_PAUSED, GST_STATE_NULL,
1080 GST_STATE_PAUSED, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_NULL,
1081 GST_STATE_PAUSED, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_NULL,
1082 GST_STATE_PAUSED, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
1083 GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
1084 GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
1085 GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
1088 gst_init (NULL, NULL);
1090 pipeline = gst_pipeline_new ("pipeline");
1092 agg = gst_check_setup_element ("testaggregator");
1093 sink = gst_check_setup_element ("fakesink");
1095 fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
1096 fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
1097 fail_unless (gst_element_link (agg, sink));
1099 for (i = 0; i < num_srcs; i++) {
1100 src = gst_element_factory_make ("fakesrc", NULL);
1101 g_object_set (src, "sizetype", 2, "sizemax", 4, NULL);
1102 fail_unless (gst_bin_add (GST_BIN (pipeline), src));
1103 fail_unless (gst_element_link (src, agg));
1106 bus = gst_element_get_bus (pipeline);
1107 fail_if (bus == NULL);
1109 wanted_state = wanted_states[state_i++];
1110 state_return = gst_element_set_state (pipeline, wanted_state);
1112 while (state_i < G_N_ELEMENTS (wanted_states) && carry_on) {
1113 if (state_return == GST_STATE_CHANGE_SUCCESS && ready) {
1114 wanted_state = wanted_states[state_i++];
1115 fail_unless (gst_element_set_state (pipeline, wanted_state),
1116 GST_STATE_CHANGE_SUCCESS);
1117 GST_INFO ("Wanted state: %s", gst_element_state_get_name (wanted_state));
1120 message = gst_bus_poll (bus, GST_MESSAGE_ANY, GST_SECOND / 10);
1122 switch (GST_MESSAGE_TYPE (message)) {
1123 case GST_MESSAGE_EOS:
1125 /* we should check if we really finished here */
1126 GST_WARNING ("Got an EOS");
1130 case GST_MESSAGE_STATE_CHANGED:
1134 if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipeline)) {
1135 gst_message_parse_state_changed (message, NULL, &new, NULL);
1137 if (new != wanted_state) {
1142 GST_DEBUG ("State %s reached",
1143 gst_element_state_get_name (wanted_state));
1144 wanted_state = wanted_states[state_i++];
1145 GST_DEBUG ("Wanted state: %s",
1146 gst_element_state_get_name (wanted_state));
1147 state_return = gst_element_set_state (pipeline, wanted_state);
1148 fail_unless (state_return == GST_STATE_CHANGE_SUCCESS ||
1149 state_return == GST_STATE_CHANGE_ASYNC);
1155 case GST_MESSAGE_ERROR:
1156 GST_ERROR ("Error on the bus: %" GST_PTR_FORMAT, message);
1162 gst_message_unref (message);
1166 gst_element_set_state (pipeline, GST_STATE_NULL);
1167 gst_object_unref (bus);
1168 gst_object_unref (pipeline);
1174 gst_aggregator_suite (void)
1179 gst_test_aggregator_plugin_register ();
1181 suite = suite_create ("GstAggregator");
1183 general = tcase_create ("general");
1184 suite_add_tcase (suite, general);
1185 tcase_add_test (general, test_aggregate);
1186 tcase_add_test (general, test_aggregate_eos);
1187 tcase_add_test (general, test_aggregate_gap);
1188 tcase_add_test (general, test_flushing_seek);
1189 tcase_add_test (general, test_infinite_seek);
1190 tcase_add_test (general, test_infinite_seek_50_src);
1191 tcase_add_test (general, test_infinite_seek_50_src_live);
1192 tcase_add_test (general, test_linear_pipeline);
1193 tcase_add_test (general, test_two_src_pipeline);
1194 tcase_add_test (general, test_timeout_pipeline);
1195 tcase_add_test (general, test_timeout_pipeline_with_wait);
1196 tcase_add_test (general, test_add_remove);
1197 tcase_add_test (general, test_change_state_intensive);
1202 GST_CHECK_MAIN (gst_aggregator);