Add size and rows limits for batch operations large tests (#17270)

2 files changed, 14 insertions, 7 deletions

diff --git a/ydb/core/kqp/ut/batch_operations/kqp_batch_delete_ut.cpp b/ydb/core/kqp/ut/batch_operations/kqp_batch_delete_ut.cpp
index f977ea070cc..19b5c64ad6a 100644
--- a/ydb/core/kqp/ut/batch_operations/kqp_batch_delete_ut.cpp
+++ b/ydb/core/kqp/ut/batch_operations/kqp_batch_delete_ut.cpp
@@ -236,20 +236,25 @@ Y_UNIT_TEST_SUITE(KqpBatchDelete) {
     }
 
     Y_UNIT_TEST(Large_1) {
-        for (size_t size = 1; size <= 1000; size *= 10) {
+        ui32 sizeLimit = NSan::PlainOrUnderSanitizer(1000, 100);
+        for (size_t size = 1; size <= sizeLimit; size *= 10) {
             TestLarge(size, 100);
         }
     }
 
     Y_UNIT_TEST(Large_2) {
-        for (size_t size = 100; size <= 10000; size *= 10) {
-            TestLarge(size, 10000);
+        ui32 sizeLimit = NSan::PlainOrUnderSanitizer(10000, 1000);
+        ui32 shardRows = NSan::PlainOrUnderSanitizer(10000, 5000);
+        for (size_t size = 100; size <= sizeLimit; size *= 10) {
+            TestLarge(size, shardRows);
         }
     }
 
     Y_UNIT_TEST(Large_3) {
-        for (size_t size = 1000; size <= 100000; size *= 10) {
-            TestLarge(size, 100000);
+        ui32 sizeLimit = NSan::PlainOrUnderSanitizer(100000, 10000);
+        ui32 shardRows = NSan::PlainOrUnderSanitizer(50000, 25000);
+        for (size_t size = 1000; size <= sizeLimit; size *= 10) {
+            TestLarge(size, shardRows);
         }
     }
 
diff --git a/ydb/core/kqp/ut/batch_operations/kqp_batch_update_ut.cpp b/ydb/core/kqp/ut/batch_operations/kqp_batch_update_ut.cpp
index 520841792a5..f16cd5e73e0 100644
--- a/ydb/core/kqp/ut/batch_operations/kqp_batch_update_ut.cpp
+++ b/ydb/core/kqp/ut/batch_operations/kqp_batch_update_ut.cpp
@@ -309,15 +309,17 @@ Y_UNIT_TEST_SUITE(KqpBatchUpdate) {
 
     Y_UNIT_TEST(Large_2) {
         ui32 sizeLimit = NSan::PlainOrUnderSanitizer(10000, 1000);
+        ui32 shardRows = NSan::PlainOrUnderSanitizer(10000, 5000);
         for (size_t size = 100; size <= sizeLimit; size *= 10) {
-            TestLarge(size, 10000);
+            TestLarge(size, shardRows);
         }
     }
 
     Y_UNIT_TEST(Large_3) {
         ui32 sizeLimit = NSan::PlainOrUnderSanitizer(100000, 10000);
+        ui32 shardRows = NSan::PlainOrUnderSanitizer(50000, 25000);
         for (size_t size = 1000; size <= sizeLimit; size *= 10) {
-            TestLarge(size, 100000);
+            TestLarge(size, shardRows);
         }
     }