A context manager for assertLogs() and assertNoLogs()
| 26 | |
| 27 | |
| 28 | class _AssertLogsContext(_BaseTestCaseContext): |
| 29 | """A context manager for assertLogs() and assertNoLogs() """ |
| 30 | |
| 31 | LOGGING_FORMAT = "%(levelname)s:%(name)s:%(message)s" |
| 32 | |
| 33 | def __init__(self, test_case, logger_name, level, no_logs, formatter=None): |
| 34 | _BaseTestCaseContext.__init__(self, test_case) |
| 35 | self.logger_name = logger_name |
| 36 | if level: |
| 37 | self.level = logging._nameToLevel.get(level, level) |
| 38 | else: |
| 39 | self.level = logging.INFO |
| 40 | self.msg = None |
| 41 | self.no_logs = no_logs |
| 42 | self.formatter = formatter |
| 43 | |
| 44 | def __enter__(self): |
| 45 | if isinstance(self.logger_name, logging.Logger): |
| 46 | logger = self.logger = self.logger_name |
| 47 | else: |
| 48 | logger = self.logger = logging.getLogger(self.logger_name) |
| 49 | formatter = self.formatter or logging.Formatter(self.LOGGING_FORMAT) |
| 50 | handler = _CapturingHandler() |
| 51 | handler.setLevel(self.level) |
| 52 | handler.setFormatter(formatter) |
| 53 | self.watcher = handler.watcher |
| 54 | self.old_handlers = logger.handlers[:] |
| 55 | self.old_level = logger.level |
| 56 | self.old_propagate = logger.propagate |
| 57 | logger.handlers = [handler] |
| 58 | logger.setLevel(self.level) |
| 59 | logger.propagate = False |
| 60 | if self.no_logs: |
| 61 | return |
| 62 | return handler.watcher |
| 63 | |
| 64 | def __exit__(self, exc_type, exc_value, tb): |
| 65 | self.logger.handlers = self.old_handlers |
| 66 | self.logger.propagate = self.old_propagate |
| 67 | self.logger.setLevel(self.old_level) |
| 68 | |
| 69 | if exc_type is not None: |
| 70 | # let unexpected exceptions pass through |
| 71 | return False |
| 72 | |
| 73 | if self.no_logs: |
| 74 | # assertNoLogs |
| 75 | if len(self.watcher.records) > 0: |
| 76 | self._raiseFailure( |
| 77 | "Unexpected logs found: {!r}".format( |
| 78 | self.watcher.output |
| 79 | ) |
| 80 | ) |
| 81 | |
| 82 | else: |
| 83 | # assertLogs |
| 84 | if len(self.watcher.records) == 0: |
| 85 | self._raiseFailure( |
no outgoing calls
no test coverage detected
searching dependent graphs…