ztest: move wrapped malloc functions into a separate file

Move rzalloc(), rfree(), sof_heap_alloc(), sof_heap_free() and
related function wrappers into a separate common file. Use it for
fast-get and objpool tests.

Signed-off-by: Guennadi Liakhovetski <guennadi.liakhovetski@linux.intel.com>

Author: lyakh

test/ztest/unit/common/alloc.c

@@ -0,0 +1,59 @@
+// SPDX-License-Identifier: BSD-3-Clause
+/* Copyright(c) 2026 Intel Corporation. All rights reserved. */
+
+#include <zephyr/ztest.h>
+#include <sof/common.h>
+
+#include <stddef.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+
+void *__wrap_rzalloc(uint32_t flags, size_t bytes)
+{
+	void *ret;
+	(void)flags;
+
+	ret = malloc(bytes);
+
+	zassert_not_null(ret, "Memory allocation should not fail");
+
+	memset(ret, 0, bytes);
+
+	return ret;
+}
+
+void __wrap_rfree(void *ptr)
+{
+	free(ptr);
+}
+
+struct k_heap;
+void *__wrap_sof_heap_alloc(struct k_heap *heap, uint32_t flags, size_t bytes, size_t alignment)
+{
+	void *ret;
+
+	(void)flags;
+	(void)heap;
+
+	if (alignment)
+		ret = aligned_alloc(alignment, ALIGN_UP(bytes, alignment));
+	else
+		ret = malloc(bytes);
+
+	zassert_not_null(ret, "Memory allocation should not fail");
+
+	return ret;
+}
+
+void __wrap_sof_heap_free(struct k_heap *heap, void *ptr)
+{
+	(void)heap;
+
+	free(ptr);
+}
+
+struct k_heap *__wrap_sof_sys_heap_get(void)
+{
+	return NULL;
+}

test/ztest/unit/fast-get/CMakeLists.txt

@@ -22,6 +22,7 @@ target_compile_definitions(app PRIVATE
 target_sources(app PRIVATE
 	test_fast_get_ztest.c
 	${SOF_ROOT}/zephyr/lib/fast-get.c
+	${SOF_ROOT}/test/ztest/unit/common/alloc.c
 )
 
 target_link_libraries(app PRIVATE "-Wl,--wrap=rzalloc,--wrap=rfree,--wrap=sof_heap_alloc,--wrap=sof_heap_free")

test/ztest/unit/fast-get/test_fast_get_ztest.c

@@ -56,52 +56,6 @@ static const int testdata[33][100] = {
 	{ 33 },
 };
 
-/* Mock memory allocation functions for testing purposes */
-
-void *__wrap_rzalloc(uint32_t flags, size_t bytes)
-{
-	void *ret;
-	(void)flags;
-
-	ret = malloc(bytes);
-
-	zassert_not_null(ret, "Memory allocation should not fail");
-
-	memset(ret, 0, bytes);
-
-	return ret;
-}
-
-void __wrap_rfree(void *ptr)
-{
-	free(ptr);
-}
-
-struct k_heap;
-void *__wrap_sof_heap_alloc(struct k_heap *heap, uint32_t flags, size_t bytes, size_t alignment)
-{
-	void *ret;
-
-	(void)flags;
-	(void)heap;
-
-	if (alignment)
-		ret = aligned_alloc(alignment, ALIGN_UP(bytes, alignment));
-	else
-		ret = malloc(bytes);
-
-	zassert_not_null(ret, "Memory allocation should not fail");
-
-	return ret;
-}
-
-void __wrap_sof_heap_free(struct k_heap *heap, void *ptr)
-{
-	(void)heap;
-
-	free(ptr);
-}
-
 /**
  * @brief Test basic fast_get and fast_put functionality
  *

test/ztest/unit/objpool/CMakeLists.txt

@@ -21,10 +21,11 @@ target_compile_definitions(app PRIVATE
 
 target_sources(app PRIVATE
 	test_objpool_ztest.c
+	${SOF_ROOT}/test/ztest/unit/common/alloc.c
 	${SOF_ROOT}/src/lib/objpool.c
 )
 
-target_link_libraries(app PRIVATE "-Wl,--wrap=rzalloc")
+target_link_libraries(app PRIVATE "-Wl,--wrap=rzalloc,--wrap=sof_heap_alloc,--wrap=sof_heap_free,--wrap=sof_sys_heap_get")
 
 # Add RELATIVE_FILE definitions for SOF trace functionality
 sof_append_relative_path_definitions(app)

test/ztest/unit/objpool/test_objpool_ztest.c

@@ -15,18 +15,6 @@
 #include <stdlib.h>
 #include <string.h>
 
-void *__wrap_rzalloc(uint32_t flags, size_t bytes)
-{
-	(void)flags;
-
-	void *ret = malloc(bytes);
-
-	if (ret)
-		memset(ret, 0, bytes);
-
-	return ret;
-}
-
 ZTEST(objpool_suite, test_objpool_wrong_size)
 {
 	struct objpool_head head = {.list = LIST_INIT(head.list)};