valve: Add unit tests

Add a unit test for the valve element.

diff --git a/tests/check/elements/valve.c b/tests/check/elements/valve.c
new file mode 100644 (file)
index 0000000..6c875c0
--- /dev/null
+++ b/tests/check/elements/valve.c
@@ -0,0 +1,137 @@
+/* GStreamer
+ *
+ * unit test for the valve element
+ *
+ * Copyright 2009 Collabora Ltd.
+ *  @author: Olivier Crete <olivier.crete@collabora.co.uk>
+ * Copyright 2009 Nokia Corp.
+ *
+ * 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.
+ */
+
+#include <gst/check/gstcheck.h>
+#include <gst/rtp/gstrtpbuffer.h>
+#include <gst/gst.h>
+
+
+static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS ("audio/x-raw-int"));
+
+static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS ("audio/x-raw-int"));
+
+gboolean event_received = FALSE;
+gboolean buffer_allocated = FALSE;
+
+static gboolean
+event_func (GstPad * pad, GstEvent * event)
+{
+  event_received = TRUE;
+  gst_event_unref (event);
+  return TRUE;
+}
+
+static GstFlowReturn
+bufferalloc_func (GstPad * pad, guint64 offset, guint size, GstCaps * caps,
+    GstBuffer ** buf)
+{
+  buffer_allocated = TRUE;
+  *buf = gst_buffer_new_and_alloc (size);
+  GST_BUFFER_OFFSET (*buf) = offset;
+  gst_buffer_set_caps (*buf, caps);
+
+  return GST_FLOW_OK;
+}
+
+GST_START_TEST (test_valve_basic)
+{
+  GstElement *valve;
+  GstPad *sink;
+  GstPad *src;
+  GstBuffer *buf;
+  GstCaps *caps;
+
+  valve = gst_check_setup_element ("valve");
+
+  sink = gst_check_setup_sink_pad_by_name (valve, &sinktemplate, "src");
+  src = gst_check_setup_src_pad_by_name (valve, &srctemplate, "sink");
+  gst_pad_set_event_function (sink, event_func);
+  gst_pad_set_bufferalloc_function (sink, bufferalloc_func);
+  gst_pad_set_active (src, TRUE);
+  gst_pad_set_active (sink, TRUE);
+  gst_element_set_state (valve, GST_STATE_PLAYING);
+
+  g_object_set (valve, "drop", FALSE, NULL);
+
+  fail_unless (gst_pad_push_event (src, gst_event_new_eos ()) == TRUE);
+  fail_unless (event_received == TRUE);
+  fail_unless (gst_pad_alloc_buffer (src, 0, 10, gst_caps_new_any (), &buf) ==
+      GST_FLOW_OK);
+  fail_unless (buffer_allocated == TRUE);
+  gst_buffer_unref (buf);
+  fail_unless (gst_pad_push (src, gst_buffer_new ()) == GST_FLOW_OK);
+  fail_unless (gst_pad_push (src, gst_buffer_new ()) == GST_FLOW_OK);
+  fail_unless (g_list_length (buffers) == 2);
+  caps = gst_pad_get_caps (src);
+  fail_unless (caps && gst_caps_is_equal (caps,
+          gst_pad_get_pad_template_caps (src)));
+  gst_caps_unref (caps);
+
+  gst_check_drop_buffers ();
+  event_received = buffer_allocated = FALSE;
+
+  g_object_set (valve, "drop", TRUE, NULL);
+  fail_unless (gst_pad_push_event (src, gst_event_new_eos ()) == TRUE);
+  fail_unless (event_received == FALSE);
+  fail_unless (gst_pad_alloc_buffer (src, 0, 10, gst_caps_new_any (), &buf) ==
+      GST_FLOW_OK);
+  fail_unless (buffer_allocated == FALSE);
+  gst_buffer_unref (buf);
+  fail_unless (gst_pad_push (src, gst_buffer_new ()) == GST_FLOW_OK);
+  fail_unless (gst_pad_push (src, gst_buffer_new ()) == GST_FLOW_OK);
+  fail_unless (buffers == NULL);
+  caps = gst_pad_get_caps (src);
+  fail_unless (caps && gst_caps_is_equal (caps,
+          gst_pad_get_pad_template_caps (src)));
+  gst_caps_unref (caps);
+
+  gst_pad_set_active (src, FALSE);
+  gst_pad_set_active (sink, FALSE);
+  gst_check_teardown_src_pad (valve);
+  gst_check_teardown_sink_pad (valve);
+  gst_check_teardown_element (valve);
+}
+
+GST_END_TEST;
+
+static Suite *
+valve_suite (void)
+{
+  Suite *s = suite_create ("valve");
+  TCase *tc_chain;
+
+  tc_chain = tcase_create ("valve_basic");
+  tcase_add_test (tc_chain, test_valve_basic);
+  suite_add_tcase (s, tc_chain);
+
+  return s;
+}
+
+GST_CHECK_MAIN (valve)