blob: ac4e7a7e0f53b3eefa9b08864444cfe626555114 [file] [log] [blame]
Antoine Pitrou4c8ce842013-09-01 19:51:49 +02001"""
2Tests for the threading module.
3"""
Skip Montanaro4533f602001-08-20 20:28:48 +00004
Benjamin Petersonee8712c2008-05-20 21:35:26 +00005import test.support
Serhiy Storchakaa7930372016-07-03 22:27:26 +03006from test.support import (verbose, import_module, cpython_only,
7 requires_type_collecting)
Berker Peksagce643912015-05-06 06:33:17 +03008from test.support.script_helper import assert_python_ok, assert_python_failure
Antoine Pitrou8e6e0fd2012-04-19 23:55:01 +02009
Skip Montanaro4533f602001-08-20 20:28:48 +000010import random
Guido van Rossumcd16bf62007-06-13 18:07:49 +000011import sys
Antoine Pitroua6a4dc82017-09-07 18:56:24 +020012import _thread
13import threading
Skip Montanaro4533f602001-08-20 20:28:48 +000014import time
Tim Peters84d54892005-01-08 06:03:17 +000015import unittest
Christian Heimesd3eb5a152008-02-24 00:38:49 +000016import weakref
Alexandre Vassalotti93f2cd22009-07-22 04:54:52 +000017import os
Gregory P. Smith4b129d22011-01-04 00:51:50 +000018import subprocess
Matěj Cepl608876b2019-05-23 22:30:00 +020019import signal
Skip Montanaro4533f602001-08-20 20:28:48 +000020
Antoine Pitrou557934f2009-11-06 22:41:14 +000021from test import lock_tests
Martin Panter19e69c52015-11-14 12:46:42 +000022from test import support
Antoine Pitrou557934f2009-11-06 22:41:14 +000023
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +030024
25# Between fork() and exec(), only async-safe functions are allowed (issues
26# #12316 and #11870), and fork() from a worker thread is known to trigger
27# problems with some operating systems (issue #3863): skip problematic tests
28# on platforms known to behave badly.
Victor Stinner13ff2452018-01-22 18:32:50 +010029platforms_to_skip = ('netbsd5', 'hp-ux11')
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +030030
31
Tim Peters84d54892005-01-08 06:03:17 +000032# A trivial mutable counter.
33class Counter(object):
34 def __init__(self):
35 self.value = 0
36 def inc(self):
37 self.value += 1
38 def dec(self):
39 self.value -= 1
40 def get(self):
41 return self.value
Skip Montanaro4533f602001-08-20 20:28:48 +000042
43class TestThread(threading.Thread):
Tim Peters84d54892005-01-08 06:03:17 +000044 def __init__(self, name, testcase, sema, mutex, nrunning):
45 threading.Thread.__init__(self, name=name)
46 self.testcase = testcase
47 self.sema = sema
48 self.mutex = mutex
49 self.nrunning = nrunning
50
Skip Montanaro4533f602001-08-20 20:28:48 +000051 def run(self):
Christian Heimes4fbc72b2008-03-22 00:47:35 +000052 delay = random.random() / 10000.0
Skip Montanaro4533f602001-08-20 20:28:48 +000053 if verbose:
Jeffrey Yasskinca674122008-03-29 05:06:52 +000054 print('task %s will run for %.1f usec' %
Benjamin Petersonfdbea962008-08-18 17:33:47 +000055 (self.name, delay * 1e6))
Tim Peters84d54892005-01-08 06:03:17 +000056
Christian Heimes4fbc72b2008-03-22 00:47:35 +000057 with self.sema:
58 with self.mutex:
59 self.nrunning.inc()
60 if verbose:
61 print(self.nrunning.get(), 'tasks are running')
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +020062 self.testcase.assertLessEqual(self.nrunning.get(), 3)
Tim Peters84d54892005-01-08 06:03:17 +000063
Christian Heimes4fbc72b2008-03-22 00:47:35 +000064 time.sleep(delay)
65 if verbose:
Benjamin Petersonfdbea962008-08-18 17:33:47 +000066 print('task', self.name, 'done')
Benjamin Peterson672b8032008-06-11 19:14:14 +000067
Christian Heimes4fbc72b2008-03-22 00:47:35 +000068 with self.mutex:
69 self.nrunning.dec()
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +020070 self.testcase.assertGreaterEqual(self.nrunning.get(), 0)
Christian Heimes4fbc72b2008-03-22 00:47:35 +000071 if verbose:
72 print('%s is finished. %d tasks are running' %
Benjamin Petersonfdbea962008-08-18 17:33:47 +000073 (self.name, self.nrunning.get()))
Benjamin Peterson672b8032008-06-11 19:14:14 +000074
Skip Montanaro4533f602001-08-20 20:28:48 +000075
Antoine Pitroub0e9bd42009-10-27 20:05:26 +000076class BaseTestCase(unittest.TestCase):
77 def setUp(self):
78 self._threads = test.support.threading_setup()
79
80 def tearDown(self):
81 test.support.threading_cleanup(*self._threads)
82 test.support.reap_children()
83
84
85class ThreadTests(BaseTestCase):
Skip Montanaro4533f602001-08-20 20:28:48 +000086
Tim Peters84d54892005-01-08 06:03:17 +000087 # Create a bunch of threads, let each do some work, wait until all are
88 # done.
89 def test_various_ops(self):
90 # This takes about n/3 seconds to run (about n/3 clumps of tasks,
91 # times about 1 second per clump).
92 NUMTASKS = 10
93
94 # no more than 3 of the 10 can run at once
95 sema = threading.BoundedSemaphore(value=3)
96 mutex = threading.RLock()
97 numrunning = Counter()
98
99 threads = []
100
101 for i in range(NUMTASKS):
102 t = TestThread("<thread %d>"%i, self, sema, mutex, numrunning)
103 threads.append(t)
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200104 self.assertIsNone(t.ident)
105 self.assertRegex(repr(t), r'^<TestThread\(.*, initial\)>$')
Tim Peters84d54892005-01-08 06:03:17 +0000106 t.start()
107
Jake Teslerb121f632019-05-22 08:43:17 -0700108 if hasattr(threading, 'get_native_id'):
109 native_ids = set(t.native_id for t in threads) | {threading.get_native_id()}
110 self.assertNotIn(None, native_ids)
111 self.assertEqual(len(native_ids), NUMTASKS + 1)
112
Tim Peters84d54892005-01-08 06:03:17 +0000113 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000114 print('waiting for all tasks to complete')
Tim Peters84d54892005-01-08 06:03:17 +0000115 for t in threads:
Antoine Pitrou5da7e792013-09-08 13:19:06 +0200116 t.join()
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200117 self.assertFalse(t.is_alive())
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000118 self.assertNotEqual(t.ident, 0)
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200119 self.assertIsNotNone(t.ident)
120 self.assertRegex(repr(t), r'^<TestThread\(.*, stopped -?\d+\)>$')
Tim Peters84d54892005-01-08 06:03:17 +0000121 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000122 print('all tasks done')
Tim Peters84d54892005-01-08 06:03:17 +0000123 self.assertEqual(numrunning.get(), 0)
124
Benjamin Petersond23f8222009-04-05 19:13:16 +0000125 def test_ident_of_no_threading_threads(self):
126 # The ident still must work for the main thread and dummy threads.
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200127 self.assertIsNotNone(threading.currentThread().ident)
Benjamin Petersond23f8222009-04-05 19:13:16 +0000128 def f():
129 ident.append(threading.currentThread().ident)
130 done.set()
131 done = threading.Event()
132 ident = []
Victor Stinnerff40ecd2017-09-14 13:07:24 -0700133 with support.wait_threads_exit():
134 tid = _thread.start_new_thread(f, ())
135 done.wait()
136 self.assertEqual(ident[0], tid)
Antoine Pitrouca13a0d2009-11-08 00:30:04 +0000137 # Kill the "immortal" _DummyThread
138 del threading._active[ident[0]]
Benjamin Petersond23f8222009-04-05 19:13:16 +0000139
Victor Stinner8c663fd2017-11-08 14:44:44 -0800140 # run with a small(ish) thread stack size (256 KiB)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000141 def test_various_ops_small_stack(self):
142 if verbose:
Victor Stinner8c663fd2017-11-08 14:44:44 -0800143 print('with 256 KiB thread stack size...')
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000144 try:
145 threading.stack_size(262144)
Georg Brandl2067bfd2008-05-25 13:05:15 +0000146 except _thread.error:
Alexandre Vassalotti93f2cd22009-07-22 04:54:52 +0000147 raise unittest.SkipTest(
148 'platform does not support changing thread stack size')
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000149 self.test_various_ops()
150 threading.stack_size(0)
151
Victor Stinner8c663fd2017-11-08 14:44:44 -0800152 # run with a large thread stack size (1 MiB)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000153 def test_various_ops_large_stack(self):
154 if verbose:
Victor Stinner8c663fd2017-11-08 14:44:44 -0800155 print('with 1 MiB thread stack size...')
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000156 try:
157 threading.stack_size(0x100000)
Georg Brandl2067bfd2008-05-25 13:05:15 +0000158 except _thread.error:
Alexandre Vassalotti93f2cd22009-07-22 04:54:52 +0000159 raise unittest.SkipTest(
160 'platform does not support changing thread stack size')
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000161 self.test_various_ops()
162 threading.stack_size(0)
163
Tim Peters711906e2005-01-08 07:30:42 +0000164 def test_foreign_thread(self):
165 # Check that a "foreign" thread can use the threading module.
166 def f(mutex):
Antoine Pitroub0872682009-11-09 16:08:16 +0000167 # Calling current_thread() forces an entry for the foreign
Tim Peters711906e2005-01-08 07:30:42 +0000168 # thread to get made in the threading._active map.
Antoine Pitroub0872682009-11-09 16:08:16 +0000169 threading.current_thread()
Tim Peters711906e2005-01-08 07:30:42 +0000170 mutex.release()
171
172 mutex = threading.Lock()
173 mutex.acquire()
Victor Stinnerff40ecd2017-09-14 13:07:24 -0700174 with support.wait_threads_exit():
175 tid = _thread.start_new_thread(f, (mutex,))
176 # Wait for the thread to finish.
177 mutex.acquire()
Benjamin Peterson577473f2010-01-19 00:09:57 +0000178 self.assertIn(tid, threading._active)
Ezio Melottie9615932010-01-24 19:26:24 +0000179 self.assertIsInstance(threading._active[tid], threading._DummyThread)
Xiang Zhangf3a9fab2017-02-27 11:01:30 +0800180 #Issue 29376
181 self.assertTrue(threading._active[tid].is_alive())
182 self.assertRegex(repr(threading._active[tid]), '_DummyThread')
Tim Peters711906e2005-01-08 07:30:42 +0000183 del threading._active[tid]
Tim Peters84d54892005-01-08 06:03:17 +0000184
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000185 # PyThreadState_SetAsyncExc() is a CPython-only gimmick, not (currently)
186 # exposed at the Python level. This test relies on ctypes to get at it.
187 def test_PyThreadState_SetAsyncExc(self):
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200188 ctypes = import_module("ctypes")
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000189
190 set_async_exc = ctypes.pythonapi.PyThreadState_SetAsyncExc
Serhiy Storchakaaefa7eb2017-03-23 15:48:39 +0200191 set_async_exc.argtypes = (ctypes.c_ulong, ctypes.py_object)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000192
193 class AsyncExc(Exception):
194 pass
195
196 exception = ctypes.py_object(AsyncExc)
197
Antoine Pitroube4d8092009-10-18 18:27:17 +0000198 # First check it works when setting the exception from the same thread.
Victor Stinner2a129742011-05-30 23:02:52 +0200199 tid = threading.get_ident()
Serhiy Storchakaaefa7eb2017-03-23 15:48:39 +0200200 self.assertIsInstance(tid, int)
201 self.assertGreater(tid, 0)
Antoine Pitroube4d8092009-10-18 18:27:17 +0000202
203 try:
Serhiy Storchakaaefa7eb2017-03-23 15:48:39 +0200204 result = set_async_exc(tid, exception)
Antoine Pitroube4d8092009-10-18 18:27:17 +0000205 # The exception is async, so we might have to keep the VM busy until
206 # it notices.
207 while True:
208 pass
209 except AsyncExc:
210 pass
211 else:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000212 # This code is unreachable but it reflects the intent. If we wanted
213 # to be smarter the above loop wouldn't be infinite.
Antoine Pitroube4d8092009-10-18 18:27:17 +0000214 self.fail("AsyncExc not raised")
215 try:
216 self.assertEqual(result, 1) # one thread state modified
217 except UnboundLocalError:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000218 # The exception was raised too quickly for us to get the result.
Antoine Pitroube4d8092009-10-18 18:27:17 +0000219 pass
220
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000221 # `worker_started` is set by the thread when it's inside a try/except
222 # block waiting to catch the asynchronously set AsyncExc exception.
223 # `worker_saw_exception` is set by the thread upon catching that
224 # exception.
225 worker_started = threading.Event()
226 worker_saw_exception = threading.Event()
227
228 class Worker(threading.Thread):
229 def run(self):
Victor Stinner2a129742011-05-30 23:02:52 +0200230 self.id = threading.get_ident()
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000231 self.finished = False
232
233 try:
234 while True:
235 worker_started.set()
236 time.sleep(0.1)
237 except AsyncExc:
238 self.finished = True
239 worker_saw_exception.set()
240
241 t = Worker()
Benjamin Petersonfdbea962008-08-18 17:33:47 +0000242 t.daemon = True # so if this fails, we don't hang Python at shutdown
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000243 t.start()
244 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000245 print(" started worker thread")
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000246
247 # Try a thread id that doesn't make sense.
248 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000249 print(" trying nonsensical thread id")
Serhiy Storchakaaefa7eb2017-03-23 15:48:39 +0200250 result = set_async_exc(-1, exception)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000251 self.assertEqual(result, 0) # no thread states modified
252
253 # Now raise an exception in the worker thread.
254 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000255 print(" waiting for worker thread to get started")
Benjamin Petersond23f8222009-04-05 19:13:16 +0000256 ret = worker_started.wait()
257 self.assertTrue(ret)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000258 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000259 print(" verifying worker hasn't exited")
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200260 self.assertFalse(t.finished)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000261 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000262 print(" attempting to raise asynch exception in worker")
Serhiy Storchakaaefa7eb2017-03-23 15:48:39 +0200263 result = set_async_exc(t.id, exception)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000264 self.assertEqual(result, 1) # one thread state modified
265 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000266 print(" waiting for worker to say it caught the exception")
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000267 worker_saw_exception.wait(timeout=10)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000268 self.assertTrue(t.finished)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000269 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000270 print(" all OK -- joining worker")
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000271 if t.finished:
272 t.join()
273 # else the thread is still running, and we have no way to kill it
274
Gregory P. Smith3fdd9642010-02-28 18:57:46 +0000275 def test_limbo_cleanup(self):
276 # Issue 7481: Failure to start thread should cleanup the limbo map.
277 def fail_new_thread(*args):
278 raise threading.ThreadError()
279 _start_new_thread = threading._start_new_thread
280 threading._start_new_thread = fail_new_thread
281 try:
282 t = threading.Thread(target=lambda: None)
Gregory P. Smithf50f1682010-03-01 03:13:36 +0000283 self.assertRaises(threading.ThreadError, t.start)
284 self.assertFalse(
285 t in threading._limbo,
286 "Failed to cleanup _limbo map on failure of Thread.start().")
Gregory P. Smith3fdd9642010-02-28 18:57:46 +0000287 finally:
288 threading._start_new_thread = _start_new_thread
289
Christian Heimes7d2ff882007-11-30 14:35:04 +0000290 def test_finalize_runnning_thread(self):
291 # Issue 1402: the PyGILState_Ensure / _Release functions may be called
292 # very late on python exit: on deallocation of a running thread for
293 # example.
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200294 import_module("ctypes")
Christian Heimes7d2ff882007-11-30 14:35:04 +0000295
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200296 rc, out, err = assert_python_failure("-c", """if 1:
Georg Brandl2067bfd2008-05-25 13:05:15 +0000297 import ctypes, sys, time, _thread
Christian Heimes7d2ff882007-11-30 14:35:04 +0000298
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000299 # This lock is used as a simple event variable.
Georg Brandl2067bfd2008-05-25 13:05:15 +0000300 ready = _thread.allocate_lock()
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000301 ready.acquire()
302
Christian Heimes7d2ff882007-11-30 14:35:04 +0000303 # Module globals are cleared before __del__ is run
304 # So we save the functions in class dict
305 class C:
306 ensure = ctypes.pythonapi.PyGILState_Ensure
307 release = ctypes.pythonapi.PyGILState_Release
308 def __del__(self):
309 state = self.ensure()
310 self.release(state)
311
312 def waitingThread():
313 x = C()
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000314 ready.release()
Christian Heimes7d2ff882007-11-30 14:35:04 +0000315 time.sleep(100)
316
Georg Brandl2067bfd2008-05-25 13:05:15 +0000317 _thread.start_new_thread(waitingThread, ())
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000318 ready.acquire() # Be sure the other thread is waiting.
Christian Heimes7d2ff882007-11-30 14:35:04 +0000319 sys.exit(42)
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200320 """)
Christian Heimes7d2ff882007-11-30 14:35:04 +0000321 self.assertEqual(rc, 42)
322
Neal Norwitzf5c7c2e2008-04-05 04:47:45 +0000323 def test_finalize_with_trace(self):
324 # Issue1733757
325 # Avoid a deadlock when sys.settrace steps into threading._shutdown
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200326 assert_python_ok("-c", """if 1:
Neal Norwitzf5c7c2e2008-04-05 04:47:45 +0000327 import sys, threading
328
329 # A deadlock-killer, to prevent the
330 # testsuite to hang forever
331 def killer():
332 import os, time
333 time.sleep(2)
334 print('program blocked; aborting')
335 os._exit(2)
336 t = threading.Thread(target=killer)
Benjamin Petersonfdbea962008-08-18 17:33:47 +0000337 t.daemon = True
Neal Norwitzf5c7c2e2008-04-05 04:47:45 +0000338 t.start()
339
340 # This is the trace function
341 def func(frame, event, arg):
Benjamin Peterson672b8032008-06-11 19:14:14 +0000342 threading.current_thread()
Neal Norwitzf5c7c2e2008-04-05 04:47:45 +0000343 return func
344
345 sys.settrace(func)
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200346 """)
Neal Norwitzf5c7c2e2008-04-05 04:47:45 +0000347
Antoine Pitrou011bd622009-10-20 21:52:47 +0000348 def test_join_nondaemon_on_shutdown(self):
349 # Issue 1722344
350 # Raising SystemExit skipped threading._shutdown
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200351 rc, out, err = assert_python_ok("-c", """if 1:
Antoine Pitrou011bd622009-10-20 21:52:47 +0000352 import threading
353 from time import sleep
354
355 def child():
356 sleep(1)
357 # As a non-daemon thread we SHOULD wake up and nothing
358 # should be torn down yet
359 print("Woke up, sleep function is:", sleep)
360
361 threading.Thread(target=child).start()
362 raise SystemExit
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200363 """)
364 self.assertEqual(out.strip(),
Antoine Pitrou899d1c62009-10-23 21:55:36 +0000365 b"Woke up, sleep function is: <built-in function sleep>")
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200366 self.assertEqual(err, b"")
Neal Norwitzf5c7c2e2008-04-05 04:47:45 +0000367
Christian Heimes1af737c2008-01-23 08:24:23 +0000368 def test_enumerate_after_join(self):
369 # Try hard to trigger #1703448: a thread is still returned in
370 # threading.enumerate() after it has been join()ed.
371 enum = threading.enumerate
Antoine Pitrouc3b07572009-11-13 22:19:19 +0000372 old_interval = sys.getswitchinterval()
Christian Heimes1af737c2008-01-23 08:24:23 +0000373 try:
Jeffrey Yasskinca674122008-03-29 05:06:52 +0000374 for i in range(1, 100):
Antoine Pitrouc3b07572009-11-13 22:19:19 +0000375 sys.setswitchinterval(i * 0.0002)
Christian Heimes1af737c2008-01-23 08:24:23 +0000376 t = threading.Thread(target=lambda: None)
377 t.start()
378 t.join()
379 l = enum()
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000380 self.assertNotIn(t, l,
Christian Heimes1af737c2008-01-23 08:24:23 +0000381 "#1703448 triggered after %d trials: %s" % (i, l))
382 finally:
Antoine Pitrouc3b07572009-11-13 22:19:19 +0000383 sys.setswitchinterval(old_interval)
Christian Heimes1af737c2008-01-23 08:24:23 +0000384
Christian Heimesd3eb5a152008-02-24 00:38:49 +0000385 def test_no_refcycle_through_target(self):
386 class RunSelfFunction(object):
387 def __init__(self, should_raise):
388 # The links in this refcycle from Thread back to self
389 # should be cleaned up when the thread completes.
390 self.should_raise = should_raise
391 self.thread = threading.Thread(target=self._run,
392 args=(self,),
393 kwargs={'yet_another':self})
394 self.thread.start()
395
396 def _run(self, other_ref, yet_another):
397 if self.should_raise:
398 raise SystemExit
399
400 cyclic_object = RunSelfFunction(should_raise=False)
401 weak_cyclic_object = weakref.ref(cyclic_object)
402 cyclic_object.thread.join()
403 del cyclic_object
Raymond Hettinger7beae8a2011-01-06 05:34:17 +0000404 self.assertIsNone(weak_cyclic_object(),
Ezio Melottib3aedd42010-11-20 19:04:17 +0000405 msg=('%d references still around' %
406 sys.getrefcount(weak_cyclic_object())))
Christian Heimesd3eb5a152008-02-24 00:38:49 +0000407
408 raising_cyclic_object = RunSelfFunction(should_raise=True)
409 weak_raising_cyclic_object = weakref.ref(raising_cyclic_object)
410 raising_cyclic_object.thread.join()
411 del raising_cyclic_object
Raymond Hettinger7beae8a2011-01-06 05:34:17 +0000412 self.assertIsNone(weak_raising_cyclic_object(),
Ezio Melottib3aedd42010-11-20 19:04:17 +0000413 msg=('%d references still around' %
414 sys.getrefcount(weak_raising_cyclic_object())))
Christian Heimesd3eb5a152008-02-24 00:38:49 +0000415
Benjamin Petersonb3085c92008-09-01 23:09:31 +0000416 def test_old_threading_api(self):
417 # Just a quick sanity check to make sure the old method names are
418 # still present
Benjamin Petersonf0923f52008-08-18 22:10:13 +0000419 t = threading.Thread()
Benjamin Petersonb3085c92008-09-01 23:09:31 +0000420 t.isDaemon()
421 t.setDaemon(True)
422 t.getName()
423 t.setName("name")
Dong-hee Na89669ff2019-01-17 21:14:45 +0900424 with self.assertWarnsRegex(DeprecationWarning, 'use is_alive()'):
425 t.isAlive()
Benjamin Petersonb3085c92008-09-01 23:09:31 +0000426 e = threading.Event()
427 e.isSet()
428 threading.activeCount()
Benjamin Petersonf0923f52008-08-18 22:10:13 +0000429
Brian Curtin81a4a6a2010-07-23 16:30:10 +0000430 def test_repr_daemon(self):
431 t = threading.Thread()
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200432 self.assertNotIn('daemon', repr(t))
Brian Curtin81a4a6a2010-07-23 16:30:10 +0000433 t.daemon = True
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200434 self.assertIn('daemon', repr(t))
Brett Cannon3f5f2262010-07-23 15:50:52 +0000435
luzpaza5293b42017-11-05 07:37:50 -0600436 def test_daemon_param(self):
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +0000437 t = threading.Thread()
438 self.assertFalse(t.daemon)
439 t = threading.Thread(daemon=False)
440 self.assertFalse(t.daemon)
441 t = threading.Thread(daemon=True)
442 self.assertTrue(t.daemon)
443
Antoine Pitrou8e6e0fd2012-04-19 23:55:01 +0200444 @unittest.skipUnless(hasattr(os, 'fork'), 'test needs fork()')
445 def test_dummy_thread_after_fork(self):
446 # Issue #14308: a dummy thread in the active list doesn't mess up
447 # the after-fork mechanism.
448 code = """if 1:
449 import _thread, threading, os, time
450
451 def background_thread(evt):
452 # Creates and registers the _DummyThread instance
453 threading.current_thread()
454 evt.set()
455 time.sleep(10)
456
457 evt = threading.Event()
458 _thread.start_new_thread(background_thread, (evt,))
459 evt.wait()
460 assert threading.active_count() == 2, threading.active_count()
461 if os.fork() == 0:
462 assert threading.active_count() == 1, threading.active_count()
463 os._exit(0)
464 else:
465 os.wait()
466 """
467 _, out, err = assert_python_ok("-c", code)
468 self.assertEqual(out, b'')
469 self.assertEqual(err, b'')
470
Charles-François Natali9939cc82013-08-30 23:32:53 +0200471 @unittest.skipUnless(hasattr(os, 'fork'), "needs os.fork()")
472 def test_is_alive_after_fork(self):
473 # Try hard to trigger #18418: is_alive() could sometimes be True on
474 # threads that vanished after a fork.
475 old_interval = sys.getswitchinterval()
476 self.addCleanup(sys.setswitchinterval, old_interval)
477
478 # Make the bug more likely to manifest.
Xavier de Gayecb9ab0f2016-12-08 12:21:00 +0100479 test.support.setswitchinterval(1e-6)
Charles-François Natali9939cc82013-08-30 23:32:53 +0200480
481 for i in range(20):
482 t = threading.Thread(target=lambda: None)
483 t.start()
Charles-François Natali9939cc82013-08-30 23:32:53 +0200484 pid = os.fork()
485 if pid == 0:
Victor Stinnerf8d05b32017-05-17 11:58:50 -0700486 os._exit(11 if t.is_alive() else 10)
Charles-François Natali9939cc82013-08-30 23:32:53 +0200487 else:
Victor Stinnerf8d05b32017-05-17 11:58:50 -0700488 t.join()
489
Charles-François Natali9939cc82013-08-30 23:32:53 +0200490 pid, status = os.waitpid(pid, 0)
Victor Stinnerf8d05b32017-05-17 11:58:50 -0700491 self.assertTrue(os.WIFEXITED(status))
492 self.assertEqual(10, os.WEXITSTATUS(status))
Charles-François Natali9939cc82013-08-30 23:32:53 +0200493
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +0300494 def test_main_thread(self):
495 main = threading.main_thread()
496 self.assertEqual(main.name, 'MainThread')
497 self.assertEqual(main.ident, threading.current_thread().ident)
498 self.assertEqual(main.ident, threading.get_ident())
499
500 def f():
501 self.assertNotEqual(threading.main_thread().ident,
502 threading.current_thread().ident)
503 th = threading.Thread(target=f)
504 th.start()
505 th.join()
506
507 @unittest.skipUnless(hasattr(os, 'fork'), "test needs os.fork()")
508 @unittest.skipUnless(hasattr(os, 'waitpid'), "test needs os.waitpid()")
509 def test_main_thread_after_fork(self):
510 code = """if 1:
511 import os, threading
512
513 pid = os.fork()
514 if pid == 0:
515 main = threading.main_thread()
516 print(main.name)
517 print(main.ident == threading.current_thread().ident)
518 print(main.ident == threading.get_ident())
519 else:
520 os.waitpid(pid, 0)
521 """
522 _, out, err = assert_python_ok("-c", code)
523 data = out.decode().replace('\r', '')
524 self.assertEqual(err, b"")
525 self.assertEqual(data, "MainThread\nTrue\nTrue\n")
526
527 @unittest.skipIf(sys.platform in platforms_to_skip, "due to known OS bug")
528 @unittest.skipUnless(hasattr(os, 'fork'), "test needs os.fork()")
529 @unittest.skipUnless(hasattr(os, 'waitpid'), "test needs os.waitpid()")
530 def test_main_thread_after_fork_from_nonmain_thread(self):
531 code = """if 1:
532 import os, threading, sys
533
534 def f():
535 pid = os.fork()
536 if pid == 0:
537 main = threading.main_thread()
538 print(main.name)
539 print(main.ident == threading.current_thread().ident)
540 print(main.ident == threading.get_ident())
541 # stdout is fully buffered because not a tty,
542 # we have to flush before exit.
543 sys.stdout.flush()
544 else:
545 os.waitpid(pid, 0)
546
547 th = threading.Thread(target=f)
548 th.start()
549 th.join()
550 """
551 _, out, err = assert_python_ok("-c", code)
552 data = out.decode().replace('\r', '')
553 self.assertEqual(err, b"")
554 self.assertEqual(data, "Thread-1\nTrue\nTrue\n")
555
Zackery Spytz65d2f8c2018-10-12 02:31:21 -0600556 @requires_type_collecting
Antoine Pitrou1023dbb2017-10-02 16:42:15 +0200557 def test_main_thread_during_shutdown(self):
558 # bpo-31516: current_thread() should still point to the main thread
559 # at shutdown
560 code = """if 1:
561 import gc, threading
562
563 main_thread = threading.current_thread()
564 assert main_thread is threading.main_thread() # sanity check
565
566 class RefCycle:
567 def __init__(self):
568 self.cycle = self
569
570 def __del__(self):
571 print("GC:",
572 threading.current_thread() is main_thread,
573 threading.main_thread() is main_thread,
574 threading.enumerate() == [main_thread])
575
576 RefCycle()
577 gc.collect() # sanity check
578 x = RefCycle()
579 """
580 _, out, err = assert_python_ok("-c", code)
581 data = out.decode()
582 self.assertEqual(err, b"")
583 self.assertEqual(data.splitlines(),
584 ["GC: True True True"] * 2)
585
Victor Stinnere40a97a2019-06-13 13:44:23 +0200586 def test_finalization_shutdown(self):
587 # bpo-36402: Py_Finalize() calls threading._shutdown() which must wait
588 # until Python thread states of all non-daemon threads get deleted.
589 #
590 # Test similar to SubinterpThreadingTests.test_threads_join_2(), but
591 # test the finalization of the main interpreter.
592 code = """if 1:
593 import os
594 import threading
595 import time
596 import random
597
598 def random_sleep():
599 seconds = random.random() * 0.010
600 time.sleep(seconds)
601
602 class Sleeper:
603 def __del__(self):
604 random_sleep()
605
606 tls = threading.local()
607
608 def f():
609 # Sleep a bit so that the thread is still running when
610 # Py_Finalize() is called.
611 random_sleep()
612 tls.x = Sleeper()
613 random_sleep()
614
615 threading.Thread(target=f).start()
616 random_sleep()
617 """
618 rc, out, err = assert_python_ok("-c", code)
619 self.assertEqual(err, b"")
620
Antoine Pitrou7b476992013-09-07 23:38:37 +0200621 def test_tstate_lock(self):
622 # Test an implementation detail of Thread objects.
623 started = _thread.allocate_lock()
624 finish = _thread.allocate_lock()
625 started.acquire()
626 finish.acquire()
627 def f():
628 started.release()
629 finish.acquire()
630 time.sleep(0.01)
631 # The tstate lock is None until the thread is started
632 t = threading.Thread(target=f)
633 self.assertIs(t._tstate_lock, None)
634 t.start()
635 started.acquire()
636 self.assertTrue(t.is_alive())
637 # The tstate lock can't be acquired when the thread is running
638 # (or suspended).
639 tstate_lock = t._tstate_lock
640 self.assertFalse(tstate_lock.acquire(timeout=0), False)
641 finish.release()
642 # When the thread ends, the state_lock can be successfully
643 # acquired.
644 self.assertTrue(tstate_lock.acquire(timeout=5), False)
645 # But is_alive() is still True: we hold _tstate_lock now, which
646 # prevents is_alive() from knowing the thread's end-of-life C code
647 # is done.
648 self.assertTrue(t.is_alive())
649 # Let is_alive() find out the C code is done.
650 tstate_lock.release()
651 self.assertFalse(t.is_alive())
652 # And verify the thread disposed of _tstate_lock.
Serhiy Storchaka8c0f0c52016-03-14 10:28:59 +0200653 self.assertIsNone(t._tstate_lock)
Victor Stinnerb8c7be22017-09-14 13:05:21 -0700654 t.join()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200655
Tim Peters72460fa2013-09-09 18:48:24 -0500656 def test_repr_stopped(self):
657 # Verify that "stopped" shows up in repr(Thread) appropriately.
658 started = _thread.allocate_lock()
659 finish = _thread.allocate_lock()
660 started.acquire()
661 finish.acquire()
662 def f():
663 started.release()
664 finish.acquire()
665 t = threading.Thread(target=f)
666 t.start()
667 started.acquire()
668 self.assertIn("started", repr(t))
669 finish.release()
670 # "stopped" should appear in the repr in a reasonable amount of time.
671 # Implementation detail: as of this writing, that's trivially true
672 # if .join() is called, and almost trivially true if .is_alive() is
673 # called. The detail we're testing here is that "stopped" shows up
674 # "all on its own".
675 LOOKING_FOR = "stopped"
676 for i in range(500):
677 if LOOKING_FOR in repr(t):
678 break
679 time.sleep(0.01)
680 self.assertIn(LOOKING_FOR, repr(t)) # we waited at least 5 seconds
Victor Stinnerb8c7be22017-09-14 13:05:21 -0700681 t.join()
Christian Heimes1af737c2008-01-23 08:24:23 +0000682
Tim Peters7634e1c2013-10-08 20:55:51 -0500683 def test_BoundedSemaphore_limit(self):
Tim Peters3d1b7a02013-10-08 21:29:27 -0500684 # BoundedSemaphore should raise ValueError if released too often.
685 for limit in range(1, 10):
686 bs = threading.BoundedSemaphore(limit)
687 threads = [threading.Thread(target=bs.acquire)
688 for _ in range(limit)]
689 for t in threads:
690 t.start()
691 for t in threads:
692 t.join()
693 threads = [threading.Thread(target=bs.release)
694 for _ in range(limit)]
695 for t in threads:
696 t.start()
697 for t in threads:
698 t.join()
699 self.assertRaises(ValueError, bs.release)
Tim Peters7634e1c2013-10-08 20:55:51 -0500700
Serhiy Storchakaf28ba362014-02-07 10:10:55 +0200701 @cpython_only
Victor Stinnerfdeb6ec2013-12-13 02:01:38 +0100702 def test_frame_tstate_tracing(self):
703 # Issue #14432: Crash when a generator is created in a C thread that is
704 # destroyed while the generator is still used. The issue was that a
705 # generator contains a frame, and the frame kept a reference to the
706 # Python state of the destroyed C thread. The crash occurs when a trace
707 # function is setup.
708
709 def noop_trace(frame, event, arg):
710 # no operation
711 return noop_trace
712
713 def generator():
714 while 1:
Berker Peksag4882cac2015-04-14 09:30:01 +0300715 yield "generator"
Victor Stinnerfdeb6ec2013-12-13 02:01:38 +0100716
717 def callback():
718 if callback.gen is None:
719 callback.gen = generator()
720 return next(callback.gen)
721 callback.gen = None
722
723 old_trace = sys.gettrace()
724 sys.settrace(noop_trace)
725 try:
726 # Install a trace function
727 threading.settrace(noop_trace)
728
729 # Create a generator in a C thread which exits after the call
Serhiy Storchakaf28ba362014-02-07 10:10:55 +0200730 import _testcapi
Victor Stinnerfdeb6ec2013-12-13 02:01:38 +0100731 _testcapi.call_in_temporary_c_thread(callback)
732
733 # Call the generator in a different Python thread, check that the
734 # generator didn't keep a reference to the destroyed thread state
735 for test in range(3):
736 # The trace function is still called here
737 callback()
738 finally:
739 sys.settrace(old_trace)
740
Victor Stinnere40a97a2019-06-13 13:44:23 +0200741 @cpython_only
742 def test_shutdown_locks(self):
743 for daemon in (False, True):
744 with self.subTest(daemon=daemon):
745 event = threading.Event()
746 thread = threading.Thread(target=event.wait, daemon=daemon)
747
748 # Thread.start() must add lock to _shutdown_locks,
749 # but only for non-daemon thread
750 thread.start()
751 tstate_lock = thread._tstate_lock
752 if not daemon:
753 self.assertIn(tstate_lock, threading._shutdown_locks)
754 else:
755 self.assertNotIn(tstate_lock, threading._shutdown_locks)
756
757 # unblock the thread and join it
758 event.set()
759 thread.join()
760
761 # Thread._stop() must remove tstate_lock from _shutdown_locks.
762 # Daemon threads must never add it to _shutdown_locks.
763 self.assertNotIn(tstate_lock, threading._shutdown_locks)
764
Victor Stinner45956b92013-11-12 16:37:55 +0100765
Antoine Pitroub0e9bd42009-10-27 20:05:26 +0000766class ThreadJoinOnShutdown(BaseTestCase):
Jesse Nollera8513972008-07-17 16:49:17 +0000767
768 def _run_and_join(self, script):
769 script = """if 1:
770 import sys, os, time, threading
771
772 # a thread, which waits for the main program to terminate
773 def joiningfunc(mainthread):
774 mainthread.join()
775 print('end of thread')
Antoine Pitrou5fe291f2008-09-06 23:00:03 +0000776 # stdout is fully buffered because not a tty, we have to flush
777 # before exit.
778 sys.stdout.flush()
Jesse Nollera8513972008-07-17 16:49:17 +0000779 \n""" + script
780
Antoine Pitrouc4d78642011-05-05 20:17:32 +0200781 rc, out, err = assert_python_ok("-c", script)
782 data = out.decode().replace('\r', '')
Benjamin Petersonad703dc2008-07-17 17:02:57 +0000783 self.assertEqual(data, "end of main\nend of thread\n")
Jesse Nollera8513972008-07-17 16:49:17 +0000784
785 def test_1_join_on_shutdown(self):
786 # The usual case: on exit, wait for a non-daemon thread
787 script = """if 1:
788 import os
789 t = threading.Thread(target=joiningfunc,
790 args=(threading.current_thread(),))
791 t.start()
792 time.sleep(0.1)
793 print('end of main')
794 """
795 self._run_and_join(script)
796
Alexandre Vassalotti93f2cd22009-07-22 04:54:52 +0000797 @unittest.skipUnless(hasattr(os, 'fork'), "needs os.fork()")
Victor Stinner26d31862011-07-01 14:26:24 +0200798 @unittest.skipIf(sys.platform in platforms_to_skip, "due to known OS bug")
Jesse Nollera8513972008-07-17 16:49:17 +0000799 def test_2_join_in_forked_process(self):
800 # Like the test above, but from a forked interpreter
Jesse Nollera8513972008-07-17 16:49:17 +0000801 script = """if 1:
802 childpid = os.fork()
803 if childpid != 0:
804 os.waitpid(childpid, 0)
805 sys.exit(0)
806
807 t = threading.Thread(target=joiningfunc,
808 args=(threading.current_thread(),))
809 t.start()
810 print('end of main')
811 """
812 self._run_and_join(script)
813
Alexandre Vassalotti93f2cd22009-07-22 04:54:52 +0000814 @unittest.skipUnless(hasattr(os, 'fork'), "needs os.fork()")
Victor Stinner26d31862011-07-01 14:26:24 +0200815 @unittest.skipIf(sys.platform in platforms_to_skip, "due to known OS bug")
Antoine Pitrou5fe291f2008-09-06 23:00:03 +0000816 def test_3_join_in_forked_from_thread(self):
Jesse Nollera8513972008-07-17 16:49:17 +0000817 # Like the test above, but fork() was called from a worker thread
818 # In the forked process, the main Thread object must be marked as stopped.
Alexandre Vassalotti93f2cd22009-07-22 04:54:52 +0000819
Jesse Nollera8513972008-07-17 16:49:17 +0000820 script = """if 1:
821 main_thread = threading.current_thread()
822 def worker():
823 childpid = os.fork()
824 if childpid != 0:
825 os.waitpid(childpid, 0)
826 sys.exit(0)
827
828 t = threading.Thread(target=joiningfunc,
829 args=(main_thread,))
830 print('end of main')
831 t.start()
832 t.join() # Should not block: main_thread is already stopped
833
834 w = threading.Thread(target=worker)
835 w.start()
836 """
837 self._run_and_join(script)
838
Victor Stinner26d31862011-07-01 14:26:24 +0200839 @unittest.skipIf(sys.platform in platforms_to_skip, "due to known OS bug")
Tim Petersc363a232013-09-08 18:44:40 -0500840 def test_4_daemon_threads(self):
Antoine Pitrou0d5e52d2011-05-04 20:02:30 +0200841 # Check that a daemon thread cannot crash the interpreter on shutdown
842 # by manipulating internal structures that are being disposed of in
843 # the main thread.
844 script = """if True:
845 import os
846 import random
847 import sys
848 import time
849 import threading
850
851 thread_has_run = set()
852
853 def random_io():
854 '''Loop for a while sleeping random tiny amounts and doing some I/O.'''
Antoine Pitrou0d5e52d2011-05-04 20:02:30 +0200855 while True:
Serhiy Storchaka5b10b982019-03-05 10:06:26 +0200856 with open(os.__file__, 'rb') as in_f:
857 stuff = in_f.read(200)
858 with open(os.devnull, 'wb') as null_f:
859 null_f.write(stuff)
860 time.sleep(random.random() / 1995)
Antoine Pitrou0d5e52d2011-05-04 20:02:30 +0200861 thread_has_run.add(threading.current_thread())
862
863 def main():
864 count = 0
865 for _ in range(40):
866 new_thread = threading.Thread(target=random_io)
867 new_thread.daemon = True
868 new_thread.start()
869 count += 1
870 while len(thread_has_run) < count:
871 time.sleep(0.001)
872 # Trigger process shutdown
873 sys.exit(0)
874
875 main()
876 """
877 rc, out, err = assert_python_ok('-c', script)
878 self.assertFalse(err)
879
Charles-François Natali6d0d24e2012-02-02 20:31:42 +0100880 @unittest.skipUnless(hasattr(os, 'fork'), "needs os.fork()")
Charles-François Natalib2c9e9a2012-02-08 21:29:11 +0100881 @unittest.skipIf(sys.platform in platforms_to_skip, "due to known OS bug")
Charles-François Natali6d0d24e2012-02-02 20:31:42 +0100882 def test_reinit_tls_after_fork(self):
883 # Issue #13817: fork() would deadlock in a multithreaded program with
884 # the ad-hoc TLS implementation.
885
886 def do_fork_and_wait():
887 # just fork a child process and wait it
888 pid = os.fork()
889 if pid > 0:
890 os.waitpid(pid, 0)
891 else:
892 os._exit(0)
893
894 # start a bunch of threads that will fork() child processes
895 threads = []
896 for i in range(16):
897 t = threading.Thread(target=do_fork_and_wait)
898 threads.append(t)
899 t.start()
900
901 for t in threads:
902 t.join()
903
Antoine Pitrou8408cea2013-05-05 23:47:09 +0200904 @unittest.skipUnless(hasattr(os, 'fork'), "needs os.fork()")
905 def test_clear_threads_states_after_fork(self):
906 # Issue #17094: check that threads states are cleared after fork()
907
908 # start a bunch of threads
909 threads = []
910 for i in range(16):
911 t = threading.Thread(target=lambda : time.sleep(0.3))
912 threads.append(t)
913 t.start()
914
915 pid = os.fork()
916 if pid == 0:
917 # check that threads states have been cleared
918 if len(sys._current_frames()) == 1:
919 os._exit(0)
920 else:
921 os._exit(1)
922 else:
923 _, status = os.waitpid(pid, 0)
924 self.assertEqual(0, status)
925
926 for t in threads:
927 t.join()
928
Jesse Nollera8513972008-07-17 16:49:17 +0000929
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200930class SubinterpThreadingTests(BaseTestCase):
931
932 def test_threads_join(self):
933 # Non-daemon threads should be joined at subinterpreter shutdown
934 # (issue #18808)
935 r, w = os.pipe()
936 self.addCleanup(os.close, r)
937 self.addCleanup(os.close, w)
938 code = r"""if 1:
939 import os
Victor Stinnere40a97a2019-06-13 13:44:23 +0200940 import random
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200941 import threading
942 import time
943
Victor Stinnere40a97a2019-06-13 13:44:23 +0200944 def random_sleep():
945 seconds = random.random() * 0.010
946 time.sleep(seconds)
947
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200948 def f():
949 # Sleep a bit so that the thread is still running when
950 # Py_EndInterpreter is called.
Victor Stinnere40a97a2019-06-13 13:44:23 +0200951 random_sleep()
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200952 os.write(%d, b"x")
Victor Stinnere40a97a2019-06-13 13:44:23 +0200953
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200954 threading.Thread(target=f).start()
Victor Stinnere40a97a2019-06-13 13:44:23 +0200955 random_sleep()
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200956 """ % (w,)
Victor Stinnered3b0bc2013-11-23 12:27:24 +0100957 ret = test.support.run_in_subinterp(code)
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +0200958 self.assertEqual(ret, 0)
959 # The thread was joined properly.
960 self.assertEqual(os.read(r, 1), b"x")
961
Antoine Pitrou7b476992013-09-07 23:38:37 +0200962 def test_threads_join_2(self):
963 # Same as above, but a delay gets introduced after the thread's
964 # Python code returned but before the thread state is deleted.
965 # To achieve this, we register a thread-local object which sleeps
966 # a bit when deallocated.
967 r, w = os.pipe()
968 self.addCleanup(os.close, r)
969 self.addCleanup(os.close, w)
970 code = r"""if 1:
971 import os
Victor Stinnere40a97a2019-06-13 13:44:23 +0200972 import random
Antoine Pitrou7b476992013-09-07 23:38:37 +0200973 import threading
974 import time
975
Victor Stinnere40a97a2019-06-13 13:44:23 +0200976 def random_sleep():
977 seconds = random.random() * 0.010
978 time.sleep(seconds)
979
Antoine Pitrou7b476992013-09-07 23:38:37 +0200980 class Sleeper:
981 def __del__(self):
Victor Stinnere40a97a2019-06-13 13:44:23 +0200982 random_sleep()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200983
984 tls = threading.local()
985
986 def f():
987 # Sleep a bit so that the thread is still running when
988 # Py_EndInterpreter is called.
Victor Stinnere40a97a2019-06-13 13:44:23 +0200989 random_sleep()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200990 tls.x = Sleeper()
991 os.write(%d, b"x")
Victor Stinnere40a97a2019-06-13 13:44:23 +0200992
Antoine Pitrou7b476992013-09-07 23:38:37 +0200993 threading.Thread(target=f).start()
Victor Stinnere40a97a2019-06-13 13:44:23 +0200994 random_sleep()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200995 """ % (w,)
Victor Stinnered3b0bc2013-11-23 12:27:24 +0100996 ret = test.support.run_in_subinterp(code)
Antoine Pitrou7b476992013-09-07 23:38:37 +0200997 self.assertEqual(ret, 0)
998 # The thread was joined properly.
999 self.assertEqual(os.read(r, 1), b"x")
1000
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02001001 @cpython_only
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +02001002 def test_daemon_threads_fatal_error(self):
1003 subinterp_code = r"""if 1:
1004 import os
1005 import threading
1006 import time
1007
1008 def f():
1009 # Make sure the daemon thread is still running when
1010 # Py_EndInterpreter is called.
1011 time.sleep(10)
1012 threading.Thread(target=f, daemon=True).start()
1013 """
1014 script = r"""if 1:
1015 import _testcapi
1016
1017 _testcapi.run_in_subinterp(%r)
1018 """ % (subinterp_code,)
Antoine Pitrou77e904e2013-10-08 23:04:32 +02001019 with test.support.SuppressCrashReport():
1020 rc, out, err = assert_python_failure("-c", script)
Antoine Pitrou7eaf3f72013-08-25 19:48:18 +02001021 self.assertIn("Fatal Python error: Py_EndInterpreter: "
1022 "not the last thread", err.decode())
1023
1024
Antoine Pitroub0e9bd42009-10-27 20:05:26 +00001025class ThreadingExceptionTests(BaseTestCase):
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001026 # A RuntimeError should be raised if Thread.start() is called
1027 # multiple times.
1028 def test_start_thread_again(self):
1029 thread = threading.Thread()
1030 thread.start()
1031 self.assertRaises(RuntimeError, thread.start)
Victor Stinnerb8c7be22017-09-14 13:05:21 -07001032 thread.join()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001033
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001034 def test_joining_current_thread(self):
Benjamin Peterson672b8032008-06-11 19:14:14 +00001035 current_thread = threading.current_thread()
1036 self.assertRaises(RuntimeError, current_thread.join);
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001037
1038 def test_joining_inactive_thread(self):
1039 thread = threading.Thread()
1040 self.assertRaises(RuntimeError, thread.join)
1041
1042 def test_daemonize_active_thread(self):
1043 thread = threading.Thread()
1044 thread.start()
Benjamin Petersonfdbea962008-08-18 17:33:47 +00001045 self.assertRaises(RuntimeError, setattr, thread, "daemon", True)
Victor Stinnerb8c7be22017-09-14 13:05:21 -07001046 thread.join()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001047
Antoine Pitroufcf81fd2011-02-28 22:03:34 +00001048 def test_releasing_unacquired_lock(self):
1049 lock = threading.Lock()
1050 self.assertRaises(RuntimeError, lock.release)
1051
Ned Deily9a7c5242011-05-28 00:19:56 -07001052 def test_recursion_limit(self):
1053 # Issue 9670
1054 # test that excessive recursion within a non-main thread causes
1055 # an exception rather than crashing the interpreter on platforms
1056 # like Mac OS X or FreeBSD which have small default stack sizes
1057 # for threads
1058 script = """if True:
1059 import threading
1060
1061 def recurse():
1062 return recurse()
1063
1064 def outer():
1065 try:
1066 recurse()
Yury Selivanovf488fb42015-07-03 01:04:23 -04001067 except RecursionError:
Ned Deily9a7c5242011-05-28 00:19:56 -07001068 pass
1069
1070 w = threading.Thread(target=outer)
1071 w.start()
1072 w.join()
1073 print('end of main thread')
1074 """
1075 expected_output = "end of main thread\n"
1076 p = subprocess.Popen([sys.executable, "-c", script],
Antoine Pitroub8b6a682012-06-29 19:40:35 +02001077 stdout=subprocess.PIPE, stderr=subprocess.PIPE)
Ned Deily9a7c5242011-05-28 00:19:56 -07001078 stdout, stderr = p.communicate()
1079 data = stdout.decode().replace('\r', '')
Antoine Pitroub8b6a682012-06-29 19:40:35 +02001080 self.assertEqual(p.returncode, 0, "Unexpected error: " + stderr.decode())
Ned Deily9a7c5242011-05-28 00:19:56 -07001081 self.assertEqual(data, expected_output)
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001082
Serhiy Storchaka52005c22014-09-21 22:08:13 +03001083 def test_print_exception(self):
1084 script = r"""if True:
1085 import threading
1086 import time
1087
1088 running = False
1089 def run():
1090 global running
1091 running = True
1092 while running:
1093 time.sleep(0.01)
1094 1/0
1095 t = threading.Thread(target=run)
1096 t.start()
1097 while not running:
1098 time.sleep(0.01)
1099 running = False
1100 t.join()
1101 """
1102 rc, out, err = assert_python_ok("-c", script)
1103 self.assertEqual(out, b'')
1104 err = err.decode()
1105 self.assertIn("Exception in thread", err)
1106 self.assertIn("Traceback (most recent call last):", err)
1107 self.assertIn("ZeroDivisionError", err)
1108 self.assertNotIn("Unhandled exception", err)
1109
Serhiy Storchakaa7930372016-07-03 22:27:26 +03001110 @requires_type_collecting
Serhiy Storchaka52005c22014-09-21 22:08:13 +03001111 def test_print_exception_stderr_is_none_1(self):
1112 script = r"""if True:
1113 import sys
1114 import threading
1115 import time
1116
1117 running = False
1118 def run():
1119 global running
1120 running = True
1121 while running:
1122 time.sleep(0.01)
1123 1/0
1124 t = threading.Thread(target=run)
1125 t.start()
1126 while not running:
1127 time.sleep(0.01)
1128 sys.stderr = None
1129 running = False
1130 t.join()
1131 """
1132 rc, out, err = assert_python_ok("-c", script)
1133 self.assertEqual(out, b'')
1134 err = err.decode()
1135 self.assertIn("Exception in thread", err)
1136 self.assertIn("Traceback (most recent call last):", err)
1137 self.assertIn("ZeroDivisionError", err)
1138 self.assertNotIn("Unhandled exception", err)
1139
1140 def test_print_exception_stderr_is_none_2(self):
1141 script = r"""if True:
1142 import sys
1143 import threading
1144 import time
1145
1146 running = False
1147 def run():
1148 global running
1149 running = True
1150 while running:
1151 time.sleep(0.01)
1152 1/0
1153 sys.stderr = None
1154 t = threading.Thread(target=run)
1155 t.start()
1156 while not running:
1157 time.sleep(0.01)
1158 running = False
1159 t.join()
1160 """
1161 rc, out, err = assert_python_ok("-c", script)
1162 self.assertEqual(out, b'')
1163 self.assertNotIn("Unhandled exception", err.decode())
1164
Victor Stinnereec93312016-08-18 18:13:10 +02001165 def test_bare_raise_in_brand_new_thread(self):
1166 def bare_raise():
1167 raise
1168
1169 class Issue27558(threading.Thread):
1170 exc = None
1171
1172 def run(self):
1173 try:
1174 bare_raise()
1175 except Exception as exc:
1176 self.exc = exc
1177
1178 thread = Issue27558()
1179 thread.start()
1180 thread.join()
1181 self.assertIsNotNone(thread.exc)
1182 self.assertIsInstance(thread.exc, RuntimeError)
Victor Stinner3d284c02017-08-19 01:54:42 +02001183 # explicitly break the reference cycle to not leak a dangling thread
1184 thread.exc = None
Serhiy Storchaka52005c22014-09-21 22:08:13 +03001185
Victor Stinnercd590a72019-05-28 00:39:52 +02001186
1187class ThreadRunFail(threading.Thread):
1188 def run(self):
1189 raise ValueError("run failed")
1190
1191
1192class ExceptHookTests(BaseTestCase):
1193 def test_excepthook(self):
1194 with support.captured_output("stderr") as stderr:
1195 thread = ThreadRunFail(name="excepthook thread")
1196 thread.start()
1197 thread.join()
1198
1199 stderr = stderr.getvalue().strip()
1200 self.assertIn(f'Exception in thread {thread.name}:\n', stderr)
1201 self.assertIn('Traceback (most recent call last):\n', stderr)
1202 self.assertIn(' raise ValueError("run failed")', stderr)
1203 self.assertIn('ValueError: run failed', stderr)
1204
1205 @support.cpython_only
1206 def test_excepthook_thread_None(self):
1207 # threading.excepthook called with thread=None: log the thread
1208 # identifier in this case.
1209 with support.captured_output("stderr") as stderr:
1210 try:
1211 raise ValueError("bug")
1212 except Exception as exc:
1213 args = threading.ExceptHookArgs([*sys.exc_info(), None])
Victor Stinnercdce0572019-06-02 23:08:41 +02001214 try:
1215 threading.excepthook(args)
1216 finally:
1217 # Explicitly break a reference cycle
1218 args = None
Victor Stinnercd590a72019-05-28 00:39:52 +02001219
1220 stderr = stderr.getvalue().strip()
1221 self.assertIn(f'Exception in thread {threading.get_ident()}:\n', stderr)
1222 self.assertIn('Traceback (most recent call last):\n', stderr)
1223 self.assertIn(' raise ValueError("bug")', stderr)
1224 self.assertIn('ValueError: bug', stderr)
1225
1226 def test_system_exit(self):
1227 class ThreadExit(threading.Thread):
1228 def run(self):
1229 sys.exit(1)
1230
1231 # threading.excepthook() silently ignores SystemExit
1232 with support.captured_output("stderr") as stderr:
1233 thread = ThreadExit()
1234 thread.start()
1235 thread.join()
1236
1237 self.assertEqual(stderr.getvalue(), '')
1238
1239 def test_custom_excepthook(self):
1240 args = None
1241
1242 def hook(hook_args):
1243 nonlocal args
1244 args = hook_args
1245
1246 try:
1247 with support.swap_attr(threading, 'excepthook', hook):
1248 thread = ThreadRunFail()
1249 thread.start()
1250 thread.join()
1251
1252 self.assertEqual(args.exc_type, ValueError)
1253 self.assertEqual(str(args.exc_value), 'run failed')
1254 self.assertEqual(args.exc_traceback, args.exc_value.__traceback__)
1255 self.assertIs(args.thread, thread)
1256 finally:
1257 # Break reference cycle
1258 args = None
1259
1260 def test_custom_excepthook_fail(self):
1261 def threading_hook(args):
1262 raise ValueError("threading_hook failed")
1263
1264 err_str = None
1265
1266 def sys_hook(exc_type, exc_value, exc_traceback):
1267 nonlocal err_str
1268 err_str = str(exc_value)
1269
1270 with support.swap_attr(threading, 'excepthook', threading_hook), \
1271 support.swap_attr(sys, 'excepthook', sys_hook), \
1272 support.captured_output('stderr') as stderr:
1273 thread = ThreadRunFail()
1274 thread.start()
1275 thread.join()
1276
1277 self.assertEqual(stderr.getvalue(),
1278 'Exception in threading.excepthook:\n')
1279 self.assertEqual(err_str, 'threading_hook failed')
1280
1281
R David Murray19aeb432013-03-30 17:19:38 -04001282class TimerTests(BaseTestCase):
1283
1284 def setUp(self):
1285 BaseTestCase.setUp(self)
1286 self.callback_args = []
1287 self.callback_event = threading.Event()
1288
1289 def test_init_immutable_default_args(self):
1290 # Issue 17435: constructor defaults were mutable objects, they could be
1291 # mutated via the object attributes and affect other Timer objects.
1292 timer1 = threading.Timer(0.01, self._callback_spy)
1293 timer1.start()
1294 self.callback_event.wait()
1295 timer1.args.append("blah")
1296 timer1.kwargs["foo"] = "bar"
1297 self.callback_event.clear()
1298 timer2 = threading.Timer(0.01, self._callback_spy)
1299 timer2.start()
1300 self.callback_event.wait()
1301 self.assertEqual(len(self.callback_args), 2)
1302 self.assertEqual(self.callback_args, [((), {}), ((), {})])
Victor Stinnerda3e5cf2017-09-15 05:37:42 -07001303 timer1.join()
1304 timer2.join()
R David Murray19aeb432013-03-30 17:19:38 -04001305
1306 def _callback_spy(self, *args, **kwargs):
1307 self.callback_args.append((args[:], kwargs.copy()))
1308 self.callback_event.set()
1309
Antoine Pitrou557934f2009-11-06 22:41:14 +00001310class LockTests(lock_tests.LockTests):
1311 locktype = staticmethod(threading.Lock)
1312
Antoine Pitrou434736a2009-11-10 18:46:01 +00001313class PyRLockTests(lock_tests.RLockTests):
1314 locktype = staticmethod(threading._PyRLock)
1315
Charles-François Natali6b671b22012-01-28 11:36:04 +01001316@unittest.skipIf(threading._CRLock is None, 'RLock not implemented in C')
Antoine Pitrou434736a2009-11-10 18:46:01 +00001317class CRLockTests(lock_tests.RLockTests):
1318 locktype = staticmethod(threading._CRLock)
Antoine Pitrou557934f2009-11-06 22:41:14 +00001319
1320class EventTests(lock_tests.EventTests):
1321 eventtype = staticmethod(threading.Event)
1322
1323class ConditionAsRLockTests(lock_tests.RLockTests):
Serhiy Storchaka6a7b3a72016-04-17 08:32:47 +03001324 # Condition uses an RLock by default and exports its API.
Antoine Pitrou557934f2009-11-06 22:41:14 +00001325 locktype = staticmethod(threading.Condition)
1326
1327class ConditionTests(lock_tests.ConditionTests):
1328 condtype = staticmethod(threading.Condition)
1329
1330class SemaphoreTests(lock_tests.SemaphoreTests):
1331 semtype = staticmethod(threading.Semaphore)
1332
1333class BoundedSemaphoreTests(lock_tests.BoundedSemaphoreTests):
1334 semtype = staticmethod(threading.BoundedSemaphore)
1335
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +00001336class BarrierTests(lock_tests.BarrierTests):
1337 barriertype = staticmethod(threading.Barrier)
Antoine Pitrou557934f2009-11-06 22:41:14 +00001338
Matěj Cepl608876b2019-05-23 22:30:00 +02001339
Martin Panter19e69c52015-11-14 12:46:42 +00001340class MiscTestCase(unittest.TestCase):
1341 def test__all__(self):
1342 extra = {"ThreadError"}
1343 blacklist = {'currentThread', 'activeCount'}
1344 support.check__all__(self, threading, ('threading', '_thread'),
1345 extra=extra, blacklist=blacklist)
1346
Matěj Cepl608876b2019-05-23 22:30:00 +02001347
1348class InterruptMainTests(unittest.TestCase):
1349 def test_interrupt_main_subthread(self):
1350 # Calling start_new_thread with a function that executes interrupt_main
1351 # should raise KeyboardInterrupt upon completion.
1352 def call_interrupt():
1353 _thread.interrupt_main()
1354 t = threading.Thread(target=call_interrupt)
1355 with self.assertRaises(KeyboardInterrupt):
1356 t.start()
1357 t.join()
1358 t.join()
1359
1360 def test_interrupt_main_mainthread(self):
1361 # Make sure that if interrupt_main is called in main thread that
1362 # KeyboardInterrupt is raised instantly.
1363 with self.assertRaises(KeyboardInterrupt):
1364 _thread.interrupt_main()
1365
1366 def test_interrupt_main_noerror(self):
1367 handler = signal.getsignal(signal.SIGINT)
1368 try:
1369 # No exception should arise.
1370 signal.signal(signal.SIGINT, signal.SIG_IGN)
1371 _thread.interrupt_main()
1372
1373 signal.signal(signal.SIGINT, signal.SIG_DFL)
1374 _thread.interrupt_main()
1375 finally:
1376 # Restore original handler
1377 signal.signal(signal.SIGINT, handler)
1378
1379
Tim Peters84d54892005-01-08 06:03:17 +00001380if __name__ == "__main__":
R David Murray19aeb432013-03-30 17:19:38 -04001381 unittest.main()