mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-27 02:30:35 +00:00
3ee4896bef
Otherwise it's not guaranteed that buffers are actually on disk after pushing them, and reading the file via g_file_get_contents() might not include them yet.
503 lines
20 KiB
C
503 lines
20 KiB
C
/* GStreamer unit test for the filesink element
|
|
*
|
|
* Copyright (C) 2006 Thomas Vander Stichele <thomas at apestaart dot org>
|
|
* Copyright (C) 2007 Tim-Philipp Müller <tim centricular net>
|
|
*
|
|
* 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., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <glib.h>
|
|
#include <glib/gstdio.h>
|
|
|
|
#include <gst/check/gstcheck.h>
|
|
|
|
static GstPad *mysrcpad;
|
|
static gboolean sync_buffers = FALSE;
|
|
|
|
static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS_ANY);
|
|
|
|
static GstElement *
|
|
setup_filesink (void)
|
|
{
|
|
GstElement *filesink;
|
|
|
|
GST_DEBUG ("setup_filesink");
|
|
filesink = gst_check_setup_element ("filesink");
|
|
mysrcpad = gst_check_setup_src_pad (filesink, &srctemplate);
|
|
gst_pad_set_active (mysrcpad, TRUE);
|
|
return filesink;
|
|
}
|
|
|
|
static void
|
|
cleanup_filesink (GstElement * filesink)
|
|
{
|
|
gst_pad_set_active (mysrcpad, FALSE);
|
|
gst_check_teardown_src_pad (filesink);
|
|
gst_check_teardown_element (filesink);
|
|
}
|
|
|
|
#if 0
|
|
/* this queries via the element vfunc, which is currently not implemented */
|
|
#define CHECK_QUERY_POSITION(filesink,format,position) \
|
|
G_STMT_START { \
|
|
gint64 pos; \
|
|
fail_unless (gst_element_query_position (filesink, format, &pos)); \
|
|
fail_unless_equals_int (pos, position); \
|
|
} G_STMT_END
|
|
#else
|
|
#define CHECK_QUERY_POSITION(filesink,format,position) \
|
|
G_STMT_START { \
|
|
GstPad *pad; \
|
|
gint64 pos; \
|
|
pad = gst_element_get_static_pad (filesink, "sink"); \
|
|
fail_unless (gst_pad_query_position (pad, format, &pos)); \
|
|
fail_unless_equals_int (pos, position); \
|
|
gst_object_unref (pad); \
|
|
} G_STMT_END
|
|
#endif
|
|
|
|
#define PUSH_BYTES(num_bytes) \
|
|
G_STMT_START { \
|
|
GstBuffer *buf = gst_buffer_new_and_alloc(num_bytes); \
|
|
GRand *rand = g_rand_new_with_seed (num_bytes); \
|
|
GstMapInfo info; \
|
|
guint i; \
|
|
if (sync_buffers) \
|
|
GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_SYNC_AFTER); \
|
|
fail_unless (gst_buffer_map (buf, &info, GST_MAP_WRITE)); \
|
|
for (i = 0; i < num_bytes; ++i) \
|
|
((guint8 *)info.data)[i] = (g_rand_int (rand) >> 24) & 0xff; \
|
|
gst_buffer_unmap (buf, &info); \
|
|
fail_unless_equals_int (gst_pad_push (mysrcpad, buf), GST_FLOW_OK); \
|
|
g_rand_free (rand); \
|
|
} G_STMT_END
|
|
|
|
/* Push Buffer with num_mem_blocks memory block each of size num_bytes*/
|
|
#define PUSH_BUFFER_WITH_MULTIPLE_MEM_BLOCKS(num_mem_blocks, num_bytes) \
|
|
G_STMT_START { \
|
|
GstBuffer *buf = gst_buffer_new(); \
|
|
guint i; \
|
|
if (sync_buffers) \
|
|
GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_SYNC_AFTER); \
|
|
for (i = 0; i < num_mem_blocks; ++i){ \
|
|
GstMapInfo info; \
|
|
GstMemory* mem_block = gst_allocator_alloc(NULL,num_bytes,NULL); \
|
|
GRand *rand = g_rand_new_with_seed (num_bytes); \
|
|
guint j; \
|
|
fail_unless (gst_memory_map (mem_block, &info, GST_MAP_WRITE)); \
|
|
for (j = 0; j < num_bytes; ++j) \
|
|
((guint8 *)info.data)[j] = (g_rand_int (rand) >> 24) & 0xff; \
|
|
gst_memory_unmap (mem_block, &info); \
|
|
gst_buffer_append_memory(buf,mem_block); \
|
|
g_rand_free (rand); \
|
|
} \
|
|
fail_unless_equals_int (gst_pad_push (mysrcpad, buf), GST_FLOW_OK); \
|
|
} G_STMT_END
|
|
|
|
/* Push Buffer List with num_buffers buffers each containing num_mem_blocks
|
|
* memory blocks of size num_bytes */
|
|
#define PUSH_BUFFER_LIST_WITH_MULTI_MEM_BLOCKS_BUFFERS(num_buffers, num_mem_blocks, num_bytes) \
|
|
G_STMT_START { \
|
|
guint i; \
|
|
GstBufferList* buf_list = gst_buffer_list_new(); \
|
|
for(i = 0; i < num_buffers; ++i){ \
|
|
GstBuffer *buf = gst_buffer_new(); \
|
|
guint j; \
|
|
if (sync_buffers) \
|
|
GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_SYNC_AFTER); \
|
|
for (j = 0; j < num_mem_blocks; ++j){ \
|
|
GstMapInfo info; \
|
|
GstMemory* mem_block = gst_allocator_alloc(NULL,num_bytes,NULL); \
|
|
GRand *rand = g_rand_new_with_seed (num_bytes); \
|
|
guint k; \
|
|
fail_unless (gst_memory_map (mem_block, &info, GST_MAP_WRITE)); \
|
|
for (k = 0; k < num_bytes; ++k) \
|
|
((guint8 *)info.data)[k] = (g_rand_int (rand) >> 24) & 0xff; \
|
|
gst_memory_unmap (mem_block, &info); \
|
|
gst_buffer_append_memory(buf,mem_block); \
|
|
g_rand_free (rand); \
|
|
} \
|
|
gst_buffer_list_add(buf_list,buf); \
|
|
} \
|
|
fail_unless_equals_int (gst_pad_push_list (mysrcpad, buf_list), GST_FLOW_OK); \
|
|
} G_STMT_END
|
|
|
|
/* Push buffer_list containing num_buffers number of buffers with size
|
|
* num_bytes bytes
|
|
* Example: PUSH_BUFFER_LIST(2,10) will push the buffer list containing
|
|
* 2 buffers with size 10 bytes each */
|
|
#define PUSH_BUFFER_LIST(num_buffers, num_bytes) \
|
|
G_STMT_START { \
|
|
guint i; \
|
|
GstBufferList* buf_list = gst_buffer_list_new(); \
|
|
for(i = 0; i < num_buffers; ++i){ \
|
|
GstBuffer *buf = gst_buffer_new_and_alloc(num_bytes); \
|
|
GRand *rand = g_rand_new_with_seed (num_bytes); \
|
|
GstMapInfo info; \
|
|
guint j; \
|
|
if (sync_buffers) \
|
|
GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_SYNC_AFTER); \
|
|
fail_unless (gst_buffer_map (buf, &info, GST_MAP_WRITE)); \
|
|
for (j = 0; j < num_bytes; ++j) \
|
|
((guint8 *)info.data)[j] = (g_rand_int (rand) >> 24) & 0xff; \
|
|
gst_buffer_unmap (buf, &info); \
|
|
gst_buffer_list_add(buf_list,buf); \
|
|
g_rand_free (rand); \
|
|
} \
|
|
fail_unless_equals_int (gst_pad_push_list (mysrcpad, buf_list), GST_FLOW_OK); \
|
|
} G_STMT_END
|
|
|
|
#define CHECK_WRITTEN_BYTES(offset,written,file_size) \
|
|
G_STMT_START { \
|
|
gchar *data = NULL; \
|
|
gsize len; \
|
|
fail_unless (g_file_get_contents (tmp_fn, &data, &len, NULL), \
|
|
"Failed to read in newly-created file '%s'", tmp_fn); \
|
|
fail_unless_equals_int (len, file_size); \
|
|
{ \
|
|
/* we wrote <written> bytes at position 0 */ \
|
|
GRand *rand = g_rand_new_with_seed (written); \
|
|
guint i; \
|
|
for (i = 0; i < written; ++i) { \
|
|
guint8 byte_written = *(((guint8 *) data) + offset + i); \
|
|
\
|
|
fail_unless_equals_int (byte_written, g_rand_int (rand) >> 24); \
|
|
} \
|
|
g_rand_free (rand); \
|
|
} \
|
|
g_free (data); \
|
|
} G_STMT_END
|
|
|
|
static gchar *
|
|
create_temporary_file (void)
|
|
{
|
|
const gchar *tmpdir;
|
|
gchar *tmp_fn;
|
|
gint fd;
|
|
|
|
tmpdir = g_get_tmp_dir ();
|
|
if (tmpdir == NULL)
|
|
return NULL;
|
|
|
|
/* this is just silly, but gcc warns if we try to use tpmnam() */
|
|
tmp_fn = g_build_filename (tmpdir, "gstreamer-filesink-test-XXXXXX", NULL);
|
|
fd = g_mkstemp (tmp_fn);
|
|
if (fd < 0) {
|
|
GST_ERROR ("can't create temp file %s: %s", tmp_fn, g_strerror (errno));
|
|
g_free (tmp_fn);
|
|
return NULL;
|
|
}
|
|
/* don't want the file, just a filename (hence silly, see above) */
|
|
g_close (fd, NULL);
|
|
g_remove (tmp_fn);
|
|
|
|
return tmp_fn;
|
|
}
|
|
|
|
/* TODO: we don't check that the data is actually written to the right
|
|
* position after a seek */
|
|
GST_START_TEST (test_seeking)
|
|
{
|
|
GstElement *filesink;
|
|
gchar *tmp_fn;
|
|
GstSegment segment;
|
|
|
|
tmp_fn = create_temporary_file ();
|
|
if (tmp_fn == NULL)
|
|
return;
|
|
filesink = setup_filesink ();
|
|
|
|
sync_buffers = TRUE;
|
|
|
|
GST_LOG ("using temp file '%s'", tmp_fn);
|
|
g_object_set (filesink, "location", tmp_fn, NULL);
|
|
|
|
fail_unless_equals_int (gst_element_set_state (filesink, GST_STATE_PLAYING),
|
|
GST_STATE_CHANGE_ASYNC);
|
|
|
|
#if 0
|
|
/* Test that filesink is seekable with a file fd */
|
|
/* filesink doesn't implement seekable query at the moment */
|
|
GstQuery *seeking_query;
|
|
gboolean seekable;
|
|
|
|
fail_unless ((seeking_query = gst_query_new_seeking (GST_FORMAT_BYTES))
|
|
!= NULL);
|
|
fail_unless (gst_element_query (filesink, seeking_query) == TRUE);
|
|
gst_query_parse_seeking (seeking_query, NULL, &seekable, NULL, NULL);
|
|
fail_unless (seekable == TRUE);
|
|
gst_query_unref (seeking_query);
|
|
#endif
|
|
|
|
fail_unless (gst_pad_push_event (mysrcpad,
|
|
gst_event_new_stream_start ("test")));
|
|
|
|
gst_segment_init (&segment, GST_FORMAT_BYTES);
|
|
fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_segment (&segment)));
|
|
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 0);
|
|
|
|
/* push buffer with size 0 and NULL data */
|
|
PUSH_BYTES (0);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 0);
|
|
|
|
PUSH_BYTES (1);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 1);
|
|
|
|
PUSH_BYTES (99);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 100);
|
|
|
|
PUSH_BYTES (8800);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 8900);
|
|
|
|
/* Push buffer list with 2 buffers each of size 50 bytes */
|
|
PUSH_BUFFER_LIST (2, 50);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 9000);
|
|
/* Push buffer list with 3 buffers each of size 10 bytes */
|
|
PUSH_BUFFER_LIST (3, 10);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 9030);
|
|
/* Check bytes written using push buffer list */
|
|
CHECK_WRITTEN_BYTES (8900, 50, 9030);
|
|
CHECK_WRITTEN_BYTES (8950, 50, 9030);
|
|
CHECK_WRITTEN_BYTES (9000, 10, 9030);
|
|
CHECK_WRITTEN_BYTES (9010, 10, 9030);
|
|
CHECK_WRITTEN_BYTES (9020, 10, 9030);
|
|
|
|
/* Push buffer with 2 memory blocks each of size 20 bytes */
|
|
PUSH_BUFFER_WITH_MULTIPLE_MEM_BLOCKS (2, 20);
|
|
CHECK_WRITTEN_BYTES (9030, 20, 9070);
|
|
CHECK_WRITTEN_BYTES (9050, 20, 9070);
|
|
|
|
/* Push buffer list with 2 buffers each containing 2 memory blocks each of size 20 bytes */
|
|
PUSH_BUFFER_LIST_WITH_MULTI_MEM_BLOCKS_BUFFERS (2, 2, 20);
|
|
CHECK_WRITTEN_BYTES (9070, 20, 9150);
|
|
CHECK_WRITTEN_BYTES (9090, 20, 9150);
|
|
CHECK_WRITTEN_BYTES (9110, 20, 9150);
|
|
CHECK_WRITTEN_BYTES (9130, 20, 9150);
|
|
|
|
segment.start = 8800;
|
|
if (gst_pad_push_event (mysrcpad, gst_event_new_segment (&segment))) {
|
|
GST_LOG ("seek ok");
|
|
/* make sure that new position is reported immediately */
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 8800);
|
|
PUSH_BYTES (1);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 8801);
|
|
PUSH_BYTES (9256);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 18057);
|
|
} else {
|
|
GST_INFO ("seeking not supported for tempfile?!");
|
|
}
|
|
|
|
fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()));
|
|
|
|
fail_unless_equals_int (gst_element_set_state (filesink, GST_STATE_NULL),
|
|
GST_STATE_CHANGE_SUCCESS);
|
|
|
|
/* cleanup */
|
|
cleanup_filesink (filesink);
|
|
|
|
CHECK_WRITTEN_BYTES (8801, 9256, 18057);
|
|
|
|
sync_buffers = FALSE;
|
|
|
|
/* remove file */
|
|
g_remove (tmp_fn);
|
|
g_free (tmp_fn);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_flush)
|
|
{
|
|
GstElement *filesink;
|
|
gchar *tmp_fn;
|
|
GstSegment segment;
|
|
|
|
tmp_fn = create_temporary_file ();
|
|
if (tmp_fn == NULL)
|
|
return;
|
|
filesink = setup_filesink ();
|
|
|
|
sync_buffers = FALSE;
|
|
|
|
GST_LOG ("using temp file '%s'", tmp_fn);
|
|
g_object_set (filesink, "location", tmp_fn, NULL);
|
|
|
|
fail_unless_equals_int (gst_element_set_state (filesink, GST_STATE_PLAYING),
|
|
GST_STATE_CHANGE_ASYNC);
|
|
|
|
fail_unless (gst_pad_push_event (mysrcpad,
|
|
gst_event_new_stream_start ("test")));
|
|
|
|
gst_segment_init (&segment, GST_FORMAT_TIME);
|
|
fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_segment (&segment)));
|
|
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 0);
|
|
|
|
PUSH_BYTES (8);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 8);
|
|
|
|
fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_flush_start ()));
|
|
fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_flush_stop (TRUE)));
|
|
fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_segment (&segment)));
|
|
|
|
fail_unless_equals_int (gst_element_set_state (filesink, GST_STATE_PLAYING),
|
|
GST_STATE_CHANGE_ASYNC);
|
|
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 0);
|
|
|
|
PUSH_BYTES (4);
|
|
CHECK_QUERY_POSITION (filesink, GST_FORMAT_BYTES, 4);
|
|
|
|
cleanup_filesink (filesink);
|
|
|
|
CHECK_WRITTEN_BYTES (0, 4, 4);
|
|
|
|
g_remove (tmp_fn);
|
|
g_free (tmp_fn);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_coverage)
|
|
{
|
|
GstElement *filesink;
|
|
gchar *location;
|
|
GstBus *bus;
|
|
GstMessage *message;
|
|
|
|
filesink = setup_filesink ();
|
|
bus = gst_bus_new ();
|
|
|
|
gst_element_set_bus (filesink, bus);
|
|
|
|
g_object_set (filesink, "location", "/i/do/not/exist", NULL);
|
|
g_object_get (filesink, "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/i/do/not/exist");
|
|
g_free (location);
|
|
|
|
fail_unless_equals_int (gst_element_set_state (filesink, GST_STATE_PLAYING),
|
|
GST_STATE_CHANGE_FAILURE);
|
|
|
|
/* a state change and an error */
|
|
fail_if ((message = gst_bus_pop (bus)) == NULL);
|
|
fail_unless_message_error (message, RESOURCE, OPEN_WRITE);
|
|
gst_message_unref (message);
|
|
|
|
g_object_set (filesink, "location", NULL, NULL);
|
|
g_object_get (filesink, "location", &location, NULL);
|
|
fail_if (location);
|
|
|
|
/* cleanup */
|
|
gst_element_set_bus (filesink, NULL);
|
|
gst_object_unref (GST_OBJECT (bus));
|
|
cleanup_filesink (filesink);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
GST_START_TEST (test_uri_interface)
|
|
{
|
|
GstElement *filesink;
|
|
gchar *location;
|
|
GstBus *bus;
|
|
|
|
filesink = setup_filesink ();
|
|
bus = gst_bus_new ();
|
|
|
|
gst_element_set_bus (filesink, bus);
|
|
|
|
g_object_set (G_OBJECT (filesink), "location", "/i/do/not/exist", NULL);
|
|
g_object_get (G_OBJECT (filesink), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/i/do/not/exist");
|
|
g_free (location);
|
|
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (filesink));
|
|
fail_unless_equals_string (location, "file:///i/do/not/exist");
|
|
g_free (location);
|
|
|
|
/* should accept file:///foo/bar URIs */
|
|
fail_unless (gst_uri_handler_set_uri (GST_URI_HANDLER (filesink),
|
|
"file:///foo/bar", NULL));
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (filesink));
|
|
fail_unless_equals_string (location, "file:///foo/bar");
|
|
g_free (location);
|
|
g_object_get (G_OBJECT (filesink), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/foo/bar");
|
|
g_free (location);
|
|
|
|
/* should accept file://localhost/foo/bar URIs */
|
|
fail_unless (gst_uri_handler_set_uri (GST_URI_HANDLER (filesink),
|
|
"file://localhost/foo/baz", NULL));
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (filesink));
|
|
fail_unless_equals_string (location, "file:///foo/baz");
|
|
g_free (location);
|
|
g_object_get (G_OBJECT (filesink), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/foo/baz");
|
|
g_free (location);
|
|
|
|
/* should escape non-uri characters for the URI but not for the location */
|
|
g_object_set (G_OBJECT (filesink), "location", "/foo/b?r", NULL);
|
|
g_object_get (G_OBJECT (filesink), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "/foo/b?r");
|
|
g_free (location);
|
|
location = gst_uri_handler_get_uri (GST_URI_HANDLER (filesink));
|
|
fail_unless_equals_string (location, "file:///foo/b%3Fr");
|
|
g_free (location);
|
|
|
|
g_object_set (G_OBJECT (filesink), "location", "\".donotexist", NULL);
|
|
g_object_get (G_OBJECT (filesink), "location", &location, NULL);
|
|
fail_unless_equals_string (location, "\".donotexist");
|
|
g_free (location);
|
|
|
|
/* should fail with other hostnames */
|
|
fail_if (gst_uri_handler_set_uri (GST_URI_HANDLER (filesink),
|
|
"file://hostname/foo/foo", NULL));
|
|
|
|
/* cleanup */
|
|
gst_element_set_bus (filesink, NULL);
|
|
gst_object_unref (GST_OBJECT (bus));
|
|
cleanup_filesink (filesink);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
static Suite *
|
|
filesink_suite (void)
|
|
{
|
|
Suite *s = suite_create ("filesink");
|
|
TCase *tc_chain = tcase_create ("general");
|
|
|
|
suite_add_tcase (s, tc_chain);
|
|
|
|
tcase_add_test (tc_chain, test_coverage);
|
|
tcase_add_test (tc_chain, test_uri_interface);
|
|
tcase_add_test (tc_chain, test_seeking);
|
|
tcase_add_test (tc_chain, test_flush);
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (filesink);
|