From b3991d9dee307191f200a6b660d7264d874d3cb9 Mon Sep 17 00:00:00 2001
From: CalvinKirs <acm_master@163.com>
Date: Wed, 25 Nov 2020 13:43:14 +0800
Subject: [PATCH] Time is too small and the task is not finished

---
 .../processor/queue/TaskResponseServiceTest.java   | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java b/dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java
index 0e8156a44a..c67f477b6d 100644
--- a/dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java
+++ b/dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java
@@ -88,11 +88,17 @@ public class TaskResponseServiceTest {
 
     @After
     public void after() {
-        try {
-            Thread.sleep(60000);
-        } catch (InterruptedException ignored) {
-            Thread.currentThread().interrupt();
+
+        long startTime = System.currentTimeMillis();
+
+        long maxWaitTime = 3 * 60 * 1000;
+
+        while (System.currentTimeMillis() - maxWaitTime - startTime < 0) {
+            if (taskResponseService.getEventQueue().size() == 0) {
+                return;
+            }
         }
+
         Assert.assertEquals(0, taskResponseService.getEventQueue().size());
         taskResponseService.stop();
     }