1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
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.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)
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()
|
