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