mirror of
https://github.com/logos-messaging/logos-delivery-interop-tests.git
synced 2026-02-28 15:03:14 +00:00
03973b8897
* first commit * get message tests 1 * store tests with node restarts * reliability tests * store tests * fixes * more tests * fixes * new store tests * new store tests * adjustments * adjust tests for go-waku * postgress test * postgress test * fixes * small fixes * reliability updates * postgress test * adjsutmens for go-waku
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
import base64
|
|
import hashlib
|
|
import inspect
|
|
from time import time
|
|
import allure
|
|
import pytest
|
|
from tenacity import retry, stop_after_delay, wait_fixed
|
|
from src.libs.common import to_base64
|
|
from src.libs.custom_logger import get_custom_logger
|
|
|
|
logger = get_custom_logger(__name__)
|
|
|
|
|
|
class StepsCommon:
|
|
@pytest.fixture(scope="function", autouse=True)
|
|
def common_setup(self):
|
|
logger.debug(f"Running fixture setup: {inspect.currentframe().f_code.co_name}")
|
|
if not hasattr(self, "test_payload"):
|
|
self.test_payload = "Default Payload"
|
|
if not hasattr(self, "test_content_topic"):
|
|
self.test_content_topic = "/test/1/default/proto"
|
|
|
|
@allure.step
|
|
@retry(stop=stop_after_delay(20), wait=wait_fixed(0.5), reraise=True)
|
|
def add_node_peer(self, node, multiaddr_list, shards=[0, 1, 2, 3, 4, 5, 6, 7, 8]):
|
|
if node.is_nwaku():
|
|
for multiaddr in multiaddr_list:
|
|
node.add_peers([multiaddr])
|
|
elif node.is_gowaku():
|
|
for multiaddr in multiaddr_list:
|
|
peer_info = {"multiaddr": multiaddr, "protocols": ["/vac/waku/relay/2.0.0"], "shards": shards}
|
|
node.add_peers(peer_info)
|
|
|
|
@allure.step
|
|
def create_message(self, **kwargs):
|
|
message = {"payload": to_base64(self.test_payload), "contentTopic": self.test_content_topic, "timestamp": int(time() * 1e9)}
|
|
message.update(kwargs)
|
|
return message
|
|
|
|
@allure.step
|
|
def compute_message_hash(self, pubsub_topic, msg):
|
|
ctx = hashlib.sha256()
|
|
ctx.update(pubsub_topic.encode("utf-8"))
|
|
ctx.update(base64.b64decode(msg["payload"]))
|
|
ctx.update(msg["contentTopic"].encode("utf-8"))
|
|
if "meta" in msg:
|
|
ctx.update(base64.b64decode(msg["meta"]))
|
|
ctx.update(int(msg["timestamp"]).to_bytes(8, byteorder="big"))
|
|
hash_bytes = ctx.digest()
|
|
return base64.b64encode(hash_bytes).decode("utf-8")
|