tests/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 static gboolean have_eos = FALSE;
  28
  29 static GstPad *mysinkpad;
  30
  31 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
  32     GST_PAD_SINK,
  33     GST_PAD_ALWAYS,
  34     GST_STATIC_CAPS_ANY);
  35
  36 static gboolean
  37 event_func (GstPad * pad, GstEvent * event)
  38 {
  39   if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) {
  40     have_eos = TRUE;
  41     gst_event_unref (event);
  42     return TRUE;
  43   }
  44
  45   gst_event_unref (event);
  46   return FALSE;
  47 }
  48
  49 static GstElement *
  50 setup_fakesrc (void)
  51 {
  52   GstElement *fakesrc;
  53
  54   GST_DEBUG ("setup_fakesrc");
  55   fakesrc = gst_check_setup_element ("fakesrc");
  56   mysinkpad = gst_check_setup_sink_pad (fakesrc, &sinktemplate, NULL);
  57   gst_pad_set_event_function (mysinkpad, event_func);
  58   gst_pad_set_active (mysinkpad, TRUE);
  59   return fakesrc;
  60 }
  61
  62 static void
  63 cleanup_fakesrc (GstElement * fakesrc)
  64 {
  65   gst_pad_set_active (mysinkpad, FALSE);
  66   gst_check_teardown_sink_pad (fakesrc);
  67   gst_check_teardown_element (fakesrc);
  68 }
  69
  70 GST_START_TEST (test_num_buffers)
  71 {
  72   GstElement *src;
  73
  74   src = setup_fakesrc ();
  75   g_object_set (G_OBJECT (src), "num-buffers", 3, NULL);
  76   fail_unless (gst_element_set_state (src,
  77           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
  78       "could not set to playing");
  79
  80   while (!have_eos) {
  81     g_usleep (1000);
  82   }
  83
  84   fail_unless (g_list_length (buffers) == 3);
  85   gst_check_drop_buffers ();
  86
  87   fail_unless (gst_element_set_state (src,
  88           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
  89
  90   /* cleanup */
  91   cleanup_fakesrc (src);
  92 }
  93
  94 GST_END_TEST;
  95
  96 GST_START_TEST (test_sizetype_empty)
  97 {
  98   GstElement *src;
  99   GList *l;
 100
 101   src = setup_fakesrc ();
 102
 103   g_object_set (G_OBJECT (src), "sizetype", 1, NULL);
 104   g_object_set (G_OBJECT (src), "num-buffers", 100, NULL);
 105
 106   fail_unless (gst_element_set_state (src,
 107           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
 108       "could not set to playing");
 109
 110   while (!have_eos) {
 111     g_usleep (1000);
 112   }
 113
 114   fail_unless (g_list_length (buffers) == 100);
 115   l = buffers;
 116   while (l) {
 117     GstBuffer *buf = l->data;
 118
 119     fail_unless (GST_BUFFER_SIZE (buf) == 0);
 120     l = l->next;
 121   }
 122   gst_check_drop_buffers ();
 123
 124   fail_unless (gst_element_set_state (src,
 125           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
 126
 127   /* cleanup */
 128   cleanup_fakesrc (src);
 129 }
 130
 131 GST_END_TEST;
 132
 133 GST_START_TEST (test_sizetype_fixed)
 134 {
 135   GstElement *src;
 136   GList *l;
 137
 138   src = setup_fakesrc ();
 139
 140   g_object_set (G_OBJECT (src), "sizetype", 2, NULL);
 141   g_object_set (G_OBJECT (src), "sizemax", 8192, 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_CHANGE_SUCCESS,
 146       "could not set to playing");
 147
 148   while (!have_eos) {
 149     g_usleep (1000);
 150   }
 151
 152   fail_unless (g_list_length (buffers) == 100);
 153   l = buffers;
 154   while (l) {
 155     GstBuffer *buf = l->data;
 156
 157     fail_unless (GST_BUFFER_SIZE (buf) == 8192);
 158     l = l->next;
 159   }
 160   gst_check_drop_buffers ();
 161
 162   fail_unless (gst_element_set_state (src,
 163           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
 164
 165   /* cleanup */
 166   cleanup_fakesrc (src);
 167 }
 168
 169 GST_END_TEST;
 170
 171 GST_START_TEST (test_sizetype_random)
 172 {
 173   GstElement *src;
 174   GList *l;
 175
 176   src = setup_fakesrc ();
 177
 178   g_object_set (G_OBJECT (src), "sizetype", 3, NULL);
 179   g_object_set (G_OBJECT (src), "sizemin", 4096, NULL);
 180   g_object_set (G_OBJECT (src), "sizemax", 8192, NULL);
 181   g_object_set (G_OBJECT (src), "num-buffers", 100, NULL);
 182
 183   fail_unless (gst_element_set_state (src,
 184           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
 185       "could not set to playing");
 186
 187   while (!have_eos) {
 188     g_usleep (1000);
 189   }
 190
 191   fail_unless (g_list_length (buffers) == 100);
 192   l = buffers;
 193   while (l) {
 194     GstBuffer *buf = l->data;
 195
 196     fail_if (GST_BUFFER_SIZE (buf) > 8192);
 197     fail_if (GST_BUFFER_SIZE (buf) < 4096);
 198     l = l->next;
 199   }
 200   gst_check_drop_buffers ();
 201
 202   fail_unless (gst_element_set_state (src,
 203           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
 204
 205   /* cleanup */
 206   cleanup_fakesrc (src);
 207 }
 208
 209 GST_END_TEST;
 210
 211 GST_START_TEST (test_no_preroll)
 212 {
 213   GstElement *src;
 214   GstStateChangeReturn ret;
 215
 216   src = setup_fakesrc ();
 217
 218   g_object_set (G_OBJECT (src), "is-live", TRUE, NULL);
 219
 220   ret = gst_element_set_state (src, GST_STATE_PAUSED);
 221
 222   fail_unless (ret == GST_STATE_CHANGE_NO_PREROLL,
 223       "error going to paused the first time");
 224
 225   ret = gst_element_set_state (src, GST_STATE_PAUSED);
 226
 227   fail_unless (ret == GST_STATE_CHANGE_NO_PREROLL,
 228       "error going to paused the second time");
 229
 230   fail_unless (gst_element_set_state (src,
 231           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
 232
 233   /* cleanup */
 234   cleanup_fakesrc (src);
 235 }
 236
 237 GST_END_TEST;
 238
 239 static Suite *
 240 fakesrc_suite (void)
 241 {
 242   Suite *s = suite_create ("fakesrc");
 243   TCase *tc_chain = tcase_create ("general");
 244
 245   suite_add_tcase (s, tc_chain);
 246   tcase_add_test (tc_chain, test_num_buffers);
 247   tcase_add_test (tc_chain, test_sizetype_empty);
 248   tcase_add_test (tc_chain, test_sizetype_fixed);
 249   tcase_add_test (tc_chain, test_sizetype_random);
 250   tcase_add_test (tc_chain, test_no_preroll);
 251
 252   return s;
 253 }
 254
 255 GST_CHECK_MAIN (fakesrc);