tests/check/libs/aggregator.c

   1 /*
   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>
   6  *
   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.
  11  *
  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.
  16  *
  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.
  21  */
  22
  23 #ifdef HAVE_CONFIG_H
  24 #  include "config.h"
  25 #endif
  26
  27 #include <gst/check/gstcheck.h>
  28 #include <gst/base/gstaggregator.h>
  29
  30 /* dummy aggregator based element */
  31
  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))
  36
  37 #define fail_error_message(msg)     \
  38   G_STMT_START {        \
  39     GError *error;        \
  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);           \
  44   } G_STMT_END;
  45
  46 typedef struct _GstTestAggregator GstTestAggregator;
  47 typedef struct _GstTestAggregatorClass GstTestAggregatorClass;
  48
  49 static GType gst_test_aggregator_get_type (void);
  50
  51 #define BUFFER_DURATION 100000000       /* 10 frames per second */
  52
  53 struct _GstTestAggregator
  54 {
  55   GstAggregator parent;
  56
  57   guint64 timestamp;
  58 };
  59
  60 struct _GstTestAggregatorClass
  61 {
  62   GstAggregatorClass parent_class;
  63 };
  64
  65 static GstFlowReturn
  66 gst_test_aggregator_aggregate (GstAggregator * aggregator)
  67 {
  68   GstIterator *iter;
  69   gboolean all_eos = TRUE;
  70   GstTestAggregator *testagg;
  71   GstBuffer *buf;
  72
  73   gboolean done_iterating = FALSE;
  74
  75   testagg = GST_TEST_AGGREGATOR (aggregator);
  76
  77   iter = gst_element_iterate_sink_pads (GST_ELEMENT (testagg));
  78   while (!done_iterating) {
  79     GstBuffer *buffer;
  80     GValue value = { 0, };
  81     GstAggregatorPad *pad;
  82
  83     switch (gst_iterator_next (iter, &value)) {
  84       case GST_ITERATOR_OK:
  85         pad = g_value_get_object (&value);
  86
  87         if (pad->eos == FALSE)
  88           all_eos = FALSE;
  89         buffer = gst_aggregator_pad_steal_buffer (pad);
  90         gst_buffer_replace (&buffer, NULL);
  91
  92         g_value_reset (&value);
  93         break;
  94       case GST_ITERATOR_RESYNC:
  95         gst_iterator_resync (iter);
  96         break;
  97       case GST_ITERATOR_ERROR:
  98         GST_WARNING_OBJECT (testagg, "Sinkpads iteration error");
  99         done_iterating = TRUE;
 100         break;
 101       case GST_ITERATOR_DONE:
 102         done_iterating = TRUE;
 103         break;
 104     }
 105   }
 106   gst_iterator_free (iter);
 107
 108   if (all_eos == TRUE) {
 109     GST_INFO_OBJECT (testagg, "no data available, must be EOS");
 110     gst_pad_push_event (aggregator->srcpad, gst_event_new_eos ());
 111     return GST_FLOW_EOS;
 112   }
 113
 114   buf = gst_buffer_new ();
 115   GST_BUFFER_TIMESTAMP (buf) = testagg->timestamp;
 116   GST_BUFFER_DURATION (buf) = BUFFER_DURATION;
 117   testagg->timestamp += BUFFER_DURATION;
 118
 119   gst_aggregator_finish_buffer (aggregator, buf);
 120
 121   /* We just check finish_frame return FLOW_OK */
 122   return GST_FLOW_OK;
 123 }
 124
 125 #define gst_test_aggregator_parent_class parent_class
 126 G_DEFINE_TYPE (GstTestAggregator, gst_test_aggregator, GST_TYPE_AGGREGATOR);
 127
 128 static void
 129 gst_test_aggregator_class_init (GstTestAggregatorClass * klass)
 130 {
 131   GstElementClass *gstelement_class = (GstElementClass *) klass;
 132   GstAggregatorClass *base_aggregator_class = (GstAggregatorClass *) klass;
 133
 134   static GstStaticPadTemplate _src_template =
 135       GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
 136       GST_STATIC_CAPS_ANY);
 137
 138   static GstStaticPadTemplate _sink_template =
 139       GST_STATIC_PAD_TEMPLATE ("sink_%u", GST_PAD_SINK, GST_PAD_REQUEST,
 140       GST_STATIC_CAPS_ANY);
 141
 142   gst_element_class_add_pad_template (gstelement_class,
 143       gst_static_pad_template_get (&_src_template));
 144
 145   gst_element_class_add_pad_template (gstelement_class,
 146       gst_static_pad_template_get (&_sink_template));
 147
 148   gst_element_class_set_static_metadata (gstelement_class, "Aggregator",
 149       "Testing", "Combine N buffers", "Stefan Sauer <ensonic@users.sf.net>");
 150
 151   base_aggregator_class->aggregate =
 152       GST_DEBUG_FUNCPTR (gst_test_aggregator_aggregate);
 153 }
 154
 155 static void
 156 gst_test_aggregator_init (GstTestAggregator * self)
 157 {
 158   GstAggregator *agg = GST_AGGREGATOR (self);
 159   gst_segment_init (&agg->segment, GST_FORMAT_BYTES);
 160   self->timestamp = 0;
 161 }
 162
 163 static gboolean
 164 gst_test_aggregator_plugin_init (GstPlugin * plugin)
 165 {
 166   return gst_element_register (plugin, "testaggregator", GST_RANK_NONE,
 167       GST_TYPE_TEST_AGGREGATOR);
 168 }
 169
 170 static gboolean
 171 gst_test_aggregator_plugin_register (void)
 172 {
 173   return gst_plugin_register_static (GST_VERSION_MAJOR,
 174       GST_VERSION_MINOR,
 175       "testaggregator",
 176       "Combine buffers",
 177       gst_test_aggregator_plugin_init,
 178       VERSION, GST_LICENSE, PACKAGE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);
 179 }
 180
 181 typedef struct
 182 {
 183   GstEvent *event;
 184   GstBuffer *buffer;
 185   GstElement *aggregator;
 186   GstPad *sinkpad, *srcpad;
 187   GstFlowReturn expected_result;
 188
 189   /*                       ------------------
 190    * -----------   --------|--              |
 191    * | srcpad | -- | sinkpad |  aggregator  |
 192    * -----------   --------|--              |
 193    *                       ------------------
 194    *  This is for 1 Chain, we can have several
 195    */
 196 } ChainData;
 197
 198 typedef struct
 199 {
 200   GMainLoop *ml;
 201   GstPad *srcpad,               /* srcpad of the GstAggregator */
 202    *sinkpad;                    /* fake sinkpad to which GstAggregator.srcpad is linked */
 203   guint timeout_id;
 204   GstElement *aggregator;
 205
 206   /* -----------------|
 207    * |             ----------    -----------
 208    * | aggregator  | srcpad | -- | sinkpad |
 209    * |             ----------    -----------
 210    * -----------------|
 211    */
 212
 213   gint flush_start_events, flush_stop_events;
 214 } TestData;
 215
 216 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
 217     GST_PAD_SRC,
 218     GST_PAD_ALWAYS,
 219     GST_STATIC_CAPS_ANY);
 220
 221 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
 222     GST_PAD_SINK,
 223     GST_PAD_ALWAYS,
 224     GST_STATIC_CAPS_ANY);
 225
 226 static gpointer
 227 push_buffer (gpointer user_data)
 228 {
 229   GstFlowReturn flow;
 230   GstCaps *caps;
 231   ChainData *chain_data = (ChainData *) user_data;
 232   GstSegment segment;
 233
 234   gst_pad_push_event (chain_data->srcpad, gst_event_new_stream_start ("test"));
 235
 236   caps = gst_caps_new_empty_simple ("foo/x-bar");
 237   gst_pad_push_event (chain_data->srcpad, gst_event_new_caps (caps));
 238   gst_caps_unref (caps);
 239
 240   gst_segment_init (&segment, GST_FORMAT_TIME);
 241   gst_pad_push_event (chain_data->srcpad, gst_event_new_segment (&segment));
 242
 243   GST_DEBUG ("Pushing buffer on pad: %s:%s",
 244       GST_DEBUG_PAD_NAME (chain_data->sinkpad));
 245   flow = gst_pad_push (chain_data->srcpad, chain_data->buffer);
 246   fail_unless (flow == chain_data->expected_result,
 247       "got flow %s instead of %s on %s:%s", gst_flow_get_name (flow),
 248       gst_flow_get_name (chain_data->expected_result),
 249       GST_DEBUG_PAD_NAME (chain_data->sinkpad));
 250   chain_data->buffer = NULL;
 251
 252   return NULL;
 253 }
 254
 255 static gpointer
 256 push_event (gpointer user_data)
 257 {
 258   ChainData *chain_data = (ChainData *) user_data;
 259
 260   GST_INFO_OBJECT (chain_data->srcpad, "Pushing event: %"
 261       GST_PTR_FORMAT, chain_data->event);
 262   fail_unless (gst_pad_push_event (chain_data->srcpad,
 263           chain_data->event) == TRUE);
 264
 265   return NULL;
 266 }
 267
 268 static gboolean
 269 _aggregate_timeout (GMainLoop * ml)
 270 {
 271   g_main_loop_quit (ml);
 272
 273   fail_unless ("No buffer found on aggregator.srcpad -> TIMEOUT" == NULL);
 274
 275   return FALSE;
 276 }
 277
 278 static gboolean
 279 _quit (GMainLoop * ml)
 280 {
 281   GST_DEBUG ("QUITING ML");
 282   g_main_loop_quit (ml);
 283
 284   return G_SOURCE_REMOVE;
 285 }
 286
 287 static GstPadProbeReturn
 288 _aggregated_cb (GstPad * pad, GstPadProbeInfo * info, GMainLoop * ml)
 289 {
 290   GST_DEBUG ("SHould quit ML");
 291   g_idle_add ((GSourceFunc) _quit, ml);
 292
 293   return GST_PAD_PROBE_REMOVE;
 294 }
 295
 296 static GstPadProbeReturn
 297 downstream_probe_cb (GstPad * pad, GstPadProbeInfo * info, TestData * test)
 298 {
 299   GST_DEBUG ("PROBING ");
 300   if (info->type & GST_PAD_PROBE_TYPE_EVENT_FLUSH) {
 301     if (GST_EVENT_TYPE (GST_PAD_PROBE_INFO_EVENT (info)) ==
 302         GST_EVENT_FLUSH_START) {
 303
 304       g_atomic_int_inc (&test->flush_start_events);
 305       GST_DEBUG ("==========> FLUSH: %i", test->flush_start_events);
 306     } else if (GST_EVENT_TYPE (GST_PAD_PROBE_INFO_EVENT (info)) ==
 307         GST_EVENT_FLUSH_STOP)
 308       g_atomic_int_inc (&test->flush_stop_events);
 309   }
 310
 311   return GST_PAD_PROBE_DROP;
 312 }
 313
 314 /*
 315  * Not thread safe, will create a new ChainData which contains
 316  * an activated src pad linked to a requested sink pad of @agg, and
 317  * a newly allocated buffer ready to be pushed. Caller needs to
 318  * clear with _chain_data_clear after.
 319  */
 320 static void
 321 _chain_data_init (ChainData * data, GstElement * agg)
 322 {
 323   static gint num_src_pads = 0;
 324   gchar *pad_name = g_strdup_printf ("src%d", num_src_pads);
 325
 326   num_src_pads += 1;
 327
 328   data->srcpad = gst_pad_new_from_static_template (&srctemplate, pad_name);
 329   g_free (pad_name);
 330   gst_pad_set_active (data->srcpad, TRUE);
 331   data->aggregator = agg;
 332   data->buffer = gst_buffer_new ();
 333   data->sinkpad = gst_element_get_request_pad (agg, "sink_%u");
 334   fail_unless (GST_IS_PAD (data->sinkpad));
 335   fail_unless (gst_pad_link (data->srcpad, data->sinkpad) == GST_PAD_LINK_OK);
 336 }
 337
 338 static void
 339 _chain_data_clear (ChainData * data)
 340 {
 341   if (data->buffer)
 342     gst_buffer_unref (data->buffer);
 343   if (data->srcpad)
 344     gst_object_unref (data->srcpad);
 345   if (data->sinkpad)
 346     gst_object_unref (data->sinkpad);
 347 }
 348
 349 static void
 350 _test_data_init (TestData * test, gboolean needs_flushing)
 351 {
 352   test->aggregator = gst_element_factory_make ("testaggregator", NULL);
 353   gst_element_set_state (test->aggregator, GST_STATE_PLAYING);
 354   test->ml = g_main_loop_new (NULL, TRUE);
 355   test->srcpad = GST_AGGREGATOR (test->aggregator)->srcpad;
 356
 357   GST_DEBUG ("Srcpad: %p", test->srcpad);
 358
 359   if (needs_flushing) {
 360     static gint num_sink_pads = 0;
 361     gchar *pad_name = g_strdup_printf ("sink%d", num_sink_pads);
 362
 363     num_sink_pads += 1;
 364     test->sinkpad = gst_pad_new_from_static_template (&sinktemplate, pad_name);
 365     g_free (pad_name);
 366     fail_unless (gst_pad_link (test->srcpad, test->sinkpad) == GST_PAD_LINK_OK);
 367     gst_pad_add_probe (test->srcpad, GST_PAD_PROBE_TYPE_EVENT_DOWNSTREAM |
 368         GST_PAD_PROBE_TYPE_DATA_DOWNSTREAM |
 369         GST_PAD_PROBE_TYPE_EVENT_FLUSH,
 370         (GstPadProbeCallback) downstream_probe_cb, test, NULL);
 371   } else {
 372     gst_pad_add_probe (test->srcpad, GST_PAD_PROBE_TYPE_BUFFER,
 373         (GstPadProbeCallback) _aggregated_cb, test->ml, NULL);
 374   }
 375
 376
 377   test->timeout_id =
 378       g_timeout_add (1000, (GSourceFunc) _aggregate_timeout, test->ml);
 379 }
 380
 381 static void
 382 _test_data_clear (TestData * test)
 383 {
 384   gst_element_set_state (test->aggregator, GST_STATE_NULL);
 385   gst_object_unref (test->aggregator);
 386
 387   if (test->sinkpad)
 388     gst_object_unref (test->sinkpad);
 389
 390   g_main_loop_unref (test->ml);
 391 }
 392
 393 GST_START_TEST (test_aggregate)
 394 {
 395   GThread *thread1, *thread2;
 396
 397   ChainData data1 = { 0, };
 398   ChainData data2 = { 0, };
 399   TestData test = { 0, };
 400
 401   _test_data_init (&test, FALSE);
 402   _chain_data_init (&data1, test.aggregator);
 403   _chain_data_init (&data2, test.aggregator);
 404
 405   thread1 = g_thread_try_new ("gst-check", push_buffer, &data1, NULL);
 406   thread2 = g_thread_try_new ("gst-check", push_buffer, &data2, NULL);
 407
 408   g_main_loop_run (test.ml);
 409   g_source_remove (test.timeout_id);
 410
 411
 412   /* these will return immediately as when the data is popped the threads are
 413    * unlocked and will terminate */
 414   g_thread_join (thread1);
 415   g_thread_join (thread2);
 416
 417   _chain_data_clear (&data1);
 418   _chain_data_clear (&data2);
 419   _test_data_clear (&test);
 420 }
 421
 422 GST_END_TEST;
 423
 424 GST_START_TEST (test_aggregate_eos)
 425 {
 426   GThread *thread1, *thread2;
 427
 428   ChainData data1 = { 0, };
 429   ChainData data2 = { 0, };
 430   TestData test = { 0, };
 431
 432   _test_data_init (&test, FALSE);
 433   _chain_data_init (&data1, test.aggregator);
 434   _chain_data_init (&data2, test.aggregator);
 435
 436   data2.event = gst_event_new_eos ();
 437
 438   thread1 = g_thread_try_new ("gst-check", push_buffer, &data1, NULL);
 439   thread2 = g_thread_try_new ("gst-check", push_event, &data2, NULL);
 440
 441   g_main_loop_run (test.ml);
 442   g_source_remove (test.timeout_id);
 443
 444   /* these will return immediately as when the data is popped the threads are
 445    * unlocked and will terminate */
 446   g_thread_join (thread1);
 447   g_thread_join (thread2);
 448
 449   _chain_data_clear (&data1);
 450   _chain_data_clear (&data2);
 451   _test_data_clear (&test);
 452 }
 453
 454 GST_END_TEST;
 455
 456 #define NUM_BUFFERS 3
 457 static void
 458 handoff (GstElement * fakesink, GstBuffer * buf, GstPad * pad, guint * count)
 459 {
 460   *count = *count + 1;
 461   GST_DEBUG ("HANDOFF: %i", *count);
 462 }
 463
 464 /* Test a linear pipeline using aggregator */
 465 GST_START_TEST (test_linear_pipeline)
 466 {
 467   GstBus *bus;
 468   GstMessage *msg;
 469   GstElement *pipeline, *src, *agg, *sink;
 470
 471   gint count = 0;
 472
 473   pipeline = gst_pipeline_new ("pipeline");
 474   src = gst_check_setup_element ("fakesrc");
 475   g_object_set (src, "num-buffers", NUM_BUFFERS, "sizetype", 2, "sizemax", 4,
 476       NULL);
 477   agg = gst_check_setup_element ("testaggregator");
 478   sink = gst_check_setup_element ("fakesink");
 479   g_object_set (sink, "signal-handoffs", TRUE, NULL);
 480   g_signal_connect (sink, "handoff", (GCallback) handoff, &count);
 481
 482   fail_unless (gst_bin_add (GST_BIN (pipeline), src));
 483   fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
 484   fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
 485   fail_unless (gst_element_link (src, agg));
 486   fail_unless (gst_element_link (agg, sink));
 487
 488   bus = gst_element_get_bus (pipeline);
 489   fail_if (bus == NULL);
 490   gst_element_set_state (pipeline, GST_STATE_PLAYING);
 491
 492   msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
 493   fail_if (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_EOS);
 494   gst_message_unref (msg);
 495
 496   fail_unless_equals_int (count, NUM_BUFFERS);
 497
 498   gst_element_set_state (pipeline, GST_STATE_NULL);
 499   gst_object_unref (bus);
 500   gst_object_unref (pipeline);
 501 }
 502
 503 GST_END_TEST;
 504
 505 GST_START_TEST (test_two_src_pipeline)
 506 {
 507   GstBus *bus;
 508   GstMessage *msg;
 509   GstElement *pipeline, *src, *src1, *agg, *sink;
 510
 511   gint count = 0;
 512
 513   pipeline = gst_pipeline_new ("pipeline");
 514   src = gst_element_factory_make ("fakesrc", NULL);
 515   g_object_set (src, "num-buffers", NUM_BUFFERS, "sizetype", 2, "sizemax", 4,
 516       NULL);
 517
 518   src1 = gst_element_factory_make ("fakesrc", NULL);
 519   g_object_set (src1, "num-buffers", NUM_BUFFERS + 1, "sizetype", 2, "sizemax",
 520       4, NULL);
 521
 522   agg = gst_check_setup_element ("testaggregator");
 523   sink = gst_check_setup_element ("fakesink");
 524   g_object_set (sink, "signal-handoffs", TRUE, NULL);
 525   g_signal_connect (sink, "handoff", (GCallback) handoff, &count);
 526
 527   fail_unless (gst_bin_add (GST_BIN (pipeline), src));
 528   fail_unless (gst_bin_add (GST_BIN (pipeline), src1));
 529   fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
 530   fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
 531   fail_unless (gst_element_link (src, agg));
 532   fail_unless (gst_element_link (src1, agg));
 533   fail_unless (gst_element_link (agg, sink));
 534
 535   bus = gst_element_get_bus (pipeline);
 536   fail_if (bus == NULL);
 537   gst_element_set_state (pipeline, GST_STATE_PLAYING);
 538
 539   msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
 540   fail_if (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_EOS);
 541   gst_message_unref (msg);
 542
 543   fail_unless_equals_int (count, NUM_BUFFERS + 1);
 544
 545   gst_element_set_state (pipeline, GST_STATE_NULL);
 546   gst_object_unref (bus);
 547   gst_object_unref (pipeline);
 548 }
 549
 550 GST_END_TEST;
 551
 552 GST_START_TEST (test_flushing_seek)
 553 {
 554   GstEvent *event;
 555   GThread *thread1, *thread2;
 556
 557   ChainData data1 = { 0, };
 558   ChainData data2 = { 0, };
 559   TestData test = { 0, };
 560
 561   _test_data_init (&test, TRUE);
 562
 563   /* Queue a buffer in agg:sink_1. Then do a flushing seek and check that the
 564    * new flushing seek logic is triggered. On the first FLUSH_START call the
 565    * buffers queued in collectpads should get flushed. Only one FLUSH_START and
 566    * one FLUSH_STOP should be forwarded downstream.
 567    */
 568   _chain_data_init (&data1, test.aggregator);
 569   _chain_data_init (&data2, test.aggregator);
 570   GST_BUFFER_TIMESTAMP (data2.buffer) = 0;
 571
 572   gst_segment_init (&GST_AGGREGATOR (test.aggregator)->segment,
 573       GST_FORMAT_TIME);
 574   /* now do a successful flushing seek */
 575   event = gst_event_new_seek (1, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH,
 576       GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET, 10 * GST_SECOND);
 577   fail_unless (gst_pad_send_event (test.srcpad, event));
 578
 579   /* flushing starts once one of the upstream elements sends the first
 580    * FLUSH_START */
 581   fail_unless_equals_int (test.flush_start_events, 0);
 582   fail_unless_equals_int (test.flush_stop_events, 0);
 583
 584   /* flush ogg:sink_0. This flushs collectpads, calls ::flush() and sends
 585    * FLUSH_START downstream */
 586   GST_DEBUG ("Flushing: %s:%s", GST_DEBUG_PAD_NAME (data2.sinkpad));
 587   fail_unless (gst_pad_push_event (data2.srcpad, gst_event_new_flush_start ()));
 588
 589   /* expect this buffer to be flushed */
 590   data2.expected_result = GST_FLOW_FLUSHING;
 591   thread2 = g_thread_try_new ("gst-check", push_buffer, &data2, NULL);
 592
 593   fail_unless (gst_pad_push_event (data1.srcpad, gst_event_new_flush_start ()));
 594   fail_unless_equals_int (test.flush_start_events, 1);
 595   fail_unless_equals_int (test.flush_stop_events, 0);
 596
 597   /* the first FLUSH_STOP is not forwarded downstream */
 598   fail_unless (gst_pad_push_event (data1.srcpad,
 599           gst_event_new_flush_stop (TRUE)));
 600   fail_unless_equals_int (test.flush_start_events, 1);
 601   fail_unless_equals_int (test.flush_stop_events, 0);
 602
 603   /* at this point even the other pad agg:sink_1 should be flushing so thread2
 604    * should have stopped */
 605   g_thread_join (thread2);
 606
 607   /* push a buffer on agg:sink_0 to trigger one collect after flushing to verify
 608    * that flushing completes once all the pads have been flushed */
 609   thread1 = g_thread_try_new ("gst-check", push_buffer, &data1, NULL);
 610
 611   /* flush agg:sink_1 as well. This completes the flushing seek so a FLUSH_STOP is
 612    * sent downstream */
 613   gst_pad_push_event (data2.srcpad, gst_event_new_flush_stop (TRUE));
 614
 615   /* and the last FLUSH_STOP is forwarded downstream */
 616   fail_unless_equals_int (test.flush_start_events, 1);
 617
 618   /*  Check collected */
 619   gst_pad_add_probe (test.srcpad, GST_PAD_PROBE_TYPE_BUFFER,
 620       (GstPadProbeCallback) _aggregated_cb, test.ml, NULL);
 621
 622   data2.event = gst_event_new_eos ();
 623   thread2 = g_thread_try_new ("gst-check", push_event, &data2, NULL);
 624
 625   g_main_loop_run (test.ml);
 626   g_source_remove (test.timeout_id);
 627
 628   fail_unless_equals_int (test.flush_stop_events, 1);
 629
 630   /* these will return immediately as at this point the threads have been
 631    * unlocked and are finished */
 632   g_thread_join (thread1);
 633   g_thread_join (thread2);
 634
 635   _chain_data_clear (&data1);
 636   _chain_data_clear (&data2);
 637   _test_data_clear (&test);
 638
 639 }
 640
 641 GST_END_TEST;
 642
 643 static void
 644 infinite_seek (guint num_srcs, guint num_seeks)
 645 {
 646   GstBus *bus;
 647   GstMessage *message;
 648   GstElement *pipeline, *src, *agg, *sink;
 649
 650   gint count = 0, i;
 651   gboolean seek_res, carry_on = TRUE;
 652
 653   gst_init (NULL, NULL);
 654
 655   pipeline = gst_pipeline_new ("pipeline");
 656
 657   agg = gst_check_setup_element ("testaggregator");
 658   sink = gst_check_setup_element ("fakesink");
 659
 660   fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
 661   fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
 662   fail_unless (gst_element_link (agg, sink));
 663
 664   for (i = 0; i < num_srcs; i++) {
 665     src = gst_element_factory_make ("fakesrc", NULL);
 666     g_object_set (src, "sizetype", 2, "sizemax", 4, NULL);
 667     fail_unless (gst_bin_add (GST_BIN (pipeline), src));
 668     fail_unless (gst_element_link (src, agg));
 669   }
 670
 671   bus = gst_element_get_bus (pipeline);
 672   fail_if (bus == NULL);
 673   gst_element_set_state (pipeline, GST_STATE_PLAYING);
 674   while (count < num_seeks && carry_on) {
 675     message = gst_bus_poll (bus, GST_MESSAGE_ANY, GST_SECOND / 10);
 676     if (message) {
 677       switch (GST_MESSAGE_TYPE (message)) {
 678         case GST_MESSAGE_EOS:
 679         {
 680           /* we should check if we really finished here */
 681           GST_WARNING ("Got an EOS");
 682           carry_on = FALSE;
 683           break;
 684         }
 685         case GST_MESSAGE_STATE_CHANGED:
 686         {
 687           GstState new;
 688
 689           if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipeline)) {
 690             gst_message_parse_state_changed (message, NULL, &new, NULL);
 691
 692             if (new != GST_STATE_PLAYING)
 693               break;
 694
 695             GST_INFO ("Seeking (num: %i)", count);
 696             seek_res =
 697                 gst_element_seek_simple (sink, GST_FORMAT_BYTES,
 698                 GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, 0);
 699             GST_INFO ("seek result is : %d", seek_res);
 700             fail_unless (seek_res != 0);
 701             count++;
 702           }
 703
 704           break;
 705         }
 706         case GST_MESSAGE_ERROR:
 707           GST_ERROR ("Error on the bus: %" GST_PTR_FORMAT, message);
 708           carry_on = FALSE;
 709           fail_error_message (message);
 710           break;
 711         default:
 712           break;
 713       }
 714       gst_message_unref (message);
 715     }
 716   }
 717
 718   gst_element_set_state (pipeline, GST_STATE_NULL);
 719   gst_object_unref (bus);
 720   gst_object_unref (pipeline);
 721 }
 722
 723 GST_START_TEST (test_infinite_seek)
 724 {
 725   infinite_seek (2, 500);
 726 }
 727
 728 GST_END_TEST;
 729
 730 GST_START_TEST (test_infinite_seek_50_src)
 731 {
 732   infinite_seek (50, 100);
 733 }
 734
 735 GST_END_TEST;
 736
 737 typedef struct
 738 {
 739   GstElement *agg, *src, *pipeline;
 740   GCond *cond;
 741   GMutex *lock;
 742 } RemoveElementData;
 743
 744 static GstPadProbeReturn
 745 pad_probe_cb (GstPad * pad, GstPadProbeInfo * info, RemoveElementData * data)
 746 {
 747   GstPad *peer;
 748
 749   GST_INFO_OBJECT (pad, "Removing pad");
 750
 751   peer = gst_pad_get_peer (pad);
 752   gst_pad_unlink (pad, peer);
 753   gst_element_release_request_pad (data->agg, peer);
 754   fail_unless (gst_bin_remove (GST_BIN (data->pipeline), data->src));
 755   gst_object_unref (peer);
 756
 757   g_mutex_lock (data->lock);
 758   g_cond_broadcast (data->cond);
 759   g_mutex_unlock (data->lock);
 760
 761   return GST_PAD_PROBE_OK;
 762 }
 763
 764 GST_START_TEST (test_add_remove)
 765 {
 766   /* Used to notify that we removed the pad from  */
 767   GCond cond;
 768   GMutex lock;
 769
 770   GstBus *bus;
 771   GstState state;
 772   GstMessage *message;
 773   gboolean carry_on = TRUE;
 774   guint num_iterations = 100;
 775
 776   GstPad *pad;
 777   GstElement *pipeline, *src, *src1 = NULL, *agg, *sink;
 778
 779   gint count = 0;
 780
 781   gst_init (NULL, NULL);
 782   g_mutex_init (&lock);
 783   g_cond_init (&cond);
 784
 785   pipeline = gst_pipeline_new ("pipeline");
 786
 787   agg = gst_check_setup_element ("testaggregator");
 788   sink = gst_check_setup_element ("fakesink");
 789
 790   fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
 791   fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
 792   fail_unless (gst_element_link (agg, sink));
 793
 794   bus = gst_element_get_bus (pipeline);
 795   while (count < num_iterations) {
 796
 797     src = gst_element_factory_make ("fakesrc", NULL);
 798     g_object_set (src, "num-buffers", 100000, "sizetype", 2, "sizemax", 4,
 799         NULL);
 800     gst_element_set_locked_state (src, TRUE);
 801     fail_unless (gst_bin_add (GST_BIN (pipeline), src));
 802     fail_unless (gst_element_link (src, agg));
 803     gst_element_set_locked_state (src, FALSE);
 804     fail_unless (gst_element_sync_state_with_parent (src));
 805
 806     if (count == 0)
 807       gst_element_set_state (pipeline, GST_STATE_PLAYING);
 808
 809     /* Now make sure the seek happend */
 810     carry_on = TRUE;
 811     do {
 812       message = gst_bus_timed_pop (bus, -1);
 813       switch (GST_MESSAGE_TYPE (message)) {
 814         case GST_MESSAGE_EOS:
 815         {
 816           /* we should check if we really finished here */
 817           GST_WARNING ("Got an EOS");
 818           carry_on = FALSE;
 819           break;
 820         }
 821         case GST_MESSAGE_STATE_CHANGED:
 822         {
 823           if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipeline)) {
 824             gst_message_parse_state_changed (message, NULL, &state, NULL);
 825
 826             if (state == GST_STATE_PLAYING) {
 827               RemoveElementData data;
 828
 829               carry_on = FALSE;
 830               if (count == 0) {
 831                 GST_DEBUG ("First run, not removing any element yet");
 832
 833                 break;
 834               }
 835
 836               data.src = gst_object_ref (src1);
 837               data.agg = agg;
 838               data.lock = &lock;
 839               data.cond = &cond;
 840               data.pipeline = pipeline;
 841               pad = gst_element_get_static_pad (data.src, "src");
 842
 843               g_mutex_lock (&lock);
 844               gst_pad_add_probe (pad, GST_PAD_PROBE_TYPE_BLOCK_DOWNSTREAM,
 845                   (GstPadProbeCallback) pad_probe_cb, &data, NULL);
 846               GST_INFO ("Waiting for %" GST_PTR_FORMAT " %s", pad,
 847                   gst_element_state_get_name (GST_STATE (data.src)));
 848               g_cond_wait (&cond, &lock);
 849               g_mutex_unlock (&lock);
 850               gst_object_unref (pad);
 851
 852               /*  We can not set state from the streaming thread so we
 853                *  need to make sure that the source has been removed
 854                *  before setting its state to NULL */
 855               gst_element_set_state (data.src, GST_STATE_NULL);
 856
 857               gst_object_unref (data.src);
 858             }
 859           }
 860
 861           break;
 862         }
 863         case GST_MESSAGE_ERROR:
 864         {
 865           GST_ERROR ("Error on the bus: %" GST_PTR_FORMAT, message);
 866           carry_on = FALSE;
 867           fail_error_message (message);
 868           break;
 869         }
 870         default:
 871           break;
 872       }
 873
 874       gst_message_unref (message);
 875     } while (carry_on);
 876
 877     GST_INFO ("Seeking");
 878     fail_unless (gst_element_seek_simple (pipeline, GST_FORMAT_BYTES,
 879             GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, 0));
 880
 881     count++;
 882     src1 = src;
 883   }
 884   gst_element_set_state (pipeline, GST_STATE_NULL);
 885   gst_object_unref (bus);
 886   gst_object_unref (pipeline);
 887   g_mutex_clear (&lock);
 888   g_cond_clear (&cond);
 889 }
 890
 891 GST_END_TEST;
 892
 893 GST_START_TEST (test_change_state_intensive)
 894 {
 895   GstBus *bus;
 896   GstMessage *message;
 897   GstElement *pipeline, *src, *agg, *sink;
 898
 899   gint i, state_i = 0, num_srcs = 3;
 900   gboolean carry_on = TRUE, ready = FALSE;
 901   GstStateChangeReturn state_return;
 902   GstState wanted_state, wanted_states[] = {
 903     GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_READY,
 904     GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_READY,
 905     GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_READY,
 906     GST_STATE_PAUSED, GST_STATE_READY, GST_STATE_PAUSED, GST_STATE_READY,
 907     GST_STATE_PAUSED, GST_STATE_READY, GST_STATE_PAUSED, GST_STATE_NULL,
 908     GST_STATE_PAUSED, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_NULL,
 909     GST_STATE_PAUSED, GST_STATE_NULL, GST_STATE_PAUSED, GST_STATE_NULL,
 910     GST_STATE_PAUSED, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
 911     GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
 912     GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
 913     GST_STATE_PLAYING, GST_STATE_NULL, GST_STATE_PLAYING, GST_STATE_NULL,
 914   };
 915
 916   gst_init (NULL, NULL);
 917
 918   pipeline = gst_pipeline_new ("pipeline");
 919
 920   agg = gst_check_setup_element ("testaggregator");
 921   sink = gst_check_setup_element ("fakesink");
 922
 923   fail_unless (gst_bin_add (GST_BIN (pipeline), agg));
 924   fail_unless (gst_bin_add (GST_BIN (pipeline), sink));
 925   fail_unless (gst_element_link (agg, sink));
 926
 927   for (i = 0; i < num_srcs; i++) {
 928     src = gst_element_factory_make ("fakesrc", NULL);
 929     g_object_set (src, "sizetype", 2, "sizemax", 4, NULL);
 930     fail_unless (gst_bin_add (GST_BIN (pipeline), src));
 931     fail_unless (gst_element_link (src, agg));
 932   }
 933
 934   bus = gst_element_get_bus (pipeline);
 935   fail_if (bus == NULL);
 936
 937   wanted_state = wanted_states[state_i++];
 938   state_return = gst_element_set_state (pipeline, wanted_state);
 939
 940   while (state_i < G_N_ELEMENTS (wanted_states) && carry_on) {
 941     if (state_return == GST_STATE_CHANGE_SUCCESS && ready) {
 942       wanted_state = wanted_states[state_i++];
 943       fail_unless (gst_element_set_state (pipeline, wanted_state),
 944           GST_STATE_CHANGE_SUCCESS);
 945       GST_INFO ("Wanted state: %s", gst_element_state_get_name (wanted_state));
 946     }
 947
 948     message = gst_bus_poll (bus, GST_MESSAGE_ANY, GST_SECOND / 10);
 949     if (message) {
 950       switch (GST_MESSAGE_TYPE (message)) {
 951         case GST_MESSAGE_EOS:
 952         {
 953           /* we should check if we really finished here */
 954           GST_WARNING ("Got an EOS");
 955           carry_on = FALSE;
 956           break;
 957         }
 958         case GST_MESSAGE_STATE_CHANGED:
 959         {
 960           GstState new;
 961
 962           if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipeline)) {
 963             gst_message_parse_state_changed (message, NULL, &new, NULL);
 964
 965             if (new != wanted_state) {
 966               ready = FALSE;
 967               break;
 968             }
 969
 970             GST_DEBUG ("State %s reached",
 971                 gst_element_state_get_name (wanted_state));
 972             wanted_state = wanted_states[state_i++];
 973             GST_DEBUG ("Wanted state: %s",
 974                 gst_element_state_get_name (wanted_state));
 975             state_return = gst_element_set_state (pipeline, wanted_state);
 976             fail_unless (state_return == GST_STATE_CHANGE_SUCCESS ||
 977                 state_return == GST_STATE_CHANGE_ASYNC);
 978             ready = TRUE;
 979           }
 980
 981           break;
 982         }
 983         case GST_MESSAGE_ERROR:
 984           GST_ERROR ("Error on the bus: %" GST_PTR_FORMAT, message);
 985           carry_on = FALSE;
 986           break;
 987         default:
 988           break;
 989       }
 990       gst_message_unref (message);
 991     }
 992   }
 993
 994   gst_element_set_state (pipeline, GST_STATE_NULL);
 995   gst_object_unref (bus);
 996   gst_object_unref (pipeline);
 997 }
 998
 999 GST_END_TEST;
1000
1001 static Suite *
1002 gst_aggregator_suite (void)
1003 {
1004   Suite *suite;
1005   TCase *general;
1006
1007   gst_test_aggregator_plugin_register ();
1008
1009   suite = suite_create ("GstAggregator");
1010
1011   general = tcase_create ("general");
1012   suite_add_tcase (suite, general);
1013   tcase_add_test (general, test_aggregate);
1014   tcase_add_test (general, test_aggregate_eos);
1015   tcase_add_test (general, test_flushing_seek);
1016   tcase_add_test (general, test_infinite_seek);
1017   tcase_add_test (general, test_infinite_seek_50_src);
1018   tcase_add_test (general, test_linear_pipeline);
1019   tcase_add_test (general, test_two_src_pipeline);
1020   tcase_add_test (general, test_add_remove);
1021   tcase_add_test (general, test_change_state_intensive);
1022
1023   return suite;
1024 }
1025
1026 GST_CHECK_MAIN (gst_aggregator);