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

UT: Simplify test logic with fixtures in PyPubSubEventQueue

This commit is contained in:
Shreya Malviya 2022-08-10 15:39:14 +05:30
parent 3aff182d47
commit edad0d8cf2
1 changed files with 23 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23,19 +23,29 @@ class EventType(AbstractEvent):
pypubsub_event_queue = PyPubSubEventQueue(pub)
subscriber_1_calls = subscriber_2_calls = subscriber_1 = subscriber_2 = None
@pytest.fixture()
def subscriber_1_calls():
return []
@pytest.fixture(autouse=True, scope="function")
def reset_subscribers():
global subscriber_1, subscriber_2, subscriber_1_calls, subscriber_2_calls
subscriber_1_calls = []
subscriber_2_calls = []
subscriber_1 = lambda event, topic=pub.AUTO_TOPIC: subscriber_1_calls.append(topic.getName())
subscriber_2 = lambda event, topic=pub.AUTO_TOPIC: subscriber_2_calls.append(topic.getName())
@pytest.fixture()
def subscriber_2_calls():
return []
def test_topic_subscription():
@pytest.fixture
def subscriber_1(subscriber_1_calls):
return lambda event, topic=pub.AUTO_TOPIC: subscriber_1_calls.append(topic.getName())
@pytest.fixture
def subscriber_2(subscriber_2_calls):
return lambda event, topic=pub.AUTO_TOPIC: subscriber_2_calls.append(topic.getName())
@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_tag(EVENT_TAG_2, subscriber_1)
pypubsub_event_queue.subscribe_tag(EVENT_TAG_1, subscriber_2)
@ -60,7 +70,8 @@ def test_subscribe_all():
]
def test_subscribe_types():
@pytest.mark.usefixtures("subscriber_1", "subscriber_2", "subscriber_1_calls", "subscriber_2_calls")
def test_subscribe_types(subscriber_1, subscriber_2, subscriber_1_calls, subscriber_2_calls):
pypubsub_event_queue.subscribe_type(EventType, subscriber_1)
pypubsub_event_queue.publish(EventType)
@ -68,7 +79,8 @@ def test_subscribe_types():
assert subscriber_2_calls == []
def test_subscribe_tags():
@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)