Интегрирован wesp в сайт

WESP_REL/tests/test_system_restart_service.py

@@ -0,0 +1,32 @@
+"""Тесты автоперезагрузки и restart."""
+
+from __future__ import annotations
+
+import unittest
+
+from app.services.system_restart_service import (
+    default_systemd_restart_cmd,
+    resolve_admin_restart_cmd,
+    schedule_system_reboot,
+)
+
+
+class SystemRestartServiceTests(unittest.TestCase):
+    def test_default_systemd_unit(self) -> None:
+        cmd = default_systemd_restart_cmd({"WESP_SYSTEMD_UNIT": "myscript"})
+        self.assertEqual(cmd, "systemctl restart myscript")
+
+    def test_resolve_admin_restart_explicit(self) -> None:
+        cfg = {
+            "WESP_ADMIN_RESTART_CMD": "systemctl restart custom",
+            "WESP_SYSTEMD_UNIT": "myscript",
+        }
+        self.assertEqual(resolve_admin_restart_cmd(cfg), "systemctl restart custom")
+
+    def test_schedule_reboot_disabled(self) -> None:
+        r = schedule_system_reboot({"WESP_AUTO_REBOOT_ENABLED": "0"})
+        self.assertFalse(r.get("scheduled"))
+
+
+if __name__ == "__main__":
+    unittest.main()