tests/check/elements/queue2.c - mtk-gstreamer-debian - Git at Google

 /* GStreamer unit tests for queue2
  *
  * Copyright (C) 2011 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.
  */

 #include <gst/check/gstcheck.h>

 static GstElement *
 setup_queue2 (GstElement * pipe, GstElement * input, GstElement * output)
 {
   GstElement *queue2;
   GstPad *sinkpad, *srcpad;

   queue2 = gst_element_factory_make ("queue2", NULL);
   fail_unless (queue2 != NULL, "failed to create 'queue2' element");

   gst_bin_add (GST_BIN (pipe), queue2);
   gst_bin_add (GST_BIN (pipe), input);
   gst_bin_add (GST_BIN (pipe), output);

   sinkpad = gst_element_get_static_pad (queue2, "sink");
   fail_unless (sinkpad != NULL, "failed to get queue2 sink pad");

   srcpad = gst_element_get_static_pad (input, "src");
   fail_unless (srcpad != NULL, "failed to find src pad for input element");

   fail_unless_equals_int (GST_PAD_LINK_OK, gst_pad_link (srcpad, sinkpad));
   gst_object_unref (srcpad);
   gst_object_unref (sinkpad);

   srcpad = gst_element_get_static_pad (queue2, "src");
   fail_unless (srcpad != NULL);

   sinkpad = gst_element_get_static_pad (output, "sink");
   fail_unless (sinkpad != NULL, "failed to find sink pad of output element");

   fail_unless_equals_int (GST_PAD_LINK_OK, gst_pad_link (srcpad, sinkpad));

   gst_object_unref (srcpad);
   gst_object_unref (sinkpad);

   return queue2;
 }

 GST_START_TEST (test_simple_pipeline)
 {
   GstElement *pipe, *input, *output;
   GstMessage *msg;

   pipe = gst_pipeline_new ("pipeline");

   input = gst_element_factory_make ("fakesrc", NULL);
   fail_unless (input != NULL, "failed to create 'fakesrc' element");
   g_object_set (input, "num-buffers", 256, "sizetype", 3, NULL);

   output = gst_element_factory_make ("fakesink", NULL);
   fail_unless (output != NULL, "failed to create 'fakesink' element");

   setup_queue2 (pipe, input, output);

   gst_element_set_state (pipe, GST_STATE_PLAYING);

   msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),
       GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);

   fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR,
       "Expected EOS message, got ERROR message");
   gst_message_unref (msg);

   GST_LOG ("Got EOS, cleaning up");

   gst_element_set_state (pipe, GST_STATE_NULL);
   gst_object_unref (pipe);
 }

 GST_END_TEST;

 GST_START_TEST (test_simple_pipeline_ringbuffer)
 {
   GstElement *pipe, *queue2, *input, *output;
   GstMessage *msg;

   pipe = gst_pipeline_new ("pipeline");

   input = gst_element_factory_make ("fakesrc", NULL);
   fail_unless (input != NULL, "failed to create 'fakesrc' element");
   g_object_set (input, "num-buffers", 256, "sizetype", 3, NULL);

   output = gst_element_factory_make ("fakesink", NULL);
   fail_unless (output != NULL, "failed to create 'fakesink' element");

   queue2 = setup_queue2 (pipe, input, output);
   g_object_set (queue2, "ring-buffer-max-size", (guint64) 1024 * 50, NULL);

   gst_element_set_state (pipe, GST_STATE_PLAYING);

   msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),
       GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);

   fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR,
       "Expected EOS message, got ERROR message");
   gst_message_unref (msg);

   GST_LOG ("Got EOS, cleaning up");

   gst_element_set_state (pipe, GST_STATE_NULL);
   gst_object_unref (pipe);
 }

 GST_END_TEST;

 static void
 do_test_simple_shutdown_while_running (guint64 ring_buffer_max_size)
 {
   GstElement *pipe, *q2;
   GstElement *input;
   GstElement *output;
   GstMessage *msg;

   pipe = gst_pipeline_new ("pipeline");

   input = gst_element_factory_make ("fakesrc", NULL);
   fail_unless (input != NULL, "failed to create 'fakesrc' element");
   g_object_set (input, "format", GST_FORMAT_TIME, "sizetype", 2,
       "sizemax", 10, NULL);

   output = gst_element_factory_make ("fakesink", NULL);
   fail_unless (output != NULL, "failed to create 'fakesink' element");

   q2 = setup_queue2 (pipe, input, output);

   if (ring_buffer_max_size > 0) {
     g_object_set (q2, "ring-buffer-max-size", ring_buffer_max_size,
         "temp-template", NULL, NULL);
   }

   gst_element_set_state (pipe, GST_STATE_PAUSED);

   /* wait until pipeline is up and running */
   msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),
       GST_MESSAGE_ERROR | GST_MESSAGE_ASYNC_DONE, -1);
   fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR, "Got ERROR message");
   gst_message_unref (msg);

   GST_LOG ("pipeline is running now");
   gst_element_set_state (pipe, GST_STATE_PLAYING);
   g_usleep (G_USEC_PER_SEC / 20);

   /* now shut down only the sink, so the queue gets a wrong-state flow return */
   gst_element_set_state (output, GST_STATE_NULL);
   GST_LOG ("Cleaning up");

   gst_element_set_state (pipe, GST_STATE_NULL);
   gst_object_unref (pipe);
 }

 GST_START_TEST (test_simple_shutdown_while_running)
 {
   int i;

   /* run a couple of iterations, gives higher chance of different code paths
    * being executed at time the flush is detected (esp. useful to make sure
    * things are cleaned up properly when running under valgrind) */
   for (i = 0; i < 10; ++i) {
     do_test_simple_shutdown_while_running (0);
   }
 }

 GST_END_TEST;

 GST_START_TEST (test_simple_shutdown_while_running_ringbuffer)
 {
   int i;

   /* run a couple of iterations, gives higher chance of different code paths
    * being executed at time the flush is detected (esp. useful to make sure
    * things are cleaned up properly when running under valgrind) */
   for (i = 0; i < 10; ++i) {
     do_test_simple_shutdown_while_running (1024 * 1024);
   }
 }

 GST_END_TEST;

 GST_START_TEST (test_simple_create_destroy)
 {
   GstElement *queue2;

   queue2 = gst_element_factory_make ("queue2", NULL);
   gst_object_unref (queue2);
 }

 GST_END_TEST;

 static gpointer
 push_buffer (GstPad * sinkpad)
 {
   GstBuffer *buffer;

   buffer = gst_buffer_new_and_alloc (1 * 1024);

   gst_pad_chain (sinkpad, buffer);

   return NULL;
 }

 GST_START_TEST (test_filled_read)
 {
   GstElement *queue2;
   GstBuffer *buffer;
   GstPad *sinkpad, *srcpad;
   GThread *thread;
   GstSegment segment;

   queue2 = gst_element_factory_make ("queue2", NULL);
   sinkpad = gst_element_get_static_pad (queue2, "sink");
   srcpad = gst_element_get_static_pad (queue2, "src");

   g_object_set (queue2, "ring-buffer-max-size", (guint64) 5 * 1024,
       "use-buffering", FALSE,
       "max-size-buffers", (guint) 0, "max-size-time", (guint64) 0,
       "max-size-bytes", (guint) 4 * 1024, NULL);

   gst_pad_activate_mode (srcpad, GST_PAD_MODE_PULL, TRUE);
   gst_element_set_state (queue2, GST_STATE_PLAYING);

   gst_segment_init (&segment, GST_FORMAT_BYTES);
   gst_pad_send_event (sinkpad, gst_event_new_stream_start ("test"));
   gst_pad_send_event (sinkpad, gst_event_new_segment (&segment));

   /* fill up the buffer */
   buffer = gst_buffer_new_and_alloc (4 * 1024);
   fail_unless (gst_pad_chain (sinkpad, buffer) == GST_FLOW_OK);

   thread =
       g_thread_try_new ("gst-check", (GThreadFunc) push_buffer, sinkpad, NULL);
   fail_unless (thread != NULL);

   buffer = NULL;
   fail_unless (gst_pad_get_range (srcpad, 1024, 4 * 1024,
           &buffer) == GST_FLOW_OK);

   fail_unless (gst_buffer_get_size (buffer) == 4 * 1024);
   gst_buffer_unref (buffer);

   gst_element_set_state (queue2, GST_STATE_NULL);

   g_thread_join (thread);

   gst_object_unref (sinkpad);
   gst_object_unref (srcpad);
   gst_object_unref (queue2);
 }

 GST_END_TEST;


 static Suite *
 queue2_suite (void)
 {
   Suite *s = suite_create ("queue2");
   TCase *tc_chain = tcase_create ("general");

   suite_add_tcase (s, tc_chain);
   tcase_add_test (tc_chain, test_simple_create_destroy);
   tcase_add_test (tc_chain, test_simple_pipeline);
   tcase_add_test (tc_chain, test_simple_pipeline_ringbuffer);
   tcase_add_test (tc_chain, test_simple_shutdown_while_running);
   tcase_add_test (tc_chain, test_simple_shutdown_while_running_ringbuffer);
   tcase_add_test (tc_chain, test_filled_read);
   return s;
 }

 GST_CHECK_MAIN (queue2)
	/* GStreamer unit tests for queue2
	*
	* Copyright (C) 2011 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.
	*/

	#include <gst/check/gstcheck.h>

	static GstElement *
	setup_queue2 (GstElement * pipe, GstElement * input, GstElement * output)
	{
	GstElement *queue2;
	GstPad sinkpad, srcpad;

	queue2 = gst_element_factory_make ("queue2", NULL);
	fail_unless (queue2 != NULL, "failed to create 'queue2' element");

	gst_bin_add (GST_BIN (pipe), queue2);
	gst_bin_add (GST_BIN (pipe), input);
	gst_bin_add (GST_BIN (pipe), output);

	sinkpad = gst_element_get_static_pad (queue2, "sink");
	fail_unless (sinkpad != NULL, "failed to get queue2 sink pad");

	srcpad = gst_element_get_static_pad (input, "src");
	fail_unless (srcpad != NULL, "failed to find src pad for input element");

	fail_unless_equals_int (GST_PAD_LINK_OK, gst_pad_link (srcpad, sinkpad));
	gst_object_unref (srcpad);
	gst_object_unref (sinkpad);

	srcpad = gst_element_get_static_pad (queue2, "src");
	fail_unless (srcpad != NULL);

	sinkpad = gst_element_get_static_pad (output, "sink");
	fail_unless (sinkpad != NULL, "failed to find sink pad of output element");

	fail_unless_equals_int (GST_PAD_LINK_OK, gst_pad_link (srcpad, sinkpad));

	gst_object_unref (srcpad);
	gst_object_unref (sinkpad);

	return queue2;
	}

	GST_START_TEST (test_simple_pipeline)
	{
	GstElement pipe, input, *output;
	GstMessage *msg;

	pipe = gst_pipeline_new ("pipeline");

	input = gst_element_factory_make ("fakesrc", NULL);
	fail_unless (input != NULL, "failed to create 'fakesrc' element");
	g_object_set (input, "num-buffers", 256, "sizetype", 3, NULL);

	output = gst_element_factory_make ("fakesink", NULL);
	fail_unless (output != NULL, "failed to create 'fakesink' element");

	setup_queue2 (pipe, input, output);

	gst_element_set_state (pipe, GST_STATE_PLAYING);

	msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),
	GST_MESSAGE_EOS \| GST_MESSAGE_ERROR, -1);

	fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR,
	"Expected EOS message, got ERROR message");
	gst_message_unref (msg);

	GST_LOG ("Got EOS, cleaning up");

	gst_element_set_state (pipe, GST_STATE_NULL);
	gst_object_unref (pipe);
	}

	GST_END_TEST;

	GST_START_TEST (test_simple_pipeline_ringbuffer)
	{
	GstElement pipe, queue2, input, output;
	GstMessage *msg;

	pipe = gst_pipeline_new ("pipeline");

	input = gst_element_factory_make ("fakesrc", NULL);
	fail_unless (input != NULL, "failed to create 'fakesrc' element");
	g_object_set (input, "num-buffers", 256, "sizetype", 3, NULL);

	output = gst_element_factory_make ("fakesink", NULL);
	fail_unless (output != NULL, "failed to create 'fakesink' element");

	queue2 = setup_queue2 (pipe, input, output);
	g_object_set (queue2, "ring-buffer-max-size", (guint64) 1024 * 50, NULL);

	gst_element_set_state (pipe, GST_STATE_PLAYING);

	msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),
	GST_MESSAGE_EOS \| GST_MESSAGE_ERROR, -1);

	fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR,
	"Expected EOS message, got ERROR message");
	gst_message_unref (msg);

	GST_LOG ("Got EOS, cleaning up");

	gst_element_set_state (pipe, GST_STATE_NULL);
	gst_object_unref (pipe);
	}

	GST_END_TEST;

	static void
	do_test_simple_shutdown_while_running (guint64 ring_buffer_max_size)
	{
	GstElement pipe, q2;
	GstElement *input;
	GstElement *output;
	GstMessage *msg;

	pipe = gst_pipeline_new ("pipeline");

	input = gst_element_factory_make ("fakesrc", NULL);
	fail_unless (input != NULL, "failed to create 'fakesrc' element");
	g_object_set (input, "format", GST_FORMAT_TIME, "sizetype", 2,
	"sizemax", 10, NULL);

	output = gst_element_factory_make ("fakesink", NULL);
	fail_unless (output != NULL, "failed to create 'fakesink' element");

	q2 = setup_queue2 (pipe, input, output);

	if (ring_buffer_max_size > 0) {
	g_object_set (q2, "ring-buffer-max-size", ring_buffer_max_size,
	"temp-template", NULL, NULL);
	}

	gst_element_set_state (pipe, GST_STATE_PAUSED);

	/* wait until pipeline is up and running */
	msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),
	GST_MESSAGE_ERROR \| GST_MESSAGE_ASYNC_DONE, -1);
	fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR, "Got ERROR message");
	gst_message_unref (msg);

	GST_LOG ("pipeline is running now");
	gst_element_set_state (pipe, GST_STATE_PLAYING);
	g_usleep (G_USEC_PER_SEC / 20);

	/* now shut down only the sink, so the queue gets a wrong-state flow return */
	gst_element_set_state (output, GST_STATE_NULL);
	GST_LOG ("Cleaning up");

	gst_element_set_state (pipe, GST_STATE_NULL);
	gst_object_unref (pipe);
	}

	GST_START_TEST (test_simple_shutdown_while_running)
	{
	int i;

	/* run a couple of iterations, gives higher chance of different code paths
	* being executed at time the flush is detected (esp. useful to make sure
	* things are cleaned up properly when running under valgrind) */
	for (i = 0; i < 10; ++i) {
	do_test_simple_shutdown_while_running (0);
	}
	}

	GST_END_TEST;

	GST_START_TEST (test_simple_shutdown_while_running_ringbuffer)
	{
	int i;

	/* run a couple of iterations, gives higher chance of different code paths
	* being executed at time the flush is detected (esp. useful to make sure
	* things are cleaned up properly when running under valgrind) */
	for (i = 0; i < 10; ++i) {
	do_test_simple_shutdown_while_running (1024 * 1024);
	}
	}

	GST_END_TEST;

	GST_START_TEST (test_simple_create_destroy)
	{
	GstElement *queue2;

	queue2 = gst_element_factory_make ("queue2", NULL);
	gst_object_unref (queue2);
	}

	GST_END_TEST;

	static gpointer
	push_buffer (GstPad * sinkpad)
	{
	GstBuffer *buffer;

	buffer = gst_buffer_new_and_alloc (1 * 1024);

	gst_pad_chain (sinkpad, buffer);

	return NULL;
	}

	GST_START_TEST (test_filled_read)
	{
	GstElement *queue2;
	GstBuffer *buffer;
	GstPad sinkpad, srcpad;
	GThread *thread;
	GstSegment segment;

	queue2 = gst_element_factory_make ("queue2", NULL);
	sinkpad = gst_element_get_static_pad (queue2, "sink");
	srcpad = gst_element_get_static_pad (queue2, "src");

	g_object_set (queue2, "ring-buffer-max-size", (guint64) 5 * 1024,
	"use-buffering", FALSE,
	"max-size-buffers", (guint) 0, "max-size-time", (guint64) 0,
	"max-size-bytes", (guint) 4 * 1024, NULL);

	gst_pad_activate_mode (srcpad, GST_PAD_MODE_PULL, TRUE);
	gst_element_set_state (queue2, GST_STATE_PLAYING);

	gst_segment_init (&segment, GST_FORMAT_BYTES);
	gst_pad_send_event (sinkpad, gst_event_new_stream_start ("test"));
	gst_pad_send_event (sinkpad, gst_event_new_segment (&segment));

	/* fill up the buffer */
	buffer = gst_buffer_new_and_alloc (4 * 1024);
	fail_unless (gst_pad_chain (sinkpad, buffer) == GST_FLOW_OK);

	thread =
	g_thread_try_new ("gst-check", (GThreadFunc) push_buffer, sinkpad, NULL);
	fail_unless (thread != NULL);

	buffer = NULL;
	fail_unless (gst_pad_get_range (srcpad, 1024, 4 * 1024,
	&buffer) == GST_FLOW_OK);

	fail_unless (gst_buffer_get_size (buffer) == 4 * 1024);
	gst_buffer_unref (buffer);

	gst_element_set_state (queue2, GST_STATE_NULL);

	g_thread_join (thread);

	gst_object_unref (sinkpad);
	gst_object_unref (srcpad);
	gst_object_unref (queue2);
	}

	GST_END_TEST;


	static Suite *
	queue2_suite (void)
	{
	Suite *s = suite_create ("queue2");
	TCase *tc_chain = tcase_create ("general");

	suite_add_tcase (s, tc_chain);
	tcase_add_test (tc_chain, test_simple_create_destroy);
	tcase_add_test (tc_chain, test_simple_pipeline);
	tcase_add_test (tc_chain, test_simple_pipeline_ringbuffer);
	tcase_add_test (tc_chain, test_simple_shutdown_while_running);
	tcase_add_test (tc_chain, test_simple_shutdown_while_running_ringbuffer);
	tcase_add_test (tc_chain, test_filled_read);
	return s;
	}

	GST_CHECK_MAIN (queue2)