tests: Add unit tests for shmsink/shmsrc

tests/check/Makefile.am

@@ -162,7 +162,11 @@ uvch264_dist_data = elements/uvch264demux_data/valid_h264_jpg.mjpg \
 			elements/uvch264demux_data/valid_h264_yuy2.h264 \
 			elements/uvch264demux_data/valid_h264_yuy2.yuy2
 
-
+if USE_SHM
+check_shm=elements/shm
+else
+check_shm=
+endif
 
 VALGRIND_TO_FIX = \
 	elements/mpeg2enc \
@@ -196,6 +200,7 @@ check_PROGRAMS = \
 	$(check_kate)  \
 	$(check_opus)  \
 	$(check_curl) \
+	$(check_shm) \
 	elements/autoconvert \
 	elements/autovideoconvert \
 	elements/asfmux \

tests/check/elements/.gitignore

@@ -42,6 +42,7 @@ rganalysis
 rglimiter
 rgvolume
 schroenc
+shm
 spectrum
 timidity
 y4menc

tests/check/elements/shm.c

@@ -0,0 +1,173 @@
+/* GStreamer
+ *
+ * unit test for shm elements
+ * Copyright (C) 2013 Collabora Ltd
+ *   @author: Olivier Crete <olivier.crete@collabora.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., 51 Franklin St, Fifth Floor,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <gst/gst.h>
+#include <gst/check/gstcheck.h>
+
+
+static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS_ANY);
+
+static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS_ANY);
+
+GstElement *src, *sink;
+GstPad *sinkpad, *srcpad;
+
+static void
+setup_shm (void)
+{
+  gchar *socket_path = NULL;
+
+  sink = gst_check_setup_element ("shmsink");
+  src = gst_check_setup_element ("shmsrc");
+
+  srcpad = gst_check_setup_src_pad (sink, &src_template);
+  sinkpad = gst_check_setup_sink_pad (src, &sink_template);
+
+  g_object_set (sink, "socket-path", "shm-unit-test", NULL);
+
+  fail_unless (gst_element_set_state (sink, GST_STATE_PLAYING) ==
+      GST_STATE_CHANGE_ASYNC);
+
+  g_object_get (sink, "socket-path", &socket_path, NULL);
+  fail_unless (socket_path != NULL);
+  g_object_set (src, "socket-path", socket_path, NULL);
+  g_free (socket_path);
+
+  gst_pad_set_active (srcpad, TRUE);
+  gst_pad_set_active (sinkpad, TRUE);
+
+  fail_unless (gst_element_set_state (src, GST_STATE_PLAYING) ==
+      GST_STATE_CHANGE_SUCCESS);
+}
+
+static void
+teardown_shm (void)
+{
+  gst_check_teardown_sink_pad (src);
+  gst_check_teardown_src_pad (sink);
+  gst_check_teardown_element (src);
+  gst_check_teardown_element (sink);
+}
+
+GST_START_TEST (test_shm_sysmem_alloc)
+{
+  GstBuffer *buf;
+  GstState state, pending;
+
+  buf = gst_buffer_new_allocate (NULL, 1000, NULL);
+
+  fail_unless (gst_pad_push (srcpad, buf) == GST_FLOW_OK);
+
+  fail_unless (gst_element_get_state (sink, &state, &pending,
+          GST_CLOCK_TIME_NONE) == GST_STATE_CHANGE_SUCCESS);
+  fail_unless (state == GST_STATE_PLAYING);
+  fail_unless (pending == GST_STATE_VOID_PENDING);
+
+  g_mutex_lock (&check_mutex);
+  while (buffers == NULL)
+    g_cond_wait (&check_cond, &check_mutex);
+  g_mutex_unlock (&check_mutex);
+  fail_unless (g_list_length (buffers) == 1);
+
+  buf = buffers->data;
+  fail_unless (gst_buffer_get_size (buf) == 1000);
+
+  gst_check_drop_buffers ();
+  teardown_shm ();
+}
+
+GST_END_TEST;
+
+
+GST_START_TEST (test_shm_alloc)
+{
+  GstBuffer *buf;
+  GstQuery *query;
+  GstCaps *caps = gst_caps_new_empty_simple ("application/x-test");
+  GstAllocator *alloc;
+  GstAllocationParams params;
+  guint size;
+
+  query = gst_query_new_allocation (caps, FALSE);
+  gst_caps_unref (caps);
+
+  fail_unless (gst_pad_peer_query (srcpad, query));
+
+  fail_unless (gst_query_get_n_allocation_params (query) == 1);
+
+  gst_query_parse_nth_allocation_param (query, 0, &alloc, &params);
+  fail_unless (alloc != NULL);
+  gst_query_unref (query);
+
+  g_object_get (sink, "shm-size", &size, NULL);
+
+  /* alloc buffer of max size, this way, it will block forever it a copy
+   * is made inside shmsink*/
+  buf = gst_buffer_new_allocate (alloc, size, &params);
+
+  gst_object_unref (alloc);
+
+  fail_unless (gst_pad_push (srcpad, buf) == GST_FLOW_OK);
+
+
+  g_mutex_lock (&check_mutex);
+  while (buffers == NULL)
+    g_cond_wait (&check_cond, &check_mutex);
+  g_mutex_unlock (&check_mutex);
+  fail_unless (g_list_length (buffers) == 1);
+
+  buf = buffers->data;
+  fail_unless (gst_buffer_get_size (buf) == size);
+
+  gst_check_drop_buffers ();
+  teardown_shm ();
+}
+
+GST_END_TEST;
+
+static Suite *
+shm_suite (void)
+{
+  Suite *s = suite_create ("shm");
+  TCase *tc;
+
+  tc = tcase_create ("shm");
+  tcase_add_checked_fixture (tc, setup_shm, NULL);
+  tcase_add_test (tc, test_shm_sysmem_alloc);
+  tcase_add_test (tc, test_shm_alloc);
+  suite_add_tcase (s, tc);
+
+  return s;
+}
+
+
+GST_CHECK_MAIN (shm);