Make sure the task gets unblocked

elastic · May 28, 2024 · f235b87 · f235b87
1 parent a61e9c9
commit f235b87
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/src/test/java/org/elasticsearch/action/admin/cluster/node/tasks/TestTaskPlugin.java b/server/src/test/java/org/elasticsearch/action/admin/cluster/node/tasks/TestTaskPlugin.java
@@ -300,12 +300,13 @@ protected NodeResponse nodeOperation(NodeRequest request, Task task) {
             logger.info("Test task started on the node {}", clusterService.localNode());
             if (request.shouldBlock) {
                 try {
-                    waitUntil(() -> {
+                    boolean unblocked = waitUntil(() -> {
                         if (((CancellableTask) task).isCancelled()) {
                             throw new RuntimeException("Cancelled!");
                         }
                         return ((TestTask) task).isBlocked() == false;
                     });
+                    assert unblocked : task + " hasn't been unblocked";
                 } catch (InterruptedException ex) {
                     Thread.currentThread().interrupt();
                 }