def test_add_logging_context(self):
"""Test that :py:class:`add_logging_context` works."""
for method_name in LOG_METHOD_NAMES:
with self.subTest(name=f"logger.{method_name}"):
self.gl(method_name)("1")
self.gb(method_name).assert_called_with("1",
extra={},
stacklevel=3)
with add_logging_context(a=1):
self.gl(method_name)("2")
self.gb(method_name).assert_called_with("2",
extra=dict(a=1),
stacklevel=3)
with add_logging_context(b=2):
self.gl(method_name)("3")
self.gb(method_name).assert_called_with("3",
extra=dict(
a=1, b=2),
stacklevel=3)
with add_logging_context(a=4, c=3):
self.gl(method_name)("4")
self.gb(method_name).assert_called_with(
"4", extra=dict(a=4, b=2, c=3), stacklevel=3)
self.gl(method_name)("5")
self.gb(method_name).assert_called_with("5",
extra=dict(
a=1, b=2),
stacklevel=3)
self.gl(method_name)("6")
self.gb(method_name).assert_called_with("6",
extra=dict(a=1),
stacklevel=3)
self.gl(method_name)("7")
self.gb(method_name).assert_called_with("7",
extra={},
stacklevel=3)
self.gl(method_name)("8")
self.gb(method_name).assert_called_with("8",
extra={},
stacklevel=3)
def thread_worker(val):
self.logger.debug("1")
self.base_logger.debug.assert_called_with("1",
extra={},
stacklevel=3)
with add_logging_context(val=val):
self.logger.debug("2")
self.base_logger.debug.assert_called_with("2",
extra=dict(val=val),
stacklevel=3)
def process_worker(val):
"""See how our logger interacts with multiprocessing."""
base_logger = unittest.mock.Mock(spec=LOGGER)
logger = Logger(logger=base_logger)
logger.debug("1")
base_logger.debug.assert_called_with("1", extra={}, stacklevel=3)
with add_logging_context(val=val):
logger.debug("2")
base_logger.debug.assert_called_with("2",
extra=dict(val=val),
stacklevel=3)
def test_thread_local(self):
"""Test that our logging context is indeed thread-local."""
def thread_worker(val):
self.logger.debug("1")
self.base_logger.debug.assert_called_with("1",
extra={},
stacklevel=3)
with add_logging_context(val=val):
self.logger.debug("2")
self.base_logger.debug.assert_called_with("2",
extra=dict(val=val),
stacklevel=3)
with concurrent.futures.ThreadPoolExecutor(max_workers=2) as exc:
with add_logging_context(a=1):
list(exc.map(thread_worker, [1, 2]))
def test_process_local(self):
"""Test how our logger works with multiple processes."""
with concurrent.futures.ProcessPoolExecutor(max_workers=2) as exc:
with add_logging_context(a=1):
list(exc.map(process_worker, [1, 2]))