Commit 85b8324e authored by dk.technoboy's avatar dk.technoboy
Browse files

add BaseTaskQueueTest for UT

parent d88c9219

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

0 → 100644
+43 −0
Original line number Diff line number Diff line 
/*

 * 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();

    }

}

dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/zk/TestZk.java→dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/zk/TestZkServer.java

+1 −1
Original line number Diff line number Diff line
@@ -24,7 +24,7 @@ import org.junit.Test; 
/**

 * demo for using zkServer

 */

public class TestZk {

public class TestZkServer {



    @Before

    public void before(){

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

+2 −1
Original line number Diff line number Diff line
@@ -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);

        }