1 files changed, 61 insertions, 0 deletions
diff --git a/libs/plugin_python/tests/runner/test_diagnose.cpp b/libs/plugin_python/tests/runner/test_diagnose.cpp
new file mode 100644
index 0000000..467c9cb
--- /dev/null
+++ b/libs/plugin_python/tests/runner/test_diagnose.cpp
@@ -0,0 +1,61 @@
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+
+#include "pythonrunner.h"
+
+#include <QCoreApplication>
+
+#include <uibase/iplugindiagnose.h>
+#include <uibase/iplugingame.h>
+
+#include "MockOrganizer.h"
+
+using namespace MOBase;
+
+using ::testing::ElementsAre;
+
+TEST(IPluginDiagnose, Simple)
+{
+    const auto plugins_folder = QString(std::getenv("PLUGIN_DIR"));
+
+    auto runner = mo2::python::createPythonRunner();
+    runner->initialize();
+
+    // load objects
+    const auto objects = runner->load(plugins_folder + "/dummy-diagnose.py");
+    EXPECT_EQ(objects.size(), 3);
+
+    // load the first IPluginDiagnose
+    {
+        IPluginDiagnose* plugin = qobject_cast<IPluginDiagnose*>(objects[0]);
+        EXPECT_NE(plugin, nullptr);
+
+        ASSERT_THAT(plugin->activeProblems(), ElementsAre(1, 2));
+        EXPECT_EQ(plugin->shortDescription(1), "short-1");
+        EXPECT_EQ(plugin->fullDescription(1), "long-1");
+        EXPECT_TRUE(plugin->hasGuidedFix(1));
+        EXPECT_EQ(plugin->shortDescription(2), "short-2");
+        EXPECT_EQ(plugin->fullDescription(2), "long-2");
+        EXPECT_FALSE(plugin->hasGuidedFix(2));
+    }
+
+    // load the second one (this is cast before IPluginGame so should be before)
+    {
+        IPluginDiagnose* plugin = qobject_cast<IPluginDiagnose*>(objects[1]);
+        EXPECT_NE(plugin, nullptr);
+
+        ASSERT_THAT(plugin->activeProblems(), ElementsAre(5, 7));
+        EXPECT_EQ(plugin->shortDescription(5), "short-5");
+        EXPECT_EQ(plugin->fullDescription(5), "long-5");
+        EXPECT_FALSE(plugin->hasGuidedFix(5));
+        EXPECT_EQ(plugin->shortDescription(7), "short-7");
+        EXPECT_EQ(plugin->fullDescription(7), "long-7");
+        EXPECT_TRUE(plugin->hasGuidedFix(7));
+    }
+
+    // load the game plugin
+    {
+        IPluginGame* plugin = qobject_cast<IPluginGame*>(objects[2]);
+        EXPECT_NE(plugin, nullptr);
+    }
+}