#include "unity.h"

#include "dynamic_array.h"


void setUp(void) {}

void tearDown(void) {}


void test_array_create_without_options(void) {
    int *arr = NULL;
    array_create(arr);

    const ArrayHeader *header = array_get_header(arr);
    TEST_ASSERT_EQUAL(DYNAMIC_ARRAY_DEFAULT_ARRAY_SIZE, header->capacity);

    array_destroy(arr);
}


void test_array_destroy(void) {
    int *arr = NULL;
    array_create(arr);
    array_destroy(arr);
    TEST_ASSERT_NULL(arr);
}


void test_array_size_initial(void) {
    int *arr = NULL;
    array_create(arr);

    TEST_ASSERT_EQUAL(0, array_get_header(arr)->size);

    array_destroy(arr);
}


void test_array_push_value(void) {
    int *arr = NULL;
    array_create(arr);

    array_push_value(arr, 42);

    TEST_ASSERT_EQUAL(42, arr[0]);
    TEST_ASSERT_EQUAL(1, array_get_header(arr)->size);

    array_destroy(arr);
}


void test_array_push_multiple_values(void) {
    int *arr = NULL;
    array_create(arr);

    array_push_value(arr, 10);
    array_push_value(arr, 20);
    array_push_value(arr, 30);

    TEST_ASSERT_EQUAL(10, arr[0]);
    TEST_ASSERT_EQUAL(20, arr[1]);
    TEST_ASSERT_EQUAL(30, arr[2]);
    TEST_ASSERT_EQUAL(3, array_get_header(arr)->size);

    array_destroy(arr);
}


void test_array_pop_decrements_size(void) {
    int *arr = NULL;
    array_create(arr);

    array_push_value(arr, 42);
    array_push_value(arr, 99);
    array_pop(arr);

    TEST_ASSERT_EQUAL(1, array_get_header(arr)->size);

    array_destroy(arr);
}


void test_array_get_capacity_default(void) {
    int *arr = NULL;
    array_create(arr);

    TEST_ASSERT_EQUAL(DYNAMIC_ARRAY_DEFAULT_ARRAY_SIZE, array_get_header(arr)->capacity);

    array_destroy(arr);
}


void test_array_size_capacity_relationship(void) {
    int *arr = NULL;
    array_create(arr);

    array_push_value(arr, 1);
    array_push_value(arr, 2);
    array_push_value(arr, 3);

    const ArrayHeader *header = array_get_header(arr);
    TEST_ASSERT_EQUAL(3, header->size);
    TEST_ASSERT_LESS_OR_EQUAL(header->capacity, header->size);

    array_destroy(arr);
}


void test_array_push_beyond_capacity(void) {
    int *arr = NULL;
    array_create(arr);

    const int num_elements = DYNAMIC_ARRAY_DEFAULT_ARRAY_SIZE * 2;
    for (int i = 0; i < num_elements; i++) {
        array_push_value(arr, i);
    }

    const ArrayHeader *header = array_get_header(arr);
    TEST_ASSERT_EQUAL(num_elements, header->size);
    TEST_ASSERT_GREATER_THAN(DYNAMIC_ARRAY_DEFAULT_ARRAY_SIZE, header->capacity);

    array_destroy(arr);
}


void test_array_values_correct_after_growth(void) {
    int *arr = NULL;
    array_create(arr);

    const int num_elements = 1000;
    for (int i = 0; i < num_elements; i++) {
        array_push_value(arr, i);
    }

    TEST_ASSERT_EQUAL(num_elements, array_get_header(arr)->size);
    for (int i = 0; i < num_elements; i++) {
        TEST_ASSERT_EQUAL(i, arr[i]);
    }

    array_destroy(arr);
}


int main(void)
{
    UNITY_BEGIN();

    RUN_TEST(test_array_create_without_options);
    RUN_TEST(test_array_destroy);
    RUN_TEST(test_array_size_initial);
    RUN_TEST(test_array_push_value);
    RUN_TEST(test_array_push_multiple_values);
    RUN_TEST(test_array_pop_decrements_size);
    RUN_TEST(test_array_get_capacity_default);
    RUN_TEST(test_array_size_capacity_relationship);
    RUN_TEST(test_array_push_beyond_capacity);
    RUN_TEST(test_array_values_correct_after_growth);

    return UNITY_END();
}