diff --git a/src/frostfs_testlib/storage/controllers/shards_watcher.py b/src/frostfs_testlib/storage/controllers/shards_watcher.py index ad07ff4..3d313f1 100644 --- a/src/frostfs_testlib/storage/controllers/shards_watcher.py +++ b/src/frostfs_testlib/storage/controllers/shards_watcher.py @@ -79,9 +79,7 @@ class ShardsWatcher: assert self._is_shard_present(shard_id) shards_with_new_errors = self.get_shards_with_new_errors() - assert ( - shard_id in shards_with_new_errors - ), f"Expected shard {shard_id} to have new errors, but haven't {self.shards_snapshots[-1]}" + assert shard_id in shards_with_new_errors, f"Expected shard {shard_id} to have new errors, but haven't {self.shards_snapshots[-1]}" @wait_for_success(300, 5) def await_for_shards_have_no_new_errors(self): @@ -97,6 +95,8 @@ class ShardsWatcher: response = shards_cli.list( endpoint=self.storage_node.get_control_endpoint(), + wallet=self.storage_node.get_remote_wallet_path(), + wallet_password=self.storage_node.get_wallet_password(), json_mode=True, ) @@ -109,6 +109,8 @@ class ShardsWatcher: ) return shards_cli.set_mode( endpoint=self.storage_node.get_control_endpoint(), + wallet=self.storage_node.get_remote_wallet_path(), + wallet_password=self.storage_node.get_wallet_password(), mode=mode, id=[shard_id], clear_errors=clear_errors,