11package tech .ydb .query .impl ;
22
3+ import java .time .Duration ;
34import java .util .concurrent .CompletableFuture ;
45
56import org .junit .AfterClass ;
@@ -79,6 +80,7 @@ public static void dropAll() {
7980
8081 private Status selectWithPool (String poolName ) {
8182 ExecuteQuerySettings settings = ExecuteQuerySettings .newBuilder ()
83+ .withRequestTimeout (Duration .ofSeconds (5 ))
8284 .withExecMode (QueryExecMode .EXECUTE )
8385 .withResourcePool (poolName )
8486 .withStatsMode (QueryStatsMode .FULL )
@@ -95,7 +97,7 @@ private Status createResourcePool(String poolName) {
9597 String createPool = "CREATE RESOURCE POOL " + poolName + " WITH ("
9698 + "CONCURRENT_QUERY_LIMIT=10,"
9799 + "QUEUE_SIZE=1000,"
98- + "DATABASE_LOAD_CPU_THRESHOLD=80 );" ;
100+ + "DATABASE_LOAD_CPU_THRESHOLD=20 );" ;
99101 return retryCtx .supplyResult (s -> s .createQuery (createPool , TxMode .NONE ).execute ()).join ().getStatus ();
100102 }
101103
@@ -156,9 +158,11 @@ public void useDifferentPoolsTest() {
156158 Assert .assertTrue ("Cannot create resource pool " + create4 , create4 .isSuccess ());
157159
158160 ExecuteQuerySettings pool3 = ExecuteQuerySettings .newBuilder ()
161+ .withRequestTimeout (Duration .ofSeconds (5 ))
159162 .withResourcePool ("test_pool3" )
160163 .build ();
161164 ExecuteQuerySettings pool4 = ExecuteQuerySettings .newBuilder ()
165+ .withRequestTimeout (Duration .ofSeconds (5 ))
162166 .withResourcePool ("test_pool4" )
163167 .build ();
164168
0 commit comments