#include "test_allocator.h"
#include "wapp.h"
#include <stdlib.h>

// NOTE (Abdelrahman): Cannot query size of Arena here so it's hardcoded. Similarly, since arena
// allocation are aligned to power of 2 boundaries, the number of i32 values needed is hardcoded.
#define TEMP_BUF_SIZE (40 + sizeof(i32) * 8)

wapp_intern u8 temp_buf[TEMP_BUF_SIZE] = {};
wapp_intern Allocator temp_allocator   = {};

TestFuncResult test_arena_allocator(void) {
	Allocator allocator = wapp_mem_arena_allocator_init(4096);
	b8 result           = allocator.obj != nullptr && allocator.alloc != nullptr &&
	                      allocator.alloc_aligned != nullptr &&
	                      allocator.realloc != nullptr && allocator.realloc_aligned != nullptr &&
	                      allocator.free == nullptr;
	void *ptr           = wapp_mem_allocator_alloc(&allocator, 20);
	result              = result && (ptr != nullptr);

	wapp_mem_arena_allocator_destroy(&allocator);

	return wapp_tester_result(result);
}

TestFuncResult test_arena_allocator_with_buffer(void) {
	u8 buffer[KiB(4)] = {0};

	Allocator allocator = wapp_mem_arena_allocator_init_with_buffer(buffer, KiB(4));
	b8 result           = allocator.obj != NULL && allocator.alloc != NULL &&
	                      allocator.alloc_aligned != NULL &&
	                      allocator.realloc != NULL && allocator.realloc_aligned != NULL &&
	                      allocator.free == NULL;
	void *ptr           = wapp_mem_allocator_alloc(&allocator, 20);
	result              = result && (ptr != NULL);

	wapp_mem_arena_allocator_destroy(&allocator);

	return wapp_tester_result(result);
}

TestFuncResult test_arena_allocator_temp_begin(void) {
	temp_allocator = wapp_mem_arena_allocator_init_with_buffer(temp_buf, TEMP_BUF_SIZE);

	i32 *num1 = (i32 *)wapp_mem_allocator_alloc(&temp_allocator, sizeof(i32));
	b8 result = num1 != NULL;

	wapp_mem_arena_allocator_temp_begin(&temp_allocator);
	i32 *num2 = (i32 *)wapp_mem_allocator_alloc(&temp_allocator, sizeof(i32));
	result    = result && num2 != NULL;
	i32 *num3 = (i32 *)wapp_mem_allocator_alloc(&temp_allocator, sizeof(i32));
	result    = result && num3 == NULL;

	return wapp_tester_result(result);
}

TestFuncResult test_arena_allocator_temp_end(void) {
	wapp_mem_arena_allocator_temp_end(&temp_allocator);
	i32 *num1 = (i32 *)wapp_mem_allocator_alloc(&temp_allocator, sizeof(i32));
	b8 result = num1 != NULL;
	i32 *num2 = (i32 *)wapp_mem_allocator_alloc(&temp_allocator, sizeof(i32));
	result    = result && num2 == NULL;

	wapp_mem_arena_allocator_destroy(&temp_allocator);

	return wapp_tester_result(result);
}