| /* GStreamer unit test for multifile plugin |
| * |
| * Copyright (C) 2007 David A. Schleef <ds@schleef.org> |
| * |
| * 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 <glib/gstdio.h> |
| #include <unistd.h> |
| |
| #include <gst/check/gstcheck.h> |
| #include <gst/video/video.h> |
| #include <stdlib.h> |
| #include <unistd.h> |
| |
| static GList *mfs_messages = NULL; |
| |
| static void |
| mfs_check_next_message (const gchar * filename) |
| { |
| GstMessage *msg; |
| const gchar *msg_filename; |
| const GstStructure *structure; |
| |
| fail_unless (mfs_messages != NULL); |
| |
| msg = mfs_messages->data; |
| mfs_messages = g_list_delete_link (mfs_messages, mfs_messages); |
| |
| structure = gst_message_get_structure (msg); |
| |
| msg_filename = gst_structure_get_string (structure, "filename"); |
| |
| fail_unless (strcmp (filename, msg_filename) == 0); |
| |
| gst_message_unref (msg); |
| } |
| |
| static void |
| run_pipeline (GstElement * pipeline) |
| { |
| GstMessage *msg; |
| GstBus *bus; |
| |
| bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); |
| |
| gst_element_set_state (pipeline, GST_STATE_PAUSED); |
| gst_element_get_state (pipeline, NULL, NULL, -1); |
| gst_element_set_state (pipeline, GST_STATE_PLAYING); |
| |
| while (1) { |
| msg = |
| gst_bus_timed_pop_filtered (bus, GST_CLOCK_TIME_NONE, |
| GST_MESSAGE_EOS | GST_MESSAGE_ERROR | GST_MESSAGE_ELEMENT); |
| |
| fail_unless (msg != NULL); |
| if (msg) { |
| if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ELEMENT) { |
| if (gst_message_has_name (msg, "GstMultiFileSink")) |
| mfs_messages = g_list_append (mfs_messages, msg); |
| else |
| gst_message_unref (msg); |
| |
| continue; |
| } |
| |
| fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS); |
| gst_message_unref (msg); |
| } |
| break; |
| } |
| |
| gst_object_unref (bus); |
| gst_element_set_state (pipeline, GST_STATE_NULL); |
| } |
| |
| GST_START_TEST (test_multifilesink_key_frame) |
| { |
| GstElement *pipeline; |
| GstElement *mfs; |
| int i; |
| const gchar *tmpdir; |
| gchar *my_tmpdir; |
| gchar *template; |
| gchar *mfs_pattern; |
| |
| tmpdir = g_get_tmp_dir (); |
| template = g_build_filename (tmpdir, "multifile-test-XXXXXX", NULL); |
| my_tmpdir = g_mkdtemp (template); |
| fail_if (my_tmpdir == NULL); |
| |
| pipeline = |
| gst_parse_launch |
| ("videotestsrc num-buffers=10 ! video/x-raw,format=(string)I420,width=320,height=240 ! multifilesink name=mfs", |
| NULL); |
| fail_if (pipeline == NULL); |
| mfs = gst_bin_get_by_name (GST_BIN (pipeline), "mfs"); |
| fail_if (mfs == NULL); |
| mfs_pattern = g_build_filename (my_tmpdir, "%05d", NULL); |
| g_object_set (G_OBJECT (mfs), "location", mfs_pattern, "post-messages", TRUE, |
| NULL); |
| g_object_unref (mfs); |
| run_pipeline (pipeline); |
| gst_object_unref (pipeline); |
| |
| for (i = 0; i < 10; i++) { |
| char *s; |
| |
| s = g_strdup_printf (mfs_pattern, i); |
| fail_if (g_remove (s) != 0); |
| |
| mfs_check_next_message (s); |
| |
| g_free (s); |
| } |
| fail_if (g_remove (my_tmpdir) != 0); |
| |
| fail_unless (mfs_messages == NULL); |
| g_free (mfs_pattern); |
| g_free (my_tmpdir); |
| } |
| |
| GST_END_TEST; |
| |
| GST_START_TEST (test_multifilesink_max_files) |
| { |
| GstElement *pipeline; |
| GstElement *mfs; |
| int i; |
| const gchar *tmpdir; |
| gchar *my_tmpdir; |
| gchar *template; |
| gchar *mfs_pattern; |
| |
| tmpdir = g_get_tmp_dir (); |
| template = g_build_filename (tmpdir, "multifile-test-XXXXXX", NULL); |
| my_tmpdir = g_mkdtemp (template); |
| fail_if (my_tmpdir == NULL); |
| |
| pipeline = |
| gst_parse_launch |
| ("videotestsrc num-buffers=10 ! video/x-raw,format=(string)I420,width=320,height=240 ! multifilesink name=mfs", |
| NULL); |
| fail_if (pipeline == NULL); |
| mfs = gst_bin_get_by_name (GST_BIN (pipeline), "mfs"); |
| fail_if (mfs == NULL); |
| mfs_pattern = g_build_filename (my_tmpdir, "%05d", NULL); |
| g_object_set (G_OBJECT (mfs), "location", mfs_pattern, "max-files", 3, NULL); |
| g_object_unref (mfs); |
| run_pipeline (pipeline); |
| gst_object_unref (pipeline); |
| |
| for (i = 0; i < 7; i++) { |
| char *s; |
| |
| s = g_strdup_printf (mfs_pattern, i); |
| fail_unless (g_remove (s) != 0); |
| g_free (s); |
| } |
| for (i = 7; i < 10; i++) { |
| char *s; |
| |
| s = g_strdup_printf (mfs_pattern, i); |
| fail_if (g_remove (s) != 0); |
| g_free (s); |
| } |
| fail_if (g_remove (my_tmpdir) != 0); |
| |
| g_free (mfs_pattern); |
| g_free (my_tmpdir); |
| } |
| |
| GST_END_TEST; |
| |
| GST_START_TEST (test_multifilesink_key_unit) |
| { |
| GstElement *mfs; |
| int i; |
| const gchar *tmpdir; |
| gchar *my_tmpdir; |
| gchar *template; |
| gchar *mfs_pattern; |
| GstBuffer *buf; |
| GstPad *sink; |
| GstSegment segment; |
| GstBus *bus; |
| |
| tmpdir = g_get_tmp_dir (); |
| template = g_build_filename (tmpdir, "multifile-test-XXXXXX", NULL); |
| my_tmpdir = g_mkdtemp (template); |
| fail_if (my_tmpdir == NULL); |
| |
| mfs = gst_element_factory_make ("multifilesink", NULL); |
| fail_if (mfs == NULL); |
| mfs_pattern = g_build_filename (my_tmpdir, "%05d", NULL); |
| g_object_set (G_OBJECT (mfs), "location", mfs_pattern, "next-file", 3, |
| "post-messages", TRUE, NULL); |
| bus = gst_bus_new (); |
| gst_element_set_bus (mfs, bus); |
| fail_if (gst_element_set_state (mfs, |
| GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE); |
| |
| sink = gst_element_get_static_pad (mfs, "sink"); |
| |
| gst_pad_send_event (sink, gst_event_new_stream_start ("test")); |
| gst_segment_init (&segment, GST_FORMAT_TIME); |
| gst_pad_send_event (sink, gst_event_new_segment (&segment)); |
| |
| buf = gst_buffer_new_and_alloc (4); |
| |
| gst_buffer_fill (buf, 0, "foo", 4); |
| fail_if (gst_pad_chain (sink, gst_buffer_copy (buf)) != GST_FLOW_OK); |
| |
| gst_buffer_fill (buf, 0, "bar", 4); |
| fail_if (gst_pad_chain (sink, gst_buffer_copy (buf)) != GST_FLOW_OK); |
| |
| fail_unless (gst_pad_send_event (sink, |
| gst_video_event_new_downstream_force_key_unit (GST_CLOCK_TIME_NONE, |
| GST_CLOCK_TIME_NONE, GST_CLOCK_TIME_NONE, TRUE, 1))); |
| |
| gst_buffer_fill (buf, 0, "baz", 4); |
| fail_if (gst_pad_chain (sink, buf) != GST_FLOW_OK); |
| |
| gst_pad_send_event (sink, gst_event_new_eos ()); |
| |
| fail_if (gst_element_set_state (mfs, |
| GST_STATE_NULL) == GST_STATE_CHANGE_FAILURE); |
| gst_element_set_bus (mfs, NULL); |
| |
| for (i = 0; i < 2; i++) { |
| char *s; |
| GstMessage *msg; |
| |
| s = g_strdup_printf (mfs_pattern, i); |
| fail_if (g_remove (s) != 0); |
| |
| msg = gst_bus_pop_filtered (bus, GST_MESSAGE_ELEMENT); |
| fail_unless (msg != NULL); |
| fail_unless (gst_message_has_name (msg, "GstMultiFileSink")); |
| fail_unless (strcmp (s, |
| gst_structure_get_string (gst_message_get_structure (msg), |
| "filename")) == 0); |
| |
| gst_message_unref (msg); |
| g_free (s); |
| } |
| fail_if (g_remove (my_tmpdir) != 0); |
| |
| gst_object_unref (bus); |
| g_free (mfs_pattern); |
| g_free (my_tmpdir); |
| gst_object_unref (sink); |
| gst_object_unref (mfs); |
| } |
| |
| GST_END_TEST; |
| |
| GST_START_TEST (test_multifilesrc) |
| { |
| GstElement *pipeline; |
| GstElement *mfs; |
| int i; |
| const gchar *tmpdir; |
| gchar *my_tmpdir; |
| gchar *template; |
| gchar *mfs_pattern; |
| |
| tmpdir = g_get_tmp_dir (); |
| template = g_build_filename (tmpdir, "multifile-test-XXXXXX", NULL); |
| my_tmpdir = g_mkdtemp (template); |
| fail_if (my_tmpdir == NULL); |
| |
| pipeline = |
| gst_parse_launch |
| ("videotestsrc num-buffers=10 ! video/x-raw,format=(string)I420,width=320,height=240 ! multifilesink name=mfs", |
| NULL); |
| fail_if (pipeline == NULL); |
| mfs = gst_bin_get_by_name (GST_BIN (pipeline), "mfs"); |
| fail_if (mfs == NULL); |
| mfs_pattern = g_build_filename (my_tmpdir, "%05d", NULL); |
| g_object_set (G_OBJECT (mfs), "location", mfs_pattern, NULL); |
| g_free (mfs_pattern); |
| g_object_unref (mfs); |
| run_pipeline (pipeline); |
| gst_object_unref (pipeline); |
| |
| pipeline = |
| gst_parse_launch |
| ("multifilesrc ! video/x-raw,format=(string)I420,width=320,height=240,framerate=10/1 ! fakesink", |
| NULL); |
| fail_if (pipeline == NULL); |
| mfs = gst_bin_get_by_name (GST_BIN (pipeline), "multifilesrc0"); |
| fail_if (mfs == NULL); |
| mfs_pattern = g_build_filename (my_tmpdir, "%05d", NULL); |
| g_object_set (G_OBJECT (mfs), "location", mfs_pattern, NULL); |
| g_object_unref (mfs); |
| run_pipeline (pipeline); |
| gst_object_unref (pipeline); |
| |
| for (i = 0; i < 10; i++) { |
| char *s; |
| |
| s = g_strdup_printf (mfs_pattern, i); |
| fail_if (g_remove (s) != 0); |
| g_free (s); |
| } |
| fail_if (g_remove (my_tmpdir) != 0); |
| |
| g_free (mfs_pattern); |
| g_free (my_tmpdir); |
| } |
| |
| GST_END_TEST; |
| |
| static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", |
| GST_PAD_SINK, |
| GST_PAD_ALWAYS, |
| GST_STATIC_CAPS_ANY); |
| |
| /* make sure stop_index is honoured even if the next target file exists */ |
| GST_START_TEST (test_multifilesrc_stop_index) |
| { |
| GstElement *src; |
| GstEvent *event; |
| GstPad *sinkpad; |
| gchar *fn; |
| |
| src = gst_check_setup_element ("multifilesrc"); |
| fail_unless (src != NULL); |
| |
| fn = g_build_filename (GST_TEST_FILES_PATH, "image.jpg", NULL); |
| g_object_set (src, "location", fn, NULL); |
| g_free (fn); |
| |
| g_object_set (src, "stop-index", 5, NULL); |
| |
| sinkpad = gst_check_setup_sink_pad_by_name (src, &sinktemplate, "src"); |
| fail_unless (sinkpad != NULL); |
| gst_pad_set_active (sinkpad, TRUE); |
| |
| gst_element_set_state (src, GST_STATE_PLAYING); |
| |
| gst_element_get_state (src, NULL, NULL, -1); |
| |
| /* busy-loop for EOS */ |
| do { |
| g_usleep (G_USEC_PER_SEC / 10); |
| event = gst_pad_get_sticky_event (sinkpad, GST_EVENT_EOS, 0); |
| } while (event == NULL); |
| gst_event_unref (event); |
| |
| /* Range appears to be [ start, stop ] */ |
| fail_unless_equals_int (g_list_length (buffers), 5 + 1); |
| |
| gst_element_set_state (src, GST_STATE_NULL); |
| |
| gst_check_drop_buffers (); |
| gst_check_teardown_pad_by_name (src, "src"); |
| gst_check_teardown_element (src); |
| } |
| |
| GST_END_TEST; |
| |
| |
| static Suite * |
| multifile_suite (void) |
| { |
| Suite *s = suite_create ("multifile"); |
| TCase *tc_chain = tcase_create ("general"); |
| |
| suite_add_tcase (s, tc_chain); |
| |
| tcase_add_test (tc_chain, test_multifilesink_key_frame); |
| tcase_add_test (tc_chain, test_multifilesink_max_files); |
| tcase_add_test (tc_chain, test_multifilesink_key_unit); |
| tcase_add_test (tc_chain, test_multifilesrc); |
| tcase_add_test (tc_chain, test_multifilesrc_stop_index); |
| |
| return s; |
| } |
| |
| GST_CHECK_MAIN (multifile); |