57 lines
2.1 KiB
Python
57 lines
2.1 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_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()
|