From 96993f1555b56dc756fd2fe87690721baaababb6 Mon Sep 17 00:00:00 2001
From: Roman <zajic@zajic.net>
Date: Thu, 6 Jun 2024 18:29:09 +0200
Subject: [PATCH] test: expect two peers for two protocols/services

---
 tests/peer_connection_management/test_peer_store.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/peer_connection_management/test_peer_store.py b/tests/peer_connection_management/test_peer_store.py
index 4df68a1c..b515355f 100644
--- a/tests/peer_connection_management/test_peer_store.py
+++ b/tests/peer_connection_management/test_peer_store.py
@@ -54,10 +54,13 @@ class TestPeerStore(StepsRelay, StepsStore):
                         logger.error(f"Failed to add peer to Node {i} peer store: {ex}")
                         raise
 
-    # Related to
     @pytest.mark.skip(reason="waiting for https://github.com/waku-org/nwaku/issues/1549 resolution")
     def test_get_peers_two_protocols(self):
         self.setup_first_publishing_node(store="true", relay="true")
         self.setup_first_store_node(store="true", relay="false")
-        logger.debug(f"Node 1 connected peers {self.publishing_node1.get_peers()}")
-        logger.debug(f"Node 2 connected peers {self.store_node1.get_peers()}")
+        node1_peers = self.publishing_node1.get_peers()
+        node2_peers = self.store_node1.get_peers()
+        logger.debug(f"Node 1 connected peers {node1_peers}")
+        logger.debug(f"Node 2 connected peers {node2_peers}")
+
+        assert len(node1_peers) == 2 and len(node2_peers) == 2, f"Some nodes and/or their services are missing"