#include <ungrateful.h>

b32 sort_test_case_s32(s32 *data, s32 *expect, u64 size) {
    un_sort_quick(data, sizeof(s32), 0, size, un_sort_s32_comp_proc);
    return memcmp(data, expect, size * sizeof(s32)) == 0;
}

b32 sort_test_case_string(String *data, String *expect, u64 size) {
    un_sort_quick(data, sizeof(String), 0, size, un_sort_string_comp_proc);

    for (u64 i = 0; i < size; i++) {
        if (un_string_compare(data[i], expect[i]) != 0) return false;
    }
    return true;
}

int main() {
    un_init(UN_MB(4));
    {
        s32 data[]     = {5, 4, 3, 2, 1};
        s32 expected[] = {1, 2, 3, 4, 5};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {1, 2, 3, 4, 5};
        s32 expected[] = {1, 2, 3, 4, 5};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {3, 1, 4, 5, 2};
        s32 expected[] = {1, 2, 3, 4, 5};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {2, 3, 2, 1, 3, 4};
        s32 expected[] = {1, 2, 2, 3, 3, 4};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {42};
        s32 expected[] = {42};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {-3, 1, -2, 0, -1};
        s32 expected[] = {-3, -2, -1, 0, 1};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {7, 7, 7, 7, 7};
        s32 expected[] = {7, 7, 7, 7, 7};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        s32 data[]     = {9, 3, 7, 1, 8, 2, 5, 4, 6, 0};
        s32 expected[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
        assert(sort_test_case_s32(data, expected, sizeof(data) / sizeof(s32)));
    }
    {
        String data[]     = { UN_STR("apple"), UN_STR("banana"), UN_STR("cherry") };
        String expected[] = { UN_STR("apple"), UN_STR("banana"), UN_STR("cherry") };
        assert(sort_test_case_string(data, expected, sizeof(data) / sizeof(data[0])));
    }
    {
        String data[]     = { UN_STR("zebra"), UN_STR("elephant"), UN_STR("ant") };
        String expected[] = { UN_STR("ant"), UN_STR("elephant"), UN_STR("zebra") };
        assert(sort_test_case_string(data, expected, sizeof(data) / sizeof(data[0])));
    }
    {
        String data[]     = { UN_STR("Apple"), UN_STR("banana"), UN_STR("cherry") };
        String expected[] = { UN_STR("Apple"), UN_STR("banana"), UN_STR("cherry") };
        assert(sort_test_case_string(data, expected, sizeof(data) / sizeof(data[0])));
    }
    {
        String data[]     = { UN_STR(""), UN_STR("123"), UN_STR("!@#"), UN_STR("abc") };
        String expected[] = { UN_STR(""), UN_STR("!@#"), UN_STR("123"), UN_STR("abc") };
        assert(sort_test_case_string(data, expected, sizeof(data) / sizeof(data[0])));
    }
    {
        String data[]     = { UN_STR("hello world"), UN_STR("hello"), UN_STR("  space"), UN_STR("space") };
        String expected[] = { UN_STR("  space"), UN_STR("hello"), UN_STR("hello world"), UN_STR("space") };
        assert(sort_test_case_string(data, expected, sizeof(data) / sizeof(data[0])));
    }

    return 0;
}