tests/check/: Add unit tests for new parsers.

Original commit message from CVS:
* tests/check/Makefile.am:
* tests/check/elements/aacparse.c:
* tests/check/elements/amrparse.c:
Add unit tests for new parsers.

3 files changed, 1154 insertions, 0 deletions

diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am
index 0cd5dc49a..fd27ca281 100644
--- a/tests/check/Makefile.am
+++ b/tests/check/Makefile.am
@@ -87,6 +87,8 @@ check_PROGRAMS = \
 	$(check_ofa)        \
 	$(check_timidity)  \
 	$(check_x264enc)   \
+	elements/aacparse \
+	elements/amrparse \
 	elements/qtmux \
 	elements/selector \
 	elements/speexresample \
diff --git a/tests/check/elements/aacparse.c b/tests/check/elements/aacparse.c
new file mode 100644
index 000000000..f7de288bc
--- /dev/null
+++ b/tests/check/elements/aacparse.c
@@ -0,0 +1,498 @@
+/*
+ * GStreamer
+ *
+ * unit test for aacparse
+ *
+ * Copyright (C) 2008 Nokia Corporation. All rights reserved.
+ *
+ * Contact: Stefan Kost <stefan.kost@nokia.com>
+ *
+ * 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.
+ */
+
+#include <gst/check/gstcheck.h>
+#include "aacparse_data.h"
+
+#define SRC_CAPS_CDATA "audio/mpeg, framed=(boolean)false, codec_data=(buffer)1190"
+#define SRC_CAPS_TMPL  "audio/mpeg, framed=(boolean)false, mpegversion=(int){2,4}"
+
+#define SINK_CAPS       "audio/mpeg, framed=(boolean)true"
+#define SINK_CAPS_MPEG2 "audio/mpeg, framed=(boolean)true, mpegversion=2"
+#define SINK_CAPS_MPEG4 "audio/mpeg, framed=(boolean)true, mpegversion=4"
+#define SINK_CAPS_TMPL  "audio/mpeg, framed=(boolean)true, mpegversion=(int){2,4}"
+
+GList *buffers;
+GList *current_buf = NULL;
+
+GstPad *srcpad, *sinkpad;
+guint dataoffset = 0;
+GstClockTime ts_counter = 0;
+gint64 offset_counter = 0;
+guint buffer_counter = 0;
+
+static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SINK_CAPS_TMPL)
+    );
+
+static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SRC_CAPS_TMPL)
+    );
+
+typedef struct
+{
+  guint buffers_before_offset_skip;
+  guint offset_skip_amount;
+  const unsigned char *data_to_verify;
+  GstCaps *caps;
+} buffer_verify_data_s;
+
+/* takes a copy of the passed buffer data */
+GstBuffer *
+buffer_new (const unsigned char *buffer_data, guint size)
+{
+  GstBuffer *buffer;
+
+  buffer = gst_buffer_new_and_alloc (size);
+  if (buffer_data) {
+    memcpy (GST_BUFFER_DATA (buffer), buffer_data, size);
+  } else {
+    guint i;
+    /* Create a recognizable pattern (loop 0x00 -> 0xff) in the data block */
+    for (i = 0; i < size; i++) {
+      GST_BUFFER_DATA (buffer)[i] = i % 0x100;
+    }
+  }
+
+  gst_buffer_set_caps (buffer, GST_PAD_CAPS (srcpad));
+  GST_BUFFER_OFFSET (buffer) = dataoffset;
+  dataoffset += size;
+  return buffer;
+}
+
+
+/*
+ * Count buffer sizes together.
+ */
+static void
+buffer_count_size (void *buffer, void *user_data)
+{
+  guint *sum = (guint *) user_data;
+  *sum += GST_BUFFER_SIZE (buffer);
+}
+
+
+/*
+ * Verify that given buffer contains predefined ADTS frame.
+ */
+static void
+buffer_verify_adts (void *buffer, void *user_data)
+{
+  buffer_verify_data_s *vdata;
+
+  if (!user_data) {
+    return;
+  }
+
+  vdata = (buffer_verify_data_s *) user_data;
+
+  fail_unless (memcmp (GST_BUFFER_DATA (buffer), vdata->data_to_verify,
+          ADTS_FRAME_LEN) == 0);
+
+  fail_unless (GST_BUFFER_TIMESTAMP (buffer) == ts_counter);
+  fail_unless (GST_BUFFER_DURATION (buffer) != 0);
+
+  if (vdata->buffers_before_offset_skip) {
+    /* This is for skipping the garbage in some test cases */
+    if (buffer_counter == vdata->buffers_before_offset_skip) {
+      offset_counter += vdata->offset_skip_amount;
+    }
+  }
+  fail_unless (GST_BUFFER_OFFSET (buffer) == offset_counter);
+
+  if (vdata->caps) {
+    gchar *bcaps = gst_caps_to_string (GST_BUFFER_CAPS (buffer));
+    g_free (bcaps);
+
+    fail_unless (gst_caps_is_equal (GST_BUFFER_CAPS (buffer), vdata->caps));
+  }
+
+  ts_counter += GST_BUFFER_DURATION (buffer);
+  offset_counter += ADTS_FRAME_LEN;
+  buffer_counter++;
+}
+
+GstElement *
+setup_aacparse (const gchar * src_caps_str)
+{
+  GstElement *aacparse;
+  GstCaps *srccaps = NULL;
+  GstBus *bus;
+
+  if (src_caps_str) {
+    srccaps = gst_caps_from_string (src_caps_str);
+    fail_unless (srccaps != NULL);
+  }
+
+  aacparse = gst_check_setup_element ("aacparse");
+  srcpad = gst_check_setup_src_pad (aacparse, &srctemplate, srccaps);
+  sinkpad = gst_check_setup_sink_pad (aacparse, &sinktemplate, NULL);
+  gst_pad_set_active (srcpad, TRUE);
+  gst_pad_set_active (sinkpad, TRUE);
+
+  bus = gst_bus_new ();
+  gst_element_set_bus (aacparse, bus);
+
+  fail_unless (gst_element_set_state (aacparse,
+          GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE,
+      "could not set to playing");
+
+  if (srccaps) {
+    gst_caps_unref (srccaps);
+  }
+  ts_counter = offset_counter = buffer_counter = 0;
+  buffers = NULL;
+  return aacparse;
+}
+
+static void
+cleanup_aacparse (GstElement * aacparse)
+{
+  GstBus *bus;
+
+  /* Free parsed buffers */
+  gst_check_drop_buffers ();
+
+  bus = GST_ELEMENT_BUS (aacparse);
+  gst_bus_set_flushing (bus, TRUE);
+  gst_object_unref (bus);
+
+  gst_pad_set_active (srcpad, FALSE);
+  gst_pad_set_active (sinkpad, FALSE);
+  gst_check_teardown_src_pad (aacparse);
+  gst_check_teardown_sink_pad (aacparse);
+  gst_check_teardown_element (aacparse);
+}
+
+
+/*
+ * Test if the parser pushes data with ADIF header properly and detects the
+ * stream to MPEG4 properly.
+ */
+GST_START_TEST (test_parse_adif_normal)
+{
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  GstCaps *scaps, *sinkcaps;
+  guint datasum = 0;
+  guint i;
+
+  aacparse = setup_aacparse (NULL);
+
+  buffer = buffer_new (adif_header, ADIF_HEADER_LEN);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+
+  for (i = 0; i < 3; i++) {
+    buffer = buffer_new (NULL, 100);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* Calculate the outputted buffer sizes */
+  g_list_foreach (buffers, buffer_count_size, &datasum);
+
+  /* ADIF is not a framed format, and therefore we cannot expect the
+     same amount of output buffers as we pushed. However, all data should
+     still come through, including the header bytes */
+  fail_unless_equals_int (datasum, 3 * 100 + ADIF_HEADER_LEN);
+
+  /* Check that the negotiated caps are as expected */
+  /* For ADIF parser assumes that data is always version 4 */
+  scaps = gst_caps_from_string (SINK_CAPS_MPEG4);
+  sinkcaps = gst_pad_get_negotiated_caps (sinkpad);
+  fail_unless (gst_caps_is_equal (sinkcaps, scaps));
+  gst_caps_unref (sinkcaps);
+  gst_caps_unref (scaps);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if the parser pushes data with ADTS frames properly.
+ */
+GST_START_TEST (test_parse_adts_normal)
+{
+  buffer_verify_data_s vdata = { 0, 0, adts_frame_mpeg4, NULL };
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  guint i;
+
+  aacparse = setup_aacparse (NULL);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (adts_frame_mpeg4, ADTS_FRAME_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_adts, &vdata);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if ADTS parser drains its buffers properly. Even one single frame
+ * should be drained and pushed forward when EOS occurs. This single frame
+ * case is special, since normally the parser needs more data to be sure
+ * about stream format. But it should still push the frame forward in EOS.
+ */
+GST_START_TEST (test_parse_adts_drain_single)
+{
+  buffer_verify_data_s vdata = { 0, 0, adts_frame_mpeg4, NULL };
+  GstElement *aacparse;
+  GstBuffer *buffer;
+
+  aacparse = setup_aacparse (NULL);
+
+  buffer = buffer_new (adts_frame_mpeg4, ADTS_FRAME_LEN);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 1);
+  g_list_foreach (buffers, buffer_verify_adts, &vdata);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Make sure that parser does not drain garbage when EOS occurs.
+ */
+GST_START_TEST (test_parse_adts_drain_garbage)
+{
+  buffer_verify_data_s vdata = { 0, 0, adts_frame_mpeg4, NULL };
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  guint i;
+
+  aacparse = setup_aacparse (NULL);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (adts_frame_mpeg4, ADTS_FRAME_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  /* Push one garbage frame and then EOS */
+  buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_adts, &vdata);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if ADTS parser splits a buffer that contains two frames into two
+ * separate buffers properly.
+ */
+GST_START_TEST (test_parse_adts_split)
+{
+  buffer_verify_data_s vdata = { 0, 0, adts_frame_mpeg4, NULL };
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  guint i;
+
+  aacparse = setup_aacparse (NULL);
+
+  for (i = 0; i < 5; i++) {
+    buffer = buffer_new (adts_frame_mpeg4, ADTS_FRAME_LEN * 2);
+    memcpy (GST_BUFFER_DATA (buffer) + ADTS_FRAME_LEN,
+        adts_frame_mpeg4, ADTS_FRAME_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_adts, &vdata);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if the ADTS parser skips garbage between frames properly.
+ */
+GST_START_TEST (test_parse_adts_skip_garbage)
+{
+  buffer_verify_data_s vdata =
+      { 10, GARBAGE_FRAME_LEN, adts_frame_mpeg4, NULL };
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  guint i;
+
+  aacparse = setup_aacparse (NULL);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (adts_frame_mpeg4, ADTS_FRAME_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  /* push garbage */
+  buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (adts_frame_mpeg4, ADTS_FRAME_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 20);
+  g_list_foreach (buffers, buffer_verify_adts, &vdata);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if the src caps are set according to stream format (MPEG version).
+ */
+GST_START_TEST (test_parse_adts_detect_mpeg_version)
+{
+  buffer_verify_data_s vdata = { 0, 0, adts_frame_mpeg2, NULL };
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  GstCaps *sinkcaps;
+  guint i;
+
+  aacparse = setup_aacparse (NULL);
+
+  /* buffer_verify_adts will check if the caps are equal */
+  vdata.caps = gst_caps_from_string (SINK_CAPS_MPEG2);
+
+  for (i = 0; i < 10; i++) {
+    /* Push MPEG version 2 frames. */
+    buffer = buffer_new (adts_frame_mpeg2, ADTS_FRAME_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* Check that the negotiated caps are as expected */
+  sinkcaps = gst_pad_get_negotiated_caps (sinkpad);
+  fail_unless (gst_caps_is_equal (sinkcaps, vdata.caps));
+  gst_caps_unref (sinkcaps);
+
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_adts, &vdata);
+
+  gst_caps_unref (vdata.caps);
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if the parser handles raw stream and codec_data info properly.
+ */
+GST_START_TEST (test_parse_handle_codec_data)
+{
+  GstElement *aacparse;
+  GstBuffer *buffer;
+  GstCaps *scaps, *sinkcaps;
+  guint datasum = 0;
+  guint i;
+
+  aacparse = setup_aacparse (SRC_CAPS_CDATA);
+
+  for (i = 0; i < 10; i++) {
+    /* Push random data. It should get through since the parser should be
+       initialized because it got codec_data in the caps */
+    buffer = buffer_new (NULL, 100);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* Check that the negotiated caps are as expected */
+  /* When codec_data is present, parser assumes that data is version 4 */
+  scaps = gst_caps_from_string (SINK_CAPS_MPEG4);
+  sinkcaps = gst_pad_get_negotiated_caps (sinkpad);
+  fail_unless (gst_caps_is_equal (sinkcaps, scaps));
+  gst_caps_unref (sinkcaps);
+  gst_caps_unref (scaps);
+
+  g_list_foreach (buffers, buffer_count_size, &datasum);
+  fail_unless_equals_int (datasum, 10 * 100);
+
+  cleanup_aacparse (aacparse);
+}
+
+GST_END_TEST;
+
+
+static Suite *
+aacparse_suite ()
+{
+  Suite *s = suite_create ("aacparse");
+  TCase *tc_chain = tcase_create ("general");
+
+  suite_add_tcase (s, tc_chain);
+  /* ADIF tests */
+  tcase_add_test (tc_chain, test_parse_adif_normal);
+
+  /* ADTS tests */
+  tcase_add_test (tc_chain, test_parse_adts_normal);
+  tcase_add_test (tc_chain, test_parse_adts_drain_single);
+  tcase_add_test (tc_chain, test_parse_adts_drain_garbage);
+  tcase_add_test (tc_chain, test_parse_adts_split);
+  tcase_add_test (tc_chain, test_parse_adts_skip_garbage);
+  tcase_add_test (tc_chain, test_parse_adts_detect_mpeg_version);
+
+  /* Other tests */
+  tcase_add_test (tc_chain, test_parse_handle_codec_data);
+
+  return s;
+}
+
+
+/*
+ * TODO:
+ *   - Both push- and pull-modes need to be tested
+ *      * Pull-mode & EOS
+ */
+
+GST_CHECK_MAIN (aacparse);
diff --git a/tests/check/elements/amrparse.c b/tests/check/elements/amrparse.c
new file mode 100644
index 000000000..484fc4d70
--- /dev/null
+++ b/tests/check/elements/amrparse.c
@@ -0,0 +1,654 @@
+/*
+ * GStreamer
+ *
+ * unit test for amrparse
+ *
+ * Copyright (C) 2008 Nokia Corporation. All rights reserved.
+ *
+ * Contact: Stefan Kost <stefan.kost@nokia.com>
+ *
+ * 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.
+ */
+
+#include <gst/check/gstcheck.h>
+#include "amrparse_data.h"
+
+#define SRC_CAPS_NB  "audio/x-amr-nb-sh"
+#define SRC_CAPS_WB  "audio/x-amr-wb-sh"
+#define SRC_CAPS_ANY "ANY"
+
+#define SINK_CAPS_NB  "audio/AMR, rate=8000 , channels=1"
+#define SINK_CAPS_WB  "audio/AMR-WB, rate=16000 , channels=1"
+#define SINK_CAPS_ANY "ANY"
+
+#define AMR_FRAME_DURATION (GST_SECOND/50)
+
+GList *buffers;
+GList *current_buf = NULL;
+
+GstPad *srcpad, *sinkpad;
+guint dataoffset = 0;
+GstClockTime ts_counter = 0;
+gint64 offset_counter = 0;
+guint buffer_counter = 0;
+
+static GstStaticPadTemplate sinktemplate_nb = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SINK_CAPS_NB)
+    );
+
+static GstStaticPadTemplate sinktemplate_wb = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SINK_CAPS_WB)
+    );
+
+static GstStaticPadTemplate sinktemplate_any = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SINK_CAPS_ANY)
+    );
+
+static GstStaticPadTemplate srctemplate_nb = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SRC_CAPS_NB)
+    );
+
+static GstStaticPadTemplate srctemplate_wb = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SRC_CAPS_WB)
+    );
+
+static GstStaticPadTemplate srctemplate_any = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (SRC_CAPS_ANY)
+    );
+
+typedef struct
+{
+  guint buffers_before_offset_skip;
+  guint offset_skip_amount;
+} buffer_verify_data_s;
+
+/*
+ * Create a GstBuffer of the given data and set the caps, if not NULL.
+ */
+GstBuffer *
+buffer_new (const unsigned char *buffer_data, guint size,
+    const gchar * caps_str)
+{
+  GstBuffer *buffer;
+  GstCaps *caps;
+
+  buffer = gst_buffer_new_and_alloc (size);
+  memcpy (GST_BUFFER_DATA (buffer), buffer_data, size);
+  if (caps_str) {
+    caps = gst_caps_from_string (caps_str);
+    gst_buffer_set_caps (buffer, caps);
+    gst_caps_unref (caps);
+  }
+  GST_BUFFER_OFFSET (buffer) = dataoffset;
+  dataoffset += size;
+  return buffer;
+}
+
+
+/*
+ * Unrefs given buffer.
+ */
+static void
+buffer_unref (void *buffer, void *user_data)
+{
+  gst_buffer_unref (GST_BUFFER (buffer));
+}
+
+
+/*
+ * Verify that given buffer contains predefined AMR-NB frame.
+ */
+static void
+buffer_verify_nb (void *buffer, void *user_data)
+{
+  fail_unless (memcmp (GST_BUFFER_DATA (buffer), frame_data_nb,
+          FRAME_DATA_NB_LEN) == 0);
+  fail_unless (GST_BUFFER_TIMESTAMP (buffer) == ts_counter);
+  fail_unless (GST_BUFFER_DURATION (buffer) == AMR_FRAME_DURATION);
+  ts_counter += AMR_FRAME_DURATION;
+
+  if (user_data) {
+    buffer_verify_data_s *vdata = (buffer_verify_data_s *) user_data;
+
+    /* This is for skipping the garbage in some test cases */
+    if (buffer_counter == vdata->buffers_before_offset_skip) {
+      offset_counter += vdata->offset_skip_amount;
+    }
+  }
+  fail_unless (GST_BUFFER_OFFSET (buffer) == offset_counter);
+  offset_counter += FRAME_DATA_NB_LEN;
+  buffer_counter++;
+}
+
+
+/*
+ * Verify that given buffer contains predefined AMR-WB frame.
+ */
+static void
+buffer_verify_wb (void *buffer, void *user_data)
+{
+  fail_unless (memcmp (GST_BUFFER_DATA (buffer), frame_data_wb,
+          FRAME_DATA_WB_LEN) == 0);
+  fail_unless (GST_BUFFER_TIMESTAMP (buffer) == ts_counter);
+  fail_unless (GST_BUFFER_DURATION (buffer) == AMR_FRAME_DURATION);
+
+  if (user_data) {
+    buffer_verify_data_s *vdata = (buffer_verify_data_s *) user_data;
+
+    /* This is for skipping the garbage in some test cases */
+    if (buffer_counter == vdata->buffers_before_offset_skip) {
+      offset_counter += vdata->offset_skip_amount;
+    }
+  }
+  fail_unless (GST_BUFFER_OFFSET (buffer) == offset_counter);
+  offset_counter += FRAME_DATA_WB_LEN;
+  ts_counter += AMR_FRAME_DURATION;
+  buffer_counter++;
+}
+
+/*
+ * Create a parser and pads according to given templates.
+ */
+GstElement *
+setup_amrparse (GstStaticPadTemplate * srctemplate,
+    GstStaticPadTemplate * sinktemplate)
+{
+  GstElement *amrparse;
+  GstBus *bus;
+
+  GST_DEBUG ("setup_amrparse");
+  amrparse = gst_check_setup_element ("amrparse");
+  srcpad = gst_check_setup_src_pad (amrparse, srctemplate, NULL);
+  sinkpad = gst_check_setup_sink_pad (amrparse, sinktemplate, NULL);
+  gst_pad_set_active (srcpad, TRUE);
+  gst_pad_set_active (sinkpad, TRUE);
+
+  bus = gst_bus_new ();
+  gst_element_set_bus (amrparse, bus);
+
+  fail_unless (gst_element_set_state (amrparse,
+          GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE,
+      "could not set to playing");
+
+  ts_counter = offset_counter = buffer_counter = 0;
+  buffers = NULL;
+  return amrparse;
+}
+
+
+/*
+ * Delete parser and all related resources.
+ */
+static void
+cleanup_amrparse (GstElement * amrparse)
+{
+  GstBus *bus;
+
+  /* free parsed buffers */
+  g_list_foreach (buffers, buffer_unref, NULL);
+  g_list_free (buffers);
+  buffers = NULL;
+
+  bus = GST_ELEMENT_BUS (amrparse);
+  gst_bus_set_flushing (bus, TRUE);
+  gst_object_unref (bus);
+
+  GST_DEBUG ("cleanup_amrparse");
+  gst_pad_set_active (srcpad, FALSE);
+  gst_pad_set_active (sinkpad, FALSE);
+  gst_check_teardown_src_pad (amrparse);
+  gst_check_teardown_sink_pad (amrparse);
+  gst_check_teardown_element (amrparse);
+  srcpad = NULL;
+  sinkpad = NULL;
+}
+
+
+/*
+ * Test if NB parser manages to find all frames and pushes them forward.
+ */
+GST_START_TEST (test_parse_nb_normal)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
+
+  /* Push the header */
+  buffer = buffer_new (frame_hdr_nb, FRAME_HDR_NB_LEN, SRC_CAPS_NB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  offset_counter = FRAME_HDR_NB_LEN;
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_nb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if NB parser drains its buffers properly. Even one single buffer
+ * should be drained and pushed forward when EOS occurs. This single buffer
+ * case is special, since normally the parser needs more data to be sure
+ * about stream format. But it should still push the frame forward in EOS.
+ */
+GST_START_TEST (test_parse_nb_drain_single)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+
+  amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
+
+  buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 1);
+  g_list_foreach (buffers, buffer_verify_nb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Make sure that parser does not drain garbage when EOS occurs.
+ */
+GST_START_TEST (test_parse_nb_drain_garbage)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  /* Now push one garbage frame and then EOS */
+  buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_NB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* parser should have pushed only the valid frames */
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_nb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if NB parser splits a buffer that contains two frames into two
+ * separate buffers properly.
+ */
+GST_START_TEST (test_parse_nb_split)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
+
+  for (i = 0; i < 10; i++) {
+    /* Put two frames in one buffer */
+    buffer = buffer_new (frame_data_nb, 2 * FRAME_DATA_NB_LEN, SRC_CAPS_NB);
+    memcpy (GST_BUFFER_DATA (buffer) + FRAME_DATA_NB_LEN,
+        frame_data_nb, FRAME_DATA_NB_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 20);
+
+  /* Does output buffers contain correct frame data? */
+  g_list_foreach (buffers, buffer_verify_nb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if NB parser detects the format correctly.
+ */
+GST_START_TEST (test_parse_nb_detect_stream)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  GstCaps *caps, *mycaps;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_any, &sinktemplate_any);
+
+  /* Push the header */
+  buffer = buffer_new (frame_hdr_nb, FRAME_HDR_NB_LEN, NULL);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, NULL);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  caps = GST_PAD_CAPS (sinkpad);
+  mycaps = gst_caps_from_string (SINK_CAPS_NB);
+  fail_unless (gst_caps_is_equal (caps, mycaps));
+  gst_caps_unref (mycaps);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if NB parser skips garbage in the datastream correctly and still
+ * finds all correct frames.
+ */
+GST_START_TEST (test_parse_nb_skip_garbage)
+{
+  buffer_verify_data_s vdata = { 5, GARBAGE_FRAME_LEN };
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_nb, &sinktemplate_nb);
+
+  /* First push 5 healthy frames */
+  for (i = 0; i < 5; i++) {
+    buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  /* Then push some garbage */
+  buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_NB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+
+  /* Again, healthy frames */
+  for (i = 0; i < 5; i++) {
+    buffer = buffer_new (frame_data_nb, FRAME_DATA_NB_LEN, SRC_CAPS_NB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* Did it find all 10 healthy frames? */
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_nb, &vdata);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if WB parser manages to find all frames and pushes them forward.
+ */
+GST_START_TEST (test_parse_wb_normal)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
+
+  /* Push the header */
+  buffer = buffer_new (frame_hdr_wb, FRAME_HDR_WB_LEN, SRC_CAPS_WB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  offset_counter = FRAME_HDR_WB_LEN;
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_wb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if WB parser drains its buffers properly. Even one single buffer
+ * should be drained and pushed forward when EOS occurs. This single buffer
+ * case is special, since normally the parser needs more data to be sure
+ * about stream format. But it should still push the frame forward in EOS.
+ */
+GST_START_TEST (test_parse_wb_drain_single)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+
+  amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
+
+  buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 1);
+  g_list_foreach (buffers, buffer_verify_wb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Make sure that parser does not drain garbage when EOS occurs.
+ */
+GST_START_TEST (test_parse_wb_drain_garbage)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  /* Now push one garbage frame and then EOS */
+  buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_WB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* parser should have pushed only the valid frames */
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_wb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if WB parser splits a buffer that contains two frames into two
+ * separate buffers properly.
+ */
+GST_START_TEST (test_parse_wb_split)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
+
+  for (i = 0; i < 10; i++) {
+    /* Put two frames in one buffer */
+    buffer = buffer_new (frame_data_wb, 2 * FRAME_DATA_WB_LEN, SRC_CAPS_WB);
+    memcpy (GST_BUFFER_DATA (buffer) + FRAME_DATA_WB_LEN,
+        frame_data_wb, FRAME_DATA_WB_LEN);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  fail_unless_equals_int (g_list_length (buffers), 20);
+
+  /* Does output buffers contain correct frame data? */
+  g_list_foreach (buffers, buffer_verify_wb, NULL);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if WB parser detects the format correctly.
+ */
+GST_START_TEST (test_parse_wb_detect_stream)
+{
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  GstCaps *caps, *mycaps;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_any, &sinktemplate_any);
+
+  /* Push the header */
+  buffer = buffer_new (frame_hdr_wb, FRAME_HDR_WB_LEN, NULL);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+
+  for (i = 0; i < 10; i++) {
+    buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, NULL);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  caps = GST_PAD_CAPS (sinkpad);
+  mycaps = gst_caps_from_string (SINK_CAPS_WB);
+  fail_unless (gst_caps_is_equal (caps, mycaps));
+  gst_caps_unref (mycaps);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Test if WB parser skips garbage in the datastream correctly and still
+ * finds all correct frames.
+ */
+GST_START_TEST (test_parse_wb_skip_garbage)
+{
+  buffer_verify_data_s vdata = { 5, GARBAGE_FRAME_LEN };
+  GstElement *amrparse;
+  GstBuffer *buffer;
+  guint i;
+
+  amrparse = setup_amrparse (&srctemplate_wb, &sinktemplate_wb);
+
+  /* First push 5 healthy frames */
+  for (i = 0; i < 5; i++) {
+    buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  /* Then push some garbage */
+  buffer = buffer_new (garbage_frame, GARBAGE_FRAME_LEN, SRC_CAPS_WB);
+  fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+
+  /* Again, healthy frames */
+  for (i = 0; i < 5; i++) {
+    buffer = buffer_new (frame_data_wb, FRAME_DATA_WB_LEN, SRC_CAPS_WB);
+    fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK);
+  }
+
+  gst_pad_push_event (srcpad, gst_event_new_eos ());
+
+  /* Did it find all 10 healthy frames? */
+  fail_unless_equals_int (g_list_length (buffers), 10);
+  g_list_foreach (buffers, buffer_verify_wb, &vdata);
+
+  cleanup_amrparse (amrparse);
+}
+
+GST_END_TEST;
+
+
+/*
+ * Create test suite.
+ */
+static Suite *
+amrparse_suite ()
+{
+  Suite *s = suite_create ("amrparse");
+  TCase *tc_chain = tcase_create ("general");
+
+  suite_add_tcase (s, tc_chain);
+  /* AMR-NB tests */
+  tcase_add_test (tc_chain, test_parse_nb_normal);
+  tcase_add_test (tc_chain, test_parse_nb_drain_single);
+  tcase_add_test (tc_chain, test_parse_nb_drain_garbage);
+  tcase_add_test (tc_chain, test_parse_nb_split);
+  tcase_add_test (tc_chain, test_parse_nb_detect_stream);
+  tcase_add_test (tc_chain, test_parse_nb_skip_garbage);
+
+  /* AMR-WB tests */
+  tcase_add_test (tc_chain, test_parse_wb_normal);
+  tcase_add_test (tc_chain, test_parse_wb_drain_single);
+  tcase_add_test (tc_chain, test_parse_wb_drain_garbage);
+  tcase_add_test (tc_chain, test_parse_wb_split);
+  tcase_add_test (tc_chain, test_parse_wb_detect_stream);
+  tcase_add_test (tc_chain, test_parse_wb_skip_garbage);
+  return s;
+}
+
+/*
+ * TODO:
+ *   - Both push- and pull-modes need to be tested
+ *      * Pull-mode & EOS
+ */
+
+GST_CHECK_MAIN (amrparse);