check/elements/fakesrc.c

   1 /* GStreamer
   2  *
   3  * unit test for fakesrc
   4  *
   5  * Copyright (C) <2005> Thomas Vander Stichele <thomas at apestaart dot 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 GList *buffers = NULL;
  28 gboolean have_eos = FALSE;
  29
  30 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
  31     GST_PAD_SINK,
  32     GST_PAD_ALWAYS,
  33     GST_STATIC_CAPS_ANY);
  34
  35 GstFlowReturn
  36 chain_func (GstPad * pad, GstBuffer * buffer)
  37 {
  38   buffers = g_list_append (buffers, buffer);
  39
  40   return GST_FLOW_OK;
  41 }
  42
  43 gboolean
  44 event_func (GstPad * pad, GstEvent * event)
  45 {
  46   if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) {
  47     /* we take the lock here because it's good practice to so, even though
  48      * no buffers will be pushed anymore anyway */
  49     GST_STREAM_LOCK (pad);
  50     have_eos = TRUE;
  51     GST_STREAM_UNLOCK (pad);
  52     gst_event_unref (event);
  53     return TRUE;
  54   }
  55
  56   gst_event_unref (event);
  57   return FALSE;
  58 }
  59
  60 GstElement *
  61 setup_fakesrc ()
  62 {
  63   GstElement *src;
  64   GstPad *srcpad, *sinkpad;
  65
  66   src = gst_element_factory_make ("fakesrc", "src");
  67   fail_if (src == NULL, "Could not create a fakesrc");
  68
  69   sinkpad =
  70       gst_pad_new_from_template (gst_static_pad_template_get (&sinktemplate),
  71       "sink");
  72   fail_if (sinkpad == NULL, "Could not create a sinkpad");
  73
  74   srcpad = gst_element_get_pad (src, "src");
  75   fail_if (srcpad == NULL, "Could not get source pad from fakesrc");
  76   gst_pad_set_caps (sinkpad, NULL);
  77   gst_pad_set_chain_function (sinkpad, chain_func);
  78   gst_pad_set_event_function (sinkpad, event_func);
  79
  80   fail_unless (gst_pad_link (srcpad, sinkpad) == GST_PAD_LINK_OK,
  81       "Could not link source and sink pads");
  82   return src;
  83 }
  84
  85 void
  86 cleanup_fakesrc (GstElement * src)
  87 {
  88   GstPad *srcpad, *sinkpad;
  89
  90   fail_unless (gst_element_set_state (src, GST_STATE_NULL) == GST_STATE_SUCCESS,
  91       "could not set to null");
  92
  93   srcpad = gst_element_get_pad (src, "src");
  94   sinkpad = gst_pad_get_peer (srcpad);
  95
  96   ASSERT_OBJECT_REFCOUNT (src, "src", 1);
  97   gst_object_unref (src);
  98
  99   gst_pad_unlink (srcpad, sinkpad);
 100
 101   /* pad refs held by both creator and this function (through _get) */
 102   ASSERT_OBJECT_REFCOUNT (srcpad, "srcpad", 2);
 103   gst_object_unref (srcpad);
 104   gst_object_unref (srcpad);
 105
 106   ASSERT_OBJECT_REFCOUNT (sinkpad, "sinkpad", 2);
 107   gst_object_unref (sinkpad);
 108   gst_object_unref (sinkpad);
 109 }
 110
 111 GST_START_TEST (test_num_buffers)
 112 {
 113   GstElement *src;
 114
 115   src = setup_fakesrc ();
 116   g_object_set (G_OBJECT (src), "num-buffers", 3, NULL);
 117   fail_unless (gst_element_set_state (src,
 118           GST_STATE_PLAYING) == GST_STATE_SUCCESS, "could not set to playing");
 119
 120   while (!have_eos) {
 121     g_usleep (1000);
 122   }
 123
 124   fail_unless (g_list_length (buffers) == 3);
 125   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
 126   g_list_free (buffers);
 127
 128   /* cleanup */
 129   cleanup_fakesrc (src);
 130 }
 131
 132 GST_END_TEST;
 133
 134 GST_START_TEST (test_sizetype_empty)
 135 {
 136   GstElement *src;
 137   GList *l;
 138
 139   src = setup_fakesrc ();
 140
 141   g_object_set (G_OBJECT (src), "sizetype", 1, NULL);
 142   g_object_set (G_OBJECT (src), "num-buffers", 100, NULL);
 143
 144   fail_unless (gst_element_set_state (src,
 145           GST_STATE_PLAYING) == GST_STATE_SUCCESS, "could not set to playing");
 146
 147   while (!have_eos) {
 148     g_usleep (1000);
 149   }
 150
 151   fail_unless (g_list_length (buffers) == 100);
 152   l = buffers;
 153   while (l) {
 154     GstBuffer *buf = l->data;
 155
 156     fail_unless (GST_BUFFER_SIZE (buf) == 0);
 157     l = l->next;
 158   }
 159   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
 160   g_list_free (buffers);
 161
 162   /* cleanup */
 163   cleanup_fakesrc (src);
 164 }
 165
 166 GST_END_TEST;
 167
 168 GST_START_TEST (test_sizetype_fixed)
 169 {
 170   GstElement *src;
 171   GList *l;
 172
 173   src = setup_fakesrc ();
 174
 175   g_object_set (G_OBJECT (src), "sizetype", 2, NULL);
 176   g_object_set (G_OBJECT (src), "sizemax", 8192, NULL);
 177   g_object_set (G_OBJECT (src), "num-buffers", 100, NULL);
 178
 179   fail_unless (gst_element_set_state (src,
 180           GST_STATE_PLAYING) == GST_STATE_SUCCESS, "could not set to playing");
 181
 182   while (!have_eos) {
 183     g_usleep (1000);
 184   }
 185
 186   fail_unless (g_list_length (buffers) == 100);
 187   l = buffers;
 188   while (l) {
 189     GstBuffer *buf = l->data;
 190
 191     fail_unless (GST_BUFFER_SIZE (buf) == 8192);
 192     l = l->next;
 193   }
 194   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
 195   g_list_free (buffers);
 196
 197   /* cleanup */
 198   cleanup_fakesrc (src);
 199 }
 200
 201 GST_END_TEST;
 202
 203 GST_START_TEST (test_sizetype_random)
 204 {
 205   GstElement *src;
 206   GList *l;
 207
 208   src = setup_fakesrc ();
 209
 210   g_object_set (G_OBJECT (src), "sizetype", 3, NULL);
 211   g_object_set (G_OBJECT (src), "sizemin", 4096, NULL);
 212   g_object_set (G_OBJECT (src), "sizemax", 8192, NULL);
 213   g_object_set (G_OBJECT (src), "num-buffers", 100, NULL);
 214
 215   fail_unless (gst_element_set_state (src,
 216           GST_STATE_PLAYING) == GST_STATE_SUCCESS, "could not set to playing");
 217
 218   while (!have_eos) {
 219     g_usleep (1000);
 220   }
 221
 222   fail_unless (g_list_length (buffers) == 100);
 223   l = buffers;
 224   while (l) {
 225     GstBuffer *buf = l->data;
 226
 227     fail_if (GST_BUFFER_SIZE (buf) > 8192);
 228     fail_if (GST_BUFFER_SIZE (buf) < 4096);
 229     l = l->next;
 230   }
 231   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
 232   g_list_free (buffers);
 233
 234   /* cleanup */
 235   cleanup_fakesrc (src);
 236 }
 237
 238 GST_END_TEST;
 239
 240
 241 Suite *
 242 fakesrc_suite (void)
 243 {
 244   Suite *s = suite_create ("fakesrc");
 245   TCase *tc_chain = tcase_create ("general");
 246
 247   suite_add_tcase (s, tc_chain);
 248   tcase_add_test (tc_chain, test_num_buffers);
 249   tcase_add_test (tc_chain, test_sizetype_empty);
 250   tcase_add_test (tc_chain, test_sizetype_fixed);
 251   tcase_add_test (tc_chain, test_sizetype_random);
 252
 253   return s;
 254 }
 255
 256 int
 257 main (int argc, char **argv)
 258 {
 259   int nf;
 260
 261   Suite *s = fakesrc_suite ();
 262   SRunner *sr = srunner_create (s);
 263
 264   gst_check_init (&argc, &argv);
 265
 266   srunner_run_all (sr, CK_NORMAL);
 267   nf = srunner_ntests_failed (sr);
 268   srunner_free (sr);
 269
 270   return nf;
 271 }