def test_flush_when_unsupported_by_stream():
'''Test flush executes without error when unsupported by stream.'''
target = Mock()
stream = Stream(
stream=target,
)
stream.flush()
def test_teardown_open_stream():
'''Teardown open stream.'''
target = StringIO()
stream = Stream(stream=target)
stream.output(['test'])
stream.teardown()
assert not target.closed
assert target.getvalue() == 'test'
def test_flush_when_supported_by_stream():
'''Test flush when supported by stream.'''
target = Mock()
target.flush = Mock()
stream = Stream(
stream=target,
)
stream.flush()
assert target.flush.called
def test_output():
'''Test outputting to specified stream.'''
target = StringIO()
formatter = Mock()
formatter.format = Mock(return_value='A message')
stream = Stream(
stream=target,
formatter=formatter
)
log = Log(level='info', message='A message')
stream.handle(log)
assert target.getvalue() == 'A message'
def test_teardown_closed_stream():
'''Teardown closed stream without error.'''
target = StringIO()
stream = Stream(stream=target)
target.close()
stream.teardown()