<h2>Editing File: /lib64/python3.12/unittest/_log.py</h2><form method='post'><textarea name='content' class='file-edit'>import logging
import collections

from .case import _BaseTestCaseContext


_LoggingWatcher = collections.namedtuple(&quot;_LoggingWatcher&quot;,
                                         [&quot;records&quot;, &quot;output&quot;])

class _CapturingHandler(logging.Handler):
    &quot;&quot;&quot;
    A logging handler capturing all (raw and formatted) logging output.
    &quot;&quot;&quot;

    def __init__(self):
        logging.Handler.__init__(self)
        self.watcher = _LoggingWatcher([], [])

    def flush(self):
        pass

    def emit(self, record):
        self.watcher.records.append(record)
        msg = self.format(record)
        self.watcher.output.append(msg)


class _AssertLogsContext(_BaseTestCaseContext):
    &quot;&quot;&quot;A context manager for assertLogs() and assertNoLogs() &quot;&quot;&quot;

    LOGGING_FORMAT = &quot;%(levelname)s:%(name)s:%(message)s&quot;

    def __init__(self, test_case, logger_name, level, no_logs):
        _BaseTestCaseContext.__init__(self, test_case)
        self.logger_name = logger_name
        if level:
            self.level = logging._nameToLevel.get(level, level)
        else:
            self.level = logging.INFO
        self.msg = None
        self.no_logs = no_logs

    def __enter__(self):
        if isinstance(self.logger_name, logging.Logger):
            logger = self.logger = self.logger_name
        else:
            logger = self.logger = logging.getLogger(self.logger_name)
        formatter = logging.Formatter(self.LOGGING_FORMAT)
        handler = _CapturingHandler()
        handler.setLevel(self.level)
        handler.setFormatter(formatter)
        self.watcher = handler.watcher
        self.old_handlers = logger.handlers[:]
        self.old_level = logger.level
        self.old_propagate = logger.propagate
        logger.handlers = [handler]
        logger.setLevel(self.level)
        logger.propagate = False
        if self.no_logs:
            return
        return handler.watcher

    def __exit__(self, exc_type, exc_value, tb):
        self.logger.handlers = self.old_handlers
        self.logger.propagate = self.old_propagate
        self.logger.setLevel(self.old_level)

        if exc_type is not None:
            # let unexpected exceptions pass through
            return False

        if self.no_logs:
            # assertNoLogs
            if len(self.watcher.records) &gt; 0:
                self._raiseFailure(
                    &quot;Unexpected logs found: {!r}&quot;.format(
                        self.watcher.output
                    )
                )

        else:
            # assertLogs
            if len(self.watcher.records) == 0:
                self._raiseFailure(
                    &quot;no logs of level {} or higher triggered on {}&quot;
                    .format(logging.getLevelName(self.level), self.logger.name))
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Flib64%2Fpython3.12%2Funittest'>Back to Directory</a>