gstreamer/tests/check/libs/flowcombiner.c

249 lines
8.4 KiB
C
Raw Normal View History

/* GStreamer
*
* Copyright (C) 2014 Samsung Electronics. All rights reserved.
* Author: Thiago Santos <ts.santos@sisa.samsung.com>
*
* flowcombiner.c: Unit test for GstFlowCombiner
*
* 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 <gst/check/gstcheck.h>
#include <gst/base/gstflowcombiner.h>
static GstFlowReturn sink_flowret = GST_FLOW_OK;
#define CHECK_COMBINED_FLOWS(f1, f2, f3, expected) \
G_STMT_START { \
combiner = gst_flow_combiner_new (); \
gst_flow_combiner_add_pad (combiner, pad1); \
gst_flow_combiner_add_pad (combiner, pad2); \
gst_flow_combiner_add_pad (combiner, pad3); \
sink_flowret = f1; \
gst_pad_push (pad1, gst_buffer_new ()); \
gst_flow_combiner_update_flow (combiner, f1); \
sink_flowret = f2; \
gst_pad_push (pad2, gst_buffer_new ()); \
gst_flow_combiner_update_flow (combiner, f2); \
sink_flowret = f3; \
gst_pad_push (pad3, gst_buffer_new ()); \
ret = gst_flow_combiner_update_flow (combiner, f3); \
gst_flow_combiner_free (combiner); \
fail_unless_equals_int (ret, expected); \
} G_STMT_END
static GstFlowReturn
_sink_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
{
gst_buffer_unref (buf);
return sink_flowret;
}
GST_START_TEST (test_combined_flows)
{
GstFlowReturn ret;
GstFlowCombiner *combiner;
GstPad *pad1, *pad2, *pad3;
GstPad *peer1, *peer2, *peer3;
GstSegment segment;
pad1 = gst_pad_new ("src1", GST_PAD_SRC);
pad2 = gst_pad_new ("src2", GST_PAD_SRC);
pad3 = gst_pad_new ("src3", GST_PAD_SRC);
peer1 = gst_pad_new ("sink1", GST_PAD_SINK);
peer2 = gst_pad_new ("sink2", GST_PAD_SINK);
peer3 = gst_pad_new ("sink3", GST_PAD_SINK);
gst_pad_set_chain_function (peer1, _sink_chain);
gst_pad_set_chain_function (peer2, _sink_chain);
gst_pad_set_chain_function (peer3, _sink_chain);
gst_pad_link (pad1, peer1);
gst_pad_link (pad2, peer2);
gst_pad_link (pad3, peer3);
gst_pad_set_active (peer1, TRUE);
gst_pad_set_active (peer2, TRUE);
gst_pad_set_active (peer3, TRUE);
gst_pad_set_active (pad1, TRUE);
gst_pad_set_active (pad2, TRUE);
gst_pad_set_active (pad3, TRUE);
gst_segment_init (&segment, GST_FORMAT_BYTES);
gst_pad_push_event (pad1, gst_event_new_stream_start ("p1"));
gst_pad_push_event (pad2, gst_event_new_stream_start ("p2"));
gst_pad_push_event (pad3, gst_event_new_stream_start ("p3"));
gst_pad_push_event (pad1, gst_event_new_segment (&segment));
gst_pad_push_event (pad2, gst_event_new_segment (&segment));
gst_pad_push_event (pad3, gst_event_new_segment (&segment));
/* ok */
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_OK, GST_FLOW_OK, GST_FLOW_OK);
/* not linked */
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_NOT_LINKED, GST_FLOW_OK,
GST_FLOW_OK);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_EOS, GST_FLOW_OK, GST_FLOW_OK);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_NOT_LINKED, GST_FLOW_NOT_LINKED,
GST_FLOW_OK);
CHECK_COMBINED_FLOWS (GST_FLOW_NOT_LINKED, GST_FLOW_NOT_LINKED,
GST_FLOW_NOT_LINKED, GST_FLOW_NOT_LINKED);
/* errors */
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_ERROR, GST_FLOW_OK,
GST_FLOW_ERROR);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_CUSTOM_ERROR, GST_FLOW_OK,
GST_FLOW_CUSTOM_ERROR);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_NOT_NEGOTIATED, GST_FLOW_OK,
GST_FLOW_NOT_NEGOTIATED);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_OK, GST_FLOW_NOT_NEGOTIATED,
GST_FLOW_NOT_NEGOTIATED);
CHECK_COMBINED_FLOWS (GST_FLOW_NOT_LINKED, GST_FLOW_ERROR, GST_FLOW_OK,
GST_FLOW_ERROR);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_OK, GST_FLOW_ERROR,
GST_FLOW_ERROR);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_OK, GST_FLOW_CUSTOM_ERROR,
GST_FLOW_CUSTOM_ERROR);
/* flushing */
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_OK, GST_FLOW_FLUSHING,
GST_FLOW_FLUSHING);
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_FLUSHING, GST_FLOW_OK,
GST_FLOW_FLUSHING);
CHECK_COMBINED_FLOWS (GST_FLOW_FLUSHING, GST_FLOW_FLUSHING, GST_FLOW_FLUSHING,
GST_FLOW_FLUSHING);
/* eos */
CHECK_COMBINED_FLOWS (GST_FLOW_OK, GST_FLOW_NOT_LINKED, GST_FLOW_EOS,
GST_FLOW_OK);
CHECK_COMBINED_FLOWS (GST_FLOW_EOS, GST_FLOW_OK, GST_FLOW_EOS, GST_FLOW_OK);
CHECK_COMBINED_FLOWS (GST_FLOW_EOS, GST_FLOW_EOS, GST_FLOW_EOS, GST_FLOW_EOS);
/* eos + not-linked */
CHECK_COMBINED_FLOWS (GST_FLOW_NOT_LINKED, GST_FLOW_EOS, GST_FLOW_EOS,
GST_FLOW_EOS);
CHECK_COMBINED_FLOWS (GST_FLOW_NOT_LINKED, GST_FLOW_NOT_LINKED, GST_FLOW_EOS,
GST_FLOW_EOS);
gst_object_unref (pad1);
gst_object_unref (pad2);
gst_object_unref (pad3);
gst_object_unref (peer1);
gst_object_unref (peer2);
gst_object_unref (peer3);
}
GST_END_TEST;
GST_START_TEST (test_clear)
{
GstFlowCombiner *combiner;
GstPad *pad;
GstPad *peer;
GstSegment segment;
GstFlowReturn ret;
combiner = gst_flow_combiner_new ();
/* add a pad and make it return _FLUSHING */
pad = gst_pad_new ("src1", GST_PAD_SRC);
peer = gst_pad_new ("sink1", GST_PAD_SINK);
gst_pad_set_chain_function (peer, _sink_chain);
gst_pad_link (pad, peer);
gst_pad_set_active (peer, TRUE);
gst_pad_set_active (pad, TRUE);
gst_segment_init (&segment, GST_FORMAT_BYTES);
gst_pad_push_event (pad, gst_event_new_stream_start ("test1"));
gst_pad_push_event (pad, gst_event_new_segment (&segment));
gst_flow_combiner_add_pad (combiner, pad);
sink_flowret = GST_FLOW_FLUSHING;
fail_unless_equals_int (gst_pad_push (pad, gst_buffer_new ()),
GST_FLOW_FLUSHING);
/* the combined flow is _FLUSHING */
ret = gst_flow_combiner_update_flow (combiner, GST_FLOW_FLUSHING);
fail_unless_equals_int (ret, GST_FLOW_FLUSHING);
gst_object_unref (pad);
gst_object_unref (peer);
/* add one more pad and make it return _OK */
pad = gst_pad_new ("src2", GST_PAD_SRC);
peer = gst_pad_new ("sink2", GST_PAD_SINK);
gst_pad_set_chain_function (peer, _sink_chain);
gst_pad_link (pad, peer);
gst_pad_set_active (peer, TRUE);
gst_pad_set_active (pad, TRUE);
gst_segment_init (&segment, GST_FORMAT_BYTES);
gst_pad_push_event (pad, gst_event_new_stream_start ("test2"));
gst_pad_push_event (pad, gst_event_new_segment (&segment));
gst_flow_combiner_add_pad (combiner, pad);
sink_flowret = GST_FLOW_OK;
fail_unless_equals_int (gst_pad_push (pad, gst_buffer_new ()), GST_FLOW_OK);
/* the combined flow is _FLUSHING because of the first pad */
ret = gst_flow_combiner_update_flow (combiner, GST_FLOW_OK);
fail_unless_equals_int (ret, GST_FLOW_FLUSHING);
gst_object_unref (pad);
gst_object_unref (peer);
/* clear the combiner */
gst_flow_combiner_clear (combiner);
/* add a pad and make it return _OK */
pad = gst_pad_new ("src3", GST_PAD_SRC);
peer = gst_pad_new ("sink3", GST_PAD_SINK);
gst_pad_set_chain_function (peer, _sink_chain);
gst_pad_link (pad, peer);
gst_pad_set_active (peer, TRUE);
gst_pad_set_active (pad, TRUE);
gst_segment_init (&segment, GST_FORMAT_BYTES);
gst_pad_push_event (pad, gst_event_new_stream_start ("test3"));
gst_pad_push_event (pad, gst_event_new_segment (&segment));
gst_flow_combiner_add_pad (combiner, pad);
sink_flowret = GST_FLOW_OK;
fail_unless_equals_int (gst_pad_push (pad, gst_buffer_new ()), GST_FLOW_OK);
/* the combined flow is _OK since the other pads have been removed */
ret = gst_flow_combiner_update_flow (combiner, GST_FLOW_OK);
fail_unless_equals_int (ret, GST_FLOW_OK);
gst_object_unref (pad);
gst_object_unref (peer);
gst_flow_combiner_free (combiner);
}
GST_END_TEST;
static Suite *
flow_combiner_suite (void)
{
Suite *s = suite_create ("GstFlowCombiner");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_combined_flows);
tcase_add_test (tc_chain, test_clear);
return s;
}
GST_CHECK_MAIN (flow_combiner);