tests/check/elements/amrparse.c

   1 /*
   2  * GStreamer
   3  *
   4  * unit test for amrparse
   5  *
   6  * Copyright (C) 2008 Nokia Corporation. All rights reserved.
   7  *
   8  * Contact: Stefan Kost <stefan.kost@nokia.com>
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Library General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Library General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Library General Public
  21  * License along with this library; if not, write to the
  22  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  23  * Boston, MA 02111-1307, USA.
  24  */
  25
  26 #include <gst/check/gstcheck.h>
  27 #include "amrparse_data.h"
  28
  29 #define SRC_CAPS_NB  "audio/x-amr-nb-sh"
  30 #define SRC_CAPS_WB  "audio/x-amr-wb-sh"
  31 #define SRC_CAPS_ANY "ANY"
  32
  33 #define SINK_CAPS_NB  "audio/AMR, rate=8000 , channels=1"
  34 #define SINK_CAPS_WB  "audio/AMR-WB, rate=16000 , channels=1"
  35 #define SINK_CAPS_ANY "ANY"
  36
  37 #define AMR_FRAME_DURATION (GST_SECOND/50)
  38
  39 GList *buffers;
  40 GList *current_buf = NULL;
  41
  42 GstPad *srcpad, *sinkpad;
  43 guint dataoffset = 0;
  44 GstClockTime ts_counter = 0;
  45 gint64 offset_counter = 0;
  46 guint buffer_counter = 0;
  47
  48 static GstStaticPadTemplate sinktemplate_nb = GST_STATIC_PAD_TEMPLATE ("sink",
  49     GST_PAD_SINK,
  50     GST_PAD_ALWAYS,
  51     GST_STATIC_CAPS (SINK_CAPS_NB)
  52     );
  53
  54 static GstStaticPadTemplate sinktemplate_wb = GST_STATIC_PAD_TEMPLATE ("sink",
  55     GST_PAD_SINK,
  56     GST_PAD_ALWAYS,
  57     GST_STATIC_CAPS (SINK_CAPS_WB)
  58     );
  59
  60 static GstStaticPadTemplate sinktemplate_any = GST_STATIC_PAD_TEMPLATE ("sink",
  61     GST_PAD_SINK,
  62     GST_PAD_ALWAYS,
  63     GST_STATIC_CAPS (SINK_CAPS_ANY)
  64     );
  65
  66 static GstStaticPadTemplate srctemplate_nb = GST_STATIC_PAD_TEMPLATE ("src",
  67     GST_PAD_SRC,
  68     GST_PAD_ALWAYS,
  69     GST_STATIC_CAPS (SRC_CAPS_NB)
  70     );
  71
  72 static GstStaticPadTemplate srctemplate_wb = GST_STATIC_PAD_TEMPLATE ("src",
  73     GST_PAD_SRC,
  74     GST_PAD_ALWAYS,
  75     GST_STATIC_CAPS (SRC_CAPS_WB)
  76     );
  77
  78 static GstStaticPadTemplate srctemplate_any = GST_STATIC_PAD_TEMPLATE ("src",
  79     GST_PAD_SRC,
  80     GST_PAD_ALWAYS,
  81     GST_STATIC_CAPS (SRC_CAPS_ANY)
  82     );
  83
  84 typedef struct
  85 {
  86   guint buffers_before_offset_skip;
  87   guint offset_skip_amount;
  88 } buffer_verify_data_s;
  89
  90 /*
  91  * Create a GstBuffer of the given data and set the caps, if not NULL.
  92  */
  93 static GstBuffer *
  94 buffer_new (const unsigned char *buffer_data, guint size,
  95     const gchar * caps_str)
  96 {
  97   GstBuffer *buffer;
  98   GstCaps *caps;
  99
 100   buffer = gst_buffer_new_and_alloc (size);
 101   memcpy (GST_BUFFER_DATA (buffer), buffer_data, size);
 102   if (caps_str) {
 103     caps = gst_caps_from_string (caps_str);
 104     gst_buffer_set_caps (buffer, caps);
 105     gst_caps_unref (caps);
 106   }
 107   GST_BUFFER_OFFSET (buffer) = dataoffset;
 108   dataoffset += size;
 109   return buffer;
 110 }
 111
 112
 113 /*
 114  * Unrefs given buffer.
 115  */
 116 static void
 117 buffer_unref (void *buffer, void *user_data)
 118 {
 119   gst_buffer_unref (GST_BUFFER (buffer));
 120 }
 121
 122
 123 /*
 124  * Verify that given buffer contains predefined AMR-NB frame.
 125  */
 126 static void
 127 buffer_verify_nb (void *buffer, void *user_data)
 128 {
 129   fail_unless (memcmp (GST_BUFFER_DATA (buffer), frame_data_nb,
 130           FRAME_DATA_NB_LEN) == 0);
 131   fail_unless (GST_BUFFER_TIMESTAMP (buffer) == ts_counter);
 132   fail_unless (GST_BUFFER_DURATION (buffer) == AMR_FRAME_DURATION);
 133   ts_counter += AMR_FRAME_DURATION;
 134
 135   if (user_data) {
 136     buffer_verify_data_s *vdata = (buffer_verify_data_s *) user_data;
 137
 138     /* This is for skipping the garbage in some test cases */
 139     if (buffer_counter == vdata->buffers_before_offset_skip) {
 140       offset_counter += vdata->offset_skip_amount;
 141     }
 142   }
 143   fail_unless (GST_BUFFER_OFFSET (buffer) == offset_counter);
 144   offset_counter += FRAME_DATA_NB_LEN;
 145   buffer_counter++;
 146 }
 147
 148
 149 /*
 150  * Verify that given buffer contains predefined AMR-WB frame.
 151  */
 152 static void
 153 buffer_verify_wb (void *buffer, void *user_data)
 154 {
 155   fail_unless (memcmp (GST_BUFFER_DATA (buffer), frame_data_wb,
 156           FRAME_DATA_WB_LEN) == 0);
 157   fail_unless (GST_BUFFER_TIMESTAMP (buffer) == ts_counter);
 158   fail_unless (GST_BUFFER_DURATION (buffer) == AMR_FRAME_DURATION);
 159
 160   if (user_data) {
 161     buffer_verify_data_s *vdata = (buffer_verify_data_s *) user_data;
 162
 163     /* This is for skipping the garbage in some test cases */
 164     if (buffer_counter == vdata->buffers_before_offset_skip) {
 165       offset_counter += vdata->offset_skip_amount;
 166     }
 167   }
 168   fail_unless (GST_BUFFER_OFFSET (buffer) == offset_counter);
 169   offset_counter += FRAME_DATA_WB_LEN;
 170   ts_counter += AMR_FRAME_DURATION;
 171   buffer_counter++;
 172 }
 173
 174 /*
 175  * Create a parser and pads according to given templates.
 176  */
 177 static GstElement *
 178 setup_amrparse (GstStaticPadTemplate * srctemplate,
 179     GstStaticPadTemplate * sinktemplate)
 180 {
 181   GstElement *amrparse;
 182   GstBus *bus;
 183
 184   GST_DEBUG ("setup_amrparse");
 185   amrparse = gst_check_setup_element ("amrparse");
 186   srcpad = gst_check_setup_src_pad (amrparse, srctemplate, NULL);
 187   sinkpad = gst_check_setup_sink_pad (amrparse, sinktemplate, NULL);
 188   gst_pad_set_active (srcpad, TRUE);
 189   gst_pad_set_active (sinkpad, TRUE);
 190
 191   bus = gst_bus_new ();
 192   gst_element_set_bus (amrparse, bus);
 193
 194   fail_unless (gst_element_set_state (amrparse,
 195           GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE,
 196       "could not set to playing");
 197
 198   ts_counter = offset_counter = buffer_counter = 0;
 199   buffers = NULL;
 200   return amrparse;
 201 }
 202
 203
 204 /*
 205  * Delete parser and all related resources.
 206  */
 207 static void
 208 cleanup_amrparse (GstElement * amrparse)
 209 {
 210   GstBus *bus;
 211
 212   /* free parsed buffers */
 213   g_list_foreach (buffers, buffer_unref, NULL);
 214   g_list_free (buffers);
 215   buffers = NULL;
 216
 217   bus = GST_ELEMENT_BUS (amrparse);
 218   gst_bus_set_flushing (bus, TRUE);
 219   gst_object_unref (bus);
 220
 221   GST_DEBUG ("cleanup_amrparse");
 222   gst_pad_set_active (srcpad, FALSE);
 223   gst_pad_set_active (sinkpad, FALSE);
 224   gst_check_teardown_src_pad (amrparse);
 225   gst_check_teardown_sink_pad (amrparse);
 226   gst_check_teardown_element (amrparse);
 227   srcpad = NULL;
 228   sinkpad = NULL;
 229 }
 230
 231
 232 /*
 233  * Test if NB parser manages to find all frames and pushes them forward.
 234  */
 235 GST_START_TEST (test_parse_nb_normal)
 236 {
 237   GstElement *amrparse;
 238   GstBuffer *buffer;
 239   guint i;
 240
 241   amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
 242
 243   /* Push the header */
 244   buffer = buffer_new (frame_hdr_nb, FRAME_HDR_NB_LEN, SRC_CAPS_NB);
 245   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 246   offset_counter = FRAME_HDR_NB_LEN;
 247
 248   for (i = 0; i < 10; i++) {
 249     buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
 250     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 251   }
 252   gst_pad_push_event (srcpad, gst_event_new_eos ());
 253
 254   fail_unless_equals_int (g_list_length (buffers), 10);
 255   g_list_foreach (buffers, buffer_verify_nb, NULL);
 256
 257   cleanup_amrparse (amrparse);
 258 }
 259
 260 GST_END_TEST;
 261
 262
 263 /*
 264  * Test if NB parser drains its buffers properly. Even one single buffer
 265  * should be drained and pushed forward when EOS occurs. This single buffer
 266  * case is special, since normally the parser needs more data to be sure
 267  * about stream format. But it should still push the frame forward in EOS.
 268  */
 269 GST_START_TEST (test_parse_nb_drain_single)
 270 {
 271   GstElement *amrparse;
 272   GstBuffer *buffer;
 273
 274   amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
 275
 276   buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
 277   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 278   gst_pad_push_event (srcpad, gst_event_new_eos ());
 279
 280   fail_unless_equals_int (g_list_length (buffers), 1);
 281   g_list_foreach (buffers, buffer_verify_nb, NULL);
 282
 283   cleanup_amrparse (amrparse);
 284 }
 285
 286 GST_END_TEST;
 287
 288
 289 /*
 290  * Make sure that parser does not drain garbage when EOS occurs.
 291  */
 292 GST_START_TEST (test_parse_nb_drain_garbage)
 293 {
 294   GstElement *amrparse;
 295   GstBuffer *buffer;
 296   guint i;
 297
 298   amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
 299
 300   for (i = 0; i < 10; i++) {
 301     buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
 302     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 303   }
 304
 305   /* Now push one garbage frame and then EOS */
 306   buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_NB);
 307   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 308   gst_pad_push_event (srcpad, gst_event_new_eos ());
 309
 310   /* parser should have pushed only the valid frames */
 311   fail_unless_equals_int (g_list_length (buffers), 10);
 312   g_list_foreach (buffers, buffer_verify_nb, NULL);
 313
 314   cleanup_amrparse (amrparse);
 315 }
 316
 317 GST_END_TEST;
 318
 319
 320 /*
 321  * Test if NB parser splits a buffer that contains two frames into two
 322  * separate buffers properly.
 323  */
 324 GST_START_TEST (test_parse_nb_split)
 325 {
 326   GstElement *amrparse;
 327   GstBuffer *buffer;
 328   guint i;
 329
 330   amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
 331
 332   for (i = 0; i < 10; i++) {
 333     /* Put two frames in one buffer */
 334     buffer = buffer_new (frame_data_nb, 2 * FRAME_DATA_NB_LEN, SRC_CAPS_NB);
 335     memcpy (GST_BUFFER_DATA (buffer) + FRAME_DATA_NB_LEN,
 336         frame_data_nb, FRAME_DATA_NB_LEN);
 337     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 338   }
 339   gst_pad_push_event (srcpad, gst_event_new_eos ());
 340
 341   fail_unless_equals_int (g_list_length (buffers), 20);
 342
 343   /* Does output buffers contain correct frame data? */
 344   g_list_foreach (buffers, buffer_verify_nb, NULL);
 345
 346   cleanup_amrparse (amrparse);
 347 }
 348
 349 GST_END_TEST;
 350
 351
 352 /*
 353  * Test if NB parser detects the format correctly.
 354  */
 355 GST_START_TEST (test_parse_nb_detect_stream)
 356 {
 357   GstElement *amrparse;
 358   GstBuffer *buffer;
 359   GstCaps *caps, *mycaps;
 360   guint i;
 361
 362   amrparse = setup_amrparse (&srctemplate_any, &sinktemplate_any);
 363
 364   /* Push the header */
 365   buffer = buffer_new (frame_hdr_nb, FRAME_HDR_NB_LEN, NULL);
 366   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 367
 368   for (i = 0; i < 10; i++) {
 369     buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, NULL);
 370     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 371   }
 372   gst_pad_push_event (srcpad, gst_event_new_eos ());
 373
 374   caps = GST_PAD_CAPS (sinkpad);
 375   mycaps = gst_caps_from_string (SINK_CAPS_NB);
 376   fail_unless (gst_caps_is_equal (caps, mycaps));
 377   gst_caps_unref (mycaps);
 378
 379   cleanup_amrparse (amrparse);
 380 }
 381
 382 GST_END_TEST;
 383
 384
 385 /*
 386  * Test if NB parser skips garbage in the datastream correctly and still
 387  * finds all correct frames.
 388  */
 389 GST_START_TEST (test_parse_nb_skip_garbage)
 390 {
 391   buffer_verify_data_s vdata = { 5, GARBAGE_FRAME_LEN };
 392   GstElement *amrparse;
 393   GstBuffer *buffer;
 394   guint i;
 395
 396   amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
 397
 398   /* First push 5 healthy frames */
 399   for (i = 0; i < 5; i++) {
 400     buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
 401     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 402   }
 403
 404   /* Then push some garbage */
 405   buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_NB);
 406   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 407
 408   /* Again, healthy frames */
 409   for (i = 0; i < 5; i++) {
 410     buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
 411     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 412   }
 413
 414   gst_pad_push_event (srcpad, gst_event_new_eos ());
 415
 416   /* Did it find all 10 healthy frames? */
 417   fail_unless_equals_int (g_list_length (buffers), 10);
 418   g_list_foreach (buffers, buffer_verify_nb, &vdata);
 419
 420   cleanup_amrparse (amrparse);
 421 }
 422
 423 GST_END_TEST;
 424
 425
 426 /*
 427  * Test if WB parser manages to find all frames and pushes them forward.
 428  */
 429 GST_START_TEST (test_parse_wb_normal)
 430 {
 431   GstElement *amrparse;
 432   GstBuffer *buffer;
 433   guint i;
 434
 435   amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
 436
 437   /* Push the header */
 438   buffer = buffer_new (frame_hdr_wb, FRAME_HDR_WB_LEN, SRC_CAPS_WB);
 439   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 440   offset_counter = FRAME_HDR_WB_LEN;
 441
 442   for (i = 0; i < 10; i++) {
 443     buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
 444     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 445   }
 446   gst_pad_push_event (srcpad, gst_event_new_eos ());
 447
 448   fail_unless_equals_int (g_list_length (buffers), 10);
 449   g_list_foreach (buffers, buffer_verify_wb, NULL);
 450
 451   cleanup_amrparse (amrparse);
 452 }
 453
 454 GST_END_TEST;
 455
 456
 457 /*
 458  * Test if WB parser drains its buffers properly. Even one single buffer
 459  * should be drained and pushed forward when EOS occurs. This single buffer
 460  * case is special, since normally the parser needs more data to be sure
 461  * about stream format. But it should still push the frame forward in EOS.
 462  */
 463 GST_START_TEST (test_parse_wb_drain_single)
 464 {
 465   GstElement *amrparse;
 466   GstBuffer *buffer;
 467
 468   amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
 469
 470   buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
 471   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 472   gst_pad_push_event (srcpad, gst_event_new_eos ());
 473
 474   fail_unless_equals_int (g_list_length (buffers), 1);
 475   g_list_foreach (buffers, buffer_verify_wb, NULL);
 476
 477   cleanup_amrparse (amrparse);
 478 }
 479
 480 GST_END_TEST;
 481
 482
 483 /*
 484  * Make sure that parser does not drain garbage when EOS occurs.
 485  */
 486 GST_START_TEST (test_parse_wb_drain_garbage)
 487 {
 488   GstElement *amrparse;
 489   GstBuffer *buffer;
 490   guint i;
 491
 492   amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
 493
 494   for (i = 0; i < 10; i++) {
 495     buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
 496     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 497   }
 498
 499   /* Now push one garbage frame and then EOS */
 500   buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_WB);
 501   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 502   gst_pad_push_event (srcpad, gst_event_new_eos ());
 503
 504   /* parser should have pushed only the valid frames */
 505   fail_unless_equals_int (g_list_length (buffers), 10);
 506   g_list_foreach (buffers, buffer_verify_wb, NULL);
 507
 508   cleanup_amrparse (amrparse);
 509 }
 510
 511 GST_END_TEST;
 512
 513
 514 /*
 515  * Test if WB parser splits a buffer that contains two frames into two
 516  * separate buffers properly.
 517  */
 518 GST_START_TEST (test_parse_wb_split)
 519 {
 520   GstElement *amrparse;
 521   GstBuffer *buffer;
 522   guint i;
 523
 524   amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
 525
 526   for (i = 0; i < 10; i++) {
 527     /* Put two frames in one buffer */
 528     buffer = buffer_new (frame_data_wb, 2 * FRAME_DATA_WB_LEN, SRC_CAPS_WB);
 529     memcpy (GST_BUFFER_DATA (buffer) + FRAME_DATA_WB_LEN,
 530         frame_data_wb, FRAME_DATA_WB_LEN);
 531     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 532   }
 533   gst_pad_push_event (srcpad, gst_event_new_eos ());
 534
 535   fail_unless_equals_int (g_list_length (buffers), 20);
 536
 537   /* Does output buffers contain correct frame data? */
 538   g_list_foreach (buffers, buffer_verify_wb, NULL);
 539
 540   cleanup_amrparse (amrparse);
 541 }
 542
 543 GST_END_TEST;
 544
 545
 546 /*
 547  * Test if WB parser detects the format correctly.
 548  */
 549 GST_START_TEST (test_parse_wb_detect_stream)
 550 {
 551   GstElement *amrparse;
 552   GstBuffer *buffer;
 553   GstCaps *caps, *mycaps;
 554   guint i;
 555
 556   amrparse = setup_amrparse (&srctemplate_any, &sinktemplate_any);
 557
 558   /* Push the header */
 559   buffer = buffer_new (frame_hdr_wb, FRAME_HDR_WB_LEN, NULL);
 560   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 561
 562   for (i = 0; i < 10; i++) {
 563     buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, NULL);
 564     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 565   }
 566   gst_pad_push_event (srcpad, gst_event_new_eos ());
 567
 568   caps = GST_PAD_CAPS (sinkpad);
 569   mycaps = gst_caps_from_string (SINK_CAPS_WB);
 570   fail_unless (gst_caps_is_equal (caps, mycaps));
 571   gst_caps_unref (mycaps);
 572
 573   cleanup_amrparse (amrparse);
 574 }
 575
 576 GST_END_TEST;
 577
 578
 579 /*
 580  * Test if WB parser skips garbage in the datastream correctly and still
 581  * finds all correct frames.
 582  */
 583 GST_START_TEST (test_parse_wb_skip_garbage)
 584 {
 585   buffer_verify_data_s vdata = { 5, GARBAGE_FRAME_LEN };
 586   GstElement *amrparse;
 587   GstBuffer *buffer;
 588   guint i;
 589
 590   amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
 591
 592   /* First push 5 healthy frames */
 593   for (i = 0; i < 5; i++) {
 594     buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
 595     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 596   }
 597
 598   /* Then push some garbage */
 599   buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_WB);
 600   fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 601
 602   /* Again, healthy frames */
 603   for (i = 0; i < 5; i++) {
 604     buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
 605     fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
 606   }
 607
 608   gst_pad_push_event (srcpad, gst_event_new_eos ());
 609
 610   /* Did it find all 10 healthy frames? */
 611   fail_unless_equals_int (g_list_length (buffers), 10);
 612   g_list_foreach (buffers, buffer_verify_wb, &vdata);
 613
 614   cleanup_amrparse (amrparse);
 615 }
 616
 617 GST_END_TEST;
 618
 619
 620 /*
 621  * Create test suite.
 622  */
 623 static Suite *
 624 amrparse_suite (void)
 625 {
 626   Suite *s = suite_create ("amrparse");
 627   TCase *tc_chain = tcase_create ("general");
 628
 629   suite_add_tcase (s, tc_chain);
 630   /* AMR-NB tests */
 631   tcase_add_test (tc_chain, test_parse_nb_normal);
 632   tcase_add_test (tc_chain, test_parse_nb_drain_single);
 633   tcase_add_test (tc_chain, test_parse_nb_drain_garbage);
 634   tcase_add_test (tc_chain, test_parse_nb_split);
 635   tcase_add_test (tc_chain, test_parse_nb_detect_stream);
 636   tcase_add_test (tc_chain, test_parse_nb_skip_garbage);
 637
 638   /* AMR-WB tests */
 639   tcase_add_test (tc_chain, test_parse_wb_normal);
 640   tcase_add_test (tc_chain, test_parse_wb_drain_single);
 641   tcase_add_test (tc_chain, test_parse_wb_drain_garbage);
 642   tcase_add_test (tc_chain, test_parse_wb_split);
 643   tcase_add_test (tc_chain, test_parse_wb_detect_stream);
 644   tcase_add_test (tc_chain, test_parse_wb_skip_garbage);
 645   return s;
 646 }
 647
 648 /*
 649  * TODO:
 650  *   - Both push- and pull-modes need to be tested
 651  *      * Pull-mode & EOS
 652  */
 653
 654 GST_CHECK_MAIN (amrparse);