#ifndef CMC_TESTS_UNT_HEAP_H
#define CMC_TESTS_UNT_HEAP_H

#include "utl.h"

#define V size_t
#define PFX h
#define SNAME heap
#include "cmc/heap.h"

struct heap_fval *h_fval = &(struct heap_fval){
    .cmp = cmc_size_cmp, .cpy = NULL, .str = cmc_size_str, .free = NULL, .hash = cmc_size_hash, .pri = cmc_size_cmp
};

CMC_CREATE_UNIT(CMCHeap, true, {
    CMC_CREATE_TEST(new, {
        struct heap *h = h_new(1000000, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);
        cmc_assert_not_equals(ptr, NULL, h->buffer);
        cmc_assert_equals(size_t, 1000000, h_capacity(h));
        cmc_assert_equals(size_t, 0, h_count(h));

        h_free(h);
    });

    CMC_CREATE_TEST(new[edge_case capacity = 0], {
        struct heap *h = h_new(0, CMC_MAX_HEAP, h_fval);

        cmc_assert_equals(ptr, NULL, h);
    });

    CMC_CREATE_TEST(new[edge_case capacity = UINT64_MAX], {
        struct heap *h = h_new(UINT64_MAX, CMC_MAX_HEAP, h_fval);

        cmc_assert_equals(ptr, NULL, h);
    });

    CMC_CREATE_TEST(clear[count capacity], {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 0; i < 50; i++)
            h_insert(h, i);

        cmc_assert_equals(size_t, 50, h_count(h));

        h_clear(h);

        cmc_assert_equals(size_t, 0, h_count(h));
        cmc_assert_equals(size_t, 100, h_capacity(h));

        h_free(h);
    });

    CMC_CREATE_TEST(buffer_growth[edge_case capacity = 1], {
        struct heap *h = h_new(1, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 0; i < 50; i++)
            cmc_assert(h_insert(h, i));

        cmc_assert_equals(size_t, 50, h_count(h));

        h_free(h);
    });

    CMC_CREATE_TEST(insert, {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        cmc_assert(h_insert(h, 10));

        h_free(h);
    });

    CMC_CREATE_TEST(insert[item preservation], {
        struct heap *h = h_new(50, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 0; i <= 100; i++)
            cmc_assert(h_insert(h, i));

        size_t sum = 0;
        for (size_t i = 0; i < h_count(h); i++)
            sum += h->buffer[i];

        cmc_assert_equals(size_t, 5050, sum);

        h_free(h);
    });

    CMC_CREATE_TEST(remove, {
        struct heap *h = h_new(50, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 0; i <= 100; i++)
            cmc_assert(h_insert(h, i));

        cmc_assert_equals(size_t, 100, h_peek(h));
        cmc_assert(h_remove(h));
        cmc_assert_equals(size_t, 99, h_peek(h));

        h_free(h);
    });

    CMC_CREATE_TEST(remove[count = 0], {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        cmc_assert(!h_remove(h));

        h_free(h);
    });

    CMC_CREATE_TEST(remove[sorted], {
        struct heap *h = h_new(100, CMC_MIN_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 1; i <= 100; i++)
            cmc_assert(h_insert(h, i));

        size_t i = 0;
        size_t arr[100];
        while (!h_empty(h))
        {
            arr[i++] = h_peek(h);
            cmc_assert(h_remove(h));
        }

        cmc_assert_array_sorted_any(size_t, arr, cmc_size_cmp, 0, 99);

        h_free(h);
    });

    CMC_CREATE_TEST(peek, {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 1; i <= 100; i++)
            cmc_assert(h_insert(h, i));

        cmc_assert_equals(size_t, 100, h_peek(h));

        h_free(h);
    });

    CMC_CREATE_TEST(peek[count = 0], {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        cmc_assert_equals(size_t, (size_t){ 0 }, h_peek(h));

        h_free(h);
    });

    CMC_CREATE_TEST(flags, {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));

        h->flag = CMC_FLAG_ERROR;

        // insert
        cmc_assert(h_insert(h, 10));
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));

        h->flag = CMC_FLAG_ERROR;

        // remove
        cmc_assert(h_remove(h));
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));

        cmc_assert(!h_remove(h));
        cmc_assert_equals(int32_t, CMC_FLAG_EMPTY, h_flag(h));

        // peek
        cmc_assert(h_insert(h, 10));
        h_peek(h);
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));
        cmc_assert(h_remove(h));
        h_peek(h);
        cmc_assert_equals(int32_t, CMC_FLAG_EMPTY, h_flag(h));

        // clear
        h_clear(h);
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));

        h->flag = CMC_FLAG_ERROR;

        // contains
        h_contains(h, 10);
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));

        h->flag = CMC_FLAG_ERROR;

        // copy of
        struct heap *h2 = h_copy_of(h);
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h2));

        h->flag = CMC_FLAG_ERROR;
        h2->flag = CMC_FLAG_ERROR;

        // equals
        cmc_assert(h_equals(h, h2));
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h));
        cmc_assert_equals(int32_t, CMC_FLAG_OK, h_flag(h2));

        h_free(h);
        h_free(h2);
    });
});

