43 lines
2.5 KiB
Python
43 lines
2.5 KiB
Python
|
import pytest
|
||
|
|
from src.env_vars import ADDITIONAL_NODES, NODE_2
|
||
|
|
from src.libs.custom_logger import get_custom_logger
|
||
|
|
from src.steps.sharding import StepsSharding
|
||
|
|
|
||
|
|
logger = get_custom_logger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class TestMultipleNodes(StepsSharding):
|
||
|
|
def test_static_shard_relay(self):
|
||
|
|
self.setup_main_relay_nodes(pubsub_topic=self.test_pubsub_topic)
|
||
|
|
self.setup_optional_relay_nodes(pubsub_topic=self.test_pubsub_topic)
|
||
|
|
self.subscribe_main_relay_nodes(pubsub_topics=[self.test_pubsub_topic])
|
||
|
|
self.subscribe_optional_relay_nodes(pubsub_topics=[self.test_pubsub_topic])
|
||
|
|
self.check_published_message_reaches_relay_peer(pubsub_topic=self.test_pubsub_topic)
|
||
|
|
|
||
|
|
@pytest.mark.skipif("go-waku" in NODE_2, reason="Test works only with nwaku")
|
||
|
|
def test_static_shard_relay_10_nwaku_nodes(self):
|
||
|
|
self.setup_first_relay_node(pubsub_topic=self.test_pubsub_topic)
|
||
|
|
self.setup_nwaku_relay_nodes(num_nodes=9, pubsub_topic=self.test_pubsub_topic)
|
||
|
|
self.subscribe_main_relay_nodes(pubsub_topics=[self.test_pubsub_topic])
|
||
|
|
self.subscribe_optional_relay_nodes(pubsub_topics=[self.test_pubsub_topic])
|
||
|
|
self.check_published_message_reaches_relay_peer(pubsub_topic=self.test_pubsub_topic)
|
||
|
|
|
||
|
|
@pytest.mark.skipif(
|
||
|
|
"go-waku" in NODE_2 or "go-waku" in ADDITIONAL_NODES,
|
||
|
|
reason="Autosharding tests work only on nwaku because of https://github.com/waku-org/go-waku/issues/1061",
|
||
|
|
)
|
||
|
|
def test_auto_shard_relay(self):
|
||
|
|
self.setup_main_relay_nodes(cluster_id=self.auto_cluster, content_topic=self.test_content_topic)
|
||
|
|
self.setup_optional_relay_nodes(cluster_id=self.auto_cluster, content_topic=self.test_content_topic)
|
||
|
|
self.subscribe_main_relay_nodes(content_topics=[self.test_content_topic])
|
||
|
|
self.subscribe_optional_relay_nodes(content_topics=[self.test_content_topic])
|
||
|
|
self.check_published_message_reaches_relay_peer(content_topic=self.test_content_topic)
|
||
|
|
|
||
|
|
@pytest.mark.skipif("go-waku" in NODE_2, reason="Test works only with nwaku")
|
||
|
|
def test_auto_shard_relay_10_nwaku_nodes(self):
|
||
|
|
self.setup_first_relay_node(cluster_id=self.auto_cluster, content_topic=self.test_content_topic)
|
||
|
|
self.setup_nwaku_relay_nodes(num_nodes=8, cluster_id=self.auto_cluster, content_topic=self.test_content_topic)
|
||
|
|
self.subscribe_main_relay_nodes(content_topics=[self.test_content_topic])
|
||
|
|
self.subscribe_optional_relay_nodes(content_topics=[self.test_content_topic])
|
||
|
|
self.check_published_message_reaches_relay_peer(content_topic=self.test_content_topic)
|