From 22ccfe5f2fd4827ebdca8e5dbc821fbb3528a696 Mon Sep 17 00:00:00 2001
From: Олег <150132506+iddqdex@users.noreply.github.com>
Date: Wed, 26 Mar 2025 13:33:55 +0300
Subject: Fix return code processing (#16253)

---
 ydb/tests/olap/lib/ydb_cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ydb/tests/olap/lib/ydb_cli.py b/ydb/tests/olap/lib/ydb_cli.py
index 27e45905dce..5014bc41788 100644
--- a/ydb/tests/olap/lib/ydb_cli.py
+++ b/ydb/tests/olap/lib/ydb_cli.py
@@ -170,7 +170,7 @@ class YdbCliHelper:
                     self._add_error(f'Iteration {iter}: {msg}')
 
         def _process_returncode(self, returncode) -> None:
-            if returncode != 0 and len([x for x in filter(lambda x: x.error_message or x.warning_message, self.result.iterations.values())]) == 0:
+            if returncode != 0 and not self.result.error_message and not self.result.warning_message:
                 self._add_error(f'Invalid return code: {returncode} instead 0. stderr: {self.result.stderr}')
 
         def _load_plan(self, name: str) -> YdbCliHelper.QueryPlan:
-- 
cgit v1.3