From 0da262cc0fb6c19fd6edf780184373326383f76d Mon Sep 17 00:00:00 2001
From: "S.J.R. van Schaik" <stephan@whiteboxsystems.nl>
Date: Wed, 20 Sep 2017 14:30:38 +0200
Subject: [PATCH] test: implement __wrap_is_group_erased() and
 __wrap_find_block()

---
 source/tests/Makefile   |  2 ++
 source/tests/ftl/mock.c | 26 ++++++++++++++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/source/tests/Makefile b/source/tests/Makefile
index 28eafb8..e8de216 100644
--- a/source/tests/Makefile
+++ b/source/tests/Makefile
@@ -11,5 +11,7 @@ TEST_CFLAGS += -Dflash_read=__wrap_flash_read
 TEST_CFLAGS += -Dflash_write=__wrap_flash_write
 TEST_CFLAGS += -Dflash_is_erased=__wrap_flash_is_erased
 
+TEST_CFLAGS += -Dis_group_erased=__wrap_is_group_erased
+TEST_CFLAGS += -Dfind_block=__wrap_find_block
 TEST_CFLAGS += -Dprepare_head=__wrap_prepare_head
 TEST_CFLAGS += -Dwrite_page_desc=__wrap_write_page_desc
diff --git a/source/tests/ftl/mock.c b/source/tests/ftl/mock.c
index 8813708..131b049 100644
--- a/source/tests/ftl/mock.c
+++ b/source/tests/ftl/mock.c
@@ -9,6 +9,32 @@
 
 #include <ftl.h>
 
+int __wrap_is_group_erased(struct ftl_map *map, uint32_t group)
+{
+	check_expected_ptr(map);
+	check_expected(group);
+
+	return mock_type(int);
+}
+
+int __wrap_find_block(struct ftl_map *map, struct ftl_page_group *group,
+	uint32_t *where, uint32_t block)
+{
+	struct ftl_page_group *ret_group;
+
+	check_expected_ptr(map);
+	check_expected(block);
+
+	ret_group = mock_type(struct ftl_page_group *);
+
+	if (ret_group)
+		memcpy(group, ret_group, sizeof *group);
+
+	*where = mock_type(uint32_t);
+
+	return mock_type(int);
+}
+
 int __wrap_prepare_head(struct ftl_map *map)
 {
 	(void)map;