#include "tests/ListTests.h"

#include "data/List.h"
#include "test/Test.h"

using IntList = Core::List<int>;
using String = Core::ArrayString<128>;

template<typename T>
static String build(const T& t) {
    String s;
    CORE_TEST_FALSE(s.append(t));
    return s;
}

static void testAdd() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(5));
    CORE_TEST_EQUAL(5, list[0]);
    CORE_TEST_EQUAL(1, list.getLength());
}

static void testMultipleAdd() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(4));
    CORE_TEST_NOT_NULL(list.add(3));
    CORE_TEST_NOT_NULL(list.add(2));
    CORE_TEST_EQUAL(4, list[0]);
    CORE_TEST_EQUAL(3, list[1]);
    CORE_TEST_EQUAL(2, list[2]);
    CORE_TEST_EQUAL(3, list.getLength());
}

static void testAddReplace() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(5));
    list[0] = 3;
    CORE_TEST_EQUAL(3, list[0]);
}

static void testClear() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(5));
    CORE_TEST_NOT_NULL(list.add(4));
    list.clear();
    CORE_TEST_EQUAL(0, list.getLength());
}

static void testShrink() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(5));
    CORE_TEST_NOT_NULL(list.add(4));
    CORE_TEST_NOT_NULL(list.add(3));
    CORE_TEST_TRUE(list.getCapacity() >= 3);
    CORE_TEST_FALSE(list.shrink());
    CORE_TEST_TRUE(list.getLength() == 3);
    CORE_TEST_TRUE(list.getCapacity() == 3);
    CORE_TEST_EQUAL(5, list[0]);
    CORE_TEST_EQUAL(4, list[1]);
    CORE_TEST_EQUAL(3, list[2]);
}

static void testBigAdd() {
    IntList list;
    for(int i = 0; i < 100000; i++) {
        CORE_TEST_NOT_NULL(list.add(i));
    }
    for(int i = 0; i < list.getLength(); i++) {
        CORE_TEST_EQUAL(i, list[i]);
    }
    CORE_TEST_EQUAL(100000, list.getLength());
}

static void testCopy() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(1));
    CORE_TEST_NOT_NULL(list.add(2));
    CORE_TEST_NOT_NULL(list.add(3));

    IntList copy;
    CORE_TEST_FALSE(copy.copyFrom(list));
    CORE_TEST_EQUAL(list.getLength(), copy.getLength());
    for(int i = 0; i < copy.getLength() && i < list.getLength(); i++) {
        CORE_TEST_EQUAL(list[i], copy[i]);
    }
}

static void testMove() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(1));
    CORE_TEST_NOT_NULL(list.add(2));
    CORE_TEST_NOT_NULL(list.add(3));

    IntList move(Core::move(list));
    CORE_TEST_EQUAL(0, list.getLength());
    CORE_TEST_EQUAL(3, move.getLength());
    CORE_TEST_EQUAL(1, move[0]);
    CORE_TEST_EQUAL(2, move[1]);
    CORE_TEST_EQUAL(3, move[2]);
}

static void testMoveAssignment() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(1));
    CORE_TEST_NOT_NULL(list.add(2));
    CORE_TEST_NOT_NULL(list.add(3));

    IntList move;
    move = Core::move(list);
    CORE_TEST_EQUAL(0, list.getLength());
    CORE_TEST_EQUAL(3, move.getLength());
    CORE_TEST_EQUAL(1, move[0]);
    CORE_TEST_EQUAL(2, move[1]);
    CORE_TEST_EQUAL(3, move[2]);
}

static void testToString1() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(1));
    CORE_TEST_NOT_NULL(list.add(243));
    CORE_TEST_NOT_NULL(list.add(-423));
    CORE_TEST_EQUAL(build("[1, 243, -423]"), build(list));
}

static void testToString2() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(1));
    CORE_TEST_EQUAL(build("[1]"), build(list));
}

static void testToString3() {
    IntList list;
    CORE_TEST_EQUAL(build("[]"), build(list));
}

static void testRemoveBySwap() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(4));
    CORE_TEST_NOT_NULL(list.add(3));
    CORE_TEST_NOT_NULL(list.add(2));
    CORE_TEST_TRUE(list.removeBySwap(-1));
    CORE_TEST_FALSE(list.removeBySwap(0));
    CORE_TEST_TRUE(list.removeBySwap(2));
    CORE_TEST_EQUAL(2, list[0]);
    CORE_TEST_EQUAL(3, list[1]);
    CORE_TEST_EQUAL(2, list.getLength());
    CORE_TEST_FALSE(list.removeBySwap(1));
    CORE_TEST_EQUAL(2, list[0]);
    CORE_TEST_EQUAL(1, list.getLength());
    CORE_TEST_FALSE(list.removeBySwap(0));
    CORE_TEST_EQUAL(0, list.getLength());
    CORE_TEST_TRUE(list.removeBySwap(0));
}

static void testRemove() {
    IntList list;
    CORE_TEST_NOT_NULL(list.add(4));
    CORE_TEST_NOT_NULL(list.add(3));
    CORE_TEST_NOT_NULL(list.add(2));
    CORE_TEST_TRUE(list.remove(-1));
    CORE_TEST_FALSE(list.remove(0));
    CORE_TEST_TRUE(list.remove(2));
    CORE_TEST_EQUAL(3, list[0]);
    CORE_TEST_EQUAL(2, list[1]);
    CORE_TEST_EQUAL(2, list.getLength());
    CORE_TEST_FALSE(list.remove(1));
    CORE_TEST_EQUAL(3, list[0]);
    CORE_TEST_EQUAL(1, list.getLength());
    CORE_TEST_FALSE(list.remove(0));
    CORE_TEST_EQUAL(0, list.getLength());
    CORE_TEST_TRUE(list.remove(0));
}

static void testResize() {
    IntList list;
    CORE_TEST_FALSE(list.resize(5, 10));
    CORE_TEST_EQUAL(5, list.getLength());
    for(int i = 0; i < 5; i++) {
        CORE_TEST_EQUAL(10, list[i]);
    }
}

static void testDefaultResize() {
    IntList list;
    CORE_TEST_FALSE(list.resize(5));
    CORE_TEST_EQUAL(5, list.getLength());
    for(int i = 0; i < 5; i++) {
        CORE_TEST_EQUAL(0, list[i]);
    }
}

void Core::ListTests::test() {
    testAdd();
    testMultipleAdd();
    testAddReplace();
    testClear();
    testShrink();
    testBigAdd();
    testCopy();
    testMove();
    testMoveAssignment();
    testToString1();
    testToString2();
    testToString3();
    testRemoveBySwap();
    testRemove();
    testResize();
    testDefaultResize();
}