p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Use frozenset() for tags in test_pypubsub_event_queue.py

This commit is contained in:
Mike Salvatore 2022-08-10 12:42:37 -04:00
parent e36fb9af73
commit ac69e7d25b
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
monkey/tests/unit_tests/common/event_queue/test_pypubsub_event_queue.py
View File

@ -57,9 +57,9 @@ def event_queue() -> IEventQueue:
def test_subscribe_all(event_queue: IEventQueue, subscriber: EventSubscriber): def test_subscribe_all(event_queue: IEventQueue, subscriber: EventSubscriber):
event_queue.subscribe_all_events(subscriber) event_queue.subscribe_all_events(subscriber)
event_queue.publish(TestEvent1(tags={EVENT_TAG_1, EVENT_TAG_2})) event_queue.publish(TestEvent1(tags=frozenset({EVENT_TAG_1, EVENT_TAG_2})))
event_queue.publish(TestEvent1(tags={EVENT_TAG_2})) event_queue.publish(TestEvent1(tags=frozenset({EVENT_TAG_2})))
event_queue.publish(TestEvent1(tags={"secret_tag"})) event_queue.publish(TestEvent1(tags=frozenset({"secret_tag"})))
event_queue.publish(TestEvent2()) event_queue.publish(TestEvent2())
assert subscriber.call_count == 4 assert subscriber.call_count == 4
@ -81,8 +81,8 @@ def test_subscribe_types(event_queue: IEventQueue, subscriber: EventSubscriber,
def test_subscribe_tags_single_type(event_queue: IEventQueue, subscriber: EventSubscriber): def test_subscribe_tags_single_type(event_queue: IEventQueue, subscriber: EventSubscriber):
event_queue.subscribe_tag(EVENT_TAG_1, subscriber) event_queue.subscribe_tag(EVENT_TAG_1, subscriber)
event_queue.publish(TestEvent1(tags={EVENT_TAG_1, EVENT_TAG_2})) event_queue.publish(TestEvent1(tags=frozenset({EVENT_TAG_1, EVENT_TAG_2})))
event_queue.publish(TestEvent2(tags={EVENT_TAG_2})) event_queue.publish(TestEvent2(tags=frozenset({EVENT_TAG_2})))
assert subscriber.call_count == 1 assert subscriber.call_count == 1
assert len(subscriber.call_types) == 1 assert len(subscriber.call_types) == 1
@ -93,8 +93,8 @@ def test_subscribe_tags_single_type(event_queue: IEventQueue, subscriber: EventS
def test_subscribe_tags_multiple_types(event_queue: IEventQueue, subscriber: EventSubscriber): def test_subscribe_tags_multiple_types(event_queue: IEventQueue, subscriber: EventSubscriber):
event_queue.subscribe_tag(EVENT_TAG_2, subscriber) event_queue.subscribe_tag(EVENT_TAG_2, subscriber)
event_queue.publish(TestEvent1(tags={EVENT_TAG_1, EVENT_TAG_2})) event_queue.publish(TestEvent1(tags=frozenset({EVENT_TAG_1, EVENT_TAG_2})))
event_queue.publish(TestEvent2(tags={EVENT_TAG_2})) event_queue.publish(TestEvent2(tags=frozenset({EVENT_TAG_2})))
assert subscriber.call_count == 2 assert subscriber.call_count == 2
assert len(subscriber.call_types) == 2 assert len(subscriber.call_types) == 2
@ -106,6 +106,6 @@ def test_subscribe_tags_multiple_types(event_queue: IEventQueue, subscriber: Eve
def test_type_tag_collision(event_queue: IEventQueue, subscriber: EventSubscriber): def test_type_tag_collision(event_queue: IEventQueue, subscriber: EventSubscriber):
event_queue.subscribe_type(TestEvent1, subscriber) event_queue.subscribe_type(TestEvent1, subscriber)
event_queue.publish(TestEvent2(tags={TestEvent1.__name__})) event_queue.publish(TestEvent2(tags=frozenset({TestEvent1.__name__})))
assert subscriber.call_count == 0 assert subscriber.call_count == 0