b68daa96ef
Bring 14 files that predated the ruff-format run into line with the configured formatter (line-length 100); provably formatting-only (ruff format of HEAD == working tree, and ruff format is semantics-preserving). Also refresh two lim.image.tor._KEYGEN_SCRIPT doc references in tor_harness.py to lim.image.initramfs.keygen after the module moved. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
import pytest
|
|
|
|
from lim import device
|
|
from lim.device import Device
|
|
from lim.storage import raid1, single_drive
|
|
from lim.storage.common import StorageTarget
|
|
|
|
|
|
@pytest.fixture
|
|
def block_devices(monkeypatch):
|
|
monkeypatch.setattr(device, "is_block_device", lambda path: True)
|
|
monkeypatch.setattr(Device, "optimal_blocksize", property(lambda self: "4K"))
|
|
|
|
|
|
def test_storage_target_derives_all_paths():
|
|
target = StorageTarget(Device("sdb"))
|
|
assert target.mapper_name == "encrypteddrive-sdb"
|
|
assert target.mapper_path == "/dev/mapper/encrypteddrive-sdb"
|
|
assert target.mount_path == "/media/encrypteddrive-sdb"
|
|
assert target.partition_path == "/dev/sdb1"
|
|
|
|
|
|
def test_single_drive_setup_command_sequence(fake_runner, answers, block_devices, monkeypatch):
|
|
monkeypatch.setattr("lim.system.real_user", lambda: "kevin")
|
|
answers("sdb", "N") # device name, skip overwrite
|
|
|
|
single_drive.setup()
|
|
|
|
fdisk_calls = [
|
|
(cmd, input_text)
|
|
for kind, cmd, input_text in fake_runner.calls
|
|
if kind == "run" and cmd[0] == "fdisk"
|
|
]
|
|
assert [input_text for _, input_text in fdisk_calls] == [
|
|
single_drive.CREATE_GPT_TABLE_INPUT,
|
|
single_drive.CREATE_PARTITION_INPUT,
|
|
]
|
|
assert len(fake_runner.find("cryptsetup", "-y", "luksFormat", "/dev/sdb1")) == 1
|
|
assert len(fake_runner.find("cryptsetup", "luksOpen", "encrypteddrive-sdb")) == 1
|
|
assert len(fake_runner.find("mkfs.btrfs", "/dev/mapper/encrypteddrive-sdb")) == 1
|
|
assert len(fake_runner.find("mount", "/media/encrypteddrive-sdb")) == 1
|
|
assert len(fake_runner.find("chown", "kevin:kevin")) == 1
|
|
|
|
|
|
def test_single_drive_umount(fake_runner, answers, block_devices):
|
|
answers("sdb")
|
|
single_drive.umount()
|
|
assert len(fake_runner.find("umount", "/dev/mapper/encrypteddrive-sdb")) == 1
|
|
assert len(fake_runner.find("cryptsetup", "luksClose", "encrypteddrive-sdb")) == 1
|
|
|
|
|
|
def test_raid1_setup_uses_both_whole_devices(fake_runner, answers, block_devices):
|
|
answers("sdb", "sdc")
|
|
raid1.setup()
|
|
assert len(fake_runner.find("cryptsetup", "luksFormat", "/dev/sdb")) == 1
|
|
assert len(fake_runner.find("cryptsetup", "luksFormat", "/dev/sdc")) == 1
|
|
mkfs = fake_runner.find("mkfs.btrfs", "-m raid1", "-d raid1")
|
|
assert mkfs == [
|
|
[
|
|
"mkfs.btrfs",
|
|
"-m",
|
|
"raid1",
|
|
"-d",
|
|
"raid1",
|
|
"/dev/mapper/encrypteddrive-sdb",
|
|
"/dev/mapper/encrypteddrive-sdc",
|
|
]
|
|
]
|