在unittest期间禁用python日志记录

这是一个 context manager

from contextlib import contextmanager

@contextmanager
def disable_logger(name):
    """Temporarily disable a specific logger."""
    logger = logging.getLogger(name)
    old_value = logger.disabled
    logger.disabled = True
    try:
        yield
    finally:
        logger.disabled = old_value

然后,您可以这样使用它:

class MyTest(unittest.TestCase):

   def test_do_something(self):
      with disable_logger('mypackage.mymodule'):
          mymodule.do_something()

我看到了两种处理方法:要么删除处理程序,要么从不创建它。

首先,从中的记录器中删除处理程序 test_function

def test_function(self):
    tr = TestRun()
    tr.logger.handlers = []

全部的 但是它应该解决这个问题。

第二种方法是首先通过向添加关键字参数来控制是否生成日志 TestRun.__init__

class TestRun:
    def __init__(self, file_logs = True):
        self.test_run_id = str(uuid4())
        self.logger = logging.getLogger(__name__)
        self.logger.setLevel(logging.INFO)
        if file_logs:
            handler = logging.FileHandler('{}.log'.format(self.test_run_id))
            self.logger.addHandler(handler)

tr = TestRun(file_logs = False) .

这应该可以做到:

logging.getLogger(__name__).disabled = True