Python code coverage for Lib/test/test_asyncio/test_locks.py

#	count	content
1	n/a	"""Tests for lock.py"""
2	n/a
3	n/a	import unittest
4	n/a	from unittest import mock
5	n/a	import re
6	n/a
7	n/a	import asyncio
8	n/a	from asyncio import test_utils
9	n/a
10	n/a	STR_RGX_REPR = (
11	n/a	r'^<(?P<class>.?) object at (?P<address>.?)'
12	n/a	r'\[(?P<extras>'
13	n/a	r'(set\|unset\|locked\|unlocked)(,value:\d)?(,waiters:\d+)?'
14	n/a	r')\]>\Z'
15	n/a	)
16	n/a	RGX_REPR = re.compile(STR_RGX_REPR)
17	n/a
18	n/a
19	n/a	class LockTests(test_utils.TestCase):
20	n/a
21	n/a	def setUp(self):
22	n/a	super().setUp()
23	n/a	self.loop = self.new_test_loop()
24	n/a
25	n/a	def test_ctor_loop(self):
26	n/a	loop = mock.Mock()
27	n/a	lock = asyncio.Lock(loop=loop)
28	n/a	self.assertIs(lock._loop, loop)
29	n/a
30	n/a	lock = asyncio.Lock(loop=self.loop)
31	n/a	self.assertIs(lock._loop, self.loop)
32	n/a
33	n/a	def test_ctor_noloop(self):
34	n/a	asyncio.set_event_loop(self.loop)
35	n/a	lock = asyncio.Lock()
36	n/a	self.assertIs(lock._loop, self.loop)
37	n/a
38	n/a	def test_repr(self):
39	n/a	lock = asyncio.Lock(loop=self.loop)
40	n/a	self.assertTrue(repr(lock).endswith('[unlocked]>'))
41	n/a	self.assertTrue(RGX_REPR.match(repr(lock)))
42	n/a
43	n/a	@asyncio.coroutine
44	n/a	def acquire_lock():
45	n/a	yield from lock
46	n/a
47	n/a	self.loop.run_until_complete(acquire_lock())
48	n/a	self.assertTrue(repr(lock).endswith('[locked]>'))
49	n/a	self.assertTrue(RGX_REPR.match(repr(lock)))
50	n/a
51	n/a	def test_lock(self):
52	n/a	lock = asyncio.Lock(loop=self.loop)
53	n/a
54	n/a	@asyncio.coroutine
55	n/a	def acquire_lock():
56	n/a	return (yield from lock)
57	n/a
58	n/a	res = self.loop.run_until_complete(acquire_lock())
59	n/a
60	n/a	self.assertTrue(res)
61	n/a	self.assertTrue(lock.locked())
62	n/a
63	n/a	lock.release()
64	n/a	self.assertFalse(lock.locked())
65	n/a
66	n/a	def test_acquire(self):
67	n/a	lock = asyncio.Lock(loop=self.loop)
68	n/a	result = []
69	n/a
70	n/a	self.assertTrue(self.loop.run_until_complete(lock.acquire()))
71	n/a
72	n/a	@asyncio.coroutine
73	n/a	def c1(result):
74	n/a	if (yield from lock.acquire()):
75	n/a	result.append(1)
76	n/a	return True
77	n/a
78	n/a	@asyncio.coroutine
79	n/a	def c2(result):
80	n/a	if (yield from lock.acquire()):
81	n/a	result.append(2)
82	n/a	return True
83	n/a
84	n/a	@asyncio.coroutine
85	n/a	def c3(result):
86	n/a	if (yield from lock.acquire()):
87	n/a	result.append(3)
88	n/a	return True
89	n/a
90	n/a	t1 = asyncio.Task(c1(result), loop=self.loop)
91	n/a	t2 = asyncio.Task(c2(result), loop=self.loop)
92	n/a
93	n/a	test_utils.run_briefly(self.loop)
94	n/a	self.assertEqual([], result)
95	n/a
96	n/a	lock.release()
97	n/a	test_utils.run_briefly(self.loop)
98	n/a	self.assertEqual([1], result)
99	n/a
100	n/a	test_utils.run_briefly(self.loop)
101	n/a	self.assertEqual([1], result)
102	n/a
103	n/a	t3 = asyncio.Task(c3(result), loop=self.loop)
104	n/a
105	n/a	lock.release()
106	n/a	test_utils.run_briefly(self.loop)
107	n/a	self.assertEqual([1, 2], result)
108	n/a
109	n/a	lock.release()
110	n/a	test_utils.run_briefly(self.loop)
111	n/a	self.assertEqual([1, 2, 3], result)
112	n/a
113	n/a	self.assertTrue(t1.done())
114	n/a	self.assertTrue(t1.result())
115	n/a	self.assertTrue(t2.done())
116	n/a	self.assertTrue(t2.result())
117	n/a	self.assertTrue(t3.done())
118	n/a	self.assertTrue(t3.result())
119	n/a
120	n/a	def test_acquire_cancel(self):
121	n/a	lock = asyncio.Lock(loop=self.loop)
122	n/a	self.assertTrue(self.loop.run_until_complete(lock.acquire()))
123	n/a
124	n/a	task = asyncio.Task(lock.acquire(), loop=self.loop)
125	n/a	self.loop.call_soon(task.cancel)
126	n/a	self.assertRaises(
127	n/a	asyncio.CancelledError,
128	n/a	self.loop.run_until_complete, task)
129	n/a	self.assertFalse(lock._waiters)
130	n/a
131	n/a	def test_cancel_race(self):
132	n/a	# Several tasks:
133	n/a	# - A acquires the lock
134	n/a	# - B is blocked in acquire()
135	n/a	# - C is blocked in acquire()
136	n/a	#
137	n/a	# Now, concurrently:
138	n/a	# - B is cancelled
139	n/a	# - A releases the lock
140	n/a	#
141	n/a	# If B's waiter is marked cancelled but not yet removed from
142	n/a	# _waiters, A's release() call will crash when trying to set
143	n/a	# B's waiter; instead, it should move on to C's waiter.
144	n/a
145	n/a	# Setup: A has the lock, b and c are waiting.
146	n/a	lock = asyncio.Lock(loop=self.loop)
147	n/a
148	n/a	@asyncio.coroutine
149	n/a	def lockit(name, blocker):
150	n/a	yield from lock.acquire()
151	n/a	try:
152	n/a	if blocker is not None:
153	n/a	yield from blocker
154	n/a	finally:
155	n/a	lock.release()
156	n/a
157	n/a	fa = asyncio.Future(loop=self.loop)
158	n/a	ta = asyncio.Task(lockit('A', fa), loop=self.loop)
159	n/a	test_utils.run_briefly(self.loop)
160	n/a	self.assertTrue(lock.locked())
161	n/a	tb = asyncio.Task(lockit('B', None), loop=self.loop)
162	n/a	test_utils.run_briefly(self.loop)
163	n/a	self.assertEqual(len(lock._waiters), 1)
164	n/a	tc = asyncio.Task(lockit('C', None), loop=self.loop)
165	n/a	test_utils.run_briefly(self.loop)
166	n/a	self.assertEqual(len(lock._waiters), 2)
167	n/a
168	n/a	# Create the race and check.
169	n/a	# Without the fix this failed at the last assert.
170	n/a	fa.set_result(None)
171	n/a	tb.cancel()
172	n/a	self.assertTrue(lock._waiters[0].cancelled())
173	n/a	test_utils.run_briefly(self.loop)
174	n/a	self.assertFalse(lock.locked())
175	n/a	self.assertTrue(ta.done())
176	n/a	self.assertTrue(tb.cancelled())
177	n/a	self.assertTrue(tc.done())
178	n/a
179	n/a	def test_release_not_acquired(self):
180	n/a	lock = asyncio.Lock(loop=self.loop)
181	n/a
182	n/a	self.assertRaises(RuntimeError, lock.release)
183	n/a
184	n/a	def test_release_no_waiters(self):
185	n/a	lock = asyncio.Lock(loop=self.loop)
186	n/a	self.loop.run_until_complete(lock.acquire())
187	n/a	self.assertTrue(lock.locked())
188	n/a
189	n/a	lock.release()
190	n/a	self.assertFalse(lock.locked())
191	n/a
192	n/a	def test_context_manager(self):
193	n/a	lock = asyncio.Lock(loop=self.loop)
194	n/a
195	n/a	@asyncio.coroutine
196	n/a	def acquire_lock():
197	n/a	return (yield from lock)
198	n/a
199	n/a	with self.loop.run_until_complete(acquire_lock()):
200	n/a	self.assertTrue(lock.locked())
201	n/a
202	n/a	self.assertFalse(lock.locked())
203	n/a
204	n/a	def test_context_manager_cant_reuse(self):
205	n/a	lock = asyncio.Lock(loop=self.loop)
206	n/a
207	n/a	@asyncio.coroutine
208	n/a	def acquire_lock():
209	n/a	return (yield from lock)
210	n/a
211	n/a	# This spells "yield from lock" outside a generator.
212	n/a	cm = self.loop.run_until_complete(acquire_lock())
213	n/a	with cm:
214	n/a	self.assertTrue(lock.locked())
215	n/a
216	n/a	self.assertFalse(lock.locked())
217	n/a
218	n/a	with self.assertRaises(AttributeError):
219	n/a	with cm:
220	n/a	pass
221	n/a
222	n/a	def test_context_manager_no_yield(self):
223	n/a	lock = asyncio.Lock(loop=self.loop)
224	n/a
225	n/a	try:
226	n/a	with lock:
227	n/a	self.fail('RuntimeError is not raised in with expression')
228	n/a	except RuntimeError as err:
229	n/a	self.assertEqual(
230	n/a	str(err),
231	n/a	'"yield from" should be used as context manager expression')
232	n/a
233	n/a	self.assertFalse(lock.locked())
234	n/a
235	n/a
236	n/a	class EventTests(test_utils.TestCase):
237	n/a
238	n/a	def setUp(self):
239	n/a	super().setUp()
240	n/a	self.loop = self.new_test_loop()
241	n/a
242	n/a	def test_ctor_loop(self):
243	n/a	loop = mock.Mock()
244	n/a	ev = asyncio.Event(loop=loop)
245	n/a	self.assertIs(ev._loop, loop)
246	n/a
247	n/a	ev = asyncio.Event(loop=self.loop)
248	n/a	self.assertIs(ev._loop, self.loop)
249	n/a
250	n/a	def test_ctor_noloop(self):
251	n/a	asyncio.set_event_loop(self.loop)
252	n/a	ev = asyncio.Event()
253	n/a	self.assertIs(ev._loop, self.loop)
254	n/a
255	n/a	def test_repr(self):
256	n/a	ev = asyncio.Event(loop=self.loop)
257	n/a	self.assertTrue(repr(ev).endswith('[unset]>'))
258	n/a	match = RGX_REPR.match(repr(ev))
259	n/a	self.assertEqual(match.group('extras'), 'unset')
260	n/a
261	n/a	ev.set()
262	n/a	self.assertTrue(repr(ev).endswith('[set]>'))
263	n/a	self.assertTrue(RGX_REPR.match(repr(ev)))
264	n/a
265	n/a	ev._waiters.append(mock.Mock())
266	n/a	self.assertTrue('waiters:1' in repr(ev))
267	n/a	self.assertTrue(RGX_REPR.match(repr(ev)))
268	n/a
269	n/a	def test_wait(self):
270	n/a	ev = asyncio.Event(loop=self.loop)
271	n/a	self.assertFalse(ev.is_set())
272	n/a
273	n/a	result = []
274	n/a
275	n/a	@asyncio.coroutine
276	n/a	def c1(result):
277	n/a	if (yield from ev.wait()):
278	n/a	result.append(1)
279	n/a
280	n/a	@asyncio.coroutine
281	n/a	def c2(result):
282	n/a	if (yield from ev.wait()):
283	n/a	result.append(2)
284	n/a
285	n/a	@asyncio.coroutine
286	n/a	def c3(result):
287	n/a	if (yield from ev.wait()):
288	n/a	result.append(3)
289	n/a
290	n/a	t1 = asyncio.Task(c1(result), loop=self.loop)
291	n/a	t2 = asyncio.Task(c2(result), loop=self.loop)
292	n/a
293	n/a	test_utils.run_briefly(self.loop)
294	n/a	self.assertEqual([], result)
295	n/a
296	n/a	t3 = asyncio.Task(c3(result), loop=self.loop)
297	n/a
298	n/a	ev.set()
299	n/a	test_utils.run_briefly(self.loop)
300	n/a	self.assertEqual([3, 1, 2], result)
301	n/a
302	n/a	self.assertTrue(t1.done())
303	n/a	self.assertIsNone(t1.result())
304	n/a	self.assertTrue(t2.done())
305	n/a	self.assertIsNone(t2.result())
306	n/a	self.assertTrue(t3.done())
307	n/a	self.assertIsNone(t3.result())
308	n/a
309	n/a	def test_wait_on_set(self):
310	n/a	ev = asyncio.Event(loop=self.loop)
311	n/a	ev.set()
312	n/a
313	n/a	res = self.loop.run_until_complete(ev.wait())
314	n/a	self.assertTrue(res)
315	n/a
316	n/a	def test_wait_cancel(self):
317	n/a	ev = asyncio.Event(loop=self.loop)
318	n/a
319	n/a	wait = asyncio.Task(ev.wait(), loop=self.loop)
320	n/a	self.loop.call_soon(wait.cancel)
321	n/a	self.assertRaises(
322	n/a	asyncio.CancelledError,
323	n/a	self.loop.run_until_complete, wait)
324	n/a	self.assertFalse(ev._waiters)
325	n/a
326	n/a	def test_clear(self):
327	n/a	ev = asyncio.Event(loop=self.loop)
328	n/a	self.assertFalse(ev.is_set())
329	n/a
330	n/a	ev.set()
331	n/a	self.assertTrue(ev.is_set())
332	n/a
333	n/a	ev.clear()
334	n/a	self.assertFalse(ev.is_set())
335	n/a
336	n/a	def test_clear_with_waiters(self):
337	n/a	ev = asyncio.Event(loop=self.loop)
338	n/a	result = []
339	n/a
340	n/a	@asyncio.coroutine
341	n/a	def c1(result):
342	n/a	if (yield from ev.wait()):
343	n/a	result.append(1)
344	n/a	return True
345	n/a
346	n/a	t = asyncio.Task(c1(result), loop=self.loop)
347	n/a	test_utils.run_briefly(self.loop)
348	n/a	self.assertEqual([], result)
349	n/a
350	n/a	ev.set()
351	n/a	ev.clear()
352	n/a	self.assertFalse(ev.is_set())
353	n/a
354	n/a	ev.set()
355	n/a	ev.set()
356	n/a	self.assertEqual(1, len(ev._waiters))
357	n/a
358	n/a	test_utils.run_briefly(self.loop)
359	n/a	self.assertEqual([1], result)
360	n/a	self.assertEqual(0, len(ev._waiters))
361	n/a
362	n/a	self.assertTrue(t.done())
363	n/a	self.assertTrue(t.result())
364	n/a
365	n/a
366	n/a	class ConditionTests(test_utils.TestCase):
367	n/a
368	n/a	def setUp(self):
369	n/a	super().setUp()
370	n/a	self.loop = self.new_test_loop()
371	n/a
372	n/a	def test_ctor_loop(self):
373	n/a	loop = mock.Mock()
374	n/a	cond = asyncio.Condition(loop=loop)
375	n/a	self.assertIs(cond._loop, loop)
376	n/a
377	n/a	cond = asyncio.Condition(loop=self.loop)
378	n/a	self.assertIs(cond._loop, self.loop)
379	n/a
380	n/a	def test_ctor_noloop(self):
381	n/a	asyncio.set_event_loop(self.loop)
382	n/a	cond = asyncio.Condition()
383	n/a	self.assertIs(cond._loop, self.loop)
384	n/a
385	n/a	def test_wait(self):
386	n/a	cond = asyncio.Condition(loop=self.loop)
387	n/a	result = []
388	n/a
389	n/a	@asyncio.coroutine
390	n/a	def c1(result):
391	n/a	yield from cond.acquire()
392	n/a	if (yield from cond.wait()):
393	n/a	result.append(1)
394	n/a	return True
395	n/a
396	n/a	@asyncio.coroutine
397	n/a	def c2(result):
398	n/a	yield from cond.acquire()
399	n/a	if (yield from cond.wait()):
400	n/a	result.append(2)
401	n/a	return True
402	n/a
403	n/a	@asyncio.coroutine
404	n/a	def c3(result):
405	n/a	yield from cond.acquire()
406	n/a	if (yield from cond.wait()):
407	n/a	result.append(3)
408	n/a	return True
409	n/a
410	n/a	t1 = asyncio.Task(c1(result), loop=self.loop)
411	n/a	t2 = asyncio.Task(c2(result), loop=self.loop)
412	n/a	t3 = asyncio.Task(c3(result), loop=self.loop)
413	n/a
414	n/a	test_utils.run_briefly(self.loop)
415	n/a	self.assertEqual([], result)
416	n/a	self.assertFalse(cond.locked())
417	n/a
418	n/a	self.assertTrue(self.loop.run_until_complete(cond.acquire()))
419	n/a	cond.notify()
420	n/a	test_utils.run_briefly(self.loop)
421	n/a	self.assertEqual([], result)
422	n/a	self.assertTrue(cond.locked())
423	n/a
424	n/a	cond.release()
425	n/a	test_utils.run_briefly(self.loop)
426	n/a	self.assertEqual([1], result)
427	n/a	self.assertTrue(cond.locked())
428	n/a
429	n/a	cond.notify(2)
430	n/a	test_utils.run_briefly(self.loop)
431	n/a	self.assertEqual([1], result)
432	n/a	self.assertTrue(cond.locked())
433	n/a
434	n/a	cond.release()
435	n/a	test_utils.run_briefly(self.loop)
436	n/a	self.assertEqual([1, 2], result)
437	n/a	self.assertTrue(cond.locked())
438	n/a
439	n/a	cond.release()
440	n/a	test_utils.run_briefly(self.loop)
441	n/a	self.assertEqual([1, 2, 3], result)
442	n/a	self.assertTrue(cond.locked())
443	n/a
444	n/a	self.assertTrue(t1.done())
445	n/a	self.assertTrue(t1.result())
446	n/a	self.assertTrue(t2.done())
447	n/a	self.assertTrue(t2.result())
448	n/a	self.assertTrue(t3.done())
449	n/a	self.assertTrue(t3.result())
450	n/a
451	n/a	def test_wait_cancel(self):
452	n/a	cond = asyncio.Condition(loop=self.loop)
453	n/a	self.loop.run_until_complete(cond.acquire())
454	n/a
455	n/a	wait = asyncio.Task(cond.wait(), loop=self.loop)
456	n/a	self.loop.call_soon(wait.cancel)
457	n/a	self.assertRaises(
458	n/a	asyncio.CancelledError,
459	n/a	self.loop.run_until_complete, wait)
460	n/a	self.assertFalse(cond._waiters)
461	n/a	self.assertTrue(cond.locked())
462	n/a
463	n/a	def test_wait_cancel_contested(self):
464	n/a	cond = asyncio.Condition(loop=self.loop)
465	n/a
466	n/a	self.loop.run_until_complete(cond.acquire())
467	n/a	self.assertTrue(cond.locked())
468	n/a
469	n/a	wait_task = asyncio.Task(cond.wait(), loop=self.loop)
470	n/a	test_utils.run_briefly(self.loop)
471	n/a	self.assertFalse(cond.locked())
472	n/a
473	n/a	# Notify, but contest the lock before cancelling
474	n/a	self.loop.run_until_complete(cond.acquire())
475	n/a	self.assertTrue(cond.locked())
476	n/a	cond.notify()
477	n/a	self.loop.call_soon(wait_task.cancel)
478	n/a	self.loop.call_soon(cond.release)
479	n/a
480	n/a	try:
481	n/a	self.loop.run_until_complete(wait_task)
482	n/a	except asyncio.CancelledError:
483	n/a	# Should not happen, since no cancellation points
484	n/a	pass
485	n/a
486	n/a	self.assertTrue(cond.locked())
487	n/a
488	n/a	def test_wait_unacquired(self):
489	n/a	cond = asyncio.Condition(loop=self.loop)
490	n/a	self.assertRaises(
491	n/a	RuntimeError,
492	n/a	self.loop.run_until_complete, cond.wait())
493	n/a
494	n/a	def test_wait_for(self):
495	n/a	cond = asyncio.Condition(loop=self.loop)
496	n/a	presult = False
497	n/a
498	n/a	def predicate():
499	n/a	return presult
500	n/a
501	n/a	result = []
502	n/a
503	n/a	@asyncio.coroutine
504	n/a	def c1(result):
505	n/a	yield from cond.acquire()
506	n/a	if (yield from cond.wait_for(predicate)):
507	n/a	result.append(1)
508	n/a	cond.release()
509	n/a	return True
510	n/a
511	n/a	t = asyncio.Task(c1(result), loop=self.loop)
512	n/a
513	n/a	test_utils.run_briefly(self.loop)
514	n/a	self.assertEqual([], result)
515	n/a
516	n/a	self.loop.run_until_complete(cond.acquire())
517	n/a	cond.notify()
518	n/a	cond.release()
519	n/a	test_utils.run_briefly(self.loop)
520	n/a	self.assertEqual([], result)
521	n/a
522	n/a	presult = True
523	n/a	self.loop.run_until_complete(cond.acquire())
524	n/a	cond.notify()
525	n/a	cond.release()
526	n/a	test_utils.run_briefly(self.loop)
527	n/a	self.assertEqual([1], result)
528	n/a
529	n/a	self.assertTrue(t.done())
530	n/a	self.assertTrue(t.result())
531	n/a
532	n/a	def test_wait_for_unacquired(self):
533	n/a	cond = asyncio.Condition(loop=self.loop)
534	n/a
535	n/a	# predicate can return true immediately
536	n/a	res = self.loop.run_until_complete(cond.wait_for(lambda: [1, 2, 3]))
537	n/a	self.assertEqual([1, 2, 3], res)
538	n/a
539	n/a	self.assertRaises(
540	n/a	RuntimeError,
541	n/a	self.loop.run_until_complete,
542	n/a	cond.wait_for(lambda: False))
543	n/a
544	n/a	def test_notify(self):
545	n/a	cond = asyncio.Condition(loop=self.loop)
546	n/a	result = []
547	n/a
548	n/a	@asyncio.coroutine
549	n/a	def c1(result):
550	n/a	yield from cond.acquire()
551	n/a	if (yield from cond.wait()):
552	n/a	result.append(1)
553	n/a	cond.release()
554	n/a	return True
555	n/a
556	n/a	@asyncio.coroutine
557	n/a	def c2(result):
558	n/a	yield from cond.acquire()
559	n/a	if (yield from cond.wait()):
560	n/a	result.append(2)
561	n/a	cond.release()
562	n/a	return True
563	n/a
564	n/a	@asyncio.coroutine
565	n/a	def c3(result):
566	n/a	yield from cond.acquire()
567	n/a	if (yield from cond.wait()):
568	n/a	result.append(3)
569	n/a	cond.release()
570	n/a	return True
571	n/a
572	n/a	t1 = asyncio.Task(c1(result), loop=self.loop)
573	n/a	t2 = asyncio.Task(c2(result), loop=self.loop)
574	n/a	t3 = asyncio.Task(c3(result), loop=self.loop)
575	n/a
576	n/a	test_utils.run_briefly(self.loop)
577	n/a	self.assertEqual([], result)
578	n/a
579	n/a	self.loop.run_until_complete(cond.acquire())
580	n/a	cond.notify(1)
581	n/a	cond.release()
582	n/a	test_utils.run_briefly(self.loop)
583	n/a	self.assertEqual([1], result)
584	n/a
585	n/a	self.loop.run_until_complete(cond.acquire())
586	n/a	cond.notify(1)
587	n/a	cond.notify(2048)
588	n/a	cond.release()
589	n/a	test_utils.run_briefly(self.loop)
590	n/a	self.assertEqual([1, 2, 3], result)
591	n/a
592	n/a	self.assertTrue(t1.done())
593	n/a	self.assertTrue(t1.result())
594	n/a	self.assertTrue(t2.done())
595	n/a	self.assertTrue(t2.result())
596	n/a	self.assertTrue(t3.done())
597	n/a	self.assertTrue(t3.result())
598	n/a
599	n/a	def test_notify_all(self):
600	n/a	cond = asyncio.Condition(loop=self.loop)
601	n/a
602	n/a	result = []
603	n/a
604	n/a	@asyncio.coroutine
605	n/a	def c1(result):
606	n/a	yield from cond.acquire()
607	n/a	if (yield from cond.wait()):
608	n/a	result.append(1)
609	n/a	cond.release()
610	n/a	return True
611	n/a
612	n/a	@asyncio.coroutine
613	n/a	def c2(result):
614	n/a	yield from cond.acquire()
615	n/a	if (yield from cond.wait()):
616	n/a	result.append(2)
617	n/a	cond.release()
618	n/a	return True
619	n/a
620	n/a	t1 = asyncio.Task(c1(result), loop=self.loop)
621	n/a	t2 = asyncio.Task(c2(result), loop=self.loop)
622	n/a
623	n/a	test_utils.run_briefly(self.loop)
624	n/a	self.assertEqual([], result)
625	n/a
626	n/a	self.loop.run_until_complete(cond.acquire())
627	n/a	cond.notify_all()
628	n/a	cond.release()
629	n/a	test_utils.run_briefly(self.loop)
630	n/a	self.assertEqual([1, 2], result)
631	n/a
632	n/a	self.assertTrue(t1.done())
633	n/a	self.assertTrue(t1.result())
634	n/a	self.assertTrue(t2.done())
635	n/a	self.assertTrue(t2.result())
636	n/a
637	n/a	def test_notify_unacquired(self):
638	n/a	cond = asyncio.Condition(loop=self.loop)
639	n/a	self.assertRaises(RuntimeError, cond.notify)
640	n/a
641	n/a	def test_notify_all_unacquired(self):
642	n/a	cond = asyncio.Condition(loop=self.loop)
643	n/a	self.assertRaises(RuntimeError, cond.notify_all)
644	n/a
645	n/a	def test_repr(self):
646	n/a	cond = asyncio.Condition(loop=self.loop)
647	n/a	self.assertTrue('unlocked' in repr(cond))
648	n/a	self.assertTrue(RGX_REPR.match(repr(cond)))
649	n/a
650	n/a	self.loop.run_until_complete(cond.acquire())
651	n/a	self.assertTrue('locked' in repr(cond))
652	n/a
653	n/a	cond._waiters.append(mock.Mock())
654	n/a	self.assertTrue('waiters:1' in repr(cond))
655	n/a	self.assertTrue(RGX_REPR.match(repr(cond)))
656	n/a
657	n/a	cond._waiters.append(mock.Mock())
658	n/a	self.assertTrue('waiters:2' in repr(cond))
659	n/a	self.assertTrue(RGX_REPR.match(repr(cond)))
660	n/a
661	n/a	def test_context_manager(self):
662	n/a	cond = asyncio.Condition(loop=self.loop)
663	n/a
664	n/a	@asyncio.coroutine
665	n/a	def acquire_cond():
666	n/a	return (yield from cond)
667	n/a
668	n/a	with self.loop.run_until_complete(acquire_cond()):
669	n/a	self.assertTrue(cond.locked())
670	n/a
671	n/a	self.assertFalse(cond.locked())
672	n/a
673	n/a	def test_context_manager_no_yield(self):
674	n/a	cond = asyncio.Condition(loop=self.loop)
675	n/a
676	n/a	try:
677	n/a	with cond:
678	n/a	self.fail('RuntimeError is not raised in with expression')
679	n/a	except RuntimeError as err:
680	n/a	self.assertEqual(
681	n/a	str(err),
682	n/a	'"yield from" should be used as context manager expression')
683	n/a
684	n/a	self.assertFalse(cond.locked())
685	n/a
686	n/a	def test_explicit_lock(self):
687	n/a	lock = asyncio.Lock(loop=self.loop)
688	n/a	cond = asyncio.Condition(lock, loop=self.loop)
689	n/a
690	n/a	self.assertIs(cond._lock, lock)
691	n/a	self.assertIs(cond._loop, lock._loop)
692	n/a
693	n/a	def test_ambiguous_loops(self):
694	n/a	loop = self.new_test_loop()
695	n/a	self.addCleanup(loop.close)
696	n/a
697	n/a	lock = asyncio.Lock(loop=self.loop)
698	n/a	with self.assertRaises(ValueError):
699	n/a	asyncio.Condition(lock, loop=loop)
700	n/a
701	n/a
702	n/a	class SemaphoreTests(test_utils.TestCase):
703	n/a
704	n/a	def setUp(self):
705	n/a	super().setUp()
706	n/a	self.loop = self.new_test_loop()
707	n/a
708	n/a	def test_ctor_loop(self):
709	n/a	loop = mock.Mock()
710	n/a	sem = asyncio.Semaphore(loop=loop)
711	n/a	self.assertIs(sem._loop, loop)
712	n/a
713	n/a	sem = asyncio.Semaphore(loop=self.loop)
714	n/a	self.assertIs(sem._loop, self.loop)
715	n/a
716	n/a	def test_ctor_noloop(self):
717	n/a	asyncio.set_event_loop(self.loop)
718	n/a	sem = asyncio.Semaphore()
719	n/a	self.assertIs(sem._loop, self.loop)
720	n/a
721	n/a	def test_initial_value_zero(self):
722	n/a	sem = asyncio.Semaphore(0, loop=self.loop)
723	n/a	self.assertTrue(sem.locked())
724	n/a
725	n/a	def test_repr(self):
726	n/a	sem = asyncio.Semaphore(loop=self.loop)
727	n/a	self.assertTrue(repr(sem).endswith('[unlocked,value:1]>'))
728	n/a	self.assertTrue(RGX_REPR.match(repr(sem)))
729	n/a
730	n/a	self.loop.run_until_complete(sem.acquire())
731	n/a	self.assertTrue(repr(sem).endswith('[locked]>'))
732	n/a	self.assertTrue('waiters' not in repr(sem))
733	n/a	self.assertTrue(RGX_REPR.match(repr(sem)))
734	n/a
735	n/a	sem._waiters.append(mock.Mock())
736	n/a	self.assertTrue('waiters:1' in repr(sem))
737	n/a	self.assertTrue(RGX_REPR.match(repr(sem)))
738	n/a
739	n/a	sem._waiters.append(mock.Mock())
740	n/a	self.assertTrue('waiters:2' in repr(sem))
741	n/a	self.assertTrue(RGX_REPR.match(repr(sem)))
742	n/a
743	n/a	def test_semaphore(self):
744	n/a	sem = asyncio.Semaphore(loop=self.loop)
745	n/a	self.assertEqual(1, sem._value)
746	n/a
747	n/a	@asyncio.coroutine
748	n/a	def acquire_lock():
749	n/a	return (yield from sem)
750	n/a
751	n/a	res = self.loop.run_until_complete(acquire_lock())
752	n/a
753	n/a	self.assertTrue(res)
754	n/a	self.assertTrue(sem.locked())
755	n/a	self.assertEqual(0, sem._value)
756	n/a
757	n/a	sem.release()
758	n/a	self.assertFalse(sem.locked())
759	n/a	self.assertEqual(1, sem._value)
760	n/a
761	n/a	def test_semaphore_value(self):
762	n/a	self.assertRaises(ValueError, asyncio.Semaphore, -1)
763	n/a
764	n/a	def test_acquire(self):
765	n/a	sem = asyncio.Semaphore(3, loop=self.loop)
766	n/a	result = []
767	n/a
768	n/a	self.assertTrue(self.loop.run_until_complete(sem.acquire()))
769	n/a	self.assertTrue(self.loop.run_until_complete(sem.acquire()))
770	n/a	self.assertFalse(sem.locked())
771	n/a
772	n/a	@asyncio.coroutine
773	n/a	def c1(result):
774	n/a	yield from sem.acquire()
775	n/a	result.append(1)
776	n/a	return True
777	n/a
778	n/a	@asyncio.coroutine
779	n/a	def c2(result):
780	n/a	yield from sem.acquire()
781	n/a	result.append(2)
782	n/a	return True
783	n/a
784	n/a	@asyncio.coroutine
785	n/a	def c3(result):
786	n/a	yield from sem.acquire()
787	n/a	result.append(3)
788	n/a	return True
789	n/a
790	n/a	@asyncio.coroutine
791	n/a	def c4(result):
792	n/a	yield from sem.acquire()
793	n/a	result.append(4)
794	n/a	return True
795	n/a
796	n/a	t1 = asyncio.Task(c1(result), loop=self.loop)
797	n/a	t2 = asyncio.Task(c2(result), loop=self.loop)
798	n/a	t3 = asyncio.Task(c3(result), loop=self.loop)
799	n/a
800	n/a	test_utils.run_briefly(self.loop)
801	n/a	self.assertEqual([1], result)
802	n/a	self.assertTrue(sem.locked())
803	n/a	self.assertEqual(2, len(sem._waiters))
804	n/a	self.assertEqual(0, sem._value)
805	n/a
806	n/a	t4 = asyncio.Task(c4(result), loop=self.loop)
807	n/a
808	n/a	sem.release()
809	n/a	sem.release()
810	n/a	self.assertEqual(2, sem._value)
811	n/a
812	n/a	test_utils.run_briefly(self.loop)
813	n/a	self.assertEqual(0, sem._value)
814	n/a	self.assertEqual(3, len(result))
815	n/a	self.assertTrue(sem.locked())
816	n/a	self.assertEqual(1, len(sem._waiters))
817	n/a	self.assertEqual(0, sem._value)
818	n/a
819	n/a	self.assertTrue(t1.done())
820	n/a	self.assertTrue(t1.result())
821	n/a	race_tasks = [t2, t3, t4]
822	n/a	done_tasks = [t for t in race_tasks if t.done() and t.result()]
823	n/a	self.assertTrue(2, len(done_tasks))
824	n/a
825	n/a	# cleanup locked semaphore
826	n/a	sem.release()
827	n/a	self.loop.run_until_complete(asyncio.gather(*race_tasks))
828	n/a
829	n/a	def test_acquire_cancel(self):
830	n/a	sem = asyncio.Semaphore(loop=self.loop)
831	n/a	self.loop.run_until_complete(sem.acquire())
832	n/a
833	n/a	acquire = asyncio.Task(sem.acquire(), loop=self.loop)
834	n/a	self.loop.call_soon(acquire.cancel)
835	n/a	self.assertRaises(
836	n/a	asyncio.CancelledError,
837	n/a	self.loop.run_until_complete, acquire)
838	n/a	self.assertTrue((not sem._waiters) or
839	n/a	all(waiter.done() for waiter in sem._waiters))
840	n/a
841	n/a	def test_acquire_cancel_before_awoken(self):
842	n/a	sem = asyncio.Semaphore(value=0, loop=self.loop)
843	n/a
844	n/a	t1 = asyncio.Task(sem.acquire(), loop=self.loop)
845	n/a	t2 = asyncio.Task(sem.acquire(), loop=self.loop)
846	n/a	t3 = asyncio.Task(sem.acquire(), loop=self.loop)
847	n/a	t4 = asyncio.Task(sem.acquire(), loop=self.loop)
848	n/a
849	n/a	test_utils.run_briefly(self.loop)
850	n/a
851	n/a	sem.release()
852	n/a	t1.cancel()
853	n/a	t2.cancel()
854	n/a
855	n/a	test_utils.run_briefly(self.loop)
856	n/a	num_done = sum(t.done() for t in [t3, t4])
857	n/a	self.assertEqual(num_done, 1)
858	n/a
859	n/a	t3.cancel()
860	n/a	t4.cancel()
861	n/a	test_utils.run_briefly(self.loop)
862	n/a
863	n/a	def test_acquire_hang(self):
864	n/a	sem = asyncio.Semaphore(value=0, loop=self.loop)
865	n/a
866	n/a	t1 = asyncio.Task(sem.acquire(), loop=self.loop)
867	n/a	t2 = asyncio.Task(sem.acquire(), loop=self.loop)
868	n/a
869	n/a	test_utils.run_briefly(self.loop)
870	n/a
871	n/a	sem.release()
872	n/a	t1.cancel()
873	n/a
874	n/a	test_utils.run_briefly(self.loop)
875	n/a	self.assertTrue(sem.locked())
876	n/a
877	n/a	def test_release_not_acquired(self):
878	n/a	sem = asyncio.BoundedSemaphore(loop=self.loop)
879	n/a
880	n/a	self.assertRaises(ValueError, sem.release)
881	n/a
882	n/a	def test_release_no_waiters(self):
883	n/a	sem = asyncio.Semaphore(loop=self.loop)
884	n/a	self.loop.run_until_complete(sem.acquire())
885	n/a	self.assertTrue(sem.locked())
886	n/a
887	n/a	sem.release()
888	n/a	self.assertFalse(sem.locked())
889	n/a
890	n/a	def test_context_manager(self):
891	n/a	sem = asyncio.Semaphore(2, loop=self.loop)
892	n/a
893	n/a	@asyncio.coroutine
894	n/a	def acquire_lock():
895	n/a	return (yield from sem)
896	n/a
897	n/a	with self.loop.run_until_complete(acquire_lock()):
898	n/a	self.assertFalse(sem.locked())
899	n/a	self.assertEqual(1, sem._value)
900	n/a
901	n/a	with self.loop.run_until_complete(acquire_lock()):
902	n/a	self.assertTrue(sem.locked())
903	n/a
904	n/a	self.assertEqual(2, sem._value)
905	n/a
906	n/a	def test_context_manager_no_yield(self):
907	n/a	sem = asyncio.Semaphore(2, loop=self.loop)
908	n/a
909	n/a	try:
910	n/a	with sem:
911	n/a	self.fail('RuntimeError is not raised in with expression')
912	n/a	except RuntimeError as err:
913	n/a	self.assertEqual(
914	n/a	str(err),
915	n/a	'"yield from" should be used as context manager expression')
916	n/a
917	n/a	self.assertEqual(2, sem._value)
918	n/a
919	n/a
920	n/a	if __name__ == '__main__':
921	n/a	unittest.main()