summaryrefslogtreecommitdiff
path: root/Lib/test/test_queue.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_queue.py')
-rw-r--r--Lib/test/test_queue.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/Lib/test/test_queue.py b/Lib/test/test_queue.py
index b55dd01f22..17c1def573 100644
--- a/Lib/test/test_queue.py
+++ b/Lib/test/test_queue.py
@@ -221,7 +221,37 @@ def SimpleQueueTest(q):
_doBlockingTest(q.get, (), q.put, ('empty',))
_doBlockingTest(q.get, (True, 10), q.put, ('empty',))
+cum = 0
+cumlock = threading.Lock()
+
+def worker(q):
+ global cum
+ while True:
+ x = q.get()
+ cumlock.acquire()
+ try:
+ cum += x
+ finally:
+ cumlock.release()
+ q.task_done()
+
+def QueueJoinTest(q):
+ global cum
+ cum = 0
+ for i in (0,1):
+ t = threading.Thread(target=worker, args=(q,))
+ t.setDaemon(True)
+ t.start()
+ for i in xrange(100):
+ q.put(i)
+ q.join()
+ verify(cum==sum(range(100)), "q.join() did not block until all tasks were done")
+
def test():
+ q = Queue.Queue()
+ QueueJoinTest(q)
+ QueueJoinTest(q)
+
q = Queue.Queue(QUEUE_SIZE)
# Do it a couple of times on the same queue
SimpleQueueTest(q)