#include <stdarg.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <setjmp.h>

#include <cmocka.h>

#include <bitops.h>
#include <flash.h>
#include <ftl.h>
#include <macros.h>

size_t __wrap_flash_read(struct flash_dev *dev, uint32_t addr,
	void *data, size_t len)
{
	const void *ret_data;
	size_t ret_len;

	(void)dev;

	check_expected(addr);
	check_expected(len);

	ret_len = mock_type(size_t);
	ret_data = mock_type(const void *);

	if (len > ret_len)
		len = ret_len;

	if (len)
		memcpy(data, ret_data, len);

	return len;
}

size_t __wrap_flash_write(struct flash_dev *dev, uint32_t addr,
	const void *data, size_t len)
{
	void *ret_data;
	size_t ret_len;

	(void)dev;

	check_expected(addr);
	check_expected(len);

	ret_len = mock_type(size_t);
	ret_data = mock_type(void *);

	if (len > ret_len)
		len = ret_len;

	if (len)
		memcpy(ret_data, data, len);

	return len;
}

int __wrap_flash_is_erased(struct flash_dev *dev, uint32_t addr, size_t len)
{
	(void)dev;

	check_expected(addr);
	check_expected(len);

	return mock_type(int);
}

int test_find_block_div(void);
int test_next_upage(void);
int test_read_page_group(void);
int test_read_page_desc(void);
int test_write_page_desc(void);
int test_write_upage(void);

int main(void)
{
	int count = 0;

	count += test_find_block_div();
	count += test_next_upage();
	count += test_read_page_group();
	count += test_read_page_desc();
	count += test_write_page_desc();
	count += test_write_upage();

	return count;
}