From 5b045cc5a0745fa0c4dae0c3c81cc948939cc0cb Mon Sep 17 00:00:00 2001 From: Gabriel mermelstein Date: Thu, 30 Jan 2025 12:58:50 +0200 Subject: [PATCH] sending hash and cursor as hex --- src/libs/common.py | 12 ++++++++++++ tests/store/test_cursor.py | 6 +++--- tests/store/test_hashes.py | 4 ++-- 3 files changed, 17 insertions(+), 5 deletions(-) diff --git a/src/libs/common.py b/src/libs/common.py index b7c57cb0..cd0dfc19 100644 --- a/src/libs/common.py +++ b/src/libs/common.py @@ -26,6 +26,18 @@ def to_base64(input_data): return base64_encoded.decode() +def to_hex(input_data): + if isinstance(input_data, str): + input_bytes = input_data.encode() + elif isinstance(input_data, int): + input_bytes = str(input_data).encode() + elif isinstance(input_data, bytes): + input_bytes = input_data + else: + input_bytes = str(input_data).encode() + return "0x" + input_bytes.hex() + + def attach_allure_file(file): logger.debug(f"Attaching file {file}") allure.attach.file(file, name=os.path.basename(file), attachment_type=allure.attachment_type.TEXT) diff --git a/tests/store/test_cursor.py b/tests/store/test_cursor.py index 4e3820dd..caec4710 100644 --- a/tests/store/test_cursor.py +++ b/tests/store/test_cursor.py @@ -1,6 +1,6 @@ import pytest from src.env_vars import NODE_1, NODE_2 -from src.libs.common import to_base64 +from src.libs.common import to_base64, to_hex from src.node.store_response import StoreResponse from src.steps.store import StepsStore @@ -70,8 +70,8 @@ class TestCursor(StepsStore): def test_passing_invalid_cursor(self): for i in range(4): self.publish_message(message=self.create_message(payload=to_base64(f"Message_{i}"))) - # creating a invalid base64 cursor - cursor = to_base64("test") + # creating a invalid hex cursor + cursor = to_hex("test") for node in self.store_nodes: store_response = self.get_messages_from_store(node, page_size=100, cursor=cursor) assert not store_response.messages, "Messages found" diff --git a/tests/store/test_hashes.py b/tests/store/test_hashes.py index f296bcb3..5c3158ba 100644 --- a/tests/store/test_hashes.py +++ b/tests/store/test_hashes.py @@ -1,6 +1,6 @@ import pytest from src.env_vars import NODE_2 -from src.libs.common import to_base64 +from src.libs.common import to_base64, to_hex from src.libs.custom_logger import get_custom_logger from src.steps.store import StepsStore from src.test_data import SAMPLE_INPUTS @@ -46,7 +46,7 @@ class TestHashes(StepsStore): def test_store_with_invalid_hash(self): for i in range(4): self.publish_message(message=self.create_message(payload=to_base64(f"Message_{i}"))) - invalid_hash = to_base64("test") + invalid_hash = to_hex("test") for node in self.store_nodes: try: store_response = self.get_messages_from_store(node, hashes=invalid_hash, page_size=50)