(ignite-3) branch main updated: IGNITE-23706 Sql. Fix flaky testCursorIsClosedAfterAllDataReadWithNodeFailure (#4750)

[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 dcaad51851 IGNITE-23706 Sql. Fix flaky 
testCursorIsClosedAfterAllDataReadWithNodeFailure (#4750)
dcaad51851 is described below

commit dcaad5185167c57784c7b5d937314c99e229fa4d
Author: Evgeniy Stanilovskiy <stanilov...@gmail.com>
AuthorDate: Thu Nov 21 12:54:52 2024 +0300

    IGNITE-23706 Sql. Fix flaky 
testCursorIsClosedAfterAllDataReadWithNodeFailure (#4750)
---
 .../ignite/internal/sql/engine/exec/ExecutionServiceImplTest.java    | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git 
a/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/ExecutionServiceImplTest.java
 
b/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/ExecutionServiceImplTest.java
index 72c8cb29e0..7728fa6855 100644
--- 
a/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/ExecutionServiceImplTest.java
+++ 
b/modules/sql-engine/src/test/java/org/apache/ignite/internal/sql/engine/exec/ExecutionServiceImplTest.java
@@ -587,7 +587,10 @@ public class ExecutionServiceImplTest extends 
BaseIgniteAbstractTest {
                 // before other fragments.
                 node.taskExecutor.execute(() -> {
                     try {
-                        if (msg instanceof QueryStartResponseImpl) {
+                        // We need to block only root fragment execution, 
otherwise due to asynchronous fragments processing fragments with
+                        // different fragmentId can be processed before root 
(fragmentId=0) and block pool threads for
+                        // further processing jobs.
+                        if (msg instanceof QueryStartResponseImpl && 
((QueryStartResponseImpl) msg).fragmentId() == 0) {
                             startResponse.countDown();
                             nodeFailedLatch.await();
                         }