def test_log(self, send_message_mock):
component = Component(input_stream=BytesIO(), output_stream=BytesIO())
inputs = [("I am a robot monkey.", None, 2),
("I am a monkey who learned to talk.", 'warning', 3)]
for msg, level, storm_level in inputs:
component.output_stream.close()
component.output_stream = BytesIO()
component.log(msg, level=level)
send_message_mock.assert_called_with(component, {'command': 'log',
'msg': msg,
'level': storm_level})
def test_log(self, send_message_mock):
component = Component(input_stream=BytesIO(), output_stream=BytesIO())
inputs = [("I am a robot monkey.", None, 2),
("I am a monkey who learned to talk.", 'warning', 3)]
for msg, level, storm_level in inputs:
component.output_stream.close()
component.output_stream = component._wrap_stream(BytesIO())
component.log(msg, level=level)
send_message_mock.assert_called_with(component, {'command': 'log',
'msg': msg,
'level': storm_level})
