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

@@ -109,6 +109,7 @@ check_PROGRAMS = \
 	$(check_kate)  \
 	elements/aacparse \
 	elements/amrparse \
+	elements/autoconvert \
 	elements/asfmux \
 	elements/capssetter \
 	elements/legacyresample \

tests/check/elements/.gitignore

@@ -2,6 +2,7 @@
 aacparse
 amrparse
 asfmux
+autoconvert
 camerabin
 capssetter
 deinterleave

tests/check/elements/autoconvert.c

@@ -0,0 +1,239 @@
+/* GStreamer
+ *
+ * unit test for autoconvert element
+ * Copyright (C) 2009 Jan Schmidt <thaytan@noraisin.net>
+ *
+ * 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 <gst/gst.h>
+#include <gst/check/gstcheck.h>
+
+/* 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);