test: rtph264pay: Verify we only mark the last fragment

This commit is contained in:
Jan Alexander Steffens (heftig) 2019-01-09 15:59:16 +01:00 committed by Nicolas Dufresne
parent 06b2bbd8c7
commit 791711f9be

View file

@ -642,6 +642,47 @@ GST_START_TEST (test_rtph264pay_marker_for_au)
GST_END_TEST;
GST_START_TEST (test_rtph264pay_marker_for_fragmented_au)
{
GstHarness *h =
gst_harness_new_parse ("rtph264pay timestamp-offset=123 mtu=40");
GstFlowReturn ret;
GstBuffer *slice1, *slice2, *buffer;
GstRTPBuffer rtp = GST_RTP_BUFFER_INIT;
gint i;
gst_harness_set_src_caps_str (h,
"video/x-h264,alignment=au,stream-format=byte-stream");
slice1 = wrap_static_buffer (h264_idr_slice_1, sizeof (h264_idr_slice_1));
slice2 = wrap_static_buffer (h264_idr_slice_2, sizeof (h264_idr_slice_2));
buffer = gst_buffer_append (slice1, slice2);
ret = gst_harness_push (h, buffer);
fail_unless_equals_int (ret, GST_FLOW_OK);
fail_unless_equals_int (gst_harness_buffers_in_queue (h), 4);
for (i = 0; i < 3; i++) {
buffer = gst_harness_pull (h);
fail_unless (gst_rtp_buffer_map (buffer, GST_MAP_READ, &rtp));
fail_if (gst_rtp_buffer_get_marker (&rtp));
gst_rtp_buffer_unmap (&rtp);
gst_buffer_unref (buffer);
}
buffer = gst_harness_pull (h);
fail_unless (gst_rtp_buffer_map (buffer, GST_MAP_READ, &rtp));
fail_unless (gst_rtp_buffer_get_marker (&rtp));
gst_rtp_buffer_unmap (&rtp);
gst_buffer_unref (buffer);
gst_harness_teardown (h);
}
GST_END_TEST;
static Suite *
rtph264_suite (void)
{
@ -660,6 +701,7 @@ rtph264_suite (void)
tcase_add_test (tc_chain, test_rtph264pay_two_slices_timestamp);
tcase_add_test (tc_chain, test_rtph264pay_marker_for_flag);
tcase_add_test (tc_chain, test_rtph264pay_marker_for_au);
tcase_add_test (tc_chain, test_rtph264pay_marker_for_fragmented_au);
return s;
}