--- /dev/null
+/* GStreamer
+ *
+ * unit test for gstrtpsession
+ *
+ * Copyright (C) <2009> Wim Taymans <wim.taymans@gmail.com>
+ *
+ * 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>
+#include <gst/check/gsttestclock.h>
+
+#include <gst/rtp/gstrtpbuffer.h>
+#include <gst/rtp/gstrtcpbuffer.h>
+
+static const guint payload_size = 160;
+static const guint clock_rate = 8000;
+static const guint payload_type = 0;
+
+typedef struct
+{
+ GstElement *session;
+ GstPad *src, *rtcp_sink, *rtpsrc;
+ GstClock *clock;
+ GAsyncQueue *rtcp_queue;
+} TestData;
+
+static GstCaps *
+generate_caps (void)
+{
+ return gst_caps_new_simple ("application/x-rtp",
+ "clock-rate", G_TYPE_INT, clock_rate,
+ "payload-type", G_TYPE_INT, payload_type, NULL);
+}
+
+static GstBuffer *
+generate_test_buffer (GstClockTime gst_ts,
+ gboolean marker_bit, guint seq_num, guint32 rtp_ts, guint ssrc)
+{
+ GstBuffer *buf;
+ guint8 *payload;
+ guint i;
+ GstRTPBuffer rtp = GST_RTP_BUFFER_INIT;
+
+ buf = gst_rtp_buffer_new_allocate (payload_size, 0, 0);
+ GST_BUFFER_DTS (buf) = gst_ts;
+ GST_BUFFER_PTS (buf) = gst_ts;
+
+ gst_rtp_buffer_map (buf, GST_MAP_READWRITE, &rtp);
+ gst_rtp_buffer_set_payload_type (&rtp, payload_type);
+ gst_rtp_buffer_set_marker (&rtp, marker_bit);
+ gst_rtp_buffer_set_seq (&rtp, seq_num);
+ gst_rtp_buffer_set_timestamp (&rtp, rtp_ts);
+ gst_rtp_buffer_set_ssrc (&rtp, ssrc);
+
+ payload = gst_rtp_buffer_get_payload (&rtp);
+ for (i = 0; i < payload_size; i++)
+ payload[i] = 0xff;
+
+ gst_rtp_buffer_unmap (&rtp);
+
+ return buf;
+}
+
+static GstFlowReturn
+test_sink_pad_chain_cb (GstPad * pad, GstObject * parent, GstBuffer * buffer)
+{
+ TestData *data = gst_pad_get_element_private (pad);
+ g_async_queue_push (data->rtcp_queue, buffer);
+ GST_DEBUG ("chained");
+ return GST_FLOW_OK;
+}
+
+static GstCaps *
+pt_map_requested (GstElement * elemen, guint pt, gpointer data)
+{
+ return generate_caps ();
+}
+
+static void
+destroy_testharness (TestData * data)
+{
+ g_assert_cmpint (gst_element_set_state (data->session, GST_STATE_NULL),
+ ==, GST_STATE_CHANGE_SUCCESS);
+ gst_object_unref (data->session);
+ data->session = NULL;
+
+ gst_object_unref (data->src);
+ data->src = NULL;
+
+ gst_object_unref (data->rtcp_sink);
+ data->rtcp_sink = NULL;
+
+ gst_object_unref (data->rtpsrc);
+ data->rtpsrc = NULL;
+
+ gst_object_unref (data->clock);
+ data->clock = NULL;
+
+ g_async_queue_unref (data->rtcp_queue);
+ data->rtcp_queue = NULL;
+}
+
+static void
+setup_testharness (TestData * data)
+{
+ GstPad *rtp_sink_pad, *rtcp_src_pad, *rtp_src_pad;
+ GstSegment seg;
+ GstMiniObject *obj;
+
+ data->clock = gst_test_clock_new ();
+ GST_DEBUG ("Setting default system clock to test clock");
+ gst_system_clock_set_default (data->clock);
+ g_assert (data->clock);
+ gst_test_clock_set_time (GST_TEST_CLOCK (data->clock), 0);
+
+ data->session = gst_element_factory_make ("rtpsession", NULL);
+ g_signal_connect (data->session, "request-pt-map",
+ (GCallback) pt_map_requested, data);
+ g_assert (data->session);
+ gst_element_set_clock (data->session, data->clock);
+ g_assert_cmpint (gst_element_set_state (data->session,
+ GST_STATE_PLAYING), !=, GST_STATE_CHANGE_FAILURE);
+
+ data->rtcp_queue =
+ g_async_queue_new_full ((GDestroyNotify) gst_mini_object_unref);
+
+ /* link in the test source-pad */
+ data->src = gst_pad_new ("src", GST_PAD_SRC);
+ g_assert (data->src);
+ rtp_sink_pad = gst_element_get_request_pad (data->session, "recv_rtp_sink");
+ g_assert (rtp_sink_pad);
+ g_assert_cmpint (gst_pad_link (data->src, rtp_sink_pad), ==, GST_PAD_LINK_OK);
+ gst_object_unref (rtp_sink_pad);
+
+ data->rtpsrc = gst_pad_new ("sink", GST_PAD_SINK);
+ g_assert (data->rtpsrc);
+ rtp_src_pad = gst_element_get_static_pad (data->session, "recv_rtp_src");
+ g_assert (rtp_src_pad);
+ g_assert_cmpint (gst_pad_link (rtp_src_pad, data->rtpsrc), ==,
+ GST_PAD_LINK_OK);
+ gst_object_unref (rtp_src_pad);
+
+ /* link in the test sink-pad */
+ data->rtcp_sink = gst_pad_new ("sink", GST_PAD_SINK);
+ g_assert (data->rtcp_sink);
+ gst_pad_set_element_private (data->rtcp_sink, data);
+ gst_pad_set_caps (data->rtcp_sink, generate_caps ());
+ gst_pad_set_chain_function (data->rtcp_sink, test_sink_pad_chain_cb);
+ rtcp_src_pad = gst_element_get_request_pad (data->session, "send_rtcp_src");
+ g_assert (rtcp_src_pad);
+ g_assert_cmpint (gst_pad_link (rtcp_src_pad, data->rtcp_sink), ==,
+ GST_PAD_LINK_OK);
+ gst_object_unref (rtcp_src_pad);
+
+ g_assert (gst_pad_set_active (data->src, TRUE));
+ g_assert (gst_pad_set_active (data->rtcp_sink, TRUE));
+
+ gst_segment_init (&seg, GST_FORMAT_TIME);
+ gst_pad_push_event (data->src, gst_event_new_stream_start ("stream0"));
+ gst_pad_set_caps (data->src, generate_caps ());
+ gst_pad_push_event (data->src, gst_event_new_segment (&seg));
+
+ while ((obj = g_async_queue_try_pop (data->rtcp_queue)))
+ gst_mini_object_unref (obj);
+}
+
+GST_START_TEST (test_multiple_ssrc_rr)
+{
+ TestData data;
+ GstFlowReturn res;
+ GstClockID id;
+ GstBuffer *in_buf, *out_buf;
+ GstRTCPBuffer rtcp = GST_RTCP_BUFFER_INIT;
+ GstRTCPPacket rtcp_packet;
+ int i;
+ guint32 ssrc, exthighestseq, jitter, lsr, dlsr;
+ gint32 packetslost;
+ guint8 fractionlost;
+
+ setup_testharness (&data);
+
+ gst_test_clock_set_time (GST_TEST_CLOCK (data.clock), 10 * GST_MSECOND);
+
+ for (i = 0; i < 5; i++) {
+ GST_DEBUG ("Push %i", i);
+ in_buf =
+ generate_test_buffer (i * 20 * GST_MSECOND, FALSE, i, i * 20,
+ 0x01BADBAD);
+ res = gst_pad_push (data.src, in_buf);
+ fail_unless (res == GST_FLOW_OK || res == GST_FLOW_FLUSHING);
+
+
+ gst_test_clock_wait_for_next_pending_id (GST_TEST_CLOCK (data.clock), &id);
+ gst_test_clock_process_next_clock_id (GST_TEST_CLOCK (data.clock));
+
+ in_buf =
+ generate_test_buffer (i * 20 * GST_MSECOND, FALSE, i, i * 20,
+ 0xDEADBEEF);
+ res = gst_pad_push (data.src, in_buf);
+ fail_unless (res == GST_FLOW_OK || res == GST_FLOW_FLUSHING);
+
+ gst_test_clock_wait_for_next_pending_id (GST_TEST_CLOCK (data.clock), &id);
+ gst_test_clock_process_next_clock_id (GST_TEST_CLOCK (data.clock));
+ GST_DEBUG ("pushed %i", i);
+ gst_test_clock_set_time (GST_TEST_CLOCK (data.clock),
+ gst_clock_id_get_time (id));
+ }
+
+ gst_test_clock_set_time (GST_TEST_CLOCK (data.clock),
+ gst_clock_id_get_time (id) + (2 * GST_SECOND));
+ gst_test_clock_wait_for_next_pending_id (GST_TEST_CLOCK (data.clock), &id);
+ gst_test_clock_process_next_clock_id (GST_TEST_CLOCK (data.clock));
+
+ out_buf = g_async_queue_pop (data.rtcp_queue);
+ g_assert (out_buf != NULL);
+ g_assert (gst_rtcp_buffer_validate (out_buf));
+ gst_rtcp_buffer_map (out_buf, GST_MAP_READ, &rtcp);
+ g_assert (gst_rtcp_buffer_get_first_packet (&rtcp, &rtcp_packet));
+ g_assert (gst_rtcp_packet_get_type (&rtcp_packet) == GST_RTCP_TYPE_RR);
+ g_assert_cmpint (gst_rtcp_packet_get_rb_count (&rtcp_packet), ==, 2);
+
+ gst_rtcp_packet_get_rb (&rtcp_packet, 0, &ssrc, &fractionlost, &packetslost,
+ &exthighestseq, &jitter, &lsr, &dlsr);
+
+ g_assert_cmpint (ssrc, ==, 0x01BADBAD);
+
+ gst_rtcp_packet_get_rb (&rtcp_packet, 1, &ssrc, &fractionlost, &packetslost,
+ &exthighestseq, &jitter, &lsr, &dlsr);
+ g_assert_cmpint (ssrc, ==, 0xDEADBEEF);
+
+ destroy_testharness (&data);
+}
+
+GST_END_TEST;
+
+static Suite *
+gstrtpsession_suite (void)
+{
+ Suite *s = suite_create ("rtpsession");
+ TCase *tc_chain = tcase_create ("general");
+
+ suite_add_tcase (s, tc_chain);
+ tcase_add_test (tc_chain, test_multiple_ssrc_rr);
+
+ return s;
+}
+
+int
+main (int argc, char **argv)
+{
+ int nf;
+
+ Suite *s = gstrtpsession_suite ();
+ SRunner *sr = srunner_create (s);
+
+ gst_check_init (&argc, &argv);
+
+ srunner_run_all (sr, CK_NORMAL);
+ nf = srunner_ntests_failed (sr);
+ srunner_free (sr);
+
+ return nf;
+}