org.apache.flink.test.util

Class AbstractTestBase

java.lang.Object

org.apache.flink.util.TestLogger

org.apache.flink.test.util.AbstractTestBase

Direct Known Subclasses:

JavaProgramTestBase, MultipleProgramsTestBase

public abstract class AbstractTestBase
extends org.apache.flink.util.TestLogger

Base class for unit tests that run multiple tests and want to reuse the same Flink cluster. This saves a significant amount of time, since the startup and shutdown of the Flink clusters (including actor systems, etc) usually dominates the execution of the actual tests.

To write a unit test against this test base, simply extend it and add one or more regular test methods and retrieve the StreamExecutionEnvironment from the context:

    @Test
   public void someTest() {
       ExecutionEnvironment env = ExecutionEnvironment.getExecutionEnvironment();
       // test code
       env.execute();
   }

    @Test
   public void anotherTest() {
       StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
       // test code
       env.execute();
   }

 

Field Summary

Fields

Modifier and Type	Field and Description
static MiniClusterWithClientResource	MINI_CLUSTER_RESOURCE
static org.junit.rules.TemporaryFolder	TEMPORARY_FOLDER

Fields inherited from class org.apache.flink.util.TestLogger

log, nameProvider, watchman

Constructor Summary

Constructors

Constructor and Description
AbstractTestBase()

Method Summary

Modifier and Type	Method and Description
void	cleanupRunningJobs()
File	createAndRegisterTempFile(String fileName)
String	createTempFile(String fileName, String contents)
String	getTempDirPath(String dirName)
String	getTempFilePath(String fileName)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

MINI_CLUSTER_RESOURCE

public static final MiniClusterWithClientResource MINI_CLUSTER_RESOURCE

TEMPORARY_FOLDER

public static final org.junit.rules.TemporaryFolder TEMPORARY_FOLDER

Constructor Detail

AbstractTestBase

public AbstractTestBase()

Method Detail

cleanupRunningJobs

public final void cleanupRunningJobs()
                              throws Exception

Throws:

Exception

getTempDirPath

public String getTempDirPath(String dirName)
                      throws IOException

Throws:

IOException

getTempFilePath

public String getTempFilePath(String fileName)
                       throws IOException

Throws:

IOException

createTempFile

public String createTempFile(String fileName,
                             String contents)
                      throws IOException

Throws:

IOException

createAndRegisterTempFile

public File createAndRegisterTempFile(String fileName)
                               throws IOException

Throws:

IOException