Kaydet (Commit) 93e6b331 authored tarafından Vinay Sajip's avatar Vinay Sajip

Closes #27930: Merged fix from 3.5.

...@@ -1388,7 +1388,6 @@ if threading: ...@@ -1388,7 +1388,6 @@ if threading:
""" """
self.queue = queue self.queue = queue
self.handlers = handlers self.handlers = handlers
self._stop = threading.Event()
self._thread = None self._thread = None
self.respect_handler_level = respect_handler_level self.respect_handler_level = respect_handler_level
...@@ -1409,7 +1408,7 @@ if threading: ...@@ -1409,7 +1408,7 @@ if threading:
LogRecords to process. LogRecords to process.
""" """
self._thread = t = threading.Thread(target=self._monitor) self._thread = t = threading.Thread(target=self._monitor)
t.setDaemon(True) t.daemon = True
t.start() t.start()
def prepare(self , record): def prepare(self , record):
...@@ -1448,20 +1447,9 @@ if threading: ...@@ -1448,20 +1447,9 @@ if threading:
""" """
q = self.queue q = self.queue
has_task_done = hasattr(q, 'task_done') has_task_done = hasattr(q, 'task_done')
while not self._stop.isSet():
try:
record = self.dequeue(True)
if record is self._sentinel:
break
self.handle(record)
if has_task_done:
q.task_done()
except queue.Empty:
pass
# There might still be records in the queue.
while True: while True:
try: try:
record = self.dequeue(False) record = self.dequeue(True)
if record is self._sentinel: if record is self._sentinel:
break break
self.handle(record) self.handle(record)
...@@ -1488,7 +1476,6 @@ if threading: ...@@ -1488,7 +1476,6 @@ if threading:
Note that if you don't call this before your application exits, there Note that if you don't call this before your application exits, there
may be some records still left on the queue, which won't be processed. may be some records still left on the queue, which won't be processed.
""" """
self._stop.set()
self.enqueue_sentinel() self.enqueue_sentinel()
self._thread.join() self._thread.join()
self._thread = None self._thread = None
# Copyright 2001-2014 by Vinay Sajip. All Rights Reserved. # Copyright 2001-2016 by Vinay Sajip. All Rights Reserved.
# #
# Permission to use, copy, modify, and distribute this software and its # Permission to use, copy, modify, and distribute this software and its
# documentation for any purpose and without fee is hereby granted, # documentation for any purpose and without fee is hereby granted,
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
"""Test harness for the logging module. Run all tests. """Test harness for the logging module. Run all tests.
Copyright (C) 2001-2014 Vinay Sajip. All Rights Reserved. Copyright (C) 2001-2016 Vinay Sajip. All Rights Reserved.
""" """
import logging import logging
...@@ -26,7 +26,6 @@ import logging.config ...@@ -26,7 +26,6 @@ import logging.config
import codecs import codecs
import configparser import configparser
import datetime import datetime
import pathlib
import pickle import pickle
import io import io
import gc import gc
...@@ -309,10 +308,6 @@ class BuiltinLevelsTest(BaseTest): ...@@ -309,10 +308,6 @@ class BuiltinLevelsTest(BaseTest):
self.assertEqual(logging.getLevelName('INFO'), logging.INFO) self.assertEqual(logging.getLevelName('INFO'), logging.INFO)
self.assertEqual(logging.getLevelName(logging.INFO), 'INFO') self.assertEqual(logging.getLevelName(logging.INFO), 'INFO')
def test_issue27935(self):
fatal = logging.getLevelName('FATAL')
self.assertEqual(fatal, logging.FATAL)
class BasicFilterTest(BaseTest): class BasicFilterTest(BaseTest):
"""Test the bundled Filter class.""" """Test the bundled Filter class."""
...@@ -580,29 +575,6 @@ class HandlerTest(BaseTest): ...@@ -580,29 +575,6 @@ class HandlerTest(BaseTest):
self.assertFalse(h.shouldFlush(r)) self.assertFalse(h.shouldFlush(r))
h.close() h.close()
def test_path_objects(self):
"""
Test that Path objects are accepted as filename arguments to handlers.
See Issue #27493.
"""
fd, fn = tempfile.mkstemp()
os.close(fd)
os.unlink(fn)
pfn = pathlib.Path(fn)
cases = (
(logging.FileHandler, (pfn, 'w')),
(logging.handlers.RotatingFileHandler, (pfn, 'a')),
(logging.handlers.TimedRotatingFileHandler, (pfn, 'h')),
)
if sys.platform in ('linux', 'darwin'):
cases += ((logging.handlers.WatchedFileHandler, (pfn, 'w')),)
for cls, args in cases:
h = cls(*args)
self.assertTrue(os.path.exists(fn))
h.close()
os.unlink(fn)
@unittest.skipIf(os.name == 'nt', 'WatchedFileHandler not appropriate for Windows.') @unittest.skipIf(os.name == 'nt', 'WatchedFileHandler not appropriate for Windows.')
@unittest.skipUnless(threading, 'Threading required for this test.') @unittest.skipUnless(threading, 'Threading required for this test.')
def test_race(self): def test_race(self):
...@@ -986,7 +958,7 @@ class MemoryHandlerTest(BaseTest): ...@@ -986,7 +958,7 @@ class MemoryHandlerTest(BaseTest):
def setUp(self): def setUp(self):
BaseTest.setUp(self) BaseTest.setUp(self)
self.mem_hdlr = logging.handlers.MemoryHandler(10, logging.WARNING, self.mem_hdlr = logging.handlers.MemoryHandler(10, logging.WARNING,
self.root_hdlr) self.root_hdlr)
self.mem_logger = logging.getLogger('mem') self.mem_logger = logging.getLogger('mem')
self.mem_logger.propagate = 0 self.mem_logger.propagate = 0
self.mem_logger.addHandler(self.mem_hdlr) self.mem_logger.addHandler(self.mem_hdlr)
...@@ -1023,36 +995,6 @@ class MemoryHandlerTest(BaseTest): ...@@ -1023,36 +995,6 @@ class MemoryHandlerTest(BaseTest):
self.mem_logger.debug(self.next_message()) self.mem_logger.debug(self.next_message())
self.assert_log_lines(lines) self.assert_log_lines(lines)
def test_flush_on_close(self):
"""
Test that the flush-on-close configuration works as expected.
"""
self.mem_logger.debug(self.next_message())
self.assert_log_lines([])
self.mem_logger.info(self.next_message())
self.assert_log_lines([])
self.mem_logger.removeHandler(self.mem_hdlr)
# Default behaviour is to flush on close. Check that it happens.
self.mem_hdlr.close()
lines = [
('DEBUG', '1'),
('INFO', '2'),
]
self.assert_log_lines(lines)
# Now configure for flushing not to be done on close.
self.mem_hdlr = logging.handlers.MemoryHandler(10, logging.WARNING,
self.root_hdlr,
False)
self.mem_logger.addHandler(self.mem_hdlr)
self.mem_logger.debug(self.next_message())
self.assert_log_lines(lines) # no change
self.mem_logger.info(self.next_message())
self.assert_log_lines(lines) # no change
self.mem_logger.removeHandler(self.mem_hdlr)
self.mem_hdlr.close()
# assert that no new lines have been added
self.assert_log_lines(lines) # no change
class ExceptionFormatter(logging.Formatter): class ExceptionFormatter(logging.Formatter):
"""A special exception formatter.""" """A special exception formatter."""
...@@ -3080,6 +3022,84 @@ class QueueHandlerTest(BaseTest): ...@@ -3080,6 +3022,84 @@ class QueueHandlerTest(BaseTest):
self.assertFalse(handler.matches(levelno=logging.ERROR, message='5')) self.assertFalse(handler.matches(levelno=logging.ERROR, message='5'))
self.assertTrue(handler.matches(levelno=logging.CRITICAL, message='6')) self.assertTrue(handler.matches(levelno=logging.CRITICAL, message='6'))
if hasattr(logging.handlers, 'QueueListener'):
import multiprocessing
from unittest.mock import patch
class QueueListenerTest(BaseTest):
"""
Tests based on patch submitted for issue #27930. Ensure that
QueueListener handles all log messages.
"""
repeat = 20
@staticmethod
def setup_and_log(log_queue, ident):
"""
Creates a logger with a QueueHandler that logs to a queue read by a
QueueListener. Starts the listener, logs five messages, and stops
the listener.
"""
logger = logging.getLogger('test_logger_with_id_%s' % ident)
logger.setLevel(logging.DEBUG)
handler = logging.handlers.QueueHandler(log_queue)
logger.addHandler(handler)
listener = logging.handlers.QueueListener(log_queue)
listener.start()
logger.info('one')
logger.info('two')
logger.info('three')
logger.info('four')
logger.info('five')
listener.stop()
logger.removeHandler(handler)
handler.close()
@patch.object(logging.handlers.QueueListener, 'handle')
def test_handle_called_with_queue_queue(self, mock_handle):
for i in range(self.repeat):
log_queue = queue.Queue()
self.setup_and_log(log_queue, '%s_%s' % (self.id(), i))
self.assertEqual(mock_handle.call_count, 5 * self.repeat,
'correct number of handled log messages')
@patch.object(logging.handlers.QueueListener, 'handle')
def test_handle_called_with_mp_queue(self, mock_handle):
for i in range(self.repeat):
log_queue = multiprocessing.Queue()
self.setup_and_log(log_queue, '%s_%s' % (self.id(), i))
self.assertEqual(mock_handle.call_count, 5 * self.repeat,
'correct number of handled log messages')
@staticmethod
def get_all_from_queue(log_queue):
try:
while True:
yield log_queue.get_nowait()
except queue.Empty:
return []
def test_no_messages_in_queue_after_stop(self):
"""
Five messages are logged then the QueueListener is stopped. This
test then gets everything off the queue. Failure of this test
indicates that messages were not registered on the queue until
_after_ the QueueListener stopped.
"""
for i in range(self.repeat):
queue = multiprocessing.Queue()
self.setup_and_log(queue, '%s_%s' %(self.id(), i))
# time.sleep(1)
items = list(self.get_all_from_queue(queue))
expected = [[], [logging.handlers.QueueListener._sentinel]]
self.assertIn(items, expected,
'Found unexpected messages in queue: %s' % (
[m.msg if isinstance(m, logging.LogRecord)
else m for m in items]))
ZERO = datetime.timedelta(0) ZERO = datetime.timedelta(0)
...@@ -4219,23 +4239,12 @@ class NTEventLogHandlerTest(BaseTest): ...@@ -4219,23 +4239,12 @@ class NTEventLogHandlerTest(BaseTest):
msg = 'Record not found in event log, went back %d records' % GO_BACK msg = 'Record not found in event log, went back %d records' % GO_BACK
self.assertTrue(found, msg=msg) self.assertTrue(found, msg=msg)
class MiscTestCase(unittest.TestCase):
def test__all__(self):
blacklist = {'logThreads', 'logMultiprocessing',
'logProcesses', 'currentframe',
'PercentStyle', 'StrFormatStyle', 'StringTemplateStyle',
'Filterer', 'PlaceHolder', 'Manager', 'RootLogger',
'root'}
support.check__all__(self, logging, blacklist=blacklist)
# Set the locale to the platform-dependent default. I have no idea # Set the locale to the platform-dependent default. I have no idea
# why the test does this, but in any case we save the current locale # why the test does this, but in any case we save the current locale
# first and restore it at the end. # first and restore it at the end.
@support.run_with_locale('LC_ALL', '') @support.run_with_locale('LC_ALL', '')
def test_main(): def test_main():
support.run_unittest( tests = [
BuiltinLevelsTest, BasicFilterTest, CustomLevelsAndFiltersTest, BuiltinLevelsTest, BasicFilterTest, CustomLevelsAndFiltersTest,
HandlerTest, MemoryHandlerTest, ConfigFileTest, SocketHandlerTest, HandlerTest, MemoryHandlerTest, ConfigFileTest, SocketHandlerTest,
DatagramHandlerTest, MemoryTest, EncodingTest, WarningsTest, DatagramHandlerTest, MemoryTest, EncodingTest, WarningsTest,
...@@ -4247,7 +4256,10 @@ def test_main(): ...@@ -4247,7 +4256,10 @@ def test_main():
ExceptionTest, SysLogHandlerTest, HTTPHandlerTest, ExceptionTest, SysLogHandlerTest, HTTPHandlerTest,
NTEventLogHandlerTest, TimedRotatingFileHandlerTest, NTEventLogHandlerTest, TimedRotatingFileHandlerTest,
UnixSocketHandlerTest, UnixDatagramHandlerTest, UnixSysLogHandlerTest, UnixSocketHandlerTest, UnixDatagramHandlerTest, UnixSysLogHandlerTest,
MiscTestCase) ]
if hasattr(logging.handlers, 'QueueListener'):
tests.append(QueueListenerTest)
support.run_unittest(*tests)
if __name__ == "__main__": if __name__ == "__main__":
test_main() test_main()
...@@ -225,6 +225,9 @@ Library ...@@ -225,6 +225,9 @@ Library
- Issue #27573: exit message for code.interact is now configurable. - Issue #27573: exit message for code.interact is now configurable.
- Issue #27930: Improved behaviour of logging.handlers.QueueListener.
Thanks to Paulo Andrade and Petr Viktorin for the analysis and patch.
C API C API
----- -----
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment