Add timeout for cli commands

Signed-off-by: Andrey Berezin <a.berezin@yadro.com>

pytest_tests/helpers/complex_object_actions.py

@@ -20,21 +20,25 @@ from pytest_tests.helpers import frostfs_verbs
 from pytest_tests.helpers.cluster import Cluster, StorageNode
 from pytest_tests.helpers.frostfs_verbs import head_object
 from pytest_tests.helpers.storage_object_info import StorageObjectInfo
-from pytest_tests.resources.common import WALLET_CONFIG
+from pytest_tests.resources.common import CLI_DEFAULT_TIMEOUT, WALLET_CONFIG
 
 logger = logging.getLogger("NeoLogger")
 
 
 def get_storage_object_chunks(
-    storage_object: StorageObjectInfo, shell: Shell, cluster: Cluster
+    storage_object: StorageObjectInfo,
+    shell: Shell,
+    cluster: Cluster,
+    timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
 ) -> list[str]:
     """
     Get complex object split objects ids (no linker object)
 
     Args:
-    storage_object: storage_object to get it's chunks
-    shell: client shell to do cmd requests
-    cluster: cluster object under test
+        storage_object: storage_object to get it's chunks
+        shell: client shell to do cmd requests
+        cluster: cluster object under test
+        timeout: Timeout for an operation.
 
     Returns:
     list of object ids of complex object chunks
@@ -48,6 +52,7 @@ def get_storage_object_chunks(
             shell,
             cluster.storage_nodes,
             is_direct=False,
+            timeout=timeout,
         )
         head = head_object(
             storage_object.wallet_file_path,
@@ -55,6 +60,7 @@ def get_storage_object_chunks(
             split_object_id,
             shell,
             cluster.default_rpc_endpoint,
+            timeout=timeout,
         )
 
         chunks_object_ids = []
@@ -65,7 +71,10 @@ def get_storage_object_chunks(
 
 
 def get_complex_object_split_ranges(
-    storage_object: StorageObjectInfo, shell: Shell, cluster: Cluster
+    storage_object: StorageObjectInfo,
+    shell: Shell,
+    cluster: Cluster,
+    timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
 ) -> list[Tuple[int, int]]:
 
     """
@@ -75,9 +84,10 @@ def get_complex_object_split_ranges(
     [(0, 30), (30, 30), (60, 30), (90, 10)]
 
     Args:
-    storage_object: storage_object to get it's chunks
-    shell: client shell to do cmd requests
-    cluster: cluster object under test
+        storage_object: storage_object to get it's chunks
+        shell: client shell to do cmd requests
+        cluster: cluster object under test
+        timeout: Timeout for an operation.
 
     Returns:
     list of object ids of complex object chunks
@@ -93,6 +103,7 @@ def get_complex_object_split_ranges(
             chunk_id,
             shell,
             cluster.default_rpc_endpoint,
+            timeout=timeout,
         )
 
         length = int(head["header"]["payloadLength"])
@@ -113,6 +124,7 @@ def get_link_object(
     bearer: str = "",
     wallet_config: str = WALLET_CONFIG,
     is_direct: bool = True,
+    timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
 ):
     """
     Args:
@@ -126,6 +138,7 @@ def get_link_object(
         wallet_config (optional, str): path to the frostfs-cli config file
         is_direct: send request directly to the node or not; this flag
                    turns into `--ttl 1` key
+        timeout: Timeout for an operation.
     Returns:
         (str): Link Object ID
         When no Link Object ID is found after all Storage Nodes polling,
@@ -144,6 +157,7 @@ def get_link_object(
                 is_direct=is_direct,
                 bearer=bearer,
                 wallet_config=wallet_config,
+                timeout=timeout,
             )
             if resp["link"]:
                 return resp["link"]
@@ -155,7 +169,12 @@ def get_link_object(
 
 @allure.step("Get Last Object")
 def get_last_object(
-    wallet: str, cid: str, oid: str, shell: Shell, nodes: list[StorageNode]
+    wallet: str,
+    cid: str,
+    oid: str,
+    shell: Shell,
+    nodes: list[StorageNode],
+    timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
 ) -> Optional[str]:
     """
     Args:
@@ -165,6 +184,7 @@ def get_last_object(
         oid (str): Large Object ID
         shell: executor for cli command
         nodes: list of nodes to do search on
+        timeout: Timeout for an operation.
     Returns:
         (str): Last Object ID
         When no Last Object ID is found after all Storage Nodes polling,
@@ -174,7 +194,14 @@ def get_last_object(
         endpoint = node.get_rpc_endpoint()
         try:
             resp = frostfs_verbs.head_object(
-                wallet, cid, oid, shell=shell, endpoint=endpoint, is_raw=True, is_direct=True
+                wallet,
+                cid,
+                oid,
+                shell=shell,
+                endpoint=endpoint,
+                is_raw=True,
+                is_direct=True,
+                timeout=timeout,
             )
             if resp["lastPart"]:
                 return resp["lastPart"]