/* GStreamer * * Copyright (C) 2014 Samsung Electronics. All rights reserved. * Author: Thiago Santos * * 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 #include 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);