(ignite-3) branch main updated: IGNITE-21895 Assertion rarely raised on LimitExecutionTest.testLimit (#3531)

[zstan]

This is an automated email from the ASF dual-hosted git repository.

zstan pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/ignite-3.git


The following commit(s) were added to refs/heads/main by this push:
     new 3dd95601f1 IGNITE-21895 Assertion rarely raised on 
LimitExecutionTest.testLimit (#3531)
3dd95601f1 is described below

commit 3dd95601f1a073b274be05d0ecdbe7f76d65db36
Author: Evgeniy Stanilovskiy <stanilov...@gmail.com>
AuthorDate: Fri Apr 5 10:33:29 2024 +0300

    IGNITE-21895 Assertion rarely raised on LimitExecutionTest.testLimit (#3531)
---
 .../internal/sql/engine/exec/rel/AbstractExecutionTest.java    | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git 
a/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/rel/AbstractExecutionTest.java
 
b/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/rel/AbstractExecutionTest.java
index 5218a83be4..d97dff0a05 100644
--- 
a/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/rel/AbstractExecutionTest.java
+++ 
b/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/rel/AbstractExecutionTest.java
@@ -100,14 +100,18 @@ public abstract class AbstractExecutionTest<T> extends 
IgniteAbstractTest {
                     false,
                     0);
 
-            StripedThreadPoolExecutor stripedThreadPoolExecutor = 
(StripedThreadPoolExecutor) IgniteTestUtils.getFieldValue(
+            StripedThreadPoolExecutor stripedThreadPoolExecutor = 
IgniteTestUtils.getFieldValue(
                     taskExecutor,
                     QueryTaskExecutorImpl.class,
                     "stripedThreadPoolExecutor"
             );
-            stripedThreadPoolExecutor.shutdown();
 
-            IgniteTestUtils.setFieldValue(taskExecutor, 
"stripedThreadPoolExecutor", testExecutor);
+            // change it once on startup
+            if (!(stripedThreadPoolExecutor instanceof 
IgniteTestStripedThreadPoolExecutor)) {
+                stripedThreadPoolExecutor.shutdown();
+
+                IgniteTestUtils.setFieldValue(taskExecutor, 
"stripedThreadPoolExecutor", testExecutor);
+            }
         }
 
         FragmentDescription fragmentDesc = getFragmentDescription();