1 files changed, 62 insertions, 0 deletions
diff --git a/tests/test_plugin_flatpak_lifecycle.py b/tests/test_plugin_flatpak_lifecycle.py
new file mode 100644
index 0000000..25f09b6
--- /dev/null
+++ b/tests/test_plugin_flatpak_lifecycle.py
@@ -0,0 +1,62 @@
+import asyncio
+import sys
+import types
+import unittest
+from unittest.mock import Mock
+
+
+sys.modules.setdefault(
+    "decky",
+    types.SimpleNamespace(
+        DECKY_HOME="/decky",
+        DECKY_USER_HOME="/home/deck",
+        logger=Mock(),
+    ),
+)
+sys.modules.setdefault("tomllib", types.SimpleNamespace(loads=Mock()))
+sys.path.insert(0, "py_modules")
+
+from lsfg_vk.plugin import Plugin
+
+
+class PluginFlatpakLifecycleTests(unittest.TestCase):
+    def test_install_and_reload_ensure_global_flatpak_support(self):
+        plugin = Plugin.__new__(Plugin)
+        plugin.installation_service = Mock()
+        plugin.installation_service.install.return_value = {"success": True}
+        plugin.installation_service.check_installation.return_value = {"installed": True}
+        plugin.flatpak_service = Mock()
+        plugin.flatpak_service.ensure_plugin_support.return_value = {"success": True}
+        plugin.wrapper_service = Mock()
+        plugin.wrapper_service.repair.return_value = {"success": True}
+
+        installed = asyncio.run(plugin.install_lsfg_vk())
+        asyncio.run(plugin._main())
+
+        self.assertTrue(installed["success"])
+        self.assertEqual(plugin.flatpak_service.ensure_plugin_support.call_count, 2)
+        plugin.wrapper_service.repair.assert_called_once_with()
+
+    def test_unload_does_not_remove_persistent_flatpak_setup(self):
+        plugin = Plugin.__new__(Plugin)
+        plugin.installation_service = Mock()
+        plugin.flatpak_service = Mock()
+
+        asyncio.run(plugin._unload())
+
+        plugin.flatpak_service.remove_plugin_owned_extensions.assert_not_called()
+
+    def test_uninstall_is_the_flatpak_cleanup_boundary(self):
+        plugin = Plugin.__new__(Plugin)
+        plugin.installation_service = Mock()
+        plugin.flatpak_service = Mock()
+        plugin.flatpak_service.remove_plugin_owned_extensions.return_value = {"success": True}
+
+        asyncio.run(plugin._uninstall())
+
+        plugin.installation_service.cleanup_on_uninstall.assert_called_once_with()
+        plugin.flatpak_service.remove_plugin_owned_extensions.assert_called_once_with()
+
+
+if __name__ == "__main__":
+    unittest.main()