#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)

wp_intern u8 temp_buf[TEMP_BUF_SIZE] = {};
wp_intern WpAllocator temp_allocator   = {};

WpTestFuncResult test_arena_allocator(void) {
	WpAllocator allocator = wpMemArenaAllocatorInit(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           = wpMemAllocatorAlloc(&allocator, 20);
	result              = result && (ptr != nullptr);

	wpMemArenaAllocatorDestroy(&allocator);

	return wpTesterResult(result);
}

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

	WpAllocator allocator = wpMemArenaAllocatorInitWithBuffer(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           = wpMemAllocatorAlloc(&allocator, 20);
	result              = result && (ptr != NULL);

	wpMemArenaAllocatorDestroy(&allocator);

	return wpTesterResult(result);
}

WpTestFuncResult test_arena_allocator_temp_begin(void) {
	temp_allocator = wpMemArenaAllocatorInitWithBuffer(temp_buf, TEMP_BUF_SIZE);

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

	wpMemArenaAllocatorTempBegin(&temp_allocator);
	i32 *num2 = (i32 *)wpMemAllocatorAlloc(&temp_allocator, sizeof(i32));
	result    = result && num2 != NULL;
	i32 *num3 = (i32 *)wpMemAllocatorAlloc(&temp_allocator, sizeof(i32));
	result    = result && num3 == NULL;

	return wpTesterResult(result);
}

WpTestFuncResult test_arena_allocator_temp_end(void) {
	wpMemArenaAllocatorTempEnd(&temp_allocator);
	i32 *num1 = (i32 *)wpMemAllocatorAlloc(&temp_allocator, sizeof(i32));
	b8 result = num1 != NULL;
	i32 *num2 = (i32 *)wpMemAllocatorAlloc(&temp_allocator, sizeof(i32));
	result    = result && num2 == NULL;

	wpMemArenaAllocatorDestroy(&temp_allocator);

	return wpTesterResult(result);
}