def test_atomicdir__fails_if_path_is_file(tmp_path: Path):
already_exists_file = tmp_path / "test"
already_exists_file.write_text("")
with pytest.raises(FileExistsError):
with sh.atomicdir(already_exists_file):
pass
def test_atomicdir__does_not_overwrite_when_disabled(tmp_path: Path):
dir = Dir(tmp_path / "test", File("1"), File("2"), File("3"))
dir.mkdir()
with pytest.raises(FileExistsError):
with sh.atomicdir(dir.path, overwrite=False):
pass
def test_atomicdir(tmp_path: Path, opts: t.Dict[str, t.Any]):
dir_path = tmp_path / "test"
files = [
File("1.txt", text="1"),
File("2.txt", text="2"),
File("3.txt", text="3"),
File("a/a.txt", text="a"),
File("b/b.txt", text="b"),
File("c/c.txt", text="c"),
]
with sh.atomicdir(dir_path, **opts) as stage_path:
assert stage_path.exists()
assert not dir_path.exists()
stage_dir = Dir(stage_path, *files)
stage_dir.mkdir()
assert not dir_path.exists()
assert dir_path.exists()
created_files = stage_dir.repath(dir_path).files
for file in created_files:
assert file.path.exists()
assert file.path.read_text() == file.text
def test_atomicdir__skips_sync_when_disabled(tmp_path: Path):
dir = tmp_path / "test"
with patch_os_fsync() as mocked_os_fsync:
with sh.atomicdir(dir, skip_sync=True):
pass
assert not mocked_os_fsync.called
def test_atomicdir__syncs_dir(tmp_path: Path):
dir = tmp_path / "test"
with patch_os_fsync() as mocked_os_fsync:
with sh.atomicdir(dir):
pass
assert mocked_os_fsync.called
assert mocked_os_fsync.call_count == 2
def test_atomicdir__overwrites_when_enabled(tmp_path: Path):
dir = Dir(tmp_path / "test", File("1"), File("2"), File("3"))
dir.mkdir()
assert list(dir.path.iterdir())
with sh.atomicdir(dir.path):
pass
assert not list(dir.path.iterdir())