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

#include <containers/stackarray.h>
#include <defines.h>
#include <memory/kmemory.h>

STACKARRAY_TYPE(u8, 6);
STACKARRAY_TYPE(f32, 6);
STACKARRAY_TYPE_NAMED(const char *, string, 6);

static u8 all_stackarray_tests_after_create (void) {
	// Test a basic type first.

	stackarray_u8_6 arr = stackarray_u8_6_create();

	// Set some values
	arr.data[0] = 69;
	arr.data[2] = 42;
	arr.data[4] = 36;

	expect_should_be(69, arr.data[0]);
	expect_should_be(0, arr.data[1]);
	expect_should_be(42, arr.data[2]);
	expect_should_be(0, arr.data[3]);
	expect_should_be(36, arr.data[4]);
	expect_should_be(0, arr.data[5]);

	// Verify that it has been destroyed.
	stackarray_u8_6_destroy(&arr);

	return true;
}

static u8 stackarray_all_iterator_tests (void) {

	stackarray_iterator it;
	u32 loop_count = 0;
	stackarray_u8_6 arr = stackarray_u8_6_create();

	// Set some values
	arr.data[0] = 69;
	arr.data[2] = 42;
	arr.data[4] = 36;

	expect_should_be(69, arr.data[0]);
	expect_should_be(0, arr.data[1]);
	expect_should_be(42, arr.data[2]);
	expect_should_be(0, arr.data[3]);
	expect_should_be(36, arr.data[4]);
	expect_should_be(0, arr.data[5]);

	{
		// Try forwards iteration.
		it = arr.begin(&arr.base);
		expect_should_be(&arr.base, it.arr);
		expect_should_be(0, it.pos);
		expect_should_be(1, it.dir);
		loop_count = 0;
		for (; !it.end(&it); it.next(&it)) {
			u8 *val = it.value(&it);
			if (it.pos == 0) {
				expect_should_be(69, *val);
			} else if (it.pos == 2) {
				expect_should_be(42, *val);
			} else if (it.pos == 4) {
				expect_should_be(36, *val);
			} else {
				expect_should_be(0, *val);
			}

			loop_count++;
		}
		expect_should_be(6, loop_count);

		// Try reverse/backward iteration.
		it = arr.rbegin(&arr.base);
		expect_should_be(&arr.base, it.arr);
		expect_should_be(6 - 1, it.pos);
		expect_should_be(-1, it.dir);
		loop_count = 0;
		for (; !it.end(&it); it.next(&it)) {
			u8 *val = it.value(&it);
			if (it.pos == 0) {
				expect_should_be(69, *val);
			} else if (it.pos == 2) {
				expect_should_be(42, *val);
			} else if (it.pos == 4) {
				expect_should_be(36, *val);
			} else {
				expect_should_be(0, *val);
			}

			loop_count++;
		}
		expect_should_be(6, loop_count);
	}

	stackarray_u8_6_destroy(&arr);

	return true;
}

static u8 stackarray_string_type_test (void) {

	stackarray_string_6 arr = stackarray_string_6_create();

	// Set some data.
	arr.data[0] = "test";
	arr.data[2] = "something else";
	arr.data[4] = "ththth";

	// validate content
	expect_string_to_be("test", arr.data[0]);
	expect_string_to_be(0, arr.data[1]);
	expect_string_to_be("something else", arr.data[2]);
	expect_string_to_be(0, arr.data[3]);
	expect_string_to_be("ththth", arr.data[4]);
	expect_string_to_be(0, arr.data[5]);

	stackarray_string_6_destroy(&arr);

	return true;
}

static u8 stackarray_float_type_test (void) {

	stackarray_f32_6 arr = stackarray_f32_6_create();

	// Set some data.
	arr.data[0] = 0.1f;
	arr.data[2] = 0.2f;
	arr.data[4] = 0.3f;

	// validate content
	expect_float_to_be(0.1f, arr.data[0]);
	expect_float_to_be(0.0f, arr.data[1]);
	expect_float_to_be(0.2f, arr.data[2]);
	expect_float_to_be(0.0f, arr.data[3]);
	expect_float_to_be(0.3f, arr.data[4]);
	expect_float_to_be(0.0f, arr.data[5]);

	stackarray_f32_6_destroy(&arr);

	return true;
}

void stackarray_register_tests (void) {
	test_manager_register_test(all_stackarray_tests_after_create, "All stackarray tests after create");
	test_manager_register_test(stackarray_all_iterator_tests, "All stackarray iterator tests");
	test_manager_register_test(stackarray_string_type_test, "stackarray string type tests");
	test_manager_register_test(stackarray_float_type_test, "stackarray float type tests");
}