From 8b0c2db425563f3b773825138fcf13c03cca120a Mon Sep 17 00:00:00 2001
From: Benjamin Gaignard <benjamin.gaignard@stericsson.com>
Date: Wed, 15 Dec 2010 21:19:55 +0100
Subject: [PATCH] autocolorspace: Add unit test

---
 tests/check/Makefile.am               |   1 +
 tests/check/elements/autocolorspace.c | 139 ++++++++++++++++++++++++++
 2 files changed, 140 insertions(+)
 create mode 100644 tests/check/elements/autocolorspace.c

diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am
index d98aaecbcf..81c5862e9c 100644
--- a/tests/check/Makefile.am
+++ b/tests/check/Makefile.am
@@ -152,6 +152,7 @@ check_PROGRAMS = \
 	elements/ac3parse \
 	elements/amrparse \
 	elements/autoconvert \
+	elements/autocolorspace \
 	elements/asfmux \
 	elements/camerabin \
 	elements/dataurisrc \
diff --git a/tests/check/elements/autocolorspace.c b/tests/check/elements/autocolorspace.c
new file mode 100644
index 0000000000..51371f3b5e
--- /dev/null
+++ b/tests/check/elements/autocolorspace.c
@@ -0,0 +1,139 @@
+/* GStreamer
+ *
+ * unit test for autocolorspace element
+ * Copyright (C) 2009 Jan Schmidt <thaytan@noraisin.net>
+ * Copyright (C) 2010 ST-Ericsson SA 
+ *  @author: Benjamin Gaignard <benjamin.gaignard@stericsson.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., 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>
+
+typedef struct
+{
+  GMainLoop *loop;
+  gboolean eos;
+} OnMessageUserData;
+
+static void
+on_message_cb (GstBus * bus, GstMessage * message, gpointer user_data)
+{
+  OnMessageUserData *d = user_data;
+
+  switch (GST_MESSAGE_TYPE (message)) {
+    case GST_MESSAGE_ERROR:
+    case GST_MESSAGE_WARNING:
+      g_assert_not_reached ();
+      break;
+    case GST_MESSAGE_EOS:
+      g_main_loop_quit (d->loop);
+      d->eos = TRUE;
+      break;
+    default:
+      break;
+  }
+}
+
+static void
+run_test (const gchar * pipeline_string)
+{
+  GstElement *pipeline;
+  GstBus *bus;
+  GMainLoop *loop;
+  OnMessageUserData omud = { NULL, };
+  GstStateChangeReturn ret;
+
+  GST_DEBUG ("Testing pipeline '%s'", pipeline_string);
+
+  pipeline = gst_parse_launch (pipeline_string, NULL);
+  fail_unless (pipeline != NULL);
+  loop = g_main_loop_new (NULL, FALSE);
+
+  bus = gst_element_get_bus (pipeline);
+  fail_unless (bus != NULL);
+  gst_bus_add_signal_watch (bus);
+
+  omud.loop = loop;
+  omud.eos = FALSE;
+
+  g_signal_connect (bus, "message", (GCallback) on_message_cb, &omud);
+
+  gst_object_unref (bus);
+
+  ret = gst_element_set_state (pipeline, GST_STATE_PLAYING);
+  fail_unless (ret == GST_STATE_CHANGE_SUCCESS
+      || ret == GST_STATE_CHANGE_ASYNC);
+
+  g_main_loop_run (loop);
+
+  fail_unless (gst_element_set_state (pipeline,
+          GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS);
+
+  fail_unless (omud.eos == TRUE);
+
+  gst_object_unref (pipeline);
+  g_main_loop_unref (loop);
+
+}
+
+GST_START_TEST (test_autocolorspace_rbg2bayer)
+{
+  gchar *pipeline;
+
+  pipeline =
+      g_strdup_printf
+      ("videotestsrc num-buffers=1 ! video/x-raw-rgb,bpp=32,depth=32,width=100,height=100,framerate=10/1 ! autocolorspace ! video/x-raw-bayer,width=100,height=100,format=bggr,framerate=10/1 ! fakesink");
+
+  run_test (pipeline);
+  g_free (pipeline);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_autocolorspace_ffmpegcolorspace)
+{
+  gchar *pipeline;
+
+  pipeline =
+      g_strdup_printf
+      ("videotestsrc num-buffers=1 ! video/x-raw-rgb,bpp=32,width=100,height=100,framerate=10/1 ! autocolorspace ! video/x-raw-rgb,bpp=16,width=100,height=100,framerate=10/1 ! fakesink");
+
+  run_test (pipeline);
+  g_free (pipeline);
+}
+
+GST_END_TEST;
+
+static Suite *
+autocolorspace_suite (void)
+{
+  Suite *s = suite_create ("autocolorspace");
+  TCase *tc_basic = tcase_create ("general");
+
+  suite_add_tcase (s, tc_basic);
+  tcase_add_test (tc_basic, test_autocolorspace_rbg2bayer);
+  tcase_add_test (tc_basic, test_autocolorspace_ffmpegcolorspace);
+
+  return s;
+}
+
+GST_CHECK_MAIN (autocolorspace);