From d88c9219e4f862494ba205f3de337ffb73b6e0a3 Mon Sep 17 00:00:00 2001
From: "dk.technoboy" <guo.jiwei@immomo.com>
Date: Tue, 17 Dec 2019 11:06:02 +0800
Subject: [PATCH] updates for reference ZkServer

---
 .../dolphinscheduler/common/queue/TaskQueueImplTest.java   | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/queue/TaskQueueImplTest.java b/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/queue/TaskQueueImplTest.java
index efee627676..14e90ebcdc 100644
--- a/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/queue/TaskQueueImplTest.java
+++ b/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/queue/TaskQueueImplTest.java
@@ -19,7 +19,7 @@ package org.apache.dolphinscheduler.common.queue;
 import org.apache.dolphinscheduler.common.Constants;
 import org.apache.dolphinscheduler.common.utils.IpUtils;
 import org.apache.dolphinscheduler.common.utils.OSUtils;
-import org.apache.dolphinscheduler.common.zk.StandaloneZKServerForTest;
+import org.apache.dolphinscheduler.common.zk.ZKServer;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Ignore;
@@ -35,7 +35,7 @@ import static org.junit.Assert.assertEquals;
 /**
  * task queue test
  */
-public class TaskQueueImplTest extends StandaloneZKServerForTest {
+public class TaskQueueImplTest {
 
     private static final Logger logger = LoggerFactory.getLogger(TaskQueueImplTest.class);
 
@@ -43,7 +43,7 @@ public class TaskQueueImplTest extends StandaloneZKServerForTest {
 
     @Before
     public void before(){
-        super.before();
+        ZKServer.start();
 
         tasksQueue = TaskQueueFactory.getTaskQueueInstance();
 
@@ -57,6 +57,7 @@ public class TaskQueueImplTest extends StandaloneZKServerForTest {
     public void after(){
         //clear all data
         tasksQueue.delete();
+        ZKServer.stop();
     }