From b028574a8c1b88717610400bb27e66a9daf87954 Mon Sep 17 00:00:00 2001 From: Roman Date: Mon, 20 Jan 2025 13:48:12 +0800 Subject: [PATCH] fix: add data extraction and decoding --- src/steps/da.py | 10 +++++++++- tests/data_integrity/test_data_integrity.py | 2 +- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/steps/da.py b/src/steps/da.py index f0922a9..c749e5f 100644 --- a/src/steps/da.py +++ b/src/steps/da.py @@ -26,8 +26,16 @@ class StepsDataAvailability(StepsCommon): @allure.step def get_data_range(self, app_id, start, end): + response = [] query = prepare_get_range_request(app_id, start, end) try: - self.node2.send_get_data_range_request(query) + response = 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 diff --git a/tests/data_integrity/test_data_integrity.py b/tests/data_integrity/test_data_integrity.py index 4c2c5f4..a645e34 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 + assert DATA_TO_DISPERSE[0] == received_data[0].decode("utf-8")