CMC_CREATE_UNIT(CMCHeapIter, true, {
    CMC_CREATE_TEST(PFX##_iter_start(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_start(h);

        cmc_assert_equals(ptr, h, it.target);
        cmc_assert_equals(size_t, 0, it.cursor);
        cmc_assert_equals(bool, true, it.start);
        cmc_assert_equals(bool, true, it.end);

        cmc_assert(h_iter_at_start(&it));
        cmc_assert(h_iter_at_end(&it));

        cmc_assert(h_insert(h, 1));
        cmc_assert(h_insert(h, 2));

        it = h_iter_start(h);

        cmc_assert_equals(size_t, 0, it.cursor);
        cmc_assert_equals(bool, false, it.end);

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_end(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_end(h);

        cmc_assert_equals(ptr, h, it.target);
        cmc_assert_equals(size_t, 0, it.cursor);
        cmc_assert_equals(bool, true, it.start);
        cmc_assert_equals(bool, true, it.end);

        cmc_assert(h_iter_at_start(&it));
        cmc_assert(h_iter_at_end(&it));

        cmc_assert(h_insert(h, 1));
        cmc_assert(h_insert(h, 2));
        cmc_assert(h_insert(h, 3));

        it = h_iter_end(h);

        cmc_assert_equals(size_t, h->count - 1, it.cursor);
        cmc_assert_equals(bool, false, it.start);

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_at_start(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_start(h);

        // Empty checks
        cmc_assert(h_iter_at_start(&it));
        it = h_iter_end(h);
        cmc_assert(h_iter_at_start(&it));

        // Non-empty check
        cmc_assert(h_insert(h, 1));
        it = h_iter_end(h);
        cmc_assert(!h_iter_at_start(&it));
        it = h_iter_start(h);
        cmc_assert(h_iter_at_start(&it));

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_at_end(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_start(h);

        // Empty check
        cmc_assert(h_iter_at_end(&it));
        it = h_iter_end(h);
        cmc_assert(h_iter_at_end(&it));

        // Non-empty check
        cmc_assert(h_insert(h, 1));
        it = h_iter_end(h);
        cmc_assert(h_iter_at_end(&it));
        it = h_iter_start(h);
        cmc_assert(!h_iter_at_end(&it));

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_to_start(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_start(h);

        cmc_assert(!h_iter_to_start(&it));

        for (size_t i = 1; i <= 100; i++)
        {
            if (i % 2 == 0)
                h_insert(h, i);
            else
                h_insert(h, i);
        }

        cmc_assert_equals(size_t, 100, h->count);

        it = h_iter_end(h);
        cmc_assert(!h_iter_at_start(&it));
        cmc_assert(h_iter_at_end(&it));
        cmc_assert_equals(size_t, 99, it.cursor);

        cmc_assert(h_iter_to_start(&it));
        cmc_assert(h_iter_at_start(&it));
        cmc_assert(!h_iter_at_end(&it));
        cmc_assert_equals(size_t, 0, it.cursor);

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_to_end(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_end(h);

        cmc_assert(!h_iter_to_end(&it));

        for (size_t i = 1; i <= 100; i++)
        {
            if (i % 2 == 0)
                h_insert(h, i);
            else
                h_insert(h, i);
        }

        it = h_iter_start(h);
        cmc_assert(h_iter_at_start(&it));
        cmc_assert(!h_iter_at_end(&it));

        cmc_assert(h_iter_to_end(&it));
        cmc_assert(!h_iter_at_start(&it));
        cmc_assert(h_iter_at_end(&it));

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_next(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_start(h);

        cmc_assert(!h_iter_next(&it));

        for (size_t i = 1; i <= 1000; i++)
        {
            if (i % 2 == 0)
                h_insert(h, i);
            else
                h_insert(h, i);
        }

        size_t sum = 0;
        for (it = h_iter_start(h); !h_iter_at_end(&it); h_iter_next(&it))
        {
            sum += h_iter_value(&it);
        }

        cmc_assert_equals(size_t, 500500, sum);

        sum = 0;

        h_iter_to_start(&it);
        do
        {
            sum += h_iter_value(&it);
        } while (h_iter_next(&it));

        cmc_assert_equals(size_t, 500500, sum);

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_prev(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_end(h);

        cmc_assert(!h_iter_prev(&it));

        for (size_t i = 1; i <= 1000; i++)
        {
            if (i % 2 == 0)
                h_insert(h, i);
            else
                h_insert(h, i);
        }

        size_t sum = 0;
        for (it = h_iter_end(h); !h_iter_at_start(&it); h_iter_prev(&it))
        {
            sum += h_iter_value(&it);
        }

        cmc_assert_equals(size_t, 500500, sum);

        sum = 0;

        h_iter_to_end(&it);
        do
        {
            sum += h_iter_value(&it);
        } while (h_iter_prev(&it));

        cmc_assert_equals(size_t, 500500, sum);

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_advance(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_start(h);

        cmc_assert(!h_iter_advance(&it, 1));

        for (size_t i = 1; i <= 1000; i++)
            h_insert(h, i);

        it = h_iter_start(h);

        cmc_assert(!h_iter_advance(&it, 0));
        cmc_assert(!h_iter_advance(&it, h->count));

        size_t sum = 0;
        for (it = h_iter_start(h);;)
        {
            sum += h_iter_value(&it);

            if (!h_iter_advance(&it, 2))
                break;

            cmc_assert_equals(size_t, 0, h_iter_index(&it) % 2);
        }

        cmc_assert(h_iter_to_start(&it));
        cmc_assert(h_iter_next(&it));

        cmc_assert_equals(size_t, 1, h_iter_index(&it));

        for (;;)
        {
            sum += h_iter_value(&it);

            if (!h_iter_advance(&it, 2))
                break;

            cmc_assert_equals(size_t, 1, h_iter_index(&it) % 2);
        }

        cmc_assert_equals(size_t, 500500, sum);

        h_iter_to_start(&it);
        cmc_assert(h_iter_advance(&it, h->count - 1));

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_rewind(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_end(h);

        cmc_assert(!h_iter_rewind(&it, 1));

        for (size_t i = 0; i <= 1000; i++)
            h_insert(h, i);

        it = h_iter_end(h);

        cmc_assert(!h_iter_rewind(&it, 0));
        cmc_assert(!h_iter_rewind(&it, h->count));

        size_t sum = 0;
        for (it = h_iter_end(h);;)
        {
            sum += h_iter_value(&it);

            if (!h_iter_rewind(&it, 2))
                break;

            cmc_assert_equals(size_t, 0, h_iter_index(&it) % 2);
        }

        cmc_assert(h_iter_to_end(&it));
        cmc_assert(h_iter_prev(&it));

        cmc_assert_equals(size_t, h->count - 2, h_iter_index(&it));

        for (;;)
        {
            sum += h_iter_value(&it);

            if (!h_iter_rewind(&it, 2))
                break;

            cmc_assert_equals(size_t, 1, h_iter_index(&it) % 2);
        }

        cmc_assert_equals(size_t, 500500, sum);

        h_iter_to_end(&it);
        cmc_assert(h_iter_rewind(&it, h->count - 1));

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_go_to(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_end(h);
        cmc_assert(!h_iter_go_to(&it, 0));

        it = h_iter_start(h);
        cmc_assert(!h_iter_go_to(&it, 0));

        for (size_t i = 0; i <= 1000; i++)
            h_insert(h, i);

        cmc_assert_equals(size_t, 1001, h->count);

        it = h_iter_start(h);

        size_t sum = 0;
        for (size_t i = 0; i < 1001; i++)
        {
            h_iter_go_to(&it, i);

            sum += h_iter_value(&it);
        }

        cmc_assert_equals(size_t, 500500, sum);

        sum = 0;
        for (size_t i = 1001; i > 0; i--)
        {
            cmc_assert(h_iter_go_to(&it, i - 1));

            sum += h_iter_value(&it);
        }

        cmc_assert_equals(size_t, 500500, sum);

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_value(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        struct heap_iter it = h_iter_end(h);

        cmc_assert_equals(size_t, (size_t){ 0 }, h_iter_value(&it));

        cmc_assert(h_insert(h, 1));
        cmc_assert(h_insert(h, 3));
        cmc_assert(h_insert(h, 2));

        it = h_iter_start(h);

        cmc_assert_equals(size_t, 3, h_iter_value(&it));

        h_free(h);
    });

    CMC_CREATE_TEST(PFX##_iter_index(), {
        struct heap *h = h_new(100, CMC_MAX_HEAP, h_fval);

        cmc_assert_not_equals(ptr, NULL, h);

        for (size_t i = 0; i <= 1000; i++)
            h_insert(h, i);

        struct heap_iter it = h_iter_start(h);

        for (size_t i = 0; i < 1001; i++)
        {
            cmc_assert_equals(size_t, i, h_iter_index(&it));
            h_iter_next(&it);
        }

        it = h_iter_end(h);
        for (size_t i = 1001; i > 0; i--)
        {
            cmc_assert_equals(size_t, i - 1, h_iter_index(&it));
            h_iter_prev(&it);
        }

        h_free(h);
    });
});

#endif /* CMC_TESTS_UNT_HEAP_H */