tests/check/elements/wavpackenc.c

   1 /* GStreamer
   2  *
   3  * unit test for wavpackenc
   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., 59 Temple Place - Suite 330,
  20  * Boston, MA 02111-1307, USA.
  21  */
  22
  23 #include <unistd.h>
  24
  25 #include <gst/check/gstcheck.h>
  26
  27 /* For ease of programming we use globals to keep refs for our floating
  28  * src and sink pads we create; otherwise we always have to do get_pad,
  29  * get_peer, and then remove references in every test function */
  30 static GstPad *mysrcpad, *mysinkpad;
  31 static GstBus *bus;
  32
  33 #define RAW_CAPS_STRING "audio/x-raw-int, " \
  34                         "width = (int) 32, " \
  35                         "depth = (int) 16, " \
  36                         "channels = (int) 1, " \
  37                         "rate = (int) 44100, " \
  38                         "endianness = (int) BYTE_ORDER, " \
  39                         "signed = (boolean) true"
  40
  41 #define WAVPACK_CAPS_STRING "audio/x-wavpack, " \
  42                             "width = (int) 16, " \
  43                             "channels = (int) 1, " \
  44                             "rate = (int) 44100, " \
  45                             "framed = (boolean) true"
  46
  47 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
  48     GST_PAD_SINK,
  49     GST_PAD_ALWAYS,
  50     GST_STATIC_CAPS ("audio/x-wavpack, "
  51         "width = (int) 16, "
  52         "channels = (int) 1, "
  53         "rate = (int) 44100, " "framed = (boolean) true"));
  54
  55 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
  56     GST_PAD_SRC,
  57     GST_PAD_ALWAYS,
  58     GST_STATIC_CAPS ("audio/x-raw-int, "
  59         "width = (int) 32, "
  60         "depth = (int) 16, "
  61         "channels = (int) 1, "
  62         "rate = (int) 44100, "
  63         "endianness = (int) BYTE_ORDER, " "signed = (boolean) true"));
  64
  65 static GstElement *
  66 setup_wavpackenc (void)
  67 {
  68   GstElement *wavpackenc;
  69
  70   GST_DEBUG ("setup_wavpackenc");
  71   wavpackenc = gst_check_setup_element ("wavpackenc");
  72   mysrcpad = gst_check_setup_src_pad (wavpackenc, &srctemplate, NULL);
  73   mysinkpad = gst_check_setup_sink_pad (wavpackenc, &sinktemplate, NULL);
  74   gst_pad_set_active (mysrcpad, TRUE);
  75   gst_pad_set_active (mysinkpad, TRUE);
  76
  77   fail_unless (gst_element_set_state (wavpackenc,
  78           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
  79       "could not set to playing");
  80   bus = gst_bus_new ();
  81
  82   return wavpackenc;
  83 }
  84
  85 static void
  86 cleanup_wavpackenc (GstElement * wavpackenc)
  87 {
  88   GST_DEBUG ("cleanup_wavpackenc");
  89
  90   gst_bus_set_flushing (bus, TRUE);
  91   gst_element_set_bus (wavpackenc, NULL);
  92   gst_object_unref (GST_OBJECT (bus));
  93
  94   gst_element_set_state (wavpackenc, GST_STATE_NULL);
  95
  96   gst_pad_set_active (mysrcpad, FALSE);
  97   gst_pad_set_active (mysinkpad, FALSE);
  98   gst_check_teardown_src_pad (wavpackenc);
  99   gst_check_teardown_sink_pad (wavpackenc);
 100   gst_check_teardown_element (wavpackenc);
 101 }
 102
 103 GST_START_TEST (test_encode_silence)
 104 {
 105   GstElement *wavpackenc;
 106   GstBuffer *inbuffer, *outbuffer;
 107   GstCaps *caps;
 108   GstEvent *eos = gst_event_new_eos ();
 109   int i, num_buffers;
 110
 111   wavpackenc = setup_wavpackenc ();
 112
 113   inbuffer = gst_buffer_new_and_alloc (1000);
 114   for (i = 0; i < 1000; i++)
 115     GST_BUFFER_DATA (inbuffer)[i] = 0;
 116   caps = gst_caps_from_string (RAW_CAPS_STRING);
 117   gst_buffer_set_caps (inbuffer, caps);
 118   gst_caps_unref (caps);
 119   GST_BUFFER_TIMESTAMP (inbuffer) = 0;
 120   ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
 121   gst_buffer_ref (inbuffer);
 122
 123   gst_element_set_bus (wavpackenc, bus);
 124
 125   fail_unless_equals_int (gst_pad_push (mysrcpad, inbuffer), GST_FLOW_OK);
 126   ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
 127   gst_buffer_unref (inbuffer);
 128
 129   fail_if (gst_pad_push_event (mysrcpad, eos) != TRUE);
 130
 131   /* check first buffer */
 132   outbuffer = GST_BUFFER (buffers->data);
 133
 134   fail_if (outbuffer == NULL);
 135
 136   fail_unless_equals_int (GST_BUFFER_TIMESTAMP (outbuffer), 0);
 137   fail_unless_equals_int (GST_BUFFER_OFFSET (outbuffer), 0);
 138   fail_unless_equals_int (GST_BUFFER_DURATION (outbuffer), 5668934);
 139   fail_unless_equals_int (GST_BUFFER_OFFSET_END (outbuffer), 250);
 140
 141   fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), "wvpk", 4) == 0,
 142       "Failed to encode to valid Wavpack frames");
 143
 144   /* free all buffers */
 145   num_buffers = g_list_length (buffers);
 146
 147   for (i = 0; i < num_buffers; ++i) {
 148     outbuffer = GST_BUFFER (buffers->data);
 149     fail_if (outbuffer == NULL);
 150
 151     buffers = g_list_remove (buffers, outbuffer);
 152
 153     gst_buffer_unref (outbuffer);
 154     outbuffer = NULL;
 155   }
 156
 157   g_list_free (buffers);
 158   buffers = NULL;
 159
 160   cleanup_wavpackenc (wavpackenc);
 161 }
 162
 163 GST_END_TEST;
 164
 165 static Suite *
 166 wavpackenc_suite (void)
 167 {
 168   Suite *s = suite_create ("wavpackenc");
 169   TCase *tc_chain = tcase_create ("general");
 170
 171   suite_add_tcase (s, tc_chain);
 172   tcase_add_test (tc_chain, test_encode_silence);
 173
 174   return s;
 175 }
 176
 177 int
 178 main (int argc, char **argv)
 179 {
 180   int nf;
 181
 182   Suite *s = wavpackenc_suite ();
 183   SRunner *sr = srunner_create (s);
 184
 185   gst_check_init (&argc, &argv);
 186
 187   srunner_run_all (sr, CK_NORMAL);
 188   nf = srunner_ntests_failed (sr);
 189   srunner_free (sr);
 190
 191   return nf;
 192 }