#include "linear_allocator_tests.h"

#include <defines.h>
#include <memory/allocators/linear_allocator.h>

#include "../expect.h"
#include "../test_manager.h"

u8 linear_allocator_should_create_and_destroy (void) {
	linear_allocator alloc;
	linear_allocator_create(sizeof(u64), 0, &alloc);

	expect_should_not_be(0, alloc.memory);
	expect_should_be(sizeof(u64), alloc.total_size);
	expect_should_be(0, alloc.allocated);

	linear_allocator_destroy(&alloc);

	expect_should_be(0, alloc.memory);
	expect_should_be(0, alloc.total_size);
	expect_should_be(0, alloc.allocated);

	return true;
}

u8 linear_allocator_single_allocation_all_space (void) {
	linear_allocator alloc;
	linear_allocator_create(sizeof(u64), 0, &alloc);

	// Single allocation.
	void *block = linear_allocator_allocate(&alloc, sizeof(u64));

	// Validate it
	expect_should_not_be(0, block);
	expect_should_be(sizeof(u64), alloc.allocated);

	linear_allocator_destroy(&alloc);

	return true;
}

u8 linear_allocator_multi_allocation_all_space (void) {
	u64 max_allocs = 1024;
	linear_allocator alloc;
	linear_allocator_create(sizeof(u64) * max_allocs, 0, &alloc);

	// Multiple allocations - full.
	void *block;
	for (u64 i = 0; i < max_allocs; ++i) {
		block = linear_allocator_allocate(&alloc, sizeof(u64));
		// Validate it
		expect_should_not_be(0, block);
		expect_should_be(sizeof(u64) * (i + 1), alloc.allocated);
	}

	linear_allocator_destroy(&alloc);

	return true;
}

u8 linear_allocator_multi_allocation_over_allocate (void) {
	u64 max_allocs = 3;
	linear_allocator alloc;
	linear_allocator_create(sizeof(u64) * max_allocs, 0, &alloc);

	// Multiple allocations - full.
	void *block;
	for (u64 i = 0; i < max_allocs; ++i) {
		block = linear_allocator_allocate(&alloc, sizeof(u64));
		// Validate it
		expect_should_not_be(0, block);
		expect_should_be(sizeof(u64) * (i + 1), alloc.allocated);
	}

	KDEBUG("Note: The following error is intentionally caused by this test.");

	// Ask for one more allocation. Should error and return 0.
	block = linear_allocator_allocate(&alloc, sizeof(u64));
	// Validate it - allocated should be unchanged.
	expect_should_be(0, block);
	expect_should_be(sizeof(u64) * (max_allocs), alloc.allocated);

	linear_allocator_destroy(&alloc);

	return true;
}

u8 linear_allocator_multi_allocation_all_space_then_free (void) {
	u64 max_allocs = 1024;
	linear_allocator alloc;
	linear_allocator_create(sizeof(u64) * max_allocs, 0, &alloc);

	// Multiple allocations - full.
	void *block;
	for (u64 i = 0; i < max_allocs; ++i) {
		block = linear_allocator_allocate(&alloc, sizeof(u64));
		// Validate it
		expect_should_not_be(0, block);
		expect_should_be(sizeof(u64) * (i + 1), alloc.allocated);
	}

	// Validate that pointer is reset.
	linear_allocator_free_all(&alloc, true);
	expect_should_be(0, alloc.allocated);

	linear_allocator_destroy(&alloc);

	return true;
}

void linear_allocator_register_tests (void) {
	test_manager_register_test(linear_allocator_should_create_and_destroy, "Linear allocator should create and destroy");
	test_manager_register_test(linear_allocator_single_allocation_all_space, "Linear allocator single alloc for all space");
	test_manager_register_test(linear_allocator_multi_allocation_all_space, "Linear allocator multi alloc for all space");
	test_manager_register_test(linear_allocator_multi_allocation_over_allocate, "Linear allocator try over allocate");
	test_manager_register_test(linear_allocator_multi_allocation_all_space_then_free, "Linear allocator allocated should be 0 after free_all");
}