diff options
| -rw-r--r-- | tests/test_plugin_migration.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/tests/test_plugin_migration.py b/tests/test_plugin_migration.py index 824cbf6..7ab4621 100644 --- a/tests/test_plugin_migration.py +++ b/tests/test_plugin_migration.py @@ -17,13 +17,14 @@ class PluginMigrationTests(unittest.TestCase): ) 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 + return Plugin, decky, previous_decky, previous_tomllib, previous_plugin - def _restore(self, previous_decky, previous_tomllib): + def _restore(self, previous_decky, previous_tomllib, previous_plugin): sys.path.remove("py_modules") if previous_decky is None: sys.modules.pop("decky", None) @@ -33,9 +34,13 @@ class PluginMigrationTests(unittest.TestCase): 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 = self._load_plugin() + Plugin, decky, previous_decky, previous_tomllib, previous_plugin = self._load_plugin() try: plugin = Plugin.__new__(Plugin) plugin.installation_service = Mock() @@ -49,10 +54,10 @@ class PluginMigrationTests(unittest.TestCase): plugin.installation_service.install.assert_not_called() plugin.flatpak_service.prepare_app.assert_not_called() finally: - self._restore(previous_decky, previous_tomllib) + self._restore(previous_decky, previous_tomllib, previous_plugin) def test_uninstall_cleans_owned_flatpak_state_and_profiles(self): - Plugin, _decky, previous_decky, previous_tomllib = self._load_plugin() + Plugin, _decky, previous_decky, previous_tomllib, previous_plugin = self._load_plugin() try: plugin = Plugin.__new__(Plugin) plugin.installation_service = Mock() @@ -66,10 +71,10 @@ class PluginMigrationTests(unittest.TestCase): plugin.configuration_service.reset_all_flatpak_configs.assert_called_once_with() plugin.installation_service.cleanup_on_uninstall.assert_called_once_with() finally: - self._restore(previous_decky, previous_tomllib) + self._restore(previous_decky, previous_tomllib, previous_plugin) def test_uninstall_preserves_profiles_when_flatpak_cleanup_fails(self): - Plugin, _decky, previous_decky, previous_tomllib = self._load_plugin() + Plugin, _decky, previous_decky, previous_tomllib, previous_plugin = self._load_plugin() try: plugin = Plugin.__new__(Plugin) plugin.installation_service = Mock() @@ -82,7 +87,7 @@ class PluginMigrationTests(unittest.TestCase): plugin.configuration_service.reset_all_flatpak_configs.assert_not_called() plugin.installation_service.cleanup_on_uninstall.assert_called_once_with() finally: - self._restore(previous_decky, previous_tomllib) + self._restore(previous_decky, previous_tomllib, previous_plugin) if __name__ == "__main__": |