82 lines
4.5 KiB
Python
82 lines
4.5 KiB
Python
|
import pytest
|
||
|
|
from src.env_vars import NODE_2
|
||
|
|
from src.steps.store import StepsStore
|
||
|
|
from src.test_data import CONTENT_TOPICS_DIFFERENT_SHARDS
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.xfail("go-waku" in NODE_2, reason="Bug reported: https://github.com/waku-org/go-waku/issues/1108")
|
||
|
|
class TestTopics(StepsStore):
|
||
|
|
@pytest.fixture(scope="function", autouse=True)
|
||
|
|
def topics_setup(self, node_setup):
|
||
|
|
self.message_hash_list = []
|
||
|
|
for content_topic in CONTENT_TOPICS_DIFFERENT_SHARDS:
|
||
|
|
message = self.create_message(contentTopic=content_topic)
|
||
|
|
self.publish_message(message=message)
|
||
|
|
self.message_hash_list.append(self.compute_message_hash(self.test_pubsub_topic, message))
|
||
|
|
|
||
|
|
def test_store_with_one_content_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
for index, content_topic in enumerate(CONTENT_TOPICS_DIFFERENT_SHARDS):
|
||
|
|
store_response = node.get_store_messages(content_topics=content_topic, page_size=20, ascending="true")
|
||
|
|
assert len(store_response["messages"]) == 1, "Message count mismatch"
|
||
|
|
assert (
|
||
|
|
store_response["messages"][0]["messageHash"] == self.message_hash_list[index]
|
||
|
|
), "Incorrect messaged filtered based on content topic"
|
||
|
|
|
||
|
|
def test_store_with_multiple_content_topics(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
store_response = node.get_store_messages(
|
||
|
|
content_topics=f"{CONTENT_TOPICS_DIFFERENT_SHARDS[0]},{CONTENT_TOPICS_DIFFERENT_SHARDS[4]}", page_size=20, ascending="true"
|
||
|
|
)
|
||
|
|
assert len(store_response["messages"]) == 2, "Message count mismatch"
|
||
|
|
assert (
|
||
|
|
store_response["messages"][0]["messageHash"] == self.message_hash_list[0]
|
||
|
|
), "Incorrect messaged filtered based on multiple content topics"
|
||
|
|
assert (
|
||
|
|
store_response["messages"][1]["messageHash"] == self.message_hash_list[4]
|
||
|
|
), "Incorrect messaged filtered based on multiple content topics"
|
||
|
|
|
||
|
|
def test_store_with_unknown_content_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
store_response = node.get_store_messages(content_topics="test", page_size=20, ascending="true")
|
||
|
|
assert len(store_response["messages"]) == 0, "Message count mismatch"
|
||
|
|
|
||
|
|
def test_store_with_unknown_pubsub_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
store_response = node.get_store_messages(pubsub_topic="test", page_size=20, ascending="true")
|
||
|
|
assert len(store_response["messages"]) == 0, "Message count mismatch"
|
||
|
|
|
||
|
|
def test_store_with_both_pubsub_topic_and_content_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
for index, content_topic in enumerate(CONTENT_TOPICS_DIFFERENT_SHARDS):
|
||
|
|
store_response = node.get_store_messages(
|
||
|
|
pubsub_topic=self.test_pubsub_topic, content_topics=content_topic, page_size=20, ascending="true"
|
||
|
|
)
|
||
|
|
assert len(store_response["messages"]) == 1, "Message count mismatch"
|
||
|
|
assert (
|
||
|
|
store_response["messages"][0]["messageHash"] == self.message_hash_list[index]
|
||
|
|
), "Incorrect messaged filtered based on content topic"
|
||
|
|
|
||
|
|
def test_store_with_unknown_pubsub_topic_but_known_content_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
store_response = node.get_store_messages(
|
||
|
|
pubsub_topic="test", content_topics=CONTENT_TOPICS_DIFFERENT_SHARDS[0], page_size=20, ascending="true"
|
||
|
|
)
|
||
|
|
assert len(store_response["messages"]) == 0, "Message count mismatch"
|
||
|
|
|
||
|
|
def test_store_with_both_pubsub_topic_and_content_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
for index, content_topic in enumerate(CONTENT_TOPICS_DIFFERENT_SHARDS):
|
||
|
|
store_response = node.get_store_messages(
|
||
|
|
pubsub_topic=self.test_pubsub_topic, content_topics=content_topic, page_size=20, ascending="true"
|
||
|
|
)
|
||
|
|
assert len(store_response["messages"]) == 1, "Message count mismatch"
|
||
|
|
assert (
|
||
|
|
store_response["messages"][0]["messageHash"] == self.message_hash_list[index]
|
||
|
|
), "Incorrect messaged filtered based on content topic"
|
||
|
|
|
||
|
|
def test_store_without_pubsub_topic_and_content_topic(self):
|
||
|
|
for node in self.store_nodes:
|
||
|
|
store_response = node.get_store_messages(page_size=20, ascending="true")
|
||
|
|
assert len(store_response["messages"]) == len(CONTENT_TOPICS_DIFFERENT_SHARDS), "Message count mismatch"
|