You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
import unittest
|
|
from dataclasses import dataclass
|
|
|
|
from inputremapper.gui.message_broker import MessageBroker, MessageType
|
|
|
|
|
|
class Listener:
|
|
def __init__(self):
|
|
self.calls = []
|
|
|
|
def __call__(self, data):
|
|
self.calls.append(data)
|
|
|
|
|
|
@dataclass
|
|
class Message:
|
|
message_type: MessageType
|
|
msg: str
|
|
|
|
|
|
class TestMessageBroker(unittest.TestCase):
|
|
def test_calls_listeners(self):
|
|
"""The correct Listeners get called"""
|
|
message_broker = MessageBroker()
|
|
listener = Listener()
|
|
message_broker.subscribe(MessageType.test1, listener)
|
|
message_broker.send(Message(MessageType.test1, "foo"))
|
|
message_broker.send(Message(MessageType.test2, "bar"))
|
|
self.assertEqual(listener.calls[0], Message(MessageType.test1, "foo"))
|
|
|
|
def test_unsubscribe(self):
|
|
message_broker = MessageBroker()
|
|
listener = Listener()
|
|
message_broker.subscribe(MessageType.test1, listener)
|
|
message_broker.send(Message(MessageType.test1, "a"))
|
|
message_broker.unsubscribe(listener)
|
|
message_broker.send(Message(MessageType.test1, "b"))
|
|
self.assertEqual(len(listener.calls), 1)
|
|
self.assertEqual(listener.calls[0], Message(MessageType.test1, "a"))
|
|
|
|
def test_unsubscribe_unknown_listener(self):
|
|
"""nothing happens if we unsubscribe an unknown listener"""
|
|
message_broker = MessageBroker()
|
|
listener1 = Listener()
|
|
listener2 = Listener()
|
|
message_broker.subscribe(MessageType.test1, listener1)
|
|
message_broker.unsubscribe(listener2)
|
|
message_broker.send(Message(MessageType.test1, "a"))
|
|
self.assertEqual(listener1.calls[0], Message(MessageType.test1, "a"))
|
|
|
|
def test_preserves_order(self):
|
|
message_broker = MessageBroker()
|
|
calls = []
|
|
|
|
def listener1(_):
|
|
message_broker.send(Message(MessageType.test2, "f"))
|
|
calls.append(1)
|
|
|
|
def listener2(_):
|
|
message_broker.send(Message(MessageType.test2, "f"))
|
|
calls.append(2)
|
|
|
|
def listener3(_):
|
|
message_broker.send(Message(MessageType.test2, "f"))
|
|
calls.append(3)
|
|
|
|
def listener4(_):
|
|
calls.append(4)
|
|
|
|
message_broker.subscribe(MessageType.test1, listener1)
|
|
message_broker.subscribe(MessageType.test1, listener2)
|
|
message_broker.subscribe(MessageType.test1, listener3)
|
|
message_broker.subscribe(MessageType.test2, listener4)
|
|
message_broker.send(Message(MessageType.test1, ""))
|
|
|
|
first = calls[:3]
|
|
first.sort()
|
|
self.assertEqual([1, 2, 3], first)
|
|
self.assertEqual([4, 4, 4], calls[3:])
|