logos-messaging-interop-tests/tests/relay/test_multiple_nodes.py

import pytest
from src.libs.custom_logger import get_custom_logger
from src.steps.relay import StepsRelay

logger = get_custom_logger(__name__)


@pytest.mark.usefixtures(
    "setup_main_relay_nodes", "setup_optional_relay_nodes", "subscribe_main_relay_nodes", "subscribe_optional_relay_nodes", "relay_warm_up"
)
class TestMultipleNodes(StepsRelay):
    def test_first_node_to_start_publishes(self):
        self.wait_for_published_message_to_reach_peer()

    def test_last_node_to_start_publishes(self):
        self.wait_for_published_message_to_reach_peer(sender=self.optional_nodes[-1])