Skip to content

Commit

Permalink
Accept optional lock object in Condition ctor (tulip issue python#198)
Browse files Browse the repository at this point in the history
  • Loading branch information
asvetlov committed Jul 26, 2014
2 parents 996b671 + f21fcd0 commit a7121b8
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 3 deletions.
9 changes: 6 additions & 3 deletions Lib/asyncio/locks.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,14 +255,17 @@ class Condition:
A new Lock object is created and used as the underlying lock.
"""

def __init__(self, *, loop=None):
def __init__(self, lock=None, *, loop=None):
if loop is not None:
self._loop = loop
else:
self._loop = events.get_event_loop()

# Lock as an attribute as in threading.Condition.
lock = Lock(loop=self._loop)
if lock is None:
lock = Lock(loop=self._loop)
elif lock._loop is not self._loop:
raise ValueError("loop argument must agree with lock")

self._lock = lock
# Export the lock's locked(), acquire() and release() methods.
self.locked = lock.locked
Expand Down
12 changes: 12 additions & 0 deletions Lib/test/test_asyncio/test_locks.py
Original file line number Diff line number Diff line change
Expand Up @@ -656,6 +656,18 @@ def test_context_manager_no_yield(self):

self.assertFalse(cond.locked())

def test_explicit_lock(self):
lock = asyncio.Lock(loop=self.loop)
cond = asyncio.Condition(lock, loop=self.loop)

self.assertIs(lock._loop, cond._loop)

def test_ambiguous_loops(self):
loop = self.new_test_loop()
lock = asyncio.Lock(loop=self.loop)
with self.assertRaises(ValueError):
asyncio.Condition(lock, loop=loop)


class SemaphoreTests(test_utils.TestCase):

Expand Down

0 comments on commit a7121b8

Please sign in to comment.