tests/check/elements/wavpackparse.c

   1 /* GStreamer
   2  *
   3  * unit test for wavpackparse
   4  *
   5  * Copyright (c) 2006 Sebastian Dröge <slomo@circular-chaos.org>
   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 #include <gst/check/gstcheck.h>
  24
  25 /* For ease of programming we use globals to keep refs for our floating
  26  * src and sink pads we create; otherwise we always have to do get_pad,
  27  * get_peer, and then remove references in every test function */
  28 static GstPad *mysrcpad, *mysinkpad;
  29
  30 /* Wavpack file with 2 frames of silence */
  31 guint8 test_file[] = {
  32   0x77, 0x76, 0x70, 0x6B, 0x62, 0x00, 0x00, 0x00,       /* first frame */
  33   0x04, 0x04, 0x00, 0x00, 0x00, 0xC8, 0x00, 0x00,       /* include RIFF header */
  34   0x00, 0x00, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00,
  35   0x05, 0x18, 0x80, 0x04, 0xFF, 0xAF, 0x80, 0x60,
  36   0x21, 0x16, 0x52, 0x49, 0x46, 0x46, 0x24, 0x90,
  37   0x01, 0x00, 0x57, 0x41, 0x56, 0x45, 0x66, 0x6D,
  38   0x74, 0x20, 0x10, 0x00, 0x00, 0x00, 0x01, 0x00,
  39   0x01, 0x00, 0x44, 0xAC, 0x00, 0x00, 0x88, 0x58,
  40   0x01, 0x00, 0x02, 0x00, 0x10, 0x00, 0x64, 0x61,
  41   0x74, 0x61, 0x00, 0x90, 0x01, 0x00, 0x02, 0x00,
  42   0x03, 0x00, 0x04, 0x00, 0x05, 0x03, 0x00, 0x00,
  43   0x00, 0x00, 0x00, 0x00, 0x65, 0x02, 0x00, 0x00,
  44   0x00, 0x00, 0x8A, 0x02, 0x00, 0x00, 0xFF, 0x7F,
  45   0x00, 0xE4,
  46   0x77, 0x76, 0x70, 0x6B, 0x2E, 0x00, 0x00, 0x00,       /* second frame */
  47   0x04, 0x04, 0x00, 0x00, 0xFF, 0xFF, 0xFF, 0xFF,
  48   0x00, 0x64, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00,
  49   0x05, 0x18, 0x80, 0x04, 0xFF, 0xAF, 0x80, 0x60,
  50   0x02, 0x00, 0x03, 0x00, 0x04, 0x00, 0x05, 0x03,
  51   0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x8A, 0x02,
  52   0x00, 0x00, 0xFF, 0x7F, 0x00, 0xE4,
  53 };
  54
  55 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
  56     GST_PAD_SINK,
  57     GST_PAD_ALWAYS,
  58     GST_STATIC_CAPS ("audio/x-wavpack, "
  59         "depth = (int) 16, "
  60         "channels = (int) 1, "
  61         "rate = (int) 44100, " "framed = (boolean) TRUE"));
  62
  63 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
  64     GST_PAD_SRC,
  65     GST_PAD_ALWAYS,
  66     GST_STATIC_CAPS ("audio/x-wavpack"));
  67
  68 static GstElement *
  69 setup_wavpackparse (void)
  70 {
  71   GstElement *wavpackparse;
  72
  73   GST_DEBUG ("setup_wavpackparse");
  74
  75   wavpackparse = gst_check_setup_element ("wavpackparse");
  76   mysrcpad = gst_check_setup_src_pad (wavpackparse, &srctemplate);
  77   mysinkpad = gst_check_setup_sink_pad (wavpackparse, &sinktemplate);
  78   gst_pad_set_active (mysrcpad, TRUE);
  79   gst_pad_set_active (mysinkpad, TRUE);
  80   gst_check_setup_events (mysrcpad, wavpackparse, NULL, GST_FORMAT_BYTES);
  81
  82   return wavpackparse;
  83 }
  84
  85 static void
  86 cleanup_wavpackparse (GstElement * wavpackparse)
  87 {
  88   GST_DEBUG ("cleanup_wavpackparse");
  89   gst_element_set_state (wavpackparse, GST_STATE_NULL);
  90
  91   gst_pad_set_active (mysrcpad, FALSE);
  92   gst_pad_set_active (mysinkpad, FALSE);
  93   gst_check_teardown_src_pad (wavpackparse);
  94   gst_check_teardown_sink_pad (wavpackparse);
  95   gst_check_teardown_element (wavpackparse);
  96 }
  97
  98 GST_START_TEST (test_parsing_valid_frames)
  99 {
 100   GstElement *wavpackparse;
 101   GstBuffer *inbuffer, *outbuffer;
 102   int i, num_buffers;
 103   GstFormat format = GST_FORMAT_TIME;
 104   gint64 pos;
 105
 106   wavpackparse = setup_wavpackparse ();
 107   fail_unless (gst_element_set_state (wavpackparse,
 108           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
 109       "could not set to playing");
 110
 111   inbuffer = gst_buffer_new_and_alloc (sizeof (test_file));
 112   gst_buffer_fill (inbuffer, 0, test_file, sizeof (test_file));
 113
 114   /* should decode the buffer without problems */
 115   fail_unless_equals_int (gst_pad_push (mysrcpad, inbuffer), GST_FLOW_OK);
 116
 117   /* inform of no further data */
 118   fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()));
 119
 120   num_buffers = g_list_length (buffers);
 121   /* should get 2 buffers, each one complete wavpack frame */
 122   fail_unless_equals_int (num_buffers, 2);
 123
 124   for (i = 0; i < num_buffers; ++i) {
 125     outbuffer = GST_BUFFER (buffers->data);
 126     fail_if (outbuffer == NULL);
 127
 128     fail_unless (gst_buffer_memcmp (outbuffer, 0, "wvpk", 4) == 0,
 129         "Buffer contains no Wavpack frame");
 130     fail_unless_equals_int (GST_BUFFER_DURATION (outbuffer), 580498866);
 131
 132     switch (i) {
 133       case 0:{
 134         fail_unless_equals_int (GST_BUFFER_TIMESTAMP (outbuffer), 0);
 135         break;
 136       }
 137       case 1:{
 138         fail_unless_equals_int (GST_BUFFER_TIMESTAMP (outbuffer), 580498866);
 139         break;
 140       }
 141     }
 142
 143     buffers = g_list_remove (buffers, outbuffer);
 144
 145     gst_buffer_unref (outbuffer);
 146     outbuffer = NULL;
 147   }
 148
 149   fail_unless (gst_element_query_position (wavpackparse, format, &pos),
 150       "Position query failed");
 151   fail_unless_equals_int64 (pos, 580498866 * 2);
 152   fail_unless (gst_element_query_duration (wavpackparse, format, NULL),
 153       "Duration query failed");
 154
 155   g_list_free (buffers);
 156   buffers = NULL;
 157
 158   cleanup_wavpackparse (wavpackparse);
 159 }
 160
 161 GST_END_TEST;
 162
 163 GST_START_TEST (test_parsing_invalid_first_header)
 164 {
 165   GstElement *wavpackparse;
 166   GstBuffer *inbuffer, *outbuffer;
 167   int i, num_buffers;
 168
 169   wavpackparse = setup_wavpackparse ();
 170   fail_unless (gst_element_set_state (wavpackparse,
 171           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
 172       "could not set to playing");
 173
 174   inbuffer = gst_buffer_new_and_alloc (sizeof (test_file));
 175   gst_buffer_fill (inbuffer, 0, test_file, sizeof (test_file));
 176   gst_buffer_memset (inbuffer, 0, 'k', 1);
 177
 178   /* should decode the buffer without problems */
 179   fail_unless_equals_int (gst_pad_push (mysrcpad, inbuffer), GST_FLOW_OK);
 180
 181   /* inform of no further data */
 182   fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()));
 183
 184   num_buffers = g_list_length (buffers);
 185
 186   /* should get 1 buffers, the second non-broken one */
 187   fail_unless_equals_int (num_buffers, 1);
 188
 189   for (i = 0; i < num_buffers; ++i) {
 190     outbuffer = GST_BUFFER (buffers->data);
 191     fail_if (outbuffer == NULL);
 192
 193     fail_unless (gst_buffer_memcmp (outbuffer, 0, "wvpk", 4) == 0,
 194         "Buffer contains no Wavpack frame");
 195     fail_unless_equals_int (GST_BUFFER_DURATION (outbuffer), 580498866);
 196
 197     switch (i) {
 198       case 0:{
 199         fail_unless_equals_int (GST_BUFFER_TIMESTAMP (outbuffer), 580498866);
 200         break;
 201       }
 202     }
 203
 204     buffers = g_list_remove (buffers, outbuffer);
 205
 206     gst_buffer_unref (outbuffer);
 207     outbuffer = NULL;
 208   }
 209
 210   g_list_free (buffers);
 211   buffers = NULL;
 212
 213   cleanup_wavpackparse (wavpackparse);
 214 }
 215
 216 GST_END_TEST;
 217
 218
 219 static Suite *
 220 wavpackparse_suite (void)
 221 {
 222   Suite *s = suite_create ("wavpackparse");
 223   TCase *tc_chain = tcase_create ("general");
 224
 225   suite_add_tcase (s, tc_chain);
 226   tcase_add_test (tc_chain, test_parsing_valid_frames);
 227   tcase_add_test (tc_chain, test_parsing_invalid_first_header);
 228
 229   return s;
 230 }
 231
 232 GST_CHECK_MAIN (wavpackparse);