pollymc/tests/ResourceModel_test.cpp

#include <QAbstractItemModelTester>
#include <QTest>
#include <QTimer>

#include <Json.h>

#include <ui/pages/modplatform/ResourceModel.h>

#include "DummyResourceAPI.h"

using ResourceDownload::ResourceModel;

#define EXEC_TASK(EXEC)                                                                 \
    QEventLoop loop;                                                                    \
                                                                                        \
    connect(model, &ResourceModel::dataChanged, &loop, &QEventLoop::quit);              \
                                                                                        \
    QTimer expire_timer;                                                                \
    expire_timer.callOnTimeout(&loop, &QEventLoop::quit);                               \
    expire_timer.setSingleShot(true);                                                   \
    expire_timer.start(4000);                                                           \
                                                                                        \
    EXEC;                                                                               \
    if (model->hasActiveSearchJob())                                                    \
        loop.exec();                                                                    \
                                                                                        \
    QVERIFY2(expire_timer.isActive(), "Timer has expired. The search never finished."); \
    expire_timer.stop();                                                                \
                                                                                        \
    disconnect(model, nullptr, &loop, nullptr)

class ResourceModelTest;

class DummyResourceModel : public ResourceModel {
    Q_OBJECT

    friend class ResourceModelTest;

   public:
    DummyResourceModel() : ResourceModel(new DummyResourceAPI) {}

    [[nodiscard]] auto metaEntryBase() const -> QString override { return ""; };

    ResourceAPI::SearchArgs createSearchArguments() override { return {}; };
    ResourceAPI::VersionSearchArgs createVersionsArguments(QModelIndex&) override { return {}; };
    ResourceAPI::ProjectInfoArgs createInfoArguments(QModelIndex&) override { return {}; };

    QJsonArray documentToArray(QJsonDocument& doc) const override { return doc.object().value("hits").toArray(); }

    void loadIndexedPack(ModPlatform::IndexedPack& pack, QJsonObject& obj) override
    {
        pack.authors.append({ Json::requireString(obj, "author") });
        pack.description = Json::requireString(obj, "description");
        pack.addonId = Json::requireString(obj, "project_id");
    }
};

class ResourceModelTest : public QObject {
    Q_OBJECT
   private slots:
    void test_abstract_item_model() { [[maybe_unused]] auto tester = new QAbstractItemModelTester(new DummyResourceModel); }

    void test_search()
    {
        auto model = new DummyResourceModel;

        QVERIFY(model->m_packs.isEmpty());

        EXEC_TASK(model->search());

        QVERIFY(model->m_packs.size() == 1);
        QVERIFY(model->m_search_state == DummyResourceModel::SearchState::Finished);

        auto processed_pack = model->m_packs.at(0);
        auto search_json = DummyResourceAPI::searchRequestResult();
        auto processed_response = model->documentToArray(search_json).first().toObject();

        QVERIFY(processed_pack->addonId.toString() == Json::requireString(processed_response, "project_id"));
        QVERIFY(processed_pack->description == Json::requireString(processed_response, "description"));
        QVERIFY(processed_pack->authors.first().name == Json::requireString(processed_response, "author"));
    }
};

QTEST_GUILESS_MAIN(ResourceModelTest)

#include "ResourceModel_test.moc"

#include "moc_DummyResourceAPI.cpp"