From 00b62dc1a96585d6db1622807406b4466368163f Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 9 Aug 2022 11:56:15 +0530 Subject: [PATCH] UT: Modify PyPubSubEventQueue's tests based on changes --- .../common/event_queue/test_pypubsub_event_queue.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/common/event_queue/test_pypubsub_event_queue.py b/monkey/tests/unit_tests/common/event_queue/test_pypubsub_event_queue.py index 3a02563cc..738b91566 100644 --- a/monkey/tests/unit_tests/common/event_queue/test_pypubsub_event_queue.py +++ b/monkey/tests/unit_tests/common/event_queue/test_pypubsub_event_queue.py @@ -40,7 +40,7 @@ def test_subscribe_all(): def test_subscribe_types(): subscriber = MagicMock() - PyPubSubEventQueue.subscribe_types([EventType], subscriber) + PyPubSubEventQueue.subscribe_type(EventType, subscriber) PyPubSubEventQueue.publish(EventType) assert pypubsub_event_queue_file.pub.sendMessage.call_count == 3 @@ -50,7 +50,7 @@ def test_subscribe_types(): def test_subscribe_tags(): subscriber = MagicMock() - PyPubSubEventQueue.subscribe_tags([EVENT_TAG_2], subscriber) + PyPubSubEventQueue.subscribe_tag(EVENT_TAG_2, subscriber) PyPubSubEventQueue.publish(EventType) assert pypubsub_event_queue_file.pub.sendMessage.call_count == 3