test_legacy.py 文件源码

def log_stream():
    stream = StringIO()
    handler = logging.StreamHandler(stream)
    logger = logging.getLogger('test_logger')
    logger.setLevel(logging.WARNING)
    logger.addHandler(handler)
    prior_logger = v1.logger
    v1.logger = logger
    yield stream
    v1.logger = prior_logger
    handler.close()


# test the retry logic with a custom logger