mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-03 16:09:39 +00:00
imagefreeze: Add some unit tests
This commit is contained in:
Sebastian Dröge
Tim-Philipp Müller
parent
3e574eafe3
commit
f152f6e20f
1 changed files with 446 additions and 0 deletions
446
tests/check/elements/imagefreeze.c
Normal file
446
tests/check/elements/imagefreeze.c
Normal file
|
|
@ -0,0 +1,446 @@
|
|||
/* GStreamer
|
||||
* Copyright (C) 2010 Sebastian Dröge <sebastian.droege@collabora.co.uk>
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Library General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* Library General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Library General Public
|
||||
* License along with this library; if not, write to the
|
||||
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
||||
* Boston, MA 02111-1307, USA.
|
||||
*/
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include <gst/check/gstcheck.h>
|
||||
#include <gst/video/video.h>
|
||||
|
||||
static gboolean
|
||||
bus_handler (GstBus * bus, GstMessage * message, gpointer data)
|
||||
{
|
||||
GMainLoop *loop = (GMainLoop *) data;
|
||||
|
||||
switch (message->type) {
|
||||
case GST_MESSAGE_EOS:
|
||||
g_main_loop_quit (loop);
|
||||
break;
|
||||
case GST_MESSAGE_WARNING:
|
||||
case GST_MESSAGE_ERROR:{
|
||||
GError *gerror;
|
||||
gchar *debug;
|
||||
|
||||
if (message->type == GST_MESSAGE_WARNING)
|
||||
gst_message_parse_warning (message, &gerror, &debug);
|
||||
else
|
||||
gst_message_parse_error (message, &gerror, &debug);
|
||||
g_error ("error from %s: %s (%s)\n",
|
||||
GST_ELEMENT_NAME (GST_MESSAGE_SRC (message)), gerror->message,
|
||||
GST_STR_NULL (debug));
|
||||
gst_message_unref (message);
|
||||
g_error_free (gerror);
|
||||
g_free (debug);
|
||||
g_main_loop_quit (loop);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static GstElement *
|
||||
setup_imagefreeze (const GstCaps * caps1, const GstCaps * caps2,
|
||||
GCallback sink_handoff, gpointer sink_handoff_data)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstElement *videotestsrc, *capsfilter1, *imagefreeze, *capsfilter2, *fakesink;
|
||||
|
||||
pipeline = gst_pipeline_new ("pipeline");
|
||||
fail_unless (pipeline != NULL);
|
||||
|
||||
videotestsrc = gst_element_factory_make ("videotestsrc", NULL);
|
||||
fail_unless (videotestsrc != NULL);
|
||||
g_object_set (videotestsrc, "num-buffers", 1, NULL);
|
||||
|
||||
capsfilter1 = gst_element_factory_make ("capsfilter", NULL);
|
||||
fail_unless (capsfilter1 != NULL);
|
||||
g_object_set (capsfilter1, "caps", caps1, NULL);
|
||||
|
||||
imagefreeze = gst_element_factory_make ("imagefreeze", NULL);
|
||||
fail_unless (imagefreeze != NULL);
|
||||
|
||||
capsfilter2 = gst_element_factory_make ("capsfilter", NULL);
|
||||
fail_unless (capsfilter2 != NULL);
|
||||
g_object_set (capsfilter2, "caps", caps2, NULL);
|
||||
|
||||
fakesink = gst_element_factory_make ("fakesink", NULL);
|
||||
fail_unless (fakesink != NULL);
|
||||
g_object_set (fakesink, "signal-handoffs", TRUE, "async", FALSE, NULL);
|
||||
g_signal_connect (fakesink, "handoff", sink_handoff, sink_handoff_data);
|
||||
|
||||
gst_bin_add_many (GST_BIN (pipeline), videotestsrc, capsfilter1, imagefreeze,
|
||||
capsfilter2, fakesink, NULL);
|
||||
|
||||
fail_unless (gst_element_link_pads (videotestsrc, "src", capsfilter1,
|
||||
"sink"));
|
||||
fail_unless (gst_element_link_pads (capsfilter1, "src", imagefreeze, "sink"));
|
||||
fail_unless (gst_element_link_pads (imagefreeze, "src", capsfilter2, "sink"));
|
||||
fail_unless (gst_element_link_pads (capsfilter2, "src", fakesink, "sink"));
|
||||
|
||||
return pipeline;
|
||||
}
|
||||
|
||||
static void
|
||||
sink_handoff_cb_0_1 (GstElement * object, GstBuffer * buffer, GstPad * pad,
|
||||
gpointer user_data)
|
||||
{
|
||||
guint *n_buffers = (guint *) user_data;
|
||||
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer), 0);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), GST_CLOCK_TIME_NONE);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 0);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), 1);
|
||||
|
||||
*n_buffers = *n_buffers + 1;
|
||||
}
|
||||
|
||||
GST_START_TEST (test_imagefreeze_0_1)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstCaps *caps1, *caps2;
|
||||
GstBus *bus;
|
||||
GMainLoop *loop;
|
||||
guint n_buffers = 0;
|
||||
|
||||
caps1 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
caps2 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 0, 1, 1, 1);
|
||||
|
||||
pipeline =
|
||||
setup_imagefreeze (caps1, caps2, G_CALLBACK (sink_handoff_cb_0_1),
|
||||
&n_buffers);
|
||||
|
||||
loop = g_main_loop_new (NULL, TRUE);
|
||||
fail_unless (loop != NULL);
|
||||
|
||||
bus = gst_element_get_bus (pipeline);
|
||||
fail_unless (bus != NULL);
|
||||
gst_bus_add_watch (bus, bus_handler, loop);
|
||||
gst_object_unref (bus);
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
g_main_loop_run (loop);
|
||||
|
||||
fail_unless_equals_int (n_buffers, 1);
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_NULL);
|
||||
|
||||
g_object_unref (pipeline);
|
||||
g_main_loop_unref (loop);
|
||||
gst_caps_unref (caps1);
|
||||
gst_caps_unref (caps2);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static void
|
||||
sink_handoff_cb_25_1_0ms_400ms (GstElement * object, GstBuffer * buffer,
|
||||
GstPad * pad, gpointer user_data)
|
||||
{
|
||||
guint *n_buffers = (guint *) user_data;
|
||||
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer),
|
||||
*n_buffers * 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), *n_buffers);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer), *n_buffers + 1);
|
||||
|
||||
*n_buffers = *n_buffers + 1;
|
||||
}
|
||||
|
||||
GST_START_TEST (test_imagefreeze_25_1_0ms_400ms)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstCaps *caps1, *caps2;
|
||||
GstBus *bus;
|
||||
GMainLoop *loop;
|
||||
guint n_buffers = 0;
|
||||
|
||||
caps1 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
caps2 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
|
||||
pipeline =
|
||||
setup_imagefreeze (caps1, caps2,
|
||||
G_CALLBACK (sink_handoff_cb_25_1_0ms_400ms), &n_buffers);
|
||||
|
||||
loop = g_main_loop_new (NULL, TRUE);
|
||||
fail_unless (loop != NULL);
|
||||
|
||||
bus = gst_element_get_bus (pipeline);
|
||||
fail_unless (bus != NULL);
|
||||
gst_bus_add_watch (bus, bus_handler, loop);
|
||||
gst_object_unref (bus);
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
fail_unless (gst_element_seek (pipeline, 1.0, GST_FORMAT_TIME,
|
||||
GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET,
|
||||
400 * GST_MSECOND));
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
g_main_loop_run (loop);
|
||||
|
||||
fail_unless_equals_int (n_buffers, 10);
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_NULL);
|
||||
|
||||
g_object_unref (pipeline);
|
||||
g_main_loop_unref (loop);
|
||||
gst_caps_unref (caps1);
|
||||
gst_caps_unref (caps2);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static void
|
||||
sink_handoff_cb_25_1_200ms_400ms (GstElement * object, GstBuffer * buffer,
|
||||
GstPad * pad, gpointer user_data)
|
||||
{
|
||||
guint *n_buffers = (guint *) user_data;
|
||||
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer),
|
||||
200 * GST_MSECOND + *n_buffers * 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 5 + *n_buffers);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer),
|
||||
5 + *n_buffers + 1);
|
||||
|
||||
*n_buffers = *n_buffers + 1;
|
||||
}
|
||||
|
||||
GST_START_TEST (test_imagefreeze_25_1_200ms_400ms)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstCaps *caps1, *caps2;
|
||||
GstBus *bus;
|
||||
GMainLoop *loop;
|
||||
guint n_buffers = 0;
|
||||
|
||||
caps1 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
caps2 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
|
||||
pipeline =
|
||||
setup_imagefreeze (caps1, caps2,
|
||||
G_CALLBACK (sink_handoff_cb_25_1_200ms_400ms), &n_buffers);
|
||||
|
||||
loop = g_main_loop_new (NULL, TRUE);
|
||||
fail_unless (loop != NULL);
|
||||
|
||||
bus = gst_element_get_bus (pipeline);
|
||||
fail_unless (bus != NULL);
|
||||
gst_bus_add_watch (bus, bus_handler, loop);
|
||||
gst_object_unref (bus);
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
fail_unless (gst_element_seek (pipeline, 1.0, GST_FORMAT_TIME,
|
||||
GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 200 * GST_MSECOND,
|
||||
GST_SEEK_TYPE_SET, 400 * GST_MSECOND));
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
g_main_loop_run (loop);
|
||||
|
||||
fail_unless_equals_int (n_buffers, 5);
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_NULL);
|
||||
|
||||
g_object_unref (pipeline);
|
||||
g_main_loop_unref (loop);
|
||||
gst_caps_unref (caps1);
|
||||
gst_caps_unref (caps2);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static void
|
||||
sink_handoff_cb_25_1_400ms_0ms (GstElement * object, GstBuffer * buffer,
|
||||
GstPad * pad, gpointer user_data)
|
||||
{
|
||||
guint *n_buffers = (guint *) user_data;
|
||||
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer),
|
||||
400 * GST_MSECOND - (*n_buffers + 1) * 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 10 - (*n_buffers + 1));
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer),
|
||||
10 - (*n_buffers + 1) + 1);
|
||||
|
||||
*n_buffers = *n_buffers + 1;
|
||||
}
|
||||
|
||||
GST_START_TEST (test_imagefreeze_25_1_400ms_0ms)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstCaps *caps1, *caps2;
|
||||
GstBus *bus;
|
||||
GMainLoop *loop;
|
||||
guint n_buffers = 0;
|
||||
|
||||
caps1 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
caps2 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
|
||||
pipeline =
|
||||
setup_imagefreeze (caps1, caps2,
|
||||
G_CALLBACK (sink_handoff_cb_25_1_400ms_0ms), &n_buffers);
|
||||
|
||||
loop = g_main_loop_new (NULL, TRUE);
|
||||
fail_unless (loop != NULL);
|
||||
|
||||
bus = gst_element_get_bus (pipeline);
|
||||
fail_unless (bus != NULL);
|
||||
gst_bus_add_watch (bus, bus_handler, loop);
|
||||
gst_object_unref (bus);
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
fail_unless (gst_element_seek (pipeline, -1.0, GST_FORMAT_TIME,
|
||||
GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 0, GST_SEEK_TYPE_SET,
|
||||
400 * GST_MSECOND));
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
g_main_loop_run (loop);
|
||||
|
||||
fail_unless_equals_int (n_buffers, 10);
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_NULL);
|
||||
|
||||
g_object_unref (pipeline);
|
||||
g_main_loop_unref (loop);
|
||||
gst_caps_unref (caps1);
|
||||
gst_caps_unref (caps2);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static void
|
||||
sink_handoff_cb_25_1_220ms_380ms (GstElement * object, GstBuffer * buffer,
|
||||
GstPad * pad, gpointer user_data)
|
||||
{
|
||||
guint *n_buffers = (guint *) user_data;
|
||||
|
||||
if (*n_buffers == 0) {
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer),
|
||||
220 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 20 * GST_MSECOND);
|
||||
} else if (*n_buffers == 4) {
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer),
|
||||
360 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 20 * GST_MSECOND);
|
||||
} else {
|
||||
fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buffer),
|
||||
200 * GST_MSECOND + *n_buffers * 40 * GST_MSECOND);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_DURATION (buffer), 40 * GST_MSECOND);
|
||||
}
|
||||
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET (buffer), 5 + *n_buffers);
|
||||
fail_unless_equals_uint64 (GST_BUFFER_OFFSET_END (buffer),
|
||||
5 + *n_buffers + 1);
|
||||
|
||||
*n_buffers = *n_buffers + 1;
|
||||
}
|
||||
|
||||
GST_START_TEST (test_imagefreeze_25_1_220ms_380ms)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstCaps *caps1, *caps2;
|
||||
GstBus *bus;
|
||||
GMainLoop *loop;
|
||||
guint n_buffers = 0;
|
||||
|
||||
caps1 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
caps2 =
|
||||
gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1);
|
||||
|
||||
pipeline =
|
||||
setup_imagefreeze (caps1, caps2,
|
||||
G_CALLBACK (sink_handoff_cb_25_1_220ms_380ms), &n_buffers);
|
||||
|
||||
loop = g_main_loop_new (NULL, TRUE);
|
||||
fail_unless (loop != NULL);
|
||||
|
||||
bus = gst_element_get_bus (pipeline);
|
||||
fail_unless (bus != NULL);
|
||||
gst_bus_add_watch (bus, bus_handler, loop);
|
||||
gst_object_unref (bus);
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
fail_unless (gst_element_seek (pipeline, 1.0, GST_FORMAT_TIME,
|
||||
GST_SEEK_FLAG_FLUSH, GST_SEEK_TYPE_SET, 220 * GST_MSECOND,
|
||||
GST_SEEK_TYPE_SET, 380 * GST_MSECOND));
|
||||
|
||||
fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING),
|
||||
GST_STATE_CHANGE_SUCCESS);
|
||||
|
||||
g_main_loop_run (loop);
|
||||
|
||||
fail_unless_equals_int (n_buffers, 5);
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_NULL);
|
||||
|
||||
g_object_unref (pipeline);
|
||||
g_main_loop_unref (loop);
|
||||
gst_caps_unref (caps1);
|
||||
gst_caps_unref (caps2);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
Suite *
|
||||
imagefreeze_suite (void)
|
||||
{
|
||||
Suite *s = suite_create ("imagefreeze");
|
||||
TCase *tc_chain = tcase_create ("linear");
|
||||
|
||||
/* time out after 120s, not the default 3 */
|
||||
tcase_set_timeout (tc_chain, 120);
|
||||
|
||||
suite_add_tcase (s, tc_chain);
|
||||
tcase_add_test (tc_chain, test_imagefreeze_0_1);
|
||||
tcase_add_test (tc_chain, test_imagefreeze_25_1_0ms_400ms);
|
||||
tcase_add_test (tc_chain, test_imagefreeze_25_1_200ms_400ms);
|
||||
tcase_add_test (tc_chain, test_imagefreeze_25_1_400ms_0ms);
|
||||
tcase_add_test (tc_chain, test_imagefreeze_25_1_220ms_380ms);
|
||||
|
||||
return s;
|
||||
}
|
||||
|
||||
GST_CHECK_MAIN (imagefreeze);
|
||||
Loading…
Reference in a new issue