diff options
-rw-r--r-- | dogpile/core/legacy.py | 4 | ||||
-rw-r--r-- | tests/core/test_dogpile.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/dogpile/core/legacy.py b/dogpile/core/legacy.py index 5f8dd22..06fbea5 100644 --- a/dogpile/core/legacy.py +++ b/dogpile/core/legacy.py @@ -144,9 +144,9 @@ class SyncReaderDogpile(Dogpile): @contextlib.contextmanager def acquire(self, *arg, **kw): - with super(SyncReaderDogpile, self).acquire(*arg, **kw): + with super(SyncReaderDogpile, self).acquire(*arg, **kw) as value: self.readwritelock.acquire_read_lock() try: - yield + yield value finally: self.readwritelock.release_read_lock() diff --git a/tests/core/test_dogpile.py b/tests/core/test_dogpile.py index 9dace4f..3012f8f 100644 --- a/tests/core/test_dogpile.py +++ b/tests/core/test_dogpile.py @@ -248,7 +248,7 @@ class ConcurrencyTest(TestCase): failures[0] += 1 def check_value(value, waited): - assert value + assert value, repr(value) # time since the current resource was # created |