diff --git a/monkey/common/event_queue/pypubsub_event_queue.py b/monkey/common/event_queue/pypubsub_event_queue.py index 253e00836..d2b1aa5f5 100644 --- a/monkey/common/event_queue/pypubsub_event_queue.py +++ b/monkey/common/event_queue/pypubsub_event_queue.py @@ -6,7 +6,7 @@ from common.events import AbstractEvent from .i_event_queue import IEventQueue -INTERNAL_ALL_EVENT_TYPES_TOPIC = "internal_all_event_types" +_INTERNAL_ALL_EVENT_TYPES_TOPIC = "internal_all_event_types" class PyPubSubEventQueue(IEventQueue): @@ -15,7 +15,7 @@ class PyPubSubEventQueue(IEventQueue): def subscribe_all_events(self, subscriber: Callable[[AbstractEvent], None]): self._pypubsub_publisher.subscribe( - listener=subscriber, topicName=INTERNAL_ALL_EVENT_TYPES_TOPIC + listener=subscriber, topicName=_INTERNAL_ALL_EVENT_TYPES_TOPIC ) def subscribe_type( @@ -33,7 +33,7 @@ class PyPubSubEventQueue(IEventQueue): self._publish_to_tags_topic(event) def _publish_to_all_events_topic(self, event: AbstractEvent): - self._pypubsub_publisher.sendMessage(INTERNAL_ALL_EVENT_TYPES_TOPIC, event=event) + self._pypubsub_publisher.sendMessage(_INTERNAL_ALL_EVENT_TYPES_TOPIC, event=event) def _publish_to_type_topic(self, event: AbstractEvent): self._pypubsub_publisher.sendMessage(event.__name__, event=event) 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 aac31c0ae..188eb0566 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 @@ -3,10 +3,7 @@ from dataclasses import dataclass import pytest from pubsub import pub -from common.event_queue.pypubsub_event_queue import ( - INTERNAL_ALL_EVENT_TYPES_TOPIC, - PyPubSubEventQueue, -) +from common.event_queue.pypubsub_event_queue import PyPubSubEventQueue from common.events import AbstractEvent EVENT_TAG_1 = "event tag 1"