change bruteforce for scripting (#10246)

ydb-platform · Oct 10, 2024 · 3e8a344 · 3e8a344
1 parent ebdb387
commit 3e8a344
Showing 1 changed file with 30 additions and 5 deletions.
diff --git a/ydb/core/kqp/ut/yql/kqp_scripting_ut.cpp b/ydb/core/kqp/ut/yql/kqp_scripting_ut.cpp
@@ -698,8 +698,10 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
         auto expected = TString("[") + EXPECTED_EIGHTSHARD_VALUE1 + "]";
 
         int unsuccessStatus = 0;
+        int successResults = 0;
+        constexpr int delta = NSan::PlainOrUnderSanitizer(1, 3);
 
-        for (int i = 1; i < maxTimeoutMs; i++) {
+        for (int i = 1; i < maxTimeoutMs; i += delta) {
             auto it = client.StreamExecuteYqlScript(R"(
                 SELECT * FROM `/Root/EightShard` WHERE Text = "Value1" ORDER BY Key;
             )", getExecuteYqlRequestSettings(i)).GetValueSync();
@@ -708,6 +710,11 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
                 try {
                     auto yson = StreamResultToYson(it, true);
                     CompareYson(expected, yson);
+                    ++successResults;
+                    if (successResults == 3) {
+                        break;
+                    }
+
                 } catch (const TStreamReadError& ex) {
                     unsuccessStatus++;
                     if (ex.Status != NYdb::EStatus::CLIENT_DEADLINE_EXCEEDED && ex.Status != NYdb::EStatus::TIMEOUT) {
@@ -736,14 +743,22 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
         int maxTimeoutMs = 1000;
 
         auto getExecuteYqlRequestSettings = GetExecuteYqlRequestSettingsFn(clientTimeout, operationTimeout);
+        int successResults = 0;
 
-        for (int i = 1; i < maxTimeoutMs; i++) {
+        constexpr int delta = NSan::PlainOrUnderSanitizer(1, 3);
+
+        for (int i = 1; i < maxTimeoutMs; i += delta) {
             auto result = client.ExecuteYqlScript(R"(
                 SELECT * FROM `/Root/EightShard` WHERE Text = "Value1" ORDER BY Key;
             )", getExecuteYqlRequestSettings(i)).GetValueSync();
 
             if (result.IsSuccess()) {
                 CompareYson(EXPECTED_EIGHTSHARD_VALUE1, FormatResultSetYson(result.GetResultSet(0)));
+                successResults++;
+                if (successResults == 3) {
+                    break;
+                }
+
             } else {
                 switch (result.GetStatus()) {
                     case NYdb::EStatus::CLIENT_DEADLINE_EXCEEDED:
@@ -771,7 +786,10 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
 
         auto expected = TString("[") + EXPECTED_EIGHTSHARD_VALUE1 + "]";
 
-        for (int i = 1; i < maxTimeoutMs; i++) {
+        constexpr int delta = NSan::PlainOrUnderSanitizer(1, 5);
+        int successResults = 0;
+
+        for (int i = 1; i < maxTimeoutMs; i += delta) {
             auto it = client.StreamExecuteYqlScript(R"(
                 SELECT * FROM `/Root/EightShard` WHERE Text = "Value1" ORDER BY Key;
             )", TExecuteYqlRequestSettings().CancelAfter(TDuration::MilliSeconds(i))).GetValueSync();
@@ -780,6 +798,11 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
             try {
                 auto yson = StreamResultToYson(it, true);
                 CompareYson(expected, yson);
+                ++successResults;
+                if (successResults == 3) {
+                    break;
+                }
+
             } catch (const TStreamReadError& ex) {
                 if (ex.Status != NYdb::EStatus::CANCELLED) {
                     TStringStream msg;
@@ -813,8 +836,9 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
         };
 
         TString expected;
+        const int delta = NSan::PlainOrUnderSanitizer(1, 5);
 
-        for (int i = 1; i <= maxTimeoutMs; i++) {
+        for (int i = 1; i <= maxTimeoutMs; i += delta) {
             auto it = client.StreamExecuteYqlScript(Sprintf(R"(
                 UPSERT INTO `/Root/EightShard` (Key, Data, Text) VALUES (%lu, 100500, "newrecords");
                 COMMIT;
@@ -860,9 +884,10 @@ Y_UNIT_TEST_SUITE(KqpScripting) {
             return Sprintf(R"([[100500];[%luu];["newrecords"]])", key);
         };
 
+        constexpr int delta = NSan::PlainOrUnderSanitizer(1, 5);
         TString expected;
 
-        for (int i = 1; i <= maxTimeoutMs; i++) {
+        for (int i = 1; i <= maxTimeoutMs; i += delta) {
             auto result = client.ExecuteYqlScript(Sprintf(R"(
                 UPSERT INTO `/Root/EightShard` (Key, Data, Text) VALUES (%lu, 100500, "newrecords");
                 COMMIT;