p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Test with instances of AbstractEvent, not classes

This commit is contained in:
Mike Salvatore 2022-08-10 09:35:57 -04:00
parent 4da0cefa7d
commit d6bb56536d
1 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,7 @@
from dataclasses import dataclass
from ipaddress import IPv4Address
from typing import FrozenSet, Union
from uuid import UUID
import pytest
from pubsub import pub
@ -11,11 +14,11 @@ EVENT_TAG_2 = "event tag 2"
@dataclass(frozen=True)
class EventType(AbstractEvent):
source = "1234"
target = None
timestamp = 0.0
tags = [EVENT_TAG_1, EVENT_TAG_2]
class TestEvent(AbstractEvent):
source: UUID = "f811ad00-5a68-4437-bd51-7b5cc1768ad5"
target: Union[UUID, IPv4Address, None] = None
timestamp: float = 0.0
tags: FrozenSet = frozenset()
pypubsub_event_queue = PyPubSubEventQueue(pub)
@ -49,12 +52,12 @@ def subscriber_2(subscriber_2_calls):
@pytest.mark.usefixtures("subscriber_1", "subscriber_2", "subscriber_1_calls", "subscriber_2_calls")
def test_topic_subscription(subscriber_1, subscriber_2, subscriber_1_calls, subscriber_2_calls):
pypubsub_event_queue.subscribe_type(EventType, subscriber_1)
pypubsub_event_queue.subscribe_type(TestEvent, subscriber_1)
pypubsub_event_queue.subscribe_tag(EVENT_TAG_2, subscriber_1)
pypubsub_event_queue.subscribe_tag(EVENT_TAG_1, subscriber_2)
pypubsub_event_queue.publish(EventType)
pypubsub_event_queue.publish(TestEvent(tags={EVENT_TAG_1, EVENT_TAG_2}))
assert subscriber_1_calls == [EventType.__name__, EVENT_TAG_2]
assert subscriber_1_calls == [TestEvent.__name__, EVENT_TAG_2]
assert subscriber_2_calls == [EVENT_TAG_1]
@ -65,27 +68,27 @@ def test_subscribe_all():
subscriber_calls.append(topic.getName())
pypubsub_event_queue.subscribe_all_events(subscriber)
pypubsub_event_queue.publish(EventType)
pypubsub_event_queue.publish(TestEvent(tags={EVENT_TAG_1, EVENT_TAG_2}))
assert len(subscriber_calls) == 1
assert EventType.__name__ not in subscriber_calls
assert TestEvent.__name__ not in subscriber_calls
assert EVENT_TAG_1 not in subscriber_calls
assert EVENT_TAG_2 not in subscriber_calls
@pytest.mark.usefixtures("subscriber_1", "subscriber_1_calls")
def test_subscribe_types(subscriber_1, subscriber_1_calls):
pypubsub_event_queue.subscribe_type(EventType, subscriber_1)
pypubsub_event_queue.publish(EventType)
pypubsub_event_queue.subscribe_type(TestEvent, subscriber_1)
pypubsub_event_queue.publish(TestEvent(tags={EVENT_TAG_1, EVENT_TAG_2}))
assert subscriber_1_calls == [EventType.__name__]
assert subscriber_1_calls == [TestEvent.__name__]
@pytest.mark.usefixtures("subscriber_1", "subscriber_2", "subscriber_1_calls", "subscriber_2_calls")
def test_subscribe_tags(subscriber_1, subscriber_2, subscriber_1_calls, subscriber_2_calls):
pypubsub_event_queue.subscribe_tag(EVENT_TAG_1, subscriber_1)
pypubsub_event_queue.subscribe_tag(EVENT_TAG_2, subscriber_2)
pypubsub_event_queue.publish(EventType)
pypubsub_event_queue.publish(TestEvent(tags={EVENT_TAG_1, EVENT_TAG_2}))
assert subscriber_1_calls == [EVENT_TAG_1]
assert subscriber_2_calls == [EVENT_TAG_2]