From d6a2cf92a262e718afd81009c236b98b396682af Mon Sep 17 00:00:00 2001
From: Dmitriy Zayakin <d.zayakin@yadro.com>
Date: Thu, 11 Jan 2024 14:51:07 +0300
Subject: [PATCH] [#155] Change args to optionally

Signed-off-by: Dmitriy Zayakin <d.zayakin@yadro.com>
---
 src/frostfs_testlib/cli/frostfs_cli/shards.py | 27 +++++++------------
 1 file changed, 10 insertions(+), 17 deletions(-)

diff --git a/src/frostfs_testlib/cli/frostfs_cli/shards.py b/src/frostfs_testlib/cli/frostfs_cli/shards.py
index 6b47ac2..1727249 100644
--- a/src/frostfs_testlib/cli/frostfs_cli/shards.py
+++ b/src/frostfs_testlib/cli/frostfs_cli/shards.py
@@ -68,11 +68,7 @@ class FrostfsCliShards(CliCommand):
         return self._execute_with_password(
             "control shards set-mode",
             wallet_password,
-            **{
-                param: value
-                for param, value in locals().items()
-                if param not in ["self", "wallet_password"]
-            },
+            **{param: value for param, value in locals().items() if param not in ["self", "wallet_password"]},
         )
 
     def dump(
@@ -105,18 +101,14 @@ class FrostfsCliShards(CliCommand):
         return self._execute_with_password(
             "control shards dump",
             wallet_password,
-            **{
-                param: value
-                for param, value in locals().items()
-                if param not in ["self", "wallet_password"]
-            },
+            **{param: value for param, value in locals().items() if param not in ["self", "wallet_password"]},
         )
 
     def list(
         self,
         endpoint: str,
-        wallet: str,
-        wallet_password: str,
+        wallet: Optional[str] = None,
+        wallet_password: Optional[str] = None,
         address: Optional[str] = None,
         json_mode: bool = False,
         timeout: Optional[str] = None,
@@ -135,12 +127,13 @@ class FrostfsCliShards(CliCommand):
         Returns:
             Command's result.
         """
+        if not wallet_password:
+            return self._execute(
+                "control shards list",
+                **{param: value for param, value in locals().items() if param not in ["self"]},
+            )
         return self._execute_with_password(
             "control shards list",
             wallet_password,
-            **{
-                param: value
-                for param, value in locals().items()
-                if param not in ["self", "wallet_password"]
-            },
+            **{param: value for param, value in locals().items() if param not in ["self", "wallet_password"]},
         )