def test_recursive(lock):
with L.UpdaterLock(1) as l1:
assert l1.lock == 0
with L.UpdaterLock(1) as l2:
assert l2.lock == 0
assert lock.exists()
assert lock.read() == '%d\n' % getpid()
assert not lock.exists()
def test_error_gone(lock, capsys):
with L.UpdaterLock():
assert lock.check(file=1)
lock.remove()
out, err = capsys.readouterr()
assert "already released" in err
def test_invalid(data, lock):
lock.write(data)
with pytest.raises(L.LockingError) as exc_info:
lock = L.UpdaterLock(1)
lock.updater_lock_acquire()
assert False
assert "Invalid PID" in str(exc_info.value)
def test_parent(lock):
ppid = '%d\n' % getpid()
with L.UpdaterLock(1):
pid = fork()
if not pid:
with L.UpdaterLock(1) as l2:
assert l2.lock == 1
assert lock.exists()
assert lock.read() == ppid
_exit(0)
(cpid, status) = wait()
assert pid == cpid
assert status == 0
assert lock.exists()
assert lock.read() == ppid
def test_stale(lock, capsys):
lock.write("%d\n" % 0x7ffffff)
with L.UpdaterLock():
assert lock.exists()
assert lock.read() == '%d\n' % getpid()
out, err = capsys.readouterr()
assert out == ""
assert "Stale PID " in err
def test_error_enter(lock, capsys):
lock.write("INVALID")
with pytest.raises(SystemExit) as exc_info:
with L.UpdaterLock(1):
assert False
assert exc_info.value.code == 5
out, err = capsys.readouterr()
assert "Invalid PID" in err
def test_concurrent(lock, mocker):
fn = str(lock)
mocker.patch("os.open").side_effect = [
EnvironmentError(EEXIST, fn),
EnvironmentError(ENOENT, fn),
open(fn, O_CREAT | O_WRONLY, 0o644),
]
with L.UpdaterLock(1):
assert lock.exists()
def test_timeout(mocker, lock):
mocker.patch("os.getppid").return_value = -1
lock.write("1\n")
with pytest.raises(L.LockingError) as exc_info:
lock = L.UpdaterLock(1)
lock.updater_lock_acquire()
assert False
assert "Another updater process 1 is currently running according to " in str(exc_info.value)
def test_nested(lock, mocker):
ppid = getpid() ^ 1
mocker.patch("os.getppid").return_value = ppid
lock.write("%d\n" % ppid)
ul = L.UpdaterLock()
with ul:
assert ul.lock > 0
assert lock.exists()
def test_context(lock):
with L.UpdaterLock():
assert lock.exists()
assert lock.read() == '%d\n' % getpid()
def test_empty(lock):
lock.write("")
with L.UpdaterLock():
assert lock.exists()
assert lock.read() == '%d\n' % getpid()
def test_error_taken(lock):
with pytest.raises(EnvironmentError):
with L.UpdaterLock():
assert lock.check(file=1)
lock.remove()
lock.mkdir()
def test_error_non_file(lock):
lock.mkdir()
with pytest.raises(EnvironmentError):
L.UpdaterLock().updater_lock_acquire()
def test_error_sub_dir(tmpdir, monkeypatch):
lock = tmpdir / "sub" / "lock"
monkeypatch.setattr(L, "FN_LOCK_UP", str(lock))
with pytest.raises(EnvironmentError):
L.UpdaterLock().updater_lock_acquire()