def test_get_full_message_with_exception(trace):
"""Test getting message from exception
"""
record = Mock()
record.levelno = 50
record.exc_info = ['one', 'two']
trace.return_value = ['trace', 'two']
handler = logger.DSNHandler('dsn', 'qa', 'logging', '1.0.0')
assert handler.get_full_message(record)
trace.assert_called_once_with('one', 'two')
def test_get_full_message():
"""Test getting a full message.
"""
record = Mock()
message = 'Message'
record.msg = message
record.levelno = 10
record.exc_info = None
handler = logger.DSNHandler('dsn', 'qa', 'logging', '1.0.0')
assert handler.get_full_message(record) == message
def test_handler_creation():
"""Test the creation of the handler.
"""
dsn = 'dsn value'
environment = 'qa'
service_name = 'service_name'
service_version = '1.0.0'
handler = logger.DSNHandler(dsn, environment, service_name, service_version)
assert handler.dsn == dsn
assert handler.environment == environment
assert handler.service_name == service_name
assert handler.service_version == service_version
def test_emit_message_with_normal_exception(session):
"""Test emiting a message with a normal exception thrown from post request.
"""
session.post = Mock(side_effect=Exception())
record = Mock()
record.msg = 'Message'
record.levelno = 10
record.exc_info = None
record.correlation_id = uuid.uuid1()
handler = logger.DSNHandler('dsn', 'qa', 'logging', '1.0.0')
handler.handleError = Mock()
handler.emit(record)
handler.handleError.assert_called_with(record)
def test_session_post_called_on_message_emit(session):
"""Test session post called when a message is emitted.
"""
record = Mock()
message = 'Message'
correlation_id = uuid.uuid1()
record.levelno = 10
record.msg = message
record.created = datetime.now().timestamp()
record.exc_info = None
record.correlation_id = correlation_id
handler = logger.DSNHandler('dsn', 'qa', 'logging', '1.0.0')
handler.emit(record)
assert session.post.called
def test_emit_message_on_post_failure(session):
"""Test emiting a message with the session post triggering an exception.
"""
# SystemExit won't be an error that will be throw, it will just
# be an exception. To test this is
session.post = Mock(side_effect=SystemExit())
record = Mock()
record.msg = 'Message'
record.levelno = 10
record.created = datetime.now().timestamp()
record.exc_info = None
record.correlation_id = uuid.uuid1()
handler = logger.DSNHandler('dsn', 'qa', 'logging', '1.0.0')
with pytest.raises(SystemExit):
handler.emit(record)
