#include <QSignalSpy>
#include <QTest>

#include <memory>

#include <StatusQ/rolesrenamingmodel.h>

namespace {

class TestSourceModel : public QAbstractListModel {

public:
    explicit TestSourceModel(QList<QString> roles)
        : m_roles(std::move(roles))
    {
    }

    QVariant data(const QModelIndex& index, int role) const override
    {
        if(!index.isValid() || index.row() >= capacity)
            return {};

        return 42;
    }

    int rowCount(const QModelIndex& parent) const override
    {
        return capacity;
    }

    QHash<int, QByteArray> roleNames() const override
    {
        QHash<int, QByteArray> roles;
        roles.remove(m_roles.size());

        for (auto i = 0; i < m_roles.size(); i++)
            roles.insert(i, m_roles.at(i).toUtf8());

        return roles;
    }

private:
    static constexpr auto capacity = 5;
    QList<QString> m_roles;
};

}

class TestRolesRenamingModel: public QObject
{
    Q_OBJECT

private slots:
    void initializationWithBrokenMappingTest()
    {
        TestSourceModel sourceModel({"id", "name", "color"});
        RolesRenamingModel model;

        QQmlListProperty<RoleRename> mapping = model.mapping();

        RoleRename rename;
        rename.setFrom("someIdFrom");
        rename.setTo("someIdTo");

        mapping.append(&mapping, &rename);

        model.setSourceModel(&sourceModel);

        QTest::ignoreMessage(QtWarningMsg,
                             "RolesRenamingModel: specified source roles not "
                             "found: (\"someIdFrom\")!");

        QHash<int, QByteArray> expectedRoles = {
            {0, "id"}, {1, "name"}, {2, "color"}
        };
        QCOMPARE(model.roleNames(), expectedRoles);
    }

    void remappingTest()
    {
        TestSourceModel sourceModel({"id", "name", "color"});
        RolesRenamingModel model;

        QQmlListProperty<RoleRename> mapping = model.mapping();

        RoleRename rename_1;
        rename_1.setFrom("id");
        rename_1.setTo("tokenId");
        mapping.append(&mapping, &rename_1);

        RoleRename rename_2;
        rename_2.setFrom("name");
        rename_2.setTo("tokenName");
        mapping.append(&mapping, &rename_2);

        model.setSourceModel(&sourceModel);

        QHash<int, QByteArray> expectedRoles = {
            {0, "tokenId"}, {1, "tokenName"}, {2, "color"}
        };
        QCOMPARE(model.roleNames(), expectedRoles);
    }

    void addMappingAfterFetchingRoleNamesTest()
    {
        TestSourceModel sourceModel({"id", "name", "color"});
        RolesRenamingModel model;

        QQmlListProperty<RoleRename> mapping = model.mapping();

        RoleRename rename_1;
        rename_1.setFrom("id");
        rename_1.setTo("tokenId");
        mapping.append(&mapping, &rename_1);

        model.setSourceModel(&sourceModel);

        QHash<int, QByteArray> expectedRoles = {
            {0, "tokenId"}, {1, "name"}, {2, "color"}
        };
        QCOMPARE(model.roleNames(), expectedRoles);

        RoleRename rename_2;
        rename_2.setFrom("name");
        rename_2.setTo("tokenName");

        QTest::ignoreMessage(QtWarningMsg,
                             "RolesRenamingModel: role names mapping cannot be "
                             "modified after fetching role names!");
        mapping.append(&mapping, &rename_2);

        QCOMPARE(model.roleNames(), expectedRoles);
    }

    void duplicatedNamesTest()
    {
        TestSourceModel sourceModel({"id", "name", "color"});
        RolesRenamingModel model;

        QQmlListProperty<RoleRename> mapping = model.mapping();

        RoleRename rename_1;
        rename_1.setFrom("id");
        rename_1.setTo("name");
        mapping.append(&mapping, &rename_1);

        model.setSourceModel(&sourceModel);

        QTest::ignoreMessage(QtWarningMsg,
                             "RolesRenamingModel: model cannot contain "
                             "duplicated role names!");

        QCOMPARE(model.roleNames(), {});
    }

    void resettingFromToPropertiesTest()
    {
        RoleRename rename;

        rename.setFrom("id");
        QCOMPARE(rename.from(), "id");
        QCOMPARE(rename.to(), "");

        QTest::ignoreMessage(QtWarningMsg,
                             "RoleRename: property \"from\" is intended to be "
                             "initialized once and not changed!");
        rename.setFrom("id2");
        QCOMPARE(rename.from(), "id");
        QCOMPARE(rename.to(), "");

        rename.setTo("myId");
        QCOMPARE(rename.from(), "id");
        QCOMPARE(rename.to(), "myId");

        QTest::ignoreMessage(QtWarningMsg,
                             "RoleRename: property \"to\" is intended to be "
                             "initialized once and not changed!");
        rename.setTo("myId2");
        QCOMPARE(rename.from(), "id");
        QCOMPARE(rename.to(), "myId");
    }

    void sourceModelDeletedTest()
    {
        auto sourceModel = std::make_unique<TestSourceModel>(
                    QList<QString>{"id", "name", "color"});
        RolesRenamingModel model;

        QQmlListProperty<RoleRename> mapping = model.mapping();

        RoleRename rename_1;
        rename_1.setFrom("id");
        rename_1.setTo("tokenId");
        mapping.append(&mapping, &rename_1);

        RoleRename rename_2;
        rename_2.setFrom("name");
        rename_2.setTo("tokenName");
        mapping.append(&mapping, &rename_2);

        model.setSourceModel(sourceModel.get());

        QHash<int, QByteArray> expectedRoles = {
            {0, "tokenId"}, {1, "tokenName"}, {2, "color"}
        };
        QCOMPARE(model.roleNames(), expectedRoles);
        QCOMPARE(model.rowCount(), 5);

        QCOMPARE(model.data(model.index(0, 0), 0), 42);
        QCOMPARE(model.data(model.index(0, 0), 1), 42);
        QCOMPARE(model.data(model.index(5, 0), 0), {});
        QCOMPARE(model.data(model.index(5, 0), 1), {});

        QSignalSpy destroyedSpy(sourceModel.get(), &QObject::destroyed);
        sourceModel.reset();

        QCOMPARE(destroyedSpy.size(), 1);

        QCOMPARE(model.roleNames(), {});
        QCOMPARE(model.rowCount(), 0);

        QCOMPARE(model.roleNames(), {});
        QCOMPARE(model.data(model.index(0, 0), 0), {});
        QCOMPARE(model.data(model.index(0, 0), 1), {});
        QCOMPARE(model.data(model.index(5, 0), 0), {});
        QCOMPARE(model.data(model.index(5, 0), 1), {});
    }
};

QTEST_MAIN(TestRolesRenamingModel)
#include "tst_RolesRenamingModel.moc"