ysj7705andClaude Opus 5.5
bc45f66da5
Decky 3.3 often SIGKILLs plugins 5s after SIGTERM because its socket listener spins on EOF and starves the event loop (seen for several plugins). So: - _unload/_uninstall run synchronously (an awaited worker thread could never resume), and the daemon stop no longer waits. - A transient systemd path unit watches the plugins folder; if this plugin's folder stays gone for 20s it stops the daemon and deletes /opt/mudfish and the plugin settings, then removes itself. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
120 lines
5.2 KiB
Python
120 lines
5.2 KiB
Python
import os
|
|
import tempfile
|
|
import types
|
|
import unittest
|
|
|
|
from mudfish_decky.system import api_base, ensure_daemon, find_mudfish
|
|
|
|
|
|
def make_install(root, version):
|
|
os.makedirs(os.path.join(root, version, "bin"))
|
|
os.makedirs(os.path.join(root, version, "var"))
|
|
open(os.path.join(root, version, "bin", "mudrun-headless"), "w").close()
|
|
|
|
|
|
class SystemTest(unittest.TestCase):
|
|
def test_find_mudfish_picks_highest_version(self):
|
|
with tempfile.TemporaryDirectory() as root:
|
|
for v in ["5.11.0", "6.6.0", "6.10.1"]:
|
|
make_install(root, v)
|
|
os.makedirs(os.path.join(root, "6.99.0")) # no binary: ignored
|
|
self.assertEqual(find_mudfish(root),
|
|
(os.path.join(root, "6.10.1", "bin", "mudrun-headless"), os.path.join(root, "6.10.1", "var")))
|
|
|
|
def test_find_mudfish_none_when_missing(self):
|
|
with tempfile.TemporaryDirectory() as root:
|
|
self.assertIsNone(find_mudfish(os.path.join(root, "nope")))
|
|
|
|
def test_api_base_reads_ipport(self):
|
|
with tempfile.TemporaryDirectory() as var:
|
|
with open(os.path.join(var, ".ipport"), "w") as f:
|
|
f.write("127.0.0.1:9123")
|
|
self.assertEqual(api_base(var), "http://127.0.0.1:9123")
|
|
|
|
def test_api_base_default(self):
|
|
with tempfile.TemporaryDirectory() as var:
|
|
self.assertEqual(api_base(var), "http://127.0.0.1:8282")
|
|
|
|
def test_version_of_binary_path(self):
|
|
from mudfish_decky.system import version_of
|
|
self.assertEqual(version_of("/opt/mudfish/6.6.0/bin/mudrun-headless"), "6.6.0")
|
|
|
|
def test_stop_daemon_and_remove_nm_connection_commands(self):
|
|
from mudfish_decky.system import remove_nm_connection, stop_daemon
|
|
calls = []
|
|
run = lambda cmd, **kw: calls.append(cmd) # noqa: E731
|
|
stop_daemon(run=run)
|
|
remove_nm_connection(run=run)
|
|
self.assertEqual(calls, [["systemctl", "stop", "mudfish-decky"],
|
|
["nmcli", "connection", "delete", "mudfish-tap"]])
|
|
|
|
def test_cleanup_watcher_created_when_missing(self):
|
|
from mudfish_decky.system import CLEANUP_UNIT, ensure_cleanup_watcher
|
|
calls = []
|
|
|
|
def run(cmd, **kw):
|
|
calls.append(cmd)
|
|
return types.SimpleNamespace(returncode=3 if cmd[:2] == ["systemctl", "is-active"] else 0)
|
|
|
|
ensure_cleanup_watcher("/home/deck/homebrew/plugins/mudfish-decky",
|
|
"/home/deck/homebrew/settings/mudfish-decky", run=run)
|
|
self.assertEqual(calls[0], ["systemctl", "is-active", "--quiet", f"{CLEANUP_UNIT}.path"])
|
|
cmd = calls[1]
|
|
self.assertEqual(cmd[:3], ["systemd-run", f"--unit={CLEANUP_UNIT}",
|
|
"--path-property=PathChanged=/home/deck/homebrew/plugins"])
|
|
script = cmd[-1]
|
|
self.assertIn("sleep 20", script)
|
|
self.assertIn('[ -e "/home/deck/homebrew/plugins/mudfish-decky/plugin.json" ] && exit 0', script)
|
|
self.assertIn("rm -rf /opt/mudfish", script)
|
|
self.assertIn('rm -rf "/home/deck/homebrew/settings/mudfish-decky"', script)
|
|
self.assertIn(f"systemctl stop {CLEANUP_UNIT}.path", script)
|
|
# the plugin-dir check must come before anything destructive
|
|
self.assertLess(script.index("plugin.json"), script.index("rm -rf"))
|
|
|
|
def test_cleanup_watcher_not_recreated_when_active(self):
|
|
from mudfish_decky.system import ensure_cleanup_watcher
|
|
calls = []
|
|
|
|
def run(cmd, **kw):
|
|
calls.append(cmd)
|
|
return types.SimpleNamespace(returncode=0)
|
|
|
|
ensure_cleanup_watcher("/p/mudfish-decky", "/s/mudfish-decky", run=run)
|
|
self.assertEqual(len(calls), 1)
|
|
|
|
def test_lan_ip_from_route_get(self):
|
|
from mudfish_decky.system import lan_ip
|
|
out = "1.1.1.1 via 192.168.1.1 dev wlan0 src 192.168.1.50 uid 0 \n cache \n"
|
|
self.assertEqual(lan_ip(run=lambda cmd, **kw: types.SimpleNamespace(stdout=out, returncode=0)),
|
|
"192.168.1.50")
|
|
|
|
def test_lan_ip_none_without_route(self):
|
|
from mudfish_decky.system import lan_ip
|
|
self.assertIsNone(lan_ip(run=lambda cmd, **kw: types.SimpleNamespace(stdout="", returncode=2)))
|
|
|
|
def test_sleep_offset_is_boottime_minus_monotonic(self):
|
|
from mudfish_decky.system import sleep_offset
|
|
clocks = {"boot": 500.0, "mono": 420.0}
|
|
self.assertEqual(sleep_offset(lambda c: clocks["boot"] if c == "boot" else clocks["mono"],
|
|
boot="boot", mono="mono"), 80.0)
|
|
|
|
def test_ensure_daemon_noop_when_up(self):
|
|
calls = []
|
|
ensure_daemon("/bin/x", lambda: True, run=lambda cmd, **kw: calls.append(cmd))
|
|
self.assertEqual(calls, [])
|
|
|
|
def test_ensure_daemon_starts_transient_unit(self):
|
|
calls, ups = [], iter([False, False, True])
|
|
|
|
def run(cmd, **kw):
|
|
calls.append(cmd)
|
|
return types.SimpleNamespace(returncode=0)
|
|
|
|
ensure_daemon("/opt/mudfish/6.6.0/bin/mudrun-headless", lambda: next(ups), run=run, sleep=lambda s: None)
|
|
self.assertEqual(calls, [["systemd-run", "--unit=mudfish-decky", "--collect",
|
|
"/opt/mudfish/6.6.0/bin/mudrun-headless", "-B"]])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|