Как я могу исправить/макет logging.getlogger()

Question

Я этот код, который я хочу, чтобы тест:

log = logging.getLogger(__name__) 


class A(object): 
    def __init__(self): 
     log.debug('Init') 

, но я не могу понять, как утверждать, что log.debug была вызвана «Init»

Я пробовал патч-регистратор, но, проверяя его, я нашел только макет getLogger.

Я уверен, что он прост, но я просто не могу его изобразить!

Заранее благодарим за любую помощь!

Answer 1

Предполагая log является глобальным переменной в модуле mymod, вы хотите, чтобы дразнить фактический экземпляр, getLogger возвращенного, что и вызывает debug. Затем вы можете проверить, был ли вызван log.debug с правильным аргументом.

with mock.patch('mymod.log') as log_mock: 
    # test code 
    log_mock.debug.assert_called_with('Init') 

Answer 2

Альтернативное решение, которое позволяет убедиться, что вы используете правильный регистратор тоже:

logger = logging.getLogger('path.to.module.under.test') 
with mock.patch.object(logger, 'debug') as mock_debug: 
    run_code_under_test() 
    mock_debug.assert_called_once_with('Init')