fix: move Waku related methods to waku_node module

src/libs/common.py

@@ -38,18 +38,3 @@ def delay(num_seconds):
 
 def gen_step_id():
     return f"{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}__{str(uuid.uuid4())}"
-
-
-def peer_info2id(peer, is_nwaku=True):
-    return peer_info2multiaddr(peer, is_nwaku).split("/")[-1]
-
-
-def peer_info2multiaddr(peer, is_nwaku=True):
-    if is_nwaku:
-        return peer["multiaddr"]
-    else:
-        return peer["multiaddrs"][0]
-
-
-def multiaddr2id(multiaddr):
-    return multiaddr.split("/")[-1]

src/node/waku_node.py

@@ -39,6 +39,21 @@ def rln_credential_store_ready(creds_file_path):
         raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT), creds_file_path)
 
 
+def peer_info2multiaddr(peer, is_nwaku=True):
+    if is_nwaku:
+        return peer["multiaddr"]
+    else:
+        return peer["multiaddrs"][0]
+
+
+def peer_info2id(peer, is_nwaku=True):
+    return peer_info2multiaddr(peer, is_nwaku).split("/")[-1]
+
+
+def multiaddr2id(multiaddr):
+    return multiaddr.split("/")[-1]
+
+
 class WakuNode:
     def __init__(self, docker_image, docker_log_prefix=""):
         self._image_name = docker_image

tests/peer_connection_management/test_peer_store.py

@@ -1,7 +1,7 @@
 import pytest
 
-from src.libs.common import peer_info2id, peer_info2multiaddr, multiaddr2id, delay
 from src.libs.custom_logger import get_custom_logger
+from src.node.waku_node import peer_info2id, peer_info2multiaddr, multiaddr2id
 from src.steps.peer_store import StepsPeerStore
 from src.steps.relay import StepsRelay
 from src.steps.store import StepsStore