40 lines
1.5 KiB
Python

import pytest
from src.env_vars import NODE_1, NODE_2, STRESS_ENABLED
from src.libs.common import delay
from src.libs.custom_logger import get_custom_logger
from src.node.waku_node import WakuNode
from src.steps.filter import StepsFilter
from src.steps.light_push import StepsLightPush
from src.steps.relay import StepsRelay
from src.steps.store import StepsStore
logger = get_custom_logger(__name__)
"""
This tests will cover rendezvous protocol e2e scenarios
"""
class TestE2E(StepsRelay):
@pytest.fixture(scope="function", autouse=True)
def nodes(self):
self.node1 = WakuNode(NODE_2, f"node1_{self.test_id}")
self.node2 = WakuNode(NODE_1, f"node2_{self.test_id}")
self.node3 = WakuNode(NODE_2, f"node3_{self.test_id}")
def test_basic_rendezvous_register_and_discover(self):
self.node1.start(rendezvous="true", relay="true")
delay(5)
self.node2.start(rendezvous="true", relay="true", discv5_bootstrap_node=self.node1.get_enr_uri())
self.node1.set_relay_subscriptions([self.test_pubsub_topic])
self.node2.set_relay_subscriptions([self.test_pubsub_topic])
self.wait_for_autoconnection([self.node1, self.node2], hard_wait=30)
self.node3.start(rendezvous="true", relay="true")
self.node2.add_peers([self.node3.get_multiaddr_with_id()])
delay(60)
discovered = self.node3.get_peers()
logger.debug(f"Node3 peers count is : {len(discovered)}")
assert len(discovered) > 0, "No peers discovered via Rendezvous"