Вход в Python для обработчика StringIO - PullRequest
Новая
       18

Вход в Python для обработчика StringIO

13 голосов
/ 02 марта 2012

У меня есть тест на python, в котором я хочу проверить, работает ли логирование должным образом.Например, у меня есть функция, которая создает пользователя, и в конце записи журнала записывает в файл журнала ответ. 

logger = logging.getLogger('mylogger')
logger.setLevel(logging.DEBUG)
handler = logging.handlers.WatchedFileHandler('mylogfile.log')
formatter = logging.Formatter('%(asctime)s: %(message)s',
                              '%d/%b/%Y:%H:%M:%S %z')
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.info('Some log text')

В моем тестовом примере я хочу отправить вывод журнала в StringIO. 

class MyTest(unittest.TestCase):
    def setUp(self):
        stream = StringIO()
        self.handler = logging.StreamHandler(stream)
        log = logging.getLogger('mylogger')
        log.removeHandler(log.handlers[0])
        log.addHandler(self.handler)

    def tearDown(self):
        log = logging.getLogger('mylogger')
        log.removeHandler(self.handler)
        self.handler.close()

Проблема в том, что я не уверен, как мне проверить, работает ли мой регистратор или нет.

1 Ответ

20 голосов
/ 02 марта 2012

Вот пример, который работает, убедитесь, что вы установили уровень вашего журнала, и очистите буфер.

class MyTest(unittest.TestCase):
    def setUp(self):
        self.stream = StringIO()
        self.handler = logging.StreamHandler(self.stream)
        self.log = logging.getLogger('mylogger')
        self.log.setLevel(logging.INFO)
        for handler in self.log.handlers: 
            self.log.removeHandler(handler)
        self.log.addHandler(self.handler)
    def testLog(self):
        self.log.info("test")
        self.handler.flush()
        print '[', self.stream.getvalue(), ']'
        self.assertEqual(self.stream.getvalue(), 'test')

    def tearDown(self):
        self.log.removeHandler(self.handler)
        self.handler.close()

if __name__=='__main__':
    unittest.main()

Далее читаем, вот пример Временная запись Python Logging в строковый буфер , который показывает, как вы также можете выполнять форматирование.

...