diff --git a/tests/test_furu_locking_contention.py b/tests/test_furu_locking_contention.py index 5ef08ec9..ad1519d4 100644 --- a/tests/test_furu_locking_contention.py +++ b/tests/test_furu_locking_contention.py @@ -7,7 +7,12 @@ import furu from furu import Spec -from furu.config import _Config, _FuruDirectories, _set_config +from furu.config import ( + _Config, + _FuruDirectories, + _FuruProvenanceConfig, + _set_config, +) from furu.execution.load_or_create import _load_or_create from furu.locking import DEFAULT_ACQUIRE_POLL_INTERVAL_S from furu.result.bundle import load_result_bundle @@ -65,7 +70,7 @@ def create(self) -> int: def _worker(data_dir: str, start_evt, out_q) -> None: - _set_config(_Config(directories=_directories_for_data(data_dir))) + _set_config(_config_for_data(data_dir)) obj = SlowProbe(key=1) out_q.put(("ready", os.getpid())) start_evt.wait() @@ -77,7 +82,7 @@ def _worker(data_dir: str, start_evt, out_q) -> None: def _batch_worker(data_dir: str, keys: list[int], start_evt, out_q) -> None: - _set_config(_Config(directories=_directories_for_data(data_dir))) + _set_config(_config_for_data(data_dir)) objs = [SlowBatchProbe(key=key) for key in keys] out_q.put(("ready", os.getpid())) start_evt.wait() @@ -94,7 +99,7 @@ def _takeover_worker( release_path: str, out_q, ) -> None: - _set_config(_Config(directories=_directories_for_data(data_dir))) + _set_config(_config_for_data(data_dir)) obj = MidRunTakeoverProbe( key=1, entered_path=entered_path, @@ -107,11 +112,14 @@ def _takeover_worker( out_q.put(("err", os.getpid(), type(exc).__name__, str(exc))) -def _directories_for_data(data_dir: str) -> _FuruDirectories: +def _config_for_data(data_dir: str) -> _Config: data_path = Path(data_dir) - return _FuruDirectories( - objects=data_path, - executions=data_path.parent / "executions", + return _Config( + directories=_FuruDirectories( + objects=data_path, + executions=data_path.parent / "executions", + ), + provenance=_FuruProvenanceConfig(snapshot=False), )