Function check_tracebacks

Lib/test/test_sqlite3/util.py:45–63

Convenience context manager for testing callback tracebacks.

(self, cm, exc, exc_regex, msg_regex, obj_name)

@contextlib.contextmanager
def check_tracebacks(self, cm, exc, exc_regex, msg_regex, obj_name):
    """Convenience context manager for testing callback tracebacks."""
    sqlite3.enable_callback_tracebacks(True)
    try:
        buf = io.StringIO()
        with contextlib.redirect_stderr(buf):
            yield

        self.assertEqual(cm.unraisable.exc_type, exc)
        if exc_regex:
            msg = str(cm.unraisable.exc_value)
            self.assertIsNotNone(exc_regex.search(msg), (exc_regex, msg))
        if msg_regex:
            msg = cm.unraisable.err_msg
            self.assertIsNotNone(msg_regex.search(msg), (msg_regex, msg))
        if obj_name:
            self.assertEqual(cm.unraisable.object.__name__, obj_name)
    finally:
        sqlite3.enable_callback_tracebacks(False)


class MemoryDatabaseMixin: