#include <stdalign.h>
#include <stdlib.h>
#include <stdarg.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <setjmp.h>
#include <cmocka.h>
#include <string.h>
#include <limits.h>

#include "arena.h"

static void test_push_3_ints(void **state) {
    (void) state;

    ArenaResult value = arena_init(sizeof(int) * 3);
    assert_true(value.is_valid);
    Arena arena = value.arena;

    int int_to_push = 20;
    ArenaPointer result = arena_push(&arena, &int_to_push, sizeof(int), alignof(int));
    assert_true(result.is_valid);
    assert_int_equal(20, *(int*)arena_unwrap_pointer(result));
    assert_int_equal(20, *(int*)arena_unwrap_pointer(result));

    int_to_push = 30;
    result = arena_push(&arena, &int_to_push, sizeof(int), alignof(int));
    assert_true(result.is_valid);
    assert_int_equal(30, *(int*)arena_unwrap_pointer(result));

    int_to_push = 40;
    result = arena_push(&arena, &int_to_push, sizeof(int), alignof(int));
    assert_true(result.is_valid);
    assert_int_equal(40, *(int*)arena_unwrap_pointer(result));

    arena_destroy(&arena);
}

static void test_push_3_ints_2_doubles(void **state) {
    (void) state;

    ArenaResult value = arena_init((sizeof(int) * 3) + (sizeof(double) * 2));
    assert_true(value.is_valid);
    Arena arena = value.arena;

    int int_to_push = 20;
    ArenaPointer result = arena_push(&arena, &int_to_push, sizeof(int), alignof(int));
    assert_true(result.is_valid);
    assert_int_equal(20, *(int*)arena_unwrap_pointer(result));

    double double_to_push = 4.57;
    result = arena_push(&arena, &double_to_push, sizeof(double), alignof(double));
    assert_true(result.is_valid);
    assert_double_equal(4.57, *(double*)arena_unwrap_pointer(result), 1e-6);

    int_to_push = 30;
    result = arena_push(&arena, &int_to_push, sizeof(int), alignof(int));
    assert_true(result.is_valid);
    assert_int_equal(30, *(int*)arena_unwrap_pointer(result));

    int_to_push = 40;
    result = arena_push(&arena, &int_to_push, sizeof(int), alignof(int));
    assert_true(result.is_valid);
    assert_int_equal(40, *(int*)arena_unwrap_pointer(result));

    double_to_push = 267.33;
    result = arena_push(&arena, &double_to_push, sizeof(double), alignof(double));
    assert_true(result.is_valid);
    assert_double_equal(267.33, *(double*)arena_unwrap_pointer(result), 1e-6);

    arena_destroy(&arena);
}

static void test_init_arena_0_cap(void **state) {
    (void) state;

    ArenaResult arena = arena_init(0);
    assert_false(arena.is_valid);
    assert_int_equal(arena.err, ARENA_INVALID_SIZE);
}

static void test_arena_alloc_size_0(void **state) {
    (void) state;

    ArenaResult value = arena_init(64);
    assert_true(value.is_valid);
    Arena arena = value.arena;

    ArenaPointer pointer = arena_alloc(&arena, 0, alignof(int));
    assert_false(pointer.is_valid);
    assert_int_equal(pointer.err, ARENA_INVALID_SIZE);

    arena_destroy(&arena);
}

static void test_arena_alloc_size_max(void **state) {
    (void) state;

    ArenaResult value = arena_init(64);
    assert_true(value.is_valid);
    Arena arena = value.arena;

    ArenaPointer pointer = arena_alloc(&arena, SIZE_MAX, alignof(int));
    assert_false(pointer.is_valid);
    assert_int_equal(pointer.err, ARENA_OUT_OF_SPACE);

    arena_destroy(&arena);
}

static void test_arena_align_size_0(void **state) {
    (void) state;

    ArenaResult value = arena_init(64);
    assert_true(value.is_valid);
    Arena arena = value.arena;

    ArenaPointer pointer = arena_alloc(&arena, sizeof(int), 0);
    assert_false(pointer.is_valid);
    assert_int_equal(pointer.err, ARENA_INVALID_ALIGN);

    arena_destroy(&arena);
}

int main(void) {
    const struct CMUnitTest tests[] = {
        cmocka_unit_test(test_push_3_ints),
        cmocka_unit_test(test_push_3_ints_2_doubles),
        cmocka_unit_test(test_init_arena_0_cap),
        cmocka_unit_test(test_arena_alloc_size_0),
        cmocka_unit_test(test_arena_alloc_size_max),
        cmocka_unit_test(test_arena_align_size_0),
    };


    return cmocka_run_group_tests(tests, NULL, NULL);
}