add retries to query execution

1 files changed, 16 insertions, 1 deletions

diff --git a/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp b/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
index c671389d95..f86823c13b 100644
--- a/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
+++ b/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
@@ -11,7 +11,7 @@ using namespace NYdb;
 
 namespace {
 
-std::vector<TString> ScanQuerySelect(
+std::vector<TString> ScanQuerySelectSimple(
     NYdb::NTable::TTableClient client, const TString& tablePath,
     const std::vector<std::pair<TString, NYdb::EPrimitiveType>>& ydbSchema = TTestOlap::PublicSchema())
 {
@@ -90,6 +90,21 @@ std::vector<TString> ScanQuerySelect(
     return out;
 }
 
+std::vector<TString> ScanQuerySelect(
+    NYdb::NTable::TTableClient client, const TString& tablePath,
+    const std::vector<std::pair<TString, NYdb::EPrimitiveType>>& ydbSchema = TTestOlap::PublicSchema())
+{
+    for(ui32 iter = 0; iter < 3; iter++) {
+        try {
+            return ScanQuerySelectSimple(client, tablePath, ydbSchema);
+        } catch(...) {
+            /// o_O
+        }
+    }
+
+    return ScanQuerySelectSimple(client, tablePath, ydbSchema);
+}
+
 }
 
 Y_UNIT_TEST_SUITE(YdbTableBulkUpsertOlap) {