From 0518716df8d0f7a8d8908cba93eaaf79a83a154d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Krist=C3=B3f=20T=C3=B3th?= Date: Mon, 7 Oct 2019 13:10:37 +0200 Subject: [PATCH] Avoid waiting before first queued message --- .../frontend/message_queue_handler/message_queue_handler.py | 2 +- .../frontend/message_queue_handler/test_message_queue.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tfw/components/frontend/message_queue_handler/message_queue_handler.py b/tfw/components/frontend/message_queue_handler/message_queue_handler.py index 97f344b..60b5620 100644 --- a/tfw/components/frontend/message_queue_handler/message_queue_handler.py +++ b/tfw/components/frontend/message_queue_handler/message_queue_handler.py @@ -17,8 +17,8 @@ class MessageQueueHandler: for message in iter(self._queue.get, None): wpm = message['wpm'] if 'wpm' in message else self.wpm cps = 5 * wpm / 60 - sleep(len(message['message']) / cps) self.connector.send_message(message) + sleep(len(message['message']) / cps) def handle_event(self, message, _): for unpacked in self._generate_messages_from_queue(message): diff --git a/tfw/components/frontend/message_queue_handler/test_message_queue.py b/tfw/components/frontend/message_queue_handler/test_message_queue.py index 198c729..c0f59f3 100644 --- a/tfw/components/frontend/message_queue_handler/test_message_queue.py +++ b/tfw/components/frontend/message_queue_handler/test_message_queue.py @@ -60,7 +60,7 @@ def test_message_order(handler, queue): def test_wpm(handler, queue): handler.wpm = 10000 handler.connector.raise_event(queue) - assert not handler.connector.messages + assert len(handler.connector.messages) == 1 handler.wpm = 100000000 handler.connector.raise_event(queue) sleep(0.25)