gstreamer/tests/check/elements/imagecapturebin.c
2010-12-23 13:18:57 -03:00

245 lines
6.8 KiB
C

/* GStreamer unit test for the imagecapturebin element
* Copyright (C) 2010 Thiago Santos <thiago.sousa.santos@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.
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <stdio.h>
#include <gst/check/gstcheck.h>
#define N_BUFFERS 3
typedef struct
{
GstElement *pipe;
GstElement *src;
GstElement *icbin;
} GstImageCaptureBinTestContext;
static void
gstimagecapturebin_init_test_context (GstImageCaptureBinTestContext * ctx,
gint num_buffers)
{
fail_unless (ctx != NULL);
ctx->pipe = gst_pipeline_new ("pipeline");
fail_unless (ctx->pipe != NULL);
ctx->src = gst_element_factory_make ("videotestsrc", "src");
fail_unless (ctx->src != NULL, "Failed to create videotestsrc element");
ctx->icbin = gst_element_factory_make ("imagecapturebin", "icbin");
fail_unless (ctx->icbin != NULL, "Failed to create imagecapturebin element");
if (num_buffers > 0)
g_object_set (ctx->src, "num-buffers", num_buffers, NULL);
fail_unless (gst_bin_add (GST_BIN (ctx->pipe), ctx->src));
fail_unless (gst_bin_add (GST_BIN (ctx->pipe), ctx->icbin));
fail_unless (gst_element_link (ctx->src, ctx->icbin));
}
static void
gstimagecapturebin_unset_test_context (GstImageCaptureBinTestContext * ctx)
{
gst_element_set_state (ctx->pipe, GST_STATE_NULL);
gst_object_unref (ctx->pipe);
memset (ctx, 0, sizeof (GstImageCaptureBinTestContext));
}
static gchar *
make_test_file_name (void)
{
return g_strdup_printf ("%s" G_DIR_SEPARATOR_S
"imagecapturebintest_%%d.cap", g_get_tmp_dir ());
}
GST_START_TEST (test_simple_capture)
{
GstImageCaptureBinTestContext ctx;
GstBus *bus;
GstMessage *msg;
gchar *test_file_name;
gint i;
gstimagecapturebin_init_test_context (&ctx, N_BUFFERS);
bus = gst_element_get_bus (ctx.pipe);
test_file_name = make_test_file_name ();
g_object_set (ctx.icbin, "location", test_file_name, NULL);
fail_if (gst_element_set_state (ctx.pipe, GST_STATE_PLAYING) ==
GST_STATE_CHANGE_FAILURE);
msg = gst_bus_timed_pop_filtered (bus, GST_SECOND * 10,
GST_MESSAGE_EOS | GST_MESSAGE_ERROR);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
gst_message_unref (msg);
/* check there are N_BUFFERS files */
for (i = 0; i < N_BUFFERS; i++) {
gchar *filename;
FILE *f;
filename = g_strdup_printf (test_file_name, i);
fail_unless (g_file_test (filename, G_FILE_TEST_EXISTS));
fail_unless (g_file_test (filename, G_FILE_TEST_IS_REGULAR));
fail_if (g_file_test (filename, G_FILE_TEST_IS_SYMLINK));
/* check the file isn't empty */
f = fopen (filename, "r");
fseek (f, 0, SEEK_END);
fail_unless (ftell (f) > 0);
fclose (f);
g_free (filename);
}
gstimagecapturebin_unset_test_context (&ctx);
gst_object_unref (bus);
g_free (test_file_name);
}
GST_END_TEST;
GST_START_TEST (test_setting_encoder)
{
GstImageCaptureBinTestContext ctx;
GstBus *bus;
GstMessage *msg;
GstElement *encoder;
gchar *test_file_name;
gint i;
gstimagecapturebin_init_test_context (&ctx, N_BUFFERS);
bus = gst_element_get_bus (ctx.pipe);
test_file_name = make_test_file_name ();
g_object_set (ctx.icbin, "location", test_file_name, NULL);
encoder = gst_element_factory_make ("jpegenc", NULL);
g_object_set (ctx.icbin, "image-encoder", encoder, NULL);
fail_if (gst_element_set_state (ctx.pipe, GST_STATE_PLAYING) ==
GST_STATE_CHANGE_FAILURE);
msg = gst_bus_timed_pop_filtered (bus, GST_SECOND * 10,
GST_MESSAGE_EOS | GST_MESSAGE_ERROR);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* check there are N_BUFFERS files */
for (i = 0; i < N_BUFFERS; i++) {
gchar *filename;
FILE *f;
filename = g_strdup_printf (test_file_name, i);
fail_unless (g_file_test (filename, G_FILE_TEST_EXISTS));
fail_unless (g_file_test (filename, G_FILE_TEST_IS_REGULAR));
fail_if (g_file_test (filename, G_FILE_TEST_IS_SYMLINK));
/* check the file isn't empty */
f = fopen (filename, "r");
fseek (f, 0, SEEK_END);
fail_unless (ftell (f) > 0);
fclose (f);
g_free (filename);
}
gstimagecapturebin_unset_test_context (&ctx);
gst_object_unref (bus);
g_free (test_file_name);
}
GST_END_TEST;
GST_START_TEST (test_setting_muxer)
{
GstImageCaptureBinTestContext ctx;
GstBus *bus;
GstMessage *msg;
GstElement *encoder;
gchar *test_file_name;
gint i;
gstimagecapturebin_init_test_context (&ctx, N_BUFFERS);
bus = gst_element_get_bus (ctx.pipe);
test_file_name = make_test_file_name ();
g_object_set (ctx.icbin, "location", test_file_name, NULL);
encoder = gst_element_factory_make ("pngenc", NULL);
g_object_set (ctx.icbin, "image-encoder", encoder, NULL);
encoder = gst_element_factory_make ("identity", NULL);
g_object_set (ctx.icbin, "image-muxer", encoder, NULL);
fail_if (gst_element_set_state (ctx.pipe, GST_STATE_PLAYING) ==
GST_STATE_CHANGE_FAILURE);
msg = gst_bus_timed_pop_filtered (bus, GST_SECOND * 10,
GST_MESSAGE_EOS | GST_MESSAGE_ERROR);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
/* check there are N_BUFFERS files */
for (i = 0; i < N_BUFFERS; i++) {
gchar *filename;
FILE *f;
filename = g_strdup_printf (test_file_name, i);
fail_unless (g_file_test (filename, G_FILE_TEST_EXISTS));
fail_unless (g_file_test (filename, G_FILE_TEST_IS_REGULAR));
fail_if (g_file_test (filename, G_FILE_TEST_IS_SYMLINK));
/* check the file isn't empty */
f = fopen (filename, "r");
fseek (f, 0, SEEK_END);
fail_unless (ftell (f) > 0);
fclose (f);
g_free (filename);
}
gstimagecapturebin_unset_test_context (&ctx);
gst_object_unref (bus);
g_free (test_file_name);
}
GST_END_TEST;
static Suite *
imagecapturebin_suite (void)
{
Suite *s = suite_create ("imagecapturebin");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_simple_capture);
tcase_add_test (tc_chain, test_setting_encoder);
tcase_add_test (tc_chain, test_setting_muxer);
return s;
}
GST_CHECK_MAIN (imagecapturebin);