1 files changed, 88 insertions, 0 deletions
diff --git a/tests/ResourceModel_test.cpp b/tests/ResourceModel_test.cpp
new file mode 100644
index 00000000..716bf853
--- /dev/null
+++ b/tests/ResourceModel_test.cpp
@@ -0,0 +1,88 @@
+#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"