mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Peter Rowlands (변기호)
833e82a7ea
* drop python 3.5 support * dvc: remove fspath() use inside os.path calls * dvc: replace dvc.compat.fspath with os.fspath * tests: replace dvc.compat.fspath with os.fspath * utils: relpath() should use fspath/str * utils: os.readlink doesn't support path-like obj on Windows until 3.8
96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from dvc.lock import LockError
|
|
from dvc.path_info import PathInfo
|
|
from dvc.rwlock import (
|
|
RWLockFileCorruptedError,
|
|
RWLockFileFormatError,
|
|
_edit_rwlock,
|
|
rwlock,
|
|
)
|
|
|
|
|
|
def test_rwlock(tmp_path):
|
|
path = os.fspath(tmp_path)
|
|
foo = PathInfo("foo")
|
|
|
|
with rwlock(path, "cmd1", [foo], []):
|
|
with pytest.raises(LockError):
|
|
with rwlock(path, "cmd2", [], [foo]):
|
|
pass
|
|
|
|
with rwlock(path, "cmd1", [], [foo]):
|
|
with pytest.raises(LockError):
|
|
with rwlock(path, "cmd2", [foo], []):
|
|
pass
|
|
|
|
with rwlock(path, "cmd1", [], [foo]):
|
|
with pytest.raises(LockError):
|
|
with rwlock(path, "cmd2", [], [foo]):
|
|
pass
|
|
|
|
|
|
def test_rwlock_reentrant(tmp_path):
|
|
path = os.fspath(tmp_path)
|
|
foo = PathInfo("foo")
|
|
|
|
with rwlock(path, "cmd1", [], [foo]):
|
|
with rwlock(path, "cmd1", [], [foo]):
|
|
pass
|
|
with _edit_rwlock(path) as lock:
|
|
assert lock == {
|
|
"read": {},
|
|
"write": {"foo": {"cmd": "cmd1", "pid": os.getpid()}},
|
|
}
|
|
|
|
with rwlock(path, "cmd", [foo], []):
|
|
with rwlock(path, "cmd", [foo], []):
|
|
pass
|
|
with _edit_rwlock(path) as lock:
|
|
assert lock == {
|
|
"read": {"foo": [{"cmd": "cmd", "pid": os.getpid()}]},
|
|
"write": {},
|
|
}
|
|
|
|
|
|
def test_rwlock_subdirs(tmp_path):
|
|
path = os.fspath(tmp_path)
|
|
foo = PathInfo("foo")
|
|
subfoo = PathInfo("foo/subfoo")
|
|
|
|
with rwlock(path, "cmd1", [foo], []):
|
|
with pytest.raises(LockError, match=r"subfoo(.|\n)*cmd1"):
|
|
with rwlock(path, "cmd2", [], [subfoo]):
|
|
pass
|
|
|
|
with rwlock(path, "cmd1", [], [subfoo]):
|
|
with pytest.raises(LockError, match=r"'foo'(.|\n)*cmd1"):
|
|
with rwlock(path, "cmd2", [foo], []):
|
|
pass
|
|
|
|
with rwlock(path, "cmd1", [], [subfoo]):
|
|
with pytest.raises(LockError):
|
|
with rwlock(path, "cmd2", [], [foo]):
|
|
pass
|
|
|
|
with rwlock(path, "cmd1", [subfoo], []):
|
|
with rwlock(path, "cmd2", [foo], []):
|
|
pass
|
|
|
|
|
|
def test_broken_rwlock(tmp_path):
|
|
dir_path = os.fspath(tmp_path)
|
|
path = tmp_path / "rwlock"
|
|
|
|
path.write_text('{"broken": "format"}', encoding="utf-8")
|
|
with pytest.raises(RWLockFileFormatError):
|
|
with _edit_rwlock(dir_path):
|
|
pass
|
|
|
|
path.write_text("{broken json", encoding="utf-8")
|
|
with pytest.raises(RWLockFileCorruptedError):
|
|
with _edit_rwlock(dir_path):
|
|
pass
|