trying to fix test

1 files changed, 13 insertions, 1 deletions

diff --git a/ydb/services/ydb/ydb_olapstore_ut.cpp b/ydb/services/ydb/ydb_olapstore_ut.cpp
index 2f42612495..7919aac5a4 100644
--- a/ydb/services/ydb/ydb_olapstore_ut.cpp
+++ b/ydb/services/ydb/ydb_olapstore_ut.cpp
@@ -264,7 +264,7 @@ Y_UNIT_TEST_SUITE(YdbOlapStore) {
         }
     }
 
-    TString RunQuery(TDriver& connection, const TString& query) {
+    TString RunQuerySimple(TDriver& connection, const TString& query) {
         auto client = NYdb::NTable::TTableClient(connection);
 
         NYdb::NTable::TStreamExecScanQuerySettings execSettings;
@@ -280,6 +280,18 @@ Y_UNIT_TEST_SUITE(YdbOlapStore) {
         return result.ResultSetYson;
     }
 
+    TString RunQuery(TDriver& connection, const TString& query) {
+        for(ui32 iter = 0; iter < 3; iter++) {
+            try {
+                return RunQuerySimple(connection, query);
+            } catch(...) {
+                /// o_O
+            }
+        }
+
+        return RunQuerySimple(connection, query);
+    }
+
     // Create OLTP and OLAP tables with the same set of columns and same PK
     void CreateTestTables(const TServerSettings& settings, const TString& tableName, const TString& sharding) {
         TTestOlapTableOptions opts;