From 05aae3dd022a1253aeee3b2bfc851a088bc25fbc Mon Sep 17 00:00:00 2001
From: Seungha Yang <seungha@centricular.com>
Date: Thu, 29 Feb 2024 02:06:25 +0900
Subject: [PATCH] tests: cuda: Add buffer pool test

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/6240>
---
 .../tests/check/libs/cudamemory.c             | 57 +++++++++++++++++++
 1 file changed, 57 insertions(+)

diff --git a/subprojects/gst-plugins-bad/tests/check/libs/cudamemory.c b/subprojects/gst-plugins-bad/tests/check/libs/cudamemory.c
index 6fe0610bd5..93739bba8a 100644
--- a/subprojects/gst-plugins-bad/tests/check/libs/cudamemory.c
+++ b/subprojects/gst-plugins-bad/tests/check/libs/cudamemory.c
@@ -92,6 +92,62 @@ GST_START_TEST (test_free_active_allocator)
 
 GST_END_TEST;
 
+GST_START_TEST (test_free_buffer_after_deactivate)
+{
+  GstVideoInfo info;
+  gboolean ret;
+  GstFlowReturn flow_ret;
+  GstBufferPool *pool;
+  GstStructure *config;
+  GstCaps *caps;
+  GstBuffer *buffers[2];
+  gboolean alloc_finalized = FALSE;
+
+  gst_video_info_set_format (&info, GST_VIDEO_FORMAT_NV12, 320, 240);
+
+  caps = gst_video_info_to_caps (&info);
+  fail_unless (caps);
+
+  pool = gst_cuda_buffer_pool_new (context);
+  fail_unless (pool);
+
+  g_object_set_qdata_full (G_OBJECT (pool), memory_tester_quark,
+      &alloc_finalized, (GDestroyNotify) allocator_finalize_cb);
+
+  config = gst_buffer_pool_get_config (pool);
+  fail_unless (config);
+
+  gst_buffer_pool_config_set_params (config, caps, info.size, 0, 0);
+  gst_caps_unref (caps);
+
+  ret = gst_buffer_pool_set_config (pool, config);
+  fail_unless (ret);
+
+  ret = gst_buffer_pool_set_active (pool, TRUE);
+  fail_unless (ret);
+
+  flow_ret = gst_buffer_pool_acquire_buffer (pool, &buffers[0], NULL);
+  fail_unless (flow_ret == GST_FLOW_OK);
+
+  flow_ret = gst_buffer_pool_acquire_buffer (pool, &buffers[1], NULL);
+  fail_unless (flow_ret == GST_FLOW_OK);
+
+  ret = gst_buffer_pool_set_active (pool, FALSE);
+  fail_unless (ret);
+  fail_if (alloc_finalized);
+
+  gst_object_unref (pool);
+  fail_if (alloc_finalized);
+
+  gst_buffer_unref (buffers[0]);
+  fail_if (alloc_finalized);
+
+  gst_buffer_unref (buffers[1]);
+  fail_unless (alloc_finalized);
+}
+
+GST_END_TEST;
+
 static gboolean
 check_cuda_device (void)
 {
@@ -128,6 +184,7 @@ cudamemory_suite (void)
     return s;
 
   tcase_add_test (tc_chain, test_free_active_allocator);
+  tcase_add_test (tc_chain, test_free_buffer_after_deactivate);
 
   return s;
 }