From 1fe6354ac60f8fd651b2af2f7cc479e0e1dbf76a Mon Sep 17 00:00:00 2001
From: Roman <zajic@zajic.net>
Date: Tue, 8 Apr 2025 14:17:18 +0800
Subject: [PATCH] fix: add get_storage_block to storage

---
 src/node/nomos_node.py |  3 +++
 src/steps/storage.py   | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 37 insertions(+)
 create mode 100644 src/steps/storage.py

diff --git a/src/node/nomos_node.py b/src/node/nomos_node.py
index 106ae2c..3e7df22 100644
--- a/src/node/nomos_node.py
+++ b/src/node/nomos_node.py
@@ -160,3 +160,6 @@ class NomosNode:
 
     def send_get_commitments_request(self, data):
         return self._api.da_get_commitments(data)
+
+    def send_get_storage_block_request(self, data):
+        return self._api.storage_block(data)
diff --git a/src/steps/storage.py b/src/steps/storage.py
new file mode 100644
index 0000000..495ea0c
--- /dev/null
+++ b/src/steps/storage.py
@@ -0,0 +1,34 @@
+import allure
+from tenacity import retry, stop_after_delay, wait_fixed
+
+from src.libs.custom_logger import get_custom_logger
+from src.steps.common import StepsCommon
+
+logger = get_custom_logger(__name__)
+
+
+def prepare_get_storage_block_request(header_id):
+    query_data = {"header_id": header_id}
+    return query_data
+
+
+class StepsStorage(StepsCommon):
+    @allure.step
+    def get_storage_block(self, node, header_id, **kwargs):
+
+        timeout_duration = kwargs.get("timeout_duration", 65)
+        interval = kwargs.get("interval", 0.1)
+
+        query = prepare_get_storage_block_request(header_id)
+
+        @retry(stop=stop_after_delay(timeout_duration), wait=wait_fixed(interval), reraise=True)
+        def get_block():
+            try:
+                response = node.send_get_storage_block_request(query)
+            except Exception as ex:
+                logger.error(f"Exception while retrieving commitments: {ex}")
+                raise
+
+            return response
+
+        return get_block()