diff options
author | Victor Stinner <vstinner@python.org> | 2020-10-11 22:37:20 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-10-11 22:37:20 (GMT) |
commit | 13ff396c019d548ba181cf22c6f39309a300723c (patch) | |
tree | bb104dcbbfa9c1b8cc89a822f8a01b4b47e4c5bc /Lib/test/test_logging.py | |
parent | 47ecfd8030c391779dd9d3b4b07e6b01a8392481 (diff) | |
download | cpython-13ff396c019d548ba181cf22c6f39309a300723c.zip cpython-13ff396c019d548ba181cf22c6f39309a300723c.tar.gz cpython-13ff396c019d548ba181cf22c6f39309a300723c.tar.bz2 |
bpo-41739: Fix test_logging.test_race_between_set_target_and_flush() (GH-22655)
The test now waits until all threads complete to avoid leaking
running threads.
Also, use regular threads rather than daemon threads.
Diffstat (limited to 'Lib/test/test_logging.py')
-rw-r--r-- | Lib/test/test_logging.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/Lib/test/test_logging.py b/Lib/test/test_logging.py index 4cd8c7e..7c98e19 100644 --- a/Lib/test/test_logging.py +++ b/Lib/test/test_logging.py @@ -1164,22 +1164,27 @@ class MemoryHandlerTest(BaseTest): class MockRaceConditionHandler: def __init__(self, mem_hdlr): self.mem_hdlr = mem_hdlr + self.threads = [] def removeTarget(self): self.mem_hdlr.setTarget(None) def handle(self, msg): - t = threading.Thread(target=self.removeTarget) - t.daemon = True - t.start() + thread = threading.Thread(target=self.removeTarget) + self.threads.append(thread) + thread.start() target = MockRaceConditionHandler(self.mem_hdlr) - self.mem_hdlr.setTarget(target) + try: + self.mem_hdlr.setTarget(target) - for _ in range(10): - time.sleep(0.005) - self.mem_logger.info("not flushed") - self.mem_logger.warning("flushed") + for _ in range(10): + time.sleep(0.005) + self.mem_logger.info("not flushed") + self.mem_logger.warning("flushed") + finally: + for thread in target.threads: + threading_helper.join_thread(thread) class ExceptionFormatter(logging.Formatter): |