Merge pull request #264 from xXJSONDeruloXx/chore/migrate

the big one

1 files changed, 101 insertions, 0 deletions

diff --git a/tests/test_plugin_migration.py b/tests/test_plugin_migration.py
new file mode 100644
index 0000000..fc51470
--- /dev/null
+++ b/tests/test_plugin_migration.py
@@ -0,0 +1,101 @@
+import asyncio
+import sys
+import types
+import unittest
+from unittest.mock import Mock
+
+
+class PluginMigrationTests(unittest.TestCase):
+    def _load_plugin(self):
+        decky = types.SimpleNamespace(
+            DECKY_HOME="/decky",
+            DECKY_USER_HOME="/home/deck",
+            migrate_logs=Mock(),
+            migrate_settings=Mock(),
+            migrate_runtime=Mock(),
+            logger=Mock(),
+        )
+        previous_decky = sys.modules.get("decky")
+        previous_tomllib = sys.modules.get("tomllib")
+        previous_plugin = sys.modules.pop("lsfg_vk.plugin", None)
+        sys.modules["decky"] = decky
+        sys.modules["tomllib"] = types.SimpleNamespace(loads=Mock())
+        sys.path.insert(0, "py_modules")
+        from lsfg_vk.plugin import Plugin
+        return Plugin, decky, previous_decky, previous_tomllib, previous_plugin
+
+    def _restore(self, previous_decky, previous_tomllib, previous_plugin):
+        sys.path.remove("py_modules")
+        if previous_decky is None:
+            sys.modules.pop("decky", None)
+        else:
+            sys.modules["decky"] = previous_decky
+        if previous_tomllib is None:
+            sys.modules.pop("tomllib", None)
+        else:
+            sys.modules["tomllib"] = previous_tomllib
+        if previous_plugin is None:
+            sys.modules.pop("lsfg_vk.plugin", None)
+        else:
+            sys.modules["lsfg_vk.plugin"] = previous_plugin
+
+    def test_migration_only_runs_decky_path_migrations(self):
+        Plugin, decky, previous_decky, previous_tomllib, previous_plugin = self._load_plugin()
+        try:
+            plugin = Plugin.__new__(Plugin)
+            plugin.installation_service = Mock()
+            plugin.flatpak_service = Mock()
+
+            asyncio.run(plugin._migration())
+
+            decky.migrate_logs.assert_called_once()
+            decky.migrate_settings.assert_called_once()
+            decky.migrate_runtime.assert_called_once()
+            plugin.installation_service.install.assert_not_called()
+            plugin.flatpak_service.prepare_app.assert_not_called()
+        finally:
+            self._restore(previous_decky, previous_tomllib, previous_plugin)
+
+    def test_uninstall_cleans_owned_flatpak_state_and_profiles(self):
+        Plugin, _decky, previous_decky, previous_tomllib, previous_plugin = self._load_plugin()
+        try:
+            plugin = Plugin.__new__(Plugin)
+            plugin.installation_service = Mock()
+            plugin.flatpak_service = Mock()
+            plugin.configuration_service = Mock()
+            plugin.wrapper_service = Mock()
+            plugin.flatpak_service.remove_plugin_owned_environment.return_value = {"success": True}
+            plugin.configuration_service.reset_all_flatpak_configs.return_value = {"success": True}
+            plugin.wrapper_service.neutralize.return_value = {"success": True}
+
+            asyncio.run(plugin._uninstall())
+
+            plugin.flatpak_service.remove_plugin_owned_environment.assert_called_once_with()
+            plugin.configuration_service.reset_all_flatpak_configs.assert_called_once_with()
+            plugin.wrapper_service.neutralize.assert_called_once_with()
+            plugin.wrapper_service.purge.assert_not_called()
+            plugin.installation_service.cleanup_on_uninstall.assert_called_once_with()
+        finally:
+            self._restore(previous_decky, previous_tomllib, previous_plugin)
+
+    def test_uninstall_preserves_profiles_when_flatpak_cleanup_fails(self):
+        Plugin, _decky, previous_decky, previous_tomllib, previous_plugin = self._load_plugin()
+        try:
+            plugin = Plugin.__new__(Plugin)
+            plugin.installation_service = Mock()
+            plugin.flatpak_service = Mock()
+            plugin.configuration_service = Mock()
+            plugin.wrapper_service = Mock()
+            plugin.flatpak_service.remove_plugin_owned_environment.return_value = {"success": False, "error": "changed"}
+
+            asyncio.run(plugin._uninstall())
+
+            plugin.configuration_service.reset_all_flatpak_configs.assert_not_called()
+            plugin.wrapper_service.purge.assert_not_called()
+            plugin.installation_service.cleanup_on_uninstall.assert_not_called()
+        finally:
+            self._restore(previous_decky, previous_tomllib, previous_plugin)
+
+
+if __name__ == "__main__":
+    unittest.main()