33 lines
1000 B
Python
33 lines
1000 B
Python
"""Тесты автоперезагрузки и 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()
|