#include <gtest/gtest.h>
#include <sparse_set.hpp>

TEST(SparseSetNoType, Functionalities) {
    using SparseSet = entt::SparseSet<unsigned int>;

    SparseSet set;

    ASSERT_TRUE(set.empty());
    ASSERT_EQ(set.size(), 0u);
    ASSERT_EQ(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_FALSE(set.has(42));

    ASSERT_EQ(set.construct(42), 0u);

    ASSERT_FALSE(set.empty());
    ASSERT_EQ(set.size(), 1u);
    ASSERT_NE(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_TRUE(set.has(42));
    ASSERT_EQ(set.get(42), 0u);

    ASSERT_EQ(set.destroy(42), 0u);

    ASSERT_TRUE(set.empty());
    ASSERT_EQ(set.size(), 0u);
    ASSERT_EQ(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_FALSE(set.has(42));

    ASSERT_EQ(set.construct(42), 0u);

    set.reset();

    ASSERT_TRUE(set.empty());
    ASSERT_EQ(set.size(), 0u);
    ASSERT_EQ(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_FALSE(set.has(42));
}

TEST(SparseSetNoType, DataBeginEnd) {
    using SparseSet = entt::SparseSet<unsigned int>;

    SparseSet set;

    ASSERT_EQ(set.construct(3), 0u);
    ASSERT_EQ(set.construct(12), 1u);
    ASSERT_EQ(set.construct(42), 2u);

    ASSERT_EQ(*(set.data() + 0u), 3u);
    ASSERT_EQ(*(set.data() + 1u), 12u);
    ASSERT_EQ(*(set.data() + 2u), 42u);

    auto begin = set.begin();
    auto end = set.end();

    ASSERT_EQ(*(begin++), 42u);
    ASSERT_EQ(*(begin++), 12u);
    ASSERT_EQ(*(begin++), 3u);
    ASSERT_EQ(begin, end);

    set.reset();
}

TEST(SparseSetNoType, Swap) {
    using SparseSet = entt::SparseSet<unsigned int>;

    SparseSet set;

    ASSERT_EQ(set.construct(3), 0u);
    ASSERT_EQ(set.construct(12), 1u);

    ASSERT_EQ(*(set.data() + 0u), 3u);
    ASSERT_EQ(*(set.data() + 1u), 12u);

    set.swap(3, 12);

    ASSERT_EQ(*(set.data() + 0u), 12u);
    ASSERT_EQ(*(set.data() + 1u), 3u);

    set.reset();
}

TEST(SparseSetWithType, Functionalities) {
    using SparseSet = entt::SparseSet<unsigned int, int>;

    SparseSet set;

    ASSERT_TRUE(set.empty());
    ASSERT_EQ(set.size(), 0u);
    ASSERT_EQ(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_FALSE(set.has(42));

    ASSERT_EQ(set.construct(42, 3), 3);

    ASSERT_FALSE(set.empty());
    ASSERT_EQ(set.size(), 1u);
    ASSERT_NE(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_TRUE(set.has(42));
    ASSERT_EQ(set.get(42), 3);

    set.destroy(42);

    ASSERT_TRUE(set.empty());
    ASSERT_EQ(set.size(), 0u);
    ASSERT_EQ(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_FALSE(set.has(42));

    ASSERT_EQ(set.construct(42, 12), 12);

    set.reset();

    ASSERT_TRUE(set.empty());
    ASSERT_EQ(set.size(), 0u);
    ASSERT_EQ(set.begin(), set.end());
    ASSERT_FALSE(set.has(0));
    ASSERT_FALSE(set.has(42));
}

TEST(SparseSetWithType, RawBeginEnd) {
    using SparseSet = entt::SparseSet<unsigned int, int>;

    SparseSet set;

    ASSERT_EQ(set.construct(3, 3), 3);
    ASSERT_EQ(set.construct(12, 6), 6);
    ASSERT_EQ(set.construct(42, 9), 9);

    ASSERT_EQ(*(set.raw() + 0u), 3);
    ASSERT_EQ(*(set.raw() + 1u), 6);
    ASSERT_EQ(*(set.raw() + 2u), 9);

    auto begin = set.begin();
    auto end = set.end();

    ASSERT_EQ(set.get(*(begin++)), 9);
    ASSERT_EQ(set.get(*(begin++)), 6);
    ASSERT_EQ(set.get(*(begin++)), 3);
    ASSERT_EQ(begin, end);

    set.reset();
}

TEST(SparseSetWithType, Swap) {
    using SparseSet = entt::SparseSet<unsigned int, int>;

    SparseSet set;

    ASSERT_EQ(set.construct(3, 3), 3);
    ASSERT_EQ(set.construct(12, 6), 6);

    ASSERT_EQ(*(set.raw() + 0u), 3);
    ASSERT_EQ(*(set.raw() + 1u), 6);

    set.swap(3, 12);

    ASSERT_EQ(*(set.raw() + 0u), 6);
    ASSERT_EQ(*(set.raw() + 1u), 3);

    set.reset();
}

TEST(SparseSetWithType, SortBasic) {
    using SparseSet = entt::SparseSet<unsigned int, int>;

    SparseSet set;

    ASSERT_EQ(set.construct(12, 6), 6);
    ASSERT_EQ(set.construct(42, 9), 9);
    ASSERT_EQ(set.construct(3, 3), 3);

    set.sort([](auto lhs, auto rhs) {
        return lhs < rhs;
    });

    ASSERT_EQ(*(set.raw() + 0u), 9);
    ASSERT_EQ(*(set.raw() + 1u), 6);
    ASSERT_EQ(*(set.raw() + 2u), 3);

    auto begin = set.begin();
    auto end = set.end();

    ASSERT_EQ(set.get(*(begin++)), 3);
    ASSERT_EQ(set.get(*(begin++)), 6);
    ASSERT_EQ(set.get(*(begin++)), 9);
    ASSERT_EQ(begin, end);

    set.reset();
}

TEST(SparseSetWithType, SortAccordingTo) {
    // TODO
}