2023-11-07 18:10:25 +02:00

46 lines
2.3 KiB
Python

import logging
from src.libs.common import to_base64
from src.data_classes import MessageRpcQuery
from src.steps.relay import StepsRelay
from src.test_data import SAMPLE_INPUTS
logger = logging.getLogger(__name__)
class TestRelayPublish(StepsRelay):
def test_publish_with_various_payloads(self):
failed_payloads = []
for index, payload in enumerate(SAMPLE_INPUTS):
logger.debug("Running test with payload %s", payload["description"])
message = MessageRpcQuery(payload=to_base64(payload["value"]), contentTopic=self.test_content_topic)
try:
# The node may require a warmup period for message processing to stabilize.
# Therefore, we use the retry function for the first payload to account for this warmup.
if index == 0:
self.check_published_message_reaches_peer_with_retry(message)
else:
self.check_published_message_reaches_peer(message)
except Exception as e:
logger.error("Payload %s failed: %s", payload["description"], str(e))
failed_payloads.append(payload["description"])
assert not failed_payloads, f"Payloads failed: {failed_payloads}"
def test_publish_with_various_content_topics(self):
failed_content_topics = []
for index, content_topic in enumerate(SAMPLE_INPUTS):
logger.debug("Running test with content topic %s", content_topic["description"])
message = MessageRpcQuery(payload=to_base64(self.test_payload), contentTopic=content_topic["value"])
try:
# The node may require a warmup period for message processing to stabilize.
# Therefore, we use the retry function for the first payload to account for this warmup.
if index == 0:
self.check_published_message_reaches_peer_with_retry(message)
else:
self.check_published_message_reaches_peer(message)
except Exception as e:
logger.error("ContentTopic %s failed: %s", {content_topic["description"]}, {str(e)})
failed_content_topics.append(content_topic)
assert not failed_content_topics, f"ContentTopics failed: {failed_content_topics}"