From fca02f3dac15987502dde6439bd09b6537eca604 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= Date: Fri, 2 Nov 2018 12:21:45 +0200 Subject: [PATCH] ccextractor: Add unit test https://bugzilla.gnome.org/show_bug.cgi?id=797370 --- tests/check/Makefile.am | 10 + tests/check/elements/.gitignore | 1 + tests/check/elements/ccextractor.c | 342 +++++++++++++++++++++++++++++ tests/check/meson.build | 1 + 4 files changed, 354 insertions(+) create mode 100644 tests/check/elements/ccextractor.c diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index c6e86d4b3d..a1061a93c9 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -40,6 +40,12 @@ else check_assrender = endif +if USE_PANGO +check_ccextractor = elements/ccextractor +else +check_ccextractor = +endif + if USE_DASH check_dash = elements/dash_mpd check_dash_demux = elements/dash_demux @@ -240,6 +246,7 @@ noinst_PROGRAMS = \ check_PROGRAMS = \ generic/states \ $(check_assrender) \ + $(check_ccextractor) \ $(check_dash) \ $(check_dtls) \ $(check_faac) \ @@ -451,6 +458,9 @@ elements_camerabin_LDADD = \ $(GST_BASE_LIBS) $(GST_LIBS) $(LDADD) elements_camerabin_SOURCES = elements/camerabin.c +elements_ccextractor_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_BASE_CFLAGS) $(AM_CFLAGS) +elements_ccextractor_LDADD = $(GST_PLUGINS_BASE_LIBS) $(GST_VIDEO_LIBS) $(GST_BASE_LIBS) $(LDADD) + elements_jifmux_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(EXIF_CFLAGS) $(AM_CFLAGS) elements_jifmux_LDADD = $(GST_PLUGINS_BASE_LIBS) -lgsttag-$(GST_API_VERSION) $(GST_CHECK_LIBS) $(EXIF_LIBS) $(LDADD) elements_jifmux_SOURCES = elements/jifmux.c diff --git a/tests/check/elements/.gitignore b/tests/check/elements/.gitignore index ed045a52c1..f09b5df536 100644 --- a/tests/check/elements/.gitignore +++ b/tests/check/elements/.gitignore @@ -6,6 +6,7 @@ autoconvert autovideoconvert avwait camerabin +ccextractor compositor curlfilesink curlftpsink diff --git a/tests/check/elements/ccextractor.c b/tests/check/elements/ccextractor.c new file mode 100644 index 0000000000..4dd21a7c61 --- /dev/null +++ b/tests/check/elements/ccextractor.c @@ -0,0 +1,342 @@ +/* GStreamer + * + * Copyright (C) 2018 Sebastian Dröge + * + * 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 +#include +#include + +#include + +static GstStaticCaps foo_bar_caps = GST_STATIC_CAPS ("foo/bar"); +static GstStaticCaps cea708_cc_data_caps = +GST_STATIC_CAPS ("closedcaption/x-cea-708,format=(string) cc_data"); +static GstStaticCaps cea708_cdp_caps = +GST_STATIC_CAPS ("closedcaption/x-cea-708,format=(string) cdp"); + +GST_START_TEST (no_captions) +{ + GstHarness *h; + GstBuffer *buf, *outbuf; + GstCaps *caps; + + h = gst_harness_new ("ccextractor"); + + gst_harness_set_src_caps_str (h, "foo/bar"); + + buf = gst_buffer_new_and_alloc (128); + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + fail_unless (gst_element_get_static_pad (h->element, "caption") == NULL); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + gst_buffer_unref (buf); + gst_buffer_unref (outbuf); + + gst_harness_teardown (h); +} + +GST_END_TEST; + +static void +on_caption_pad_added (GstElement * ccextractor, GstPad * pad, GstHarness * h) +{ + fail_unless (strcmp (GST_OBJECT_NAME (pad), "caption") == 0); + gst_harness_add_element_src_pad (h, pad); +} + +GST_START_TEST (captions) +{ + GstHarness *h, *h2; + GstBuffer *buf, *outbuf; + const guint8 caption_data[] = { 0, 1, 2, 3, 4, 5, 6, 7 }; + GstCaps *caps; + + h = gst_harness_new ("ccextractor"); + h2 = gst_harness_new_with_element (h->element, NULL, NULL); + + g_signal_connect (h->element, "pad-added", G_CALLBACK (on_caption_pad_added), + h2); + + gst_harness_set_src_caps_str (h, "foo/bar"); + + buf = gst_buffer_new_and_alloc (128); + gst_buffer_add_video_caption_meta (buf, GST_VIDEO_CAPTION_TYPE_CEA708_RAW, + caption_data, sizeof (caption_data)); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + fail_unless (h2->sinkpad != NULL); + outbuf = gst_harness_pull (h2); + fail_unless (outbuf != NULL); + fail_unless (gst_buffer_memcmp (outbuf, 0, caption_data, + sizeof (caption_data)) == 0); + gst_buffer_unref (outbuf); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + caps = gst_pad_get_current_caps (h2->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&cea708_cc_data_caps))); + gst_caps_unref (caps); + + buf = gst_buffer_new_and_alloc (128); + gst_buffer_add_video_caption_meta (buf, GST_VIDEO_CAPTION_TYPE_CEA708_RAW, + caption_data, sizeof (caption_data)); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + fail_unless (h2->sinkpad != NULL); + outbuf = gst_harness_pull (h2); + fail_unless (outbuf != NULL); + fail_unless (gst_buffer_memcmp (outbuf, 0, caption_data, + sizeof (caption_data)) == 0); + gst_buffer_unref (outbuf); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + caps = gst_pad_get_current_caps (h2->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&cea708_cc_data_caps))); + gst_caps_unref (caps); + + gst_harness_teardown (h); + gst_harness_teardown (h2); +} + +GST_END_TEST; + +GST_START_TEST (no_captions_at_beginning_and_end) +{ + GstHarness *h, *h2; + GstBuffer *buf, *outbuf; + const guint8 caption_data[] = { 0, 1, 2, 3, 4, 5, 6, 7 }; + GstCaps *caps; + + h = gst_harness_new ("ccextractor"); + h2 = gst_harness_new_with_element (h->element, NULL, NULL); + + g_signal_connect (h->element, "pad-added", G_CALLBACK (on_caption_pad_added), + h2); + + gst_harness_set_src_caps_str (h, "foo/bar"); + + buf = gst_buffer_new_and_alloc (128); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + buf = gst_buffer_new_and_alloc (128); + gst_buffer_add_video_caption_meta (buf, GST_VIDEO_CAPTION_TYPE_CEA708_RAW, + caption_data, sizeof (caption_data)); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + fail_unless (h2->sinkpad != NULL); + outbuf = gst_harness_pull (h2); + fail_unless (outbuf != NULL); + fail_unless (gst_buffer_memcmp (outbuf, 0, caption_data, + sizeof (caption_data)) == 0); + gst_buffer_unref (outbuf); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + caps = gst_pad_get_current_caps (h2->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&cea708_cc_data_caps))); + gst_caps_unref (caps); + + buf = gst_buffer_new_and_alloc (128); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + fail_if (gst_harness_try_pull (h2) != NULL); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + caps = gst_pad_get_current_caps (h2->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&cea708_cc_data_caps))); + gst_caps_unref (caps); + + gst_harness_teardown (h); + gst_harness_teardown (h2); +} + +GST_END_TEST; + +GST_START_TEST (captions_format_change) +{ + GstHarness *h, *h2; + GstBuffer *buf, *outbuf; + const guint8 caption_data[] = { 0, 1, 2, 3, 4, 5, 6, 7 }; + GstCaps *caps; + + h = gst_harness_new ("ccextractor"); + h2 = gst_harness_new_with_element (h->element, NULL, NULL); + + g_signal_connect (h->element, "pad-added", G_CALLBACK (on_caption_pad_added), + h2); + + gst_harness_set_src_caps_str (h, "foo/bar"); + + buf = gst_buffer_new_and_alloc (128); + gst_buffer_add_video_caption_meta (buf, GST_VIDEO_CAPTION_TYPE_CEA708_RAW, + caption_data, sizeof (caption_data)); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + fail_unless (h2->sinkpad != NULL); + outbuf = gst_harness_pull (h2); + fail_unless (outbuf != NULL); + fail_unless (gst_buffer_memcmp (outbuf, 0, caption_data, + sizeof (caption_data)) == 0); + gst_buffer_unref (outbuf); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + caps = gst_pad_get_current_caps (h2->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&cea708_cc_data_caps))); + gst_caps_unref (caps); + + buf = gst_buffer_new_and_alloc (128); + gst_buffer_add_video_caption_meta (buf, GST_VIDEO_CAPTION_TYPE_CEA708_CDP, + caption_data, sizeof (caption_data)); + + outbuf = gst_harness_push_and_pull (h, gst_buffer_ref (buf)); + + fail_unless (outbuf != NULL); + fail_unless (outbuf == buf); + gst_buffer_unref (outbuf); + gst_buffer_unref (buf); + + fail_unless (h2->sinkpad != NULL); + outbuf = gst_harness_pull (h2); + fail_unless (outbuf != NULL); + fail_unless (gst_buffer_memcmp (outbuf, 0, caption_data, + sizeof (caption_data)) == 0); + gst_buffer_unref (outbuf); + + caps = gst_pad_get_current_caps (h->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&foo_bar_caps))); + gst_caps_unref (caps); + + caps = gst_pad_get_current_caps (h2->sinkpad); + fail_unless (caps != NULL); + fail_unless (gst_caps_can_intersect (caps, + gst_static_caps_get (&cea708_cdp_caps))); + gst_caps_unref (caps); + + gst_harness_teardown (h); + gst_harness_teardown (h2); +} + +GST_END_TEST; + +static Suite * +ccextractor_suite (void) +{ + Suite *s = suite_create ("ccextractor"); + TCase *tc = tcase_create ("general"); + + suite_add_tcase (s, tc); + + tcase_add_test (tc, no_captions); + tcase_add_test (tc, captions); + tcase_add_test (tc, no_captions_at_beginning_and_end); + tcase_add_test (tc, captions_format_change); + + return s; +} + +GST_CHECK_MAIN (ccextractor); diff --git a/tests/check/meson.build b/tests/check/meson.build index 9e3b9ed6b3..4f698394d1 100644 --- a/tests/check/meson.build +++ b/tests/check/meson.build @@ -23,6 +23,7 @@ base_tests = [ [['elements/autovideoconvert.c']], [['elements/avwait.c']], [['elements/camerabin.c']], + [['elements/ccextractor.c']], [['elements/compositor.c']], [['elements/curlhttpsink.c'], not curl_dep.found(), [curl_dep]], [['elements/curlfilesink.c'], not curl_dep.found(), [curl_dep]],