def test_log():
'''Test log method emits correct message.'''
now = 1234456789
with nested(
mock.patch(
'sawmill.logger.audit.getpass',
**{'getuser.return_value': 'thesociable'}
),
mock.patch(
'sawmill.logger.audit.time',
**{'time.return_value': now}
)
):
handler = mock.Mock()
logger = Classic(
'sawmill.test.classic',
note='A note',
_handler=handler
)
logger.log('A message', extra_info_a='Extra A')
assert handler.handle.called
log = handler.handle.call_args[0][0]
assert log == Log(
name='sawmill.test.classic',
note='A note',
message='A message',
extra_info_a='Extra A',
timestamp=now,
username='******'
)
def test_level_helper(level):
'''Test level helper correctly sets level key.'''
logger = Classic('sawmill.test.classic')
logger.log = mock.Mock()
message = 'A {0} level message.'.format(level)
getattr(logger, level)(message)
assert logger.log.call_args[0] == (message, )
assert logger.log.call_args[1] == {'level': level}