85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
|
"""Regression: the ZIP-update directory replace must never leave a half-deleted tree.
|
||
|
|
|
||
|
|
Issue #49145: on Windows the ZIP-update path did ``rmtree(dst); copytree(...)``.
|
||
|
|
A copy that failed partway (file locks / flaky I/O — the very conditions the ZIP
|
||
|
|
path exists to work around) left the directory deleted with nothing copied back,
|
||
|
|
which broke ``hermes --tui`` because ``ui-tui/`` had vanished.
|
||
|
|
|
||
|
|
``_atomic_replace_dir`` stages the new copy first and only swaps it in on full
|
||
|
|
success, so a mid-copy failure leaves the original directory intact.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from hermes_cli.main import _atomic_replace_dir
|
||
|
|
|
||
|
|
|
||
|
|
def test_atomic_replace_swaps_content_on_success(tmp_path: Path) -> None:
|
||
|
|
src = tmp_path / "src" / "ui-tui"
|
||
|
|
src.mkdir(parents=True)
|
||
|
|
(src / "new.txt").write_text("NEW")
|
||
|
|
|
||
|
|
dst = tmp_path / "install" / "ui-tui"
|
||
|
|
dst.mkdir(parents=True)
|
||
|
|
(dst / "old.txt").write_text("OLD")
|
||
|
|
|
||
|
|
_atomic_replace_dir(str(src), str(dst))
|
||
|
|
|
||
|
|
assert (dst / "new.txt").read_text() == "NEW"
|
||
|
|
assert not (dst / "old.txt").exists()
|
||
|
|
# No staging/backup siblings left behind.
|
||
|
|
assert not (dst.parent / "ui-tui.hermes-update-staging").exists()
|
||
|
|
assert not (dst.parent / "ui-tui.hermes-update-old").exists()
|
||
|
|
|
||
|
|
|
||
|
|
def test_atomic_replace_leaves_original_intact_when_copy_fails(
|
||
|
|
tmp_path: Path, monkeypatch
|
||
|
|
) -> None:
|
||
|
|
src = tmp_path / "src" / "ui-tui"
|
||
|
|
src.mkdir(parents=True)
|
||
|
|
(src / "a.txt").write_text("A")
|
||
|
|
|
||
|
|
dst = tmp_path / "install" / "ui-tui"
|
||
|
|
dst.mkdir(parents=True)
|
||
|
|
(dst / "keep.txt").write_text("PRECIOUS")
|
||
|
|
|
||
|
|
def boom(*_a, **_k):
|
||
|
|
raise OSError("[WinError 5] Access is denied")
|
||
|
|
|
||
|
|
monkeypatch.setattr(shutil, "copytree", boom)
|
||
|
|
|
||
|
|
with pytest.raises(OSError):
|
||
|
|
_atomic_replace_dir(str(src), str(dst))
|
||
|
|
|
||
|
|
# The whole point: the live directory survives a failed update untouched.
|
||
|
|
assert dst.is_dir()
|
||
|
|
assert (dst / "keep.txt").read_text() == "PRECIOUS"
|
||
|
|
assert not (dst.parent / "ui-tui.hermes-update-staging").exists()
|
||
|
|
|
||
|
|
|
||
|
|
def test_atomic_replace_clears_stale_staging_leftovers(tmp_path: Path) -> None:
|
||
|
|
"""A previously-interrupted update can leave staging/backup dirs behind."""
|
||
|
|
src = tmp_path / "src" / "ui-tui"
|
||
|
|
src.mkdir(parents=True)
|
||
|
|
(src / "new.txt").write_text("NEW")
|
||
|
|
|
||
|
|
dst = tmp_path / "install" / "ui-tui"
|
||
|
|
dst.mkdir(parents=True)
|
||
|
|
|
||
|
|
stale_staging = dst.parent / "ui-tui.hermes-update-staging"
|
||
|
|
stale_backup = dst.parent / "ui-tui.hermes-update-old"
|
||
|
|
stale_staging.mkdir()
|
||
|
|
stale_backup.mkdir()
|
||
|
|
(stale_staging / "junk").write_text("junk")
|
||
|
|
|
||
|
|
_atomic_replace_dir(str(src), str(dst))
|
||
|
|
|
||
|
|
assert (dst / "new.txt").read_text() == "NEW"
|
||
|
|
assert not stale_staging.exists()
|
||
|
|
assert not stale_backup.exists()
|