78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
|
"""Shared pytest fixtures for cura-te-ipsum tests."""
|
||
|
|
import os
|
||
|
|
import random
|
||
|
|
import string
|
||
|
|
import tempfile
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from curateipsum import backup as bk, fs
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def backup_dir(tmp_path):
|
||
|
|
"""Provide a temporary backup directory."""
|
||
|
|
return tmp_path / "backups"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def source_dir(tmp_path):
|
||
|
|
"""Provide a temporary source directory."""
|
||
|
|
src = tmp_path / "source"
|
||
|
|
src.mkdir()
|
||
|
|
return src
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def add_backup(backup_dir):
|
||
|
|
"""Factory fixture for creating test backups."""
|
||
|
|
backup_dir.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
def _add_backup(backup_name: str) -> fs.PseudoDirEntry:
|
||
|
|
backup = fs.PseudoDirEntry(os.path.join(str(backup_dir), backup_name))
|
||
|
|
os.mkdir(backup.path)
|
||
|
|
bk.set_backup_marker(backup)
|
||
|
|
|
||
|
|
fd, path = tempfile.mkstemp(prefix="backup_file_", dir=backup.path)
|
||
|
|
with open(fd, "w") as f:
|
||
|
|
f.write(''.join(random.choices(string.printable, k=128)))
|
||
|
|
return backup
|
||
|
|
|
||
|
|
return _add_backup
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def check_backup_not_empty():
|
||
|
|
"""Helper to verify backup is not empty."""
|
||
|
|
def _check(backup: fs.PseudoDirEntry) -> bool:
|
||
|
|
return bool(os.listdir(backup.path))
|
||
|
|
return _check
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def check_backups(backup_dir, check_backup_not_empty):
|
||
|
|
"""Helper to verify expected backups exist."""
|
||
|
|
def _check(expected_backups):
|
||
|
|
backups_list = os.listdir(str(backup_dir))
|
||
|
|
assert sorted(b.name for b in expected_backups) == sorted(backups_list)
|
||
|
|
for b in expected_backups:
|
||
|
|
assert check_backup_not_empty(b)
|
||
|
|
return _check
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def run_cleanup(backup_dir):
|
||
|
|
"""Helper to run cleanup_old_backups with default parameters."""
|
||
|
|
def _run(**kwargs):
|
||
|
|
cleanup_kwargs = {
|
||
|
|
"backups_dir": str(backup_dir),
|
||
|
|
"dry_run": False,
|
||
|
|
"keep_all": None,
|
||
|
|
"keep_daily": None,
|
||
|
|
"keep_weekly": None,
|
||
|
|
"keep_monthly": None,
|
||
|
|
"keep_yearly": None,
|
||
|
|
}
|
||
|
|
cleanup_kwargs.update(**kwargs)
|
||
|
|
bk.cleanup_old_backups(**cleanup_kwargs)
|
||
|
|
return _run
|