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")