better uninstall cleanup

1 files changed, 7 insertions, 1 deletions

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)