Browse Source

add BaseTaskQueueTest for UT

dk.technoboy 5 years ago
parent
commit
85b8324ee3

+ 43 - 0
dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/queue/BaseTaskQueueTest.java

@@ -0,0 +1,43 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.dolphinscheduler.common.queue;
+
+import org.apache.dolphinscheduler.common.zk.ZKServer;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+
+/**
+ * base task queue test for only start zk server once
+ */
+public class BaseTaskQueueTest {
+
+    protected static ITaskQueue tasksQueue = null;
+
+    @BeforeClass
+    public static void setup() {
+        ZKServer.start();
+        tasksQueue = TaskQueueFactory.getTaskQueueInstance();
+        //clear all data
+        tasksQueue.delete();
+    }
+
+    @AfterClass
+    public static void tearDown() {
+        tasksQueue.delete();
+        ZKServer.stop();
+    }
+}

+ 1 - 1
dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/zk/TestZk.java

@@ -24,7 +24,7 @@ import org.junit.Test;
 /**
  * demo for using zkServer
  */
-public class TestZk {
+public class TestZkServer {
 
     @Before
     public void before(){

+ 2 - 1
dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/zk/ZKServer.java

@@ -79,7 +79,7 @@ public class ZKServer {
      * @param port The port to listen on
      */
     public static void startLocalZkServer(final int port) {
-        startLocalZkServer(port, org.apache.commons.io.FileUtils.getTempDirectoryPath() + File.separator + "test-" + System.currentTimeMillis());
+        startLocalZkServer(port, org.apache.commons.io.FileUtils.getTempDirectoryPath() + "test-" + System.currentTimeMillis());
     }
 
     /**
@@ -137,6 +137,7 @@ public class ZKServer {
     public static void stop() {
         try {
             stopLocalZkServer(true);
+            logger.info("zk server stopped");
         } catch (Exception e) {
             logger.error("Failed to stop ZK ",e);
         }