diff --git a/src/steps/rln.py b/src/steps/rln.py index 34a6b2c0..1d250e66 100644 --- a/src/steps/rln.py +++ b/src/steps/rln.py @@ -1,18 +1,18 @@ -from src.env_vars import DEFAULT_NWAKU, RLN_CREDENTIALS, NODEKEY -from src.libs.common import gen_step_id, to_base64 -from src.libs.custom_logger import get_custom_logger import os import inspect import pytest import allure -from time import time -from src.node.waku_node import WakuNode, rln_credential_store_ready +from src.steps.common import StepsCommon from src.test_data import PUBSUB_TOPICS_RLN +from src.env_vars import DEFAULT_NWAKU, RLN_CREDENTIALS, NODEKEY +from src.libs.common import gen_step_id +from src.libs.custom_logger import get_custom_logger +from src.node.waku_node import WakuNode, rln_credential_store_ready logger = get_custom_logger(__name__) -class StepsRLN: +class StepsRLN(StepsCommon): test_pubsub_topic = PUBSUB_TOPICS_RLN[0] test_content_topic = "/test/1/waku-rln-relay/proto" test_payload = "RLN relay works!!" @@ -70,12 +70,6 @@ class StepsRLN: logger.error(f"Credentials at {creds_file_path} not available: {ex}") raise - @allure.step - def create_message(self, **kwargs): - message = {"payload": to_base64(self.test_payload), "contentTopic": self.test_content_topic, "timestamp": int(time() * 1e9)} - message.update(kwargs) - return message - @allure.step def publish_message(self, message=None, pubsub_topic=None, sender=None): if message is None: diff --git a/tests/relay/test_rln.py b/tests/relay/test_rln.py index a5bc5934..2938732e 100644 --- a/tests/relay/test_rln.py +++ b/tests/relay/test_rln.py @@ -14,7 +14,7 @@ logger = get_custom_logger(__name__) @pytest.mark.usefixtures("register_main_rln_relay_nodes") -class TestRelayRLN(StepsRLN, StepsRelay): +class TestRelayRLN(StepsRLN): def test_publish_with_valid_payloads_at_slow_rate(self): self.setup_main_rln_relay_nodes() self.subscribe_main_relay_nodes()