mirror of
https://github.com/RichieCahill/dotfiles.git
synced 2026-04-21 06:39:09 -04:00
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
"""Tests for safe_reboot."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
|
|
from python.tools.safe_reboot import reboot
|
|
from python.zfs.dataset import Dataset
|
|
|
|
if TYPE_CHECKING:
|
|
from pytest_mock import MockerFixture
|
|
|
|
SAFE_REBOOT = "python.tools.safe_reboot"
|
|
|
|
|
|
def create_dataset(mocker: MockerFixture, name: str, exec_state: str) -> Dataset:
|
|
"""Create a mock dataset."""
|
|
dataset = mocker.MagicMock(spec=Dataset)
|
|
dataset.name = name
|
|
dataset.exec = exec_state
|
|
return dataset
|
|
|
|
|
|
def test_reboot_reboots_when_checks_pass(mocker: MockerFixture) -> None:
|
|
"""The command should reboot when all checks pass."""
|
|
dataset = create_dataset(mocker, "root_pool/root", "on")
|
|
mocker.patch(f"{SAFE_REBOOT}.get_datasets", return_value=(dataset,))
|
|
mocker.patch(f"{SAFE_REBOOT}.drive_present", return_value=True)
|
|
mock_bash = mocker.patch(f"{SAFE_REBOOT}.bash_wrapper", return_value=("", 0))
|
|
|
|
reboot("/dev/disk/root-drive")
|
|
|
|
mock_bash.assert_called_once_with("systemctl reboot")
|
|
|
|
|
|
def test_reboot_reboots_without_drive_requirement(mocker: MockerFixture) -> None:
|
|
"""The command should reboot even when no drive is provided."""
|
|
dataset = create_dataset(mocker, "root_pool/root", "on")
|
|
mocker.patch(f"{SAFE_REBOOT}.get_datasets", return_value=(dataset,))
|
|
mock_bash = mocker.patch(f"{SAFE_REBOOT}.bash_wrapper", return_value=("", 0))
|
|
|
|
reboot(None)
|
|
|
|
mock_bash.assert_called_once_with("systemctl reboot")
|
|
|
|
|
|
def test_reboot_errors_on_non_exec_dataset(mocker: MockerFixture) -> None:
|
|
"""The command should exit when a dataset lacks exec."""
|
|
dataset = create_dataset(mocker, "root_pool/root", "off")
|
|
mocker.patch(f"{SAFE_REBOOT}.get_datasets", return_value=(dataset,))
|
|
mocker.patch(f"{SAFE_REBOOT}.drive_present", return_value=True)
|
|
mocker.patch(f"{SAFE_REBOOT}.bash_wrapper", return_value=("", 0))
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
reboot("/dev/disk/root-drive")
|
|
|
|
assert excinfo.value.code == 1
|
|
|
|
|
|
def test_reboot_errors_when_driver_missing(mocker: MockerFixture) -> None:
|
|
"""The command should exit when the requested driver is absent."""
|
|
dataset = create_dataset(mocker, "root_pool/root", "on")
|
|
mocker.patch(f"{SAFE_REBOOT}.get_datasets", return_value=(dataset,))
|
|
mocker.patch(f"{SAFE_REBOOT}.drive_present", return_value=False)
|
|
mocker.patch(f"{SAFE_REBOOT}.bash_wrapper", return_value=("", 0))
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
reboot("/dev/disk/root-drive")
|
|
|
|
assert excinfo.value.code == 1
|
|
|
|
|
|
def test_reboot_errors_when_no_datasets_found(mocker: MockerFixture) -> None:
|
|
"""The command should exit when no datasets match the prefix."""
|
|
mocker.patch(f"{SAFE_REBOOT}.get_datasets", return_value=())
|
|
mocker.patch(f"{SAFE_REBOOT}.drive_present", return_value=True)
|
|
mocker.patch(f"{SAFE_REBOOT}.bash_wrapper", return_value=("", 0))
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
reboot("/dev/disk/root-drive")
|
|
|
|
assert excinfo.value.code == 1
|
|
|
|
|
|
def test_reboot_check_only_skips_reboot(mocker: MockerFixture) -> None:
|
|
"""The command should only validate when --check-only is provided."""
|
|
dataset = create_dataset(mocker, "root_pool/root", "on")
|
|
mocker.patch(f"{SAFE_REBOOT}.get_datasets", return_value=(dataset,))
|
|
mocker.patch(f"{SAFE_REBOOT}.drive_present", return_value=True)
|
|
mock_bash = mocker.patch(f"{SAFE_REBOOT}.bash_wrapper", return_value=("", 0))
|
|
|
|
reboot("/dev/disk/root-drive", check_only=True)
|
|
|
|
mock_bash.assert_not_called()
|