#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 "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);
}


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


    return cmocka_run_group_tests(tests, NULL, NULL);
}