Merge pull request #7942 from ThomasWaldmann/lockroster-remove-fix-1.1

LockRoster.modify: no KeyError if element was already gone, fixes #7937
This commit is contained in:
TW 2023-11-18 18:22:52 +01:00 committed by GitHub
commit dd7111410f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 3 deletions

View File

@ -7,7 +7,7 @@ from . import platform
from .helpers import Error, ErrorWithTraceback
from .logger import create_logger
ADD, REMOVE = 'add', 'remove'
ADD, REMOVE, REMOVE2 = 'add', 'remove', 'remove2'
SHARED, EXCLUSIVE = 'shared', 'exclusive'
logger = create_logger(__name__)
@ -285,6 +285,11 @@ class LockRoster:
if op == ADD:
elements.add(self.id)
elif op == REMOVE:
# note: we ignore it if the element is already not present anymore.
# this has been frequently seen in teardowns involving Repository.__del__ and Repository.__exit__.
elements.discard(self.id)
elif op == REMOVE2:
# needed for callers that do not want to ignore.
elements.remove(self.id)
else:
raise ValueError('Unknown LockRoster op %r' % op)
@ -299,7 +304,7 @@ class LockRoster:
killing, self.kill_stale_locks = self.kill_stale_locks, False
try:
try:
self.modify(key, REMOVE)
self.modify(key, REMOVE2)
except KeyError:
# entry was not there, so no need to add a new one, but still update our id
self.id = new_id

View File

@ -162,7 +162,7 @@ class TestLock:
assert roster.get(SHARED) == {our_id}
assert roster.get(EXCLUSIVE) == set()
assert roster.get(SHARED) == set()
with pytest.raises(KeyError):
with pytest.raises(NotLocked):
dead_lock.release()
with Lock(lockpath, id=cant_know_if_dead_id, exclusive=True):