#include "test_allocator.h"

#include <internal/bag.h>

#include <gtest/gtest.h>

TEST(TestBag, CheckData)
{
    // Arrange
    HT_BagVoidPtr bag;
    ht_bag_void_ptr_init(&bag, 4);

    // Act
    for (intptr_t i = 0; i < 10; i++)
    {
        ht_bag_void_ptr_add(&bag, (void*)i);
    }

    // Assert
    for (intptr_t i = 0; i < 10; i++)
    {
        ASSERT_EQ(i, (intptr_t)bag.data[i]);
    }

    ASSERT_EQ(16u, bag.capacity);
    ASSERT_EQ(10u, bag.size);
    ASSERT_EQ(4u, bag.min_capacity);

    ht_bag_void_ptr_deinit(&bag);
}

TEST(TestBag, ShouldResizeIfTooManyItems)
{
    // Arrange
    HT_BagVoidPtr bag;
    ht_bag_void_ptr_init(&bag, 4);

    // Act
    for (int i = 0; i < 10; i++)
    {
        ht_bag_void_ptr_add(&bag, (void*)1);
    }

    // Assert
    ASSERT_EQ(16u, bag.capacity);
    ASSERT_EQ(10u, bag.size);
    ASSERT_EQ(4u, bag.min_capacity);

    ht_bag_void_ptr_deinit(&bag);
}

TEST(TestBag, ShouldNotResizeIfRemoveOnlyFewItems)
{
    // Arrange
    HT_BagVoidPtr bag;
    ht_bag_void_ptr_init(&bag, 4);

    for (intptr_t i = 0; i < 30; i++)
    {
        ht_bag_void_ptr_add(&bag, (void*)i);
    }

    // Act
    for (intptr_t i = 9; i >= 0; i--)
    {
        ht_bag_void_ptr_remove_nth(&bag, i);
    }

    // Assert
    ASSERT_EQ(32u, bag.capacity);
    ASSERT_EQ(20u, bag.size);
    ASSERT_EQ(4u, bag.min_capacity);

    ht_bag_void_ptr_deinit(&bag);
}

TEST(TestBag, ShouldResizeIfRemoveManyItems)
{
    // Arrange
    HT_BagVoidPtr bag;
    ht_bag_void_ptr_init(&bag, 4);

    for (intptr_t i = 0; i < 30; i++)
    {
        ht_bag_void_ptr_add(&bag, (void*)i);
    }

    // Act
    for (intptr_t i = 23; i >= 0; i--)
    {
        ht_bag_void_ptr_remove_nth(&bag, i);
    }

    // Assert
    ASSERT_EQ(16u, bag.capacity);
    ASSERT_EQ(6u, bag.size);
    ASSERT_EQ(4u, bag.min_capacity);

    ht_bag_void_ptr_deinit(&bag);
}

TEST(TestBag, InitShouldFailIfMallocReturnsNull)
{
    // Arrange
    HT_BagVoidPtr bag;
    ScopedSetAlloc allocator(ht_test_null_realloc);

    // Act
    HT_ErrorCode error_code = ht_bag_void_ptr_init(&bag, 8);

    // Assert
    ASSERT_EQ(HT_ERR_OUT_OF_MEMORY, error_code);
}