From 954b2abc47d8772211cb8ecee523900f823184e8 Mon Sep 17 00:00:00 2001 From: xXJSONDeruloXx Date: Fri, 11 Sep 2026 11:08:12 -0400 Subject: better uninstall cleanup --- tests/test_plugin_migration.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'tests/test_plugin_migration.py') diff --git a/tests/test_plugin_migration.py b/tests/test_plugin_migration.py index 7ab4621..1a7cb29 100644 --- a/tests/test_plugin_migration.py +++ b/tests/test_plugin_migration.py @@ -63,12 +63,16 @@ class PluginMigrationTests(unittest.TestCase): 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.purge.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.purge.assert_called_once_with() plugin.installation_service.cleanup_on_uninstall.assert_called_once_with() finally: self._restore(previous_decky, previous_tomllib, previous_plugin) @@ -80,12 +84,14 @@ class PluginMigrationTests(unittest.TestCase): 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.installation_service.cleanup_on_uninstall.assert_called_once_with() + 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) -- cgit v1.2.3