ccdef065df
Bash scripts were untestable and duplicated device/LUKS/mount logic; the lim/ package centralizes it behind one subprocess wrapper and a YAML image catalog (single point of truth). BREAKING CHANGE: scripts/*.sh removed. Use `lim --type <cmd>`; new types mount/umount/single-boot/raid1-boot/lock/unlock/import/export replace direct script calls. --extra is deprecated and ignored. - distributions.yml + lim/catalog.py hold the image catalog (PyYAML) - pytest suite: 102 tests with mocked subprocess (tests/unit) and a 250-line max file-length guard (tests/lint) - ruff strict (select ALL), GitHub Actions CI, Dependabot; Travis gone - Makefile: install (symlink ~/.local/bin/lim) and test targets - fixes over bash: SUDO_USER-aware chown, mmcblk/nvme partition paths, sha512 checksum support, whole-pipeline failure detection, blkid UUID fallback for pre-mounted images, conditional fstab seeding for PARTUUID/LABEL images, clean errors for missing binaries
71 lines
2.5 KiB
Python
71 lines
2.5 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",
|
|
]
|
|
]
|