From 42adfcb0879f071bd3bbdb26343b43d7c50d8830 Mon Sep 17 00:00:00 2001
From: fbarbu15 <barbu_florin_adrian@yahoo.com>
Date: Mon, 20 Nov 2023 17:04:49 +0200
Subject: [PATCH] self review

---
 .github/workflows/test.yml         | 2 +-
 src/steps/relay.py                 | 3 ++-
 tests/relay/test_multiple_nodes.py | 4 ----
 3 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 0046d6dc..64032efe 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -93,4 +93,4 @@ jobs:
         echo "- **Additonal Nodes**: ${{ env.ADDITIONAL_NODES}}" >> $GITHUB_STEP_SUMMARY
         echo "- **Protocol**: ${{ env.PROTOCOL }}" >> $GITHUB_STEP_SUMMARY
         echo "## Test Results" >> $GITHUB_STEP_SUMMARY
-        echo "Allure report will be available at: https://waku-org.github.io/waku-interop-tests/${{ github.run_number }}" >> $GITHUB_STEP_SUMMARY
\ No newline at end of file
+        echo "Allure report will be available at: https://waku-org.github.io/waku-interop-tests/${{ github.run_number }}" >> $GITHUB_STEP_SUMMARY
diff --git a/src/steps/relay.py b/src/steps/relay.py
index 53a25871..d9bafdba 100644
--- a/src/steps/relay.py
+++ b/src/steps/relay.py
@@ -55,7 +55,7 @@ class StepsRelay:
     def relay_warm_up(self):
         try:
             self.wait_for_published_message_to_reach_peer()
-            logger.info("WARM UP successful for the main nodes!!")
+            logger.info("WARM UP successful!!")
         except Exception as ex:
             raise TimeoutError(f"WARM UP FAILED WITH: {ex}")
 
@@ -76,6 +76,7 @@ class StepsRelay:
             received_message = message_rpc_response_schema.load(get_messages_response[0])
             self.assert_received_message(message, received_message)
 
+    # we need much bigger timeout in CI because we run tests in parallel there and the machine itself is slower
     @allure.step
     def wait_for_published_message_to_reach_peer(
         self, timeout_duration=120 if RUNNING_IN_CI else 20, time_between_retries=1, sender=None, peer_list=None
diff --git a/tests/relay/test_multiple_nodes.py b/tests/relay/test_multiple_nodes.py
index 9feb23bd..aa7cc121 100644
--- a/tests/relay/test_multiple_nodes.py
+++ b/tests/relay/test_multiple_nodes.py
@@ -1,10 +1,6 @@
-from time import sleep
 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")
 class TestMultipleNodes(StepsRelay):