tests/check/libs/basesink.c

   1 /* GStreamer
   2  *
   3  * Copyright (C) 2010 Alessandro Decina <alessandro.decina@collabora.co.uk>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24 #include <gst/gst.h>
  25 #include <gst/check/gstcheck.h>
  26 #include <gst/base/gstbasesink.h>
  27
  28 GST_START_TEST (basesink_last_buffer_enabled)
  29 {
  30   GstElement *src, *sink, *pipeline;
  31   GstBus *bus;
  32   GstMessage *msg;
  33   GstBuffer *last_buffer;
  34
  35   pipeline = gst_pipeline_new ("pipeline");
  36   sink = gst_element_factory_make ("fakesink", "sink");
  37   src = gst_element_factory_make ("fakesrc", "src");
  38
  39   fail_unless (gst_bin_add (GST_BIN (pipeline), src) == TRUE);
  40   fail_unless (gst_bin_add (GST_BIN (pipeline), sink) == TRUE);
  41   fail_unless (gst_element_link (src, sink) == TRUE);
  42
  43   bus = gst_element_get_bus (pipeline);
  44
  45   /* try with enable-last-buffer set to TRUE */
  46   g_object_set (src, "num-buffers", 1, NULL);
  47   fail_unless (gst_element_set_state (pipeline, GST_STATE_PLAYING)
  48       != GST_STATE_CHANGE_FAILURE);
  49   msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
  50   fail_unless (msg != NULL);
  51   fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
  52   fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
  53   gst_message_unref (msg);
  54
  55   /* last-buffer should be != NULL */
  56   g_object_get (sink, "last-buffer", &last_buffer, NULL);
  57   fail_unless (last_buffer != NULL);
  58   gst_buffer_unref (last_buffer);
  59
  60   /* set enable-last-buffer to FALSE now, this should set last-buffer to NULL */
  61   g_object_set (sink, "enable-last-buffer", FALSE, NULL);
  62   g_object_get (sink, "last-buffer", &last_buffer, NULL);
  63   fail_unless (last_buffer == NULL);
  64
  65   gst_element_set_state (pipeline, GST_STATE_NULL);
  66
  67   GST_INFO ("stopped");
  68
  69   gst_object_unref (bus);
  70   gst_object_unref (pipeline);
  71 }
  72
  73 GST_END_TEST;
  74
  75 GST_START_TEST (basesink_last_buffer_disabled)
  76 {
  77   GstElement *src, *sink, *pipeline;
  78   GstBus *bus;
  79   GstMessage *msg;
  80   GstBuffer *last_buffer;
  81
  82   pipeline = gst_pipeline_new ("pipeline");
  83   sink = gst_element_factory_make ("fakesink", "sink");
  84   src = gst_element_factory_make ("fakesrc", "src");
  85
  86   fail_unless (gst_bin_add (GST_BIN (pipeline), src) == TRUE);
  87   fail_unless (gst_bin_add (GST_BIN (pipeline), sink) == TRUE);
  88   fail_unless (gst_element_link (src, sink) == TRUE);
  89
  90   bus = gst_element_get_bus (pipeline);
  91
  92   /* set enable-last-buffer to FALSE */
  93   g_object_set (src, "num-buffers", 1, NULL);
  94   g_object_set (sink, "enable-last-buffer", FALSE, NULL);
  95   gst_element_set_state (pipeline, GST_STATE_PLAYING);
  96   msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);
  97   fail_unless (msg != NULL);
  98   fail_unless (GST_MESSAGE_TYPE (msg) != GST_MESSAGE_ERROR);
  99   fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
 100   gst_message_unref (msg);
 101
 102   /* last-buffer should be NULL */
 103   g_object_get (sink, "last-buffer", &last_buffer, NULL);
 104   fail_unless (last_buffer == NULL);
 105
 106   gst_element_set_state (pipeline, GST_STATE_NULL);
 107
 108   GST_INFO ("stopped");
 109
 110   gst_object_unref (bus);
 111   gst_object_unref (pipeline);
 112 }
 113
 114 GST_END_TEST;
 115
 116 static Suite *
 117 gst_basesrc_suite (void)
 118 {
 119   Suite *s = suite_create ("GstBaseSink");
 120   TCase *tc = tcase_create ("general");
 121
 122   suite_add_tcase (s, tc);
 123   tcase_add_test (tc, basesink_last_buffer_enabled);
 124   tcase_add_test (tc, basesink_last_buffer_disabled);
 125
 126   return s;
 127 }
 128
 129 GST_CHECK_MAIN (gst_basesrc);