tests: Add CUDA memory allocator test

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/3629>

subprojects/gst-plugins-bad/gst-libs/gst/cuda/meson.build

@@ -21,6 +21,10 @@ cuda_headers = files([
 ])
 
 gstcuda_dep = dependency('', required : false)
+cuda_stubinc = include_directories('./stub')
+gstcuda_stub_dep = declare_dependency(
+  include_directories: cuda_stubinc
+)
 
 if host_system not in ['windows', 'linux']
   subdir_done()
@@ -40,7 +44,6 @@ if host_system == 'windows'
   endforeach
 endif
 
-cuda_stubinc = include_directories('./stub')
 extra_c_args = ['-DGST_USE_UNSTABLE_API']
 
 if gstgl_dep.found()

subprojects/gst-plugins-bad/tests/check/libs/cudamemory.c

@@ -0,0 +1,135 @@
+/*
+ * GStreamer
+ * Copyright (C) 2023 Seungha Yang <seungha@centricular.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>
+#include <gst/cuda/gstcuda.h>
+#include <gst/video/video.h>
+
+static GstCudaContext *context = NULL;
+static GQuark memory_tester_quark;
+
+static void
+setup_func (void)
+{
+  context = gst_cuda_context_new (0);
+  fail_unless (GST_IS_CUDA_CONTEXT (context));
+
+  memory_tester_quark = g_quark_from_static_string ("gst-cuda-memory-tester");
+}
+
+static void
+teardown_func (void)
+{
+  gst_object_unref (context);
+}
+
+static void
+allocator_finalize_cb (gboolean * alloc_finalized)
+{
+  *alloc_finalized = TRUE;
+}
+
+GST_START_TEST (test_free_active_allocator)
+{
+  GstCudaPoolAllocator *alloc;
+  GstVideoInfo info;
+  GstMemory *mem = NULL;
+  gboolean ret;
+  GstFlowReturn flow_ret;
+  gboolean alloc_finalized = FALSE;
+
+  gst_video_info_set_format (&info, GST_VIDEO_FORMAT_NV12, 320, 240);
+
+  alloc = gst_cuda_pool_allocator_new (context, NULL, &info);
+  fail_unless (alloc);
+
+  g_object_set_qdata_full (G_OBJECT (alloc), memory_tester_quark,
+      &alloc_finalized, (GDestroyNotify) allocator_finalize_cb);
+
+  /* inactive pool should return flusing */
+  flow_ret = gst_cuda_pool_allocator_acquire_memory (alloc, &mem);
+  fail_unless (flow_ret == GST_FLOW_FLUSHING);
+  fail_if (mem);
+
+  ret = gst_cuda_allocator_set_active (GST_CUDA_ALLOCATOR (alloc), TRUE);
+  fail_unless (ret);
+
+  flow_ret = gst_cuda_pool_allocator_acquire_memory (alloc, &mem);
+  fail_unless (flow_ret == GST_FLOW_OK);
+  fail_unless (mem);
+
+  gst_object_unref (alloc);
+  /* Only memory should hold refcount at this moment */
+  fail_unless (G_OBJECT (alloc)->ref_count == 1);
+  fail_if (alloc_finalized);
+
+  /* allocator should be finalized as well */
+  gst_memory_unref (mem);
+  fail_unless (alloc_finalized);
+}
+
+GST_END_TEST;
+
+static gboolean
+check_cuda_device (void)
+{
+  GstCudaContext *context;
+
+  if (!gst_cuda_load_library ())
+    return FALSE;
+
+  context = gst_cuda_context_new (0);
+  if (context) {
+    gst_object_unref (context);
+    return TRUE;
+  }
+
+  return FALSE;
+}
+
+static Suite *
+cudamemory_suite (void)
+{
+  Suite *s;
+  TCase *tc_chain;
+
+  /* CUDA doesn't work well with fork  */
+  g_setenv ("CK_FORK", "no", TRUE);
+
+  s = suite_create ("cudamemory");
+  tc_chain = tcase_create ("general");
+
+  suite_add_tcase (s, tc_chain);
+  tcase_add_checked_fixture (tc_chain, setup_func, teardown_func);
+
+  if (!check_cuda_device ())
+    return s;
+
+  tcase_add_test (tc_chain, test_free_active_allocator);
+
+  return s;
+}
+
+GST_CHECK_MAIN (cudamemory);

subprojects/gst-plugins-bad/tests/check/meson.build

@@ -100,6 +100,7 @@ base_tests = [
   [['libs/vkimage.c'], not gstvulkan_dep.found(), [gstvulkan_dep]],
   [['libs/vkinstance.c'], not gstvulkan_dep.found(), [gstvulkan_dep]],
   [['libs/d3d11device.cpp'], not gstd3d11_dep.found(), [gstd3d11_dep]],
+  [['libs/cudamemory.c'], not gstcuda_dep.found(), [gstcuda_dep, gstcuda_stub_dep]],
 ]
 
 # FIXME: unistd dependency, unstable or not tested yet on windows