From 49a28b91615d670b636f309b1174721ad01a4048 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 10 Aug 2022 17:23:07 +0530 Subject: [PATCH] UT: Add test for subscribing to all event types in PyPubSubEventQueue --- .../common/event_queue/test_pypubsub_event_queue.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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 89cc3f08d..867673588 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 @@ -94,3 +94,12 @@ def test_subscribe_tags(subscriber_1, subscriber_2, subscriber_1_calls, subscrib assert subscriber_1_calls == [EVENT_TAG_1] assert subscriber_2_calls == [EVENT_TAG_2] + + +@pytest.mark.usefixtures("subscriber_1", "subscriber_1_calls") +def test_subscribe_all_event_types(subscriber_1, subscriber_1_calls): + pypubsub_event_queue.subscribe_all_event_types(subscriber_1) + pypubsub_event_queue.publish(EventType) + pypubsub_event_queue.publish(EventType) + + assert subscriber_1_calls == [INTERNAL_ALL_EVENT_TYPES_TOPIC, INTERNAL_ALL_EVENT_TYPES_TOPIC]