From 1fdaf6e8840e1e9ce9c0375ecd7bf01e3cdf5106 Mon Sep 17 00:00:00 2001 From: Jan Schmidt Date: Sat, 24 Oct 2009 12:54:19 +0100 Subject: [PATCH] autoconvert: Add a simple unit test Set up an autoconvert with 2 factories, and check that it can switch between them when the caps change. --- tests/check/Makefile.am | 1 + tests/check/elements/.gitignore | 1 + tests/check/elements/autoconvert.c | 239 +++++++++++++++++++++++++++++ 3 files changed, 241 insertions(+) create mode 100644 tests/check/elements/autoconvert.c diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index 26d5aad24b..8c9b902bd4 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -109,6 +109,7 @@ check_PROGRAMS = \ $(check_kate) \ elements/aacparse \ elements/amrparse \ + elements/autoconvert \ elements/asfmux \ elements/capssetter \ elements/legacyresample \ diff --git a/tests/check/elements/.gitignore b/tests/check/elements/.gitignore index 04110805a5..fb563f5af9 100644 --- a/tests/check/elements/.gitignore +++ b/tests/check/elements/.gitignore @@ -2,6 +2,7 @@ aacparse amrparse asfmux +autoconvert camerabin capssetter deinterleave diff --git a/tests/check/elements/autoconvert.c b/tests/check/elements/autoconvert.c new file mode 100644 index 0000000000..d79334c092 --- /dev/null +++ b/tests/check/elements/autoconvert.c @@ -0,0 +1,239 @@ +/* GStreamer + * + * unit test for autoconvert element + * Copyright (C) 2009 Jan Schmidt + * + * 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., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include +#include + +/* Define 2 element factories for testing with */ +typedef GstBin TestElement1; +typedef GstBinClass TestElement1Class; +typedef GstBin TestElement2; +typedef GstBinClass TestElement2Class; + +G_DEFINE_TYPE (TestElement1, test_element1, GST_TYPE_BIN); +G_DEFINE_TYPE (TestElement2, test_element2, GST_TYPE_BIN); + +static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, GST_PAD_ALWAYS, + GST_STATIC_CAPS ("test/caps,type=(int)[1,2]")); +static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, GST_PAD_ALWAYS, + GST_STATIC_CAPS ("test/caps,type=(int)[1,2]")); + +static void +setup (void) +{ + /* Register our test elements */ + fail_unless (gst_element_register (NULL, "testelement1", GST_RANK_NONE, + test_element1_get_type ())); + fail_unless (gst_element_register (NULL, "testelement2", GST_RANK_NONE, + test_element2_get_type ())); +} + +static void +teardown (void) +{ +} + +static void +set_autoconvert_factories (GstElement * autoconvert) +{ + gchar *desired_features[] = { "testelement1", "testelement2" }; + GstElementFactory *feature; + GList *factories = NULL; + gint i; + + for (i = 0; i < G_N_ELEMENTS (desired_features); i++) { + feature = + GST_ELEMENT_FACTORY_CAST (gst_default_registry_find_feature + (desired_features[i], GST_TYPE_ELEMENT_FACTORY)); + fail_if (feature == NULL, "Test element %s was not found in registry", + desired_features[i]); + factories = g_list_prepend (factories, feature); + } + + g_object_set (G_OBJECT (autoconvert), "factories", factories, NULL); +} + +typedef struct +{ + gint n; + GstCaps *caps; +} TestContext; + +static void +generate_test_buffer (GstPad * src, TestContext * ctx) +{ + GstBuffer *buf; + + if (ctx->n == 0) { + ctx->caps = gst_caps_from_string ("test/caps,type=(int)1"); + } else if (ctx->n == 10) { + gst_caps_unref (ctx->caps); + ctx->caps = gst_caps_from_string ("test/caps,type=(int)2"); + } + + buf = gst_buffer_new_and_alloc (4096); + gst_buffer_set_caps (buf, ctx->caps); + + GST_LOG ("Pushing test buffer, caps %" GST_PTR_FORMAT, ctx->caps); + fail_unless (gst_pad_push (src, buf) == GST_FLOW_OK); +} + +GST_START_TEST (test_autoconvert_simple) +{ + GstPad *test_src_pad, *test_sink_pad; + GstElement *autoconvert = gst_check_setup_element ("autoconvert"); + GstBus *bus = gst_bus_new (); + TestContext ctx = { 0 }; + + set_autoconvert_factories (autoconvert); + + test_src_pad = gst_check_setup_src_pad (autoconvert, &src_factory, NULL); + gst_pad_set_active (test_src_pad, TRUE); + test_sink_pad = gst_check_setup_sink_pad (autoconvert, &sink_factory, NULL); + gst_pad_set_active (test_sink_pad, TRUE); + + gst_element_set_state (GST_ELEMENT_CAST (autoconvert), GST_STATE_PLAYING); + + /* Push 20 items */ + for (ctx.n = 0; ctx.n < 20; ctx.n++) + generate_test_buffer (test_src_pad, &ctx); + + /* Check all the items arrived */ + fail_unless_equals_int (g_list_length (buffers), 20); + + while (TRUE) { + GstMessage *msg = gst_bus_pop (bus); + if (!msg) + break; + + GST_DEBUG ("got message %s", + gst_message_type_get_name (GST_MESSAGE_TYPE (msg))); + fail_if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR); + } + + gst_element_set_state ((GstElement *) autoconvert, GST_STATE_NULL); + + gst_bus_set_flushing (bus, TRUE); + gst_object_unref (bus); + + gst_pad_set_active (test_src_pad, FALSE); + gst_pad_set_active (test_sink_pad, FALSE); + gst_check_teardown_src_pad (autoconvert); + gst_check_teardown_sink_pad (autoconvert); + gst_check_teardown_element (autoconvert); +} + +GST_END_TEST; + +Suite * +autoconvert_suite (void) +{ + Suite *s = suite_create ("autoconvert"); + TCase *tc_basic = tcase_create ("general"); + + suite_add_tcase (s, tc_basic); + tcase_add_checked_fixture (tc_basic, setup, teardown); + tcase_add_test (tc_basic, test_autoconvert_simple); + + return s; +} + +/* Implementation of the test elements */ + +static void +configure_test_element (GstBin * bin, gchar * capsfilter) +{ + GstElement *filter; + GstElement *identity; + GstPad *pad, *ghostpad; + GstPadTemplate *test_static_templ; + + filter = gst_element_factory_make ("capsfilter", NULL); + fail_unless (filter != NULL); + gst_util_set_object_arg (G_OBJECT (filter), "caps", capsfilter); + + identity = gst_element_factory_make ("identity", NULL); + fail_unless (identity != NULL); + + gst_bin_add_many (bin, filter, identity, NULL); + fail_unless (gst_element_link_many (filter, identity, NULL) == TRUE); + + + test_static_templ = gst_static_pad_template_get (&sink_factory); + pad = gst_element_get_static_pad (filter, "sink"); + ghostpad = gst_ghost_pad_new_from_template ("sink", pad, test_static_templ); + gst_element_add_pad (GST_ELEMENT_CAST (bin), ghostpad); + gst_object_unref (pad); + gst_object_unref (test_static_templ); + + test_static_templ = gst_static_pad_template_get (&src_factory); + pad = gst_element_get_static_pad (identity, "src"); + ghostpad = gst_ghost_pad_new_from_template ("src", pad, test_static_templ); + gst_element_add_pad (GST_ELEMENT_CAST (bin), ghostpad); + gst_object_unref (pad); + gst_object_unref (test_static_templ); +} + +static void +test_element1_class_init (TestElement1Class * klass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (klass); + GstPadTemplate *src_template, *sink_template; + + src_template = gst_static_pad_template_get (&src_factory); + gst_element_class_add_pad_template (element_class, src_template); + + sink_template = gst_static_pad_template_get (&sink_factory); + gst_element_class_add_pad_template (element_class, sink_template); +} + +static void +test_element1_init (TestElement1 * elem) +{ + configure_test_element (GST_BIN_CAST (elem), "test/caps,type=(int)1"); +} + +static void +test_element2_class_init (TestElement2Class * klass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (klass); + GstPadTemplate *src_template, *sink_template; + + src_template = gst_static_pad_template_get (&src_factory); + gst_element_class_add_pad_template (element_class, src_template); + + sink_template = gst_static_pad_template_get (&sink_factory); + gst_element_class_add_pad_template (element_class, sink_template); +} + +static void +test_element2_init (TestElement2 * elem) +{ + configure_test_element (GST_BIN_CAST (elem), "test/caps,type=(int)2"); +} + +GST_CHECK_MAIN (autoconvert);