def test_process_queue_close_readers_explicitly_if_it_stops(get_reader): settings = {'foo': 'bar'} reader = get_reader.return_value streamer.process_queue(settings, 'gorillas', lambda m, s: None) reader.close.assert_called_once_with()
def test_process_queue_starts_reader(get_reader): settings = {'foo': 'bar'} reader = get_reader.return_value streamer.process_queue(settings, 'donkeys', lambda m, s: None) reader.start.assert_called_once_with(block=True)
def test_process_queue_connects_reader_on_message_to_process_message(process_message, get_reader): settings = {'foo': 'bar'} handler = Mock() reader = get_reader.return_value streamer.process_queue(settings, 'donkeys', handler) message_handler = reader.on_message.connect.call_args[1]['receiver'] message_handler(reader, 'message') process_message.assert_called_once_with(handler, reader, 'message')
def test_process_queue_creates_reader_for_topic(get_reader): settings = {'foo': 'bar'} streamer.process_queue(settings, 'donkeys', lambda m, s: None) get_reader.assert_any_call(settings, 'donkeys', ANY)