UT: Modify PyPubSubEventQueue's tests based on changes

This commit is contained in:
Shreya Malviya 2022-08-09 11:56:15 +05:30
parent 3cef54c09c
commit 00b62dc1a9
1 changed files with 2 additions and 2 deletions

View File

@ -40,7 +40,7 @@ def test_subscribe_all():
def test_subscribe_types(): def test_subscribe_types():
subscriber = MagicMock() subscriber = MagicMock()
PyPubSubEventQueue.subscribe_types([EventType], subscriber) PyPubSubEventQueue.subscribe_type(EventType, subscriber)
PyPubSubEventQueue.publish(EventType) PyPubSubEventQueue.publish(EventType)
assert pypubsub_event_queue_file.pub.sendMessage.call_count == 3 assert pypubsub_event_queue_file.pub.sendMessage.call_count == 3
@ -50,7 +50,7 @@ def test_subscribe_types():
def test_subscribe_tags(): def test_subscribe_tags():
subscriber = MagicMock() subscriber = MagicMock()
PyPubSubEventQueue.subscribe_tags([EVENT_TAG_2], subscriber) PyPubSubEventQueue.subscribe_tag(EVENT_TAG_2, subscriber)
PyPubSubEventQueue.publish(EventType) PyPubSubEventQueue.publish(EventType)
assert pypubsub_event_queue_file.pub.sendMessage.call_count == 3 assert pypubsub_event_queue_file.pub.sendMessage.call_count == 3