From 578efed1505ac70ba67e0acd40a1aa503689c5e5 Mon Sep 17 00:00:00 2001 From: Roman Date: Mon, 20 Jan 2025 14:02:18 +0800 Subject: [PATCH] fix: decode data bytes --- src/steps/da.py | 16 ++++++++-------- tests/data_integrity/test_data_integrity.py | 2 +- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/src/steps/da.py b/src/steps/da.py index c749e5f..f999eb0 100644 --- a/src/steps/da.py +++ b/src/steps/da.py @@ -1,3 +1,5 @@ +import json + import allure from src.steps.common import StepsCommon @@ -26,16 +28,14 @@ class StepsDataAvailability(StepsCommon): @allure.step def get_data_range(self, app_id, start, end): - response = [] + response_bytes = [] query = prepare_get_range_request(app_id, start, end) try: - response = self.node2.send_get_data_range_request(query) + response_bytes = self.node2.send_get_data_range_request(query) except Exception as ex: assert "Bad Request" in str(ex) or "Internal Server Error" in str(ex) - # Extract data for each index in received order - extracted_data = [] - for item in response: - extracted_data.append(item[1]) - - return extracted_data + # Extract data ss string for each index in the received order + response = response_bytes.decode("utf-8") + parsed_data = json.loads(response) + return [item[1] for item in parsed_data] diff --git a/tests/data_integrity/test_data_integrity.py b/tests/data_integrity/test_data_integrity.py index a645e34..a993ad6 100644 --- a/tests/data_integrity/test_data_integrity.py +++ b/tests/data_integrity/test_data_integrity.py @@ -15,4 +15,4 @@ class TestDataIntegrity(StepsDataAvailability): def test_da_sampling_determines_data_presence(self): self.disperse_data(DATA_TO_DISPERSE[0], [0] * 31 + [1], [0] * 8) received_data = self.get_data_range([0] * 31 + [1], [0] * 8, [0] * 7 + [5]) - assert DATA_TO_DISPERSE[0] == received_data[0].decode("utf-8") + assert DATA_TO_DISPERSE[0] == received_data[0]