62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import multiprocessing
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from tht.jobs.locking import JobAlreadyRunningError, WorkspaceJobLock
|
|
|
|
|
|
def _hold_lock(root: str, ready, release):
|
|
with WorkspaceJobLock(Path(root), "demo", "evidence"):
|
|
ready.set()
|
|
release.wait(10)
|
|
|
|
|
|
def _crash_with_lock(root: str, ready):
|
|
lock = WorkspaceJobLock(Path(root), "demo", "evidence")
|
|
lock.acquire()
|
|
ready.set()
|
|
raise SystemExit(7)
|
|
|
|
|
|
def test_same_workspace_and_job_are_exclusive_across_processes(tmp_path):
|
|
context = multiprocessing.get_context("spawn")
|
|
ready = context.Event()
|
|
release = context.Event()
|
|
process = context.Process(target=_hold_lock, args=(str(tmp_path), ready, release))
|
|
process.start()
|
|
assert ready.wait(10)
|
|
try:
|
|
with pytest.raises(JobAlreadyRunningError):
|
|
WorkspaceJobLock(tmp_path, "demo", "evidence").acquire()
|
|
finally:
|
|
release.set()
|
|
process.join(10)
|
|
assert process.exitcode == 0
|
|
|
|
|
|
def test_evidence_and_dwh_jobs_have_distinct_locks(tmp_path):
|
|
with WorkspaceJobLock(tmp_path, "demo", "evidence"):
|
|
with WorkspaceJobLock(tmp_path, "demo", "dwh"):
|
|
pass
|
|
|
|
|
|
def test_lock_keys_cannot_escape_lock_directory(tmp_path):
|
|
with pytest.raises(ValueError, match="filesystem-safe"):
|
|
WorkspaceJobLock(tmp_path, "demo", "../evidence")
|
|
|
|
|
|
def test_lock_is_recoverable_after_process_crash_without_stale_deletion(tmp_path):
|
|
context = multiprocessing.get_context("spawn")
|
|
ready = context.Event()
|
|
process = context.Process(target=_crash_with_lock, args=(str(tmp_path), ready))
|
|
process.start()
|
|
assert ready.wait(10)
|
|
process.join(10)
|
|
assert process.exitcode == 7
|
|
|
|
lock_path = WorkspaceJobLock(tmp_path, "demo", "evidence").path
|
|
assert lock_path.exists()
|
|
with WorkspaceJobLock(tmp_path, "demo", "evidence"):
|
|
assert lock_path.exists()
|