#include <library/cpp/testing/unittest/registar.h>

#include "compact_vector.h"

#include <string>

Y_UNIT_TEST_SUITE(TCompactVectorTest) {
    Y_UNIT_TEST(TestSimple1) {
    }

    Y_UNIT_TEST(TestSimple) {
        TCompactVector<ui32> vector;
        for (ui32 i = 0; i < 10000; ++i) {
            vector.PushBack(i + 20);
            UNIT_ASSERT_VALUES_EQUAL(i + 1, vector.Size());
        }
        for (ui32 i = 0; i < 10000; ++i) {
            UNIT_ASSERT_VALUES_EQUAL(i + 20, vector[i]);
        }
    }

    Y_UNIT_TEST(TestInsert) {
        TCompactVector<ui32> vector;

        for (ui32 i = 0; i < 10; ++i) {
            vector.PushBack(i + 2);
        }

        vector.Insert(vector.Begin(), 99);

        UNIT_ASSERT_VALUES_EQUAL(11u, vector.Size());
        UNIT_ASSERT_VALUES_EQUAL(99u, vector[0]);
        for (ui32 i = 0; i < 10; ++i) {
            UNIT_ASSERT_VALUES_EQUAL(i + 2, vector[i + 1]);
        }

        vector.Insert(vector.Begin() + 3, 77);

        UNIT_ASSERT_VALUES_EQUAL(12u, vector.Size());
        UNIT_ASSERT_VALUES_EQUAL(99u, vector[0]);
        UNIT_ASSERT_VALUES_EQUAL(2u, vector[1]);
        UNIT_ASSERT_VALUES_EQUAL(3u, vector[2]);
        UNIT_ASSERT_VALUES_EQUAL(77u, vector[3]);
        UNIT_ASSERT_VALUES_EQUAL(4u, vector[4]);
        UNIT_ASSERT_VALUES_EQUAL(5u, vector[5]);
        UNIT_ASSERT_VALUES_EQUAL(11u, vector[11]);
    }

    Y_UNIT_TEST(TestEmplace) {
        struct TFoo {
            explicit TFoo(int)
                : ConstructedFrom("int")
            {
            }

            explicit TFoo(char)
                : ConstructedFrom("char")
            {
            }

            std::string ConstructedFrom;
        };

        TCompactVector<TFoo> vector;
        auto& x = vector.emplace_back(123);
        UNIT_ASSERT_VALUES_EQUAL(&x, &vector.back());
        UNIT_ASSERT_VALUES_EQUAL(vector.back().ConstructedFrom, "int");
        vector.emplace_back('c');
        UNIT_ASSERT_VALUES_EQUAL(vector.back().ConstructedFrom, "char");
    }

    Y_UNIT_TEST(TestInitializerListConstructor) {
        TCompactVector<ui32> vector = { 4, 8, 10, 3, 5};
        UNIT_ASSERT_VALUES_EQUAL(5u, vector.Size());

        UNIT_ASSERT_VALUES_EQUAL(4u, vector[0]);
        UNIT_ASSERT_VALUES_EQUAL(8u, vector[1]);
        UNIT_ASSERT_VALUES_EQUAL(10u, vector[2]);
        UNIT_ASSERT_VALUES_EQUAL(3u, vector[3]);
        UNIT_ASSERT_VALUES_EQUAL(5u, vector[4]);
    }

    Y_UNIT_TEST(TestIteratorConstructor) {
        TVector<ui32> origVector = { 4, 8, 10, 3, 5};
        TCompactVector<ui32> vector(origVector.begin(), origVector.end());
        UNIT_ASSERT_VALUES_EQUAL(5u, vector.Size());

        UNIT_ASSERT_VALUES_EQUAL(4u, vector[0]);
        UNIT_ASSERT_VALUES_EQUAL(8u, vector[1]);
        UNIT_ASSERT_VALUES_EQUAL(10u, vector[2]);
        UNIT_ASSERT_VALUES_EQUAL(3u, vector[3]);
        UNIT_ASSERT_VALUES_EQUAL(5u, vector[4]);
    }

    Y_UNIT_TEST(TestInitializerListCopyOperator) {
        TCompactVector<double> vector = { 4, 8, 10, 3, 5};
        UNIT_ASSERT_VALUES_EQUAL(5u, vector.Size());

        vector = { 11, 17, 23 };
        UNIT_ASSERT_VALUES_EQUAL(3u, vector.Size());

        UNIT_ASSERT_VALUES_EQUAL(11.0, vector[0]);
        UNIT_ASSERT_VALUES_EQUAL(17.0, vector[1]);
        UNIT_ASSERT_VALUES_EQUAL(23.0, vector[2]);
    }

    Y_UNIT_TEST(TestMoveConstructor) {
        TCompactVector<ui32> vector = { 4, 8, 10, 3, 5};
        auto it = vector.Begin();

        TCompactVector<ui32> vector2(std::move(vector));
        UNIT_ASSERT_VALUES_EQUAL(it, vector2.begin());

        UNIT_ASSERT_VALUES_EQUAL(5u, vector2.Size());

        UNIT_ASSERT_VALUES_EQUAL(4u, vector2[0]);
        UNIT_ASSERT_VALUES_EQUAL(8u, vector2[1]);
        UNIT_ASSERT_VALUES_EQUAL(10u, vector2[2]);
        UNIT_ASSERT_VALUES_EQUAL(3u, vector2[3]);
        UNIT_ASSERT_VALUES_EQUAL(5u, vector2[4]);
    }

    Y_UNIT_TEST(TestReverseIterators) {
        TCompactVector<std::string> vector = {
                "мама",
                "мыла",
                "раму"
        };

        TCompactVector<std::string> reverseVector(vector.rbegin(), vector.rend());
        UNIT_ASSERT_VALUES_EQUAL(3u, reverseVector.Size());

        UNIT_ASSERT_VALUES_EQUAL("раму", reverseVector[0]);
        UNIT_ASSERT_VALUES_EQUAL("мыла", reverseVector[1]);
        UNIT_ASSERT_VALUES_EQUAL("мама", reverseVector[2]);
    }

    Y_UNIT_TEST(TestErase) {
        TCompactVector<std::string> vector = {
                "мама",
                "утром",
                "мыла",
                "раму"
        };

        vector.erase(vector.begin() + 1);
        UNIT_ASSERT_VALUES_EQUAL(3u, vector.Size());

        UNIT_ASSERT_VALUES_EQUAL("мама", vector[0]);
        UNIT_ASSERT_VALUES_EQUAL("мыла", vector[1]);
        UNIT_ASSERT_VALUES_EQUAL("раму", vector[2]);
    }

    Y_UNIT_TEST(TestCopyAssignmentOperator) {
        TCompactVector<std::string> vector;
        const TCompactVector<std::string> vector2 = {
                "мама",
                "мыла",
                "раму"
        };

        vector = vector2;

        UNIT_ASSERT_VALUES_EQUAL(3u, vector.Size());

        UNIT_ASSERT_VALUES_EQUAL("мама", vector[0]);
        UNIT_ASSERT_VALUES_EQUAL("мыла", vector[1]);
        UNIT_ASSERT_VALUES_EQUAL("раму", vector[2]);

        UNIT_ASSERT_VALUES_EQUAL(vector[0], vector2[0]);
        UNIT_ASSERT_VALUES_EQUAL(vector[1], vector2[1]);
        UNIT_ASSERT_VALUES_EQUAL(vector[2], vector2[2]);
    }

    Y_UNIT_TEST(TestComparison) {
        // UNIT_ASSERT_VALUES_EQUAL needs a specialization for Out(), so we just use bool assertions
        TCompactVector<int> vector {1, 2, 3};
        TCompactVector<int> vector2 {1, 2};
        UNIT_ASSERT(vector != vector2);  // diff size
        vector2.emplace_back(4);
        UNIT_ASSERT(vector != vector2);  // diff values
        vector2.back() = 3;
        UNIT_ASSERT(vector == vector2);
    }
}