org.apache.drill.yarn.scripts

Class ScriptUtils

java.lang.Object

org.apache.drill.yarn.scripts.ScriptUtils

public class ScriptUtils
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ScriptUtils.DrillbitRun
static class	ScriptUtils.RunResult Handy run result class to capture the information we need for testing and to do various kinds of validation on it.
static class	ScriptUtils.ScriptRunner The "business end" of the tests: runs drillbit.sh and captures results.

Field Summary

Fields

Modifier and Type	Field and Description
File	distribDir
boolean	externalLogDir
File	javaHome
static String[]	scripts Scripts we must copy from the source tree to create a simulated Drill bin directory.
static String[]	sqlLineArgs Out-of-the-box command-line arguments when launching sqlline.
static String[]	stdArgs Out-of-the-box command-line arguments when launching Drill.
static String	TEMP_DIR
File	testDir
File	testDrillHome
File	testLogDir
File	testSiteDir
static boolean	USE_SOURCE
static boolean	useSource

Method Summary

Modifier and Type	Method and Description
void	copyFile(File source, File dest)
File	createDir(File dir) Remove, then create a directory.
void	createEnvFile(File file, Map<String,String> env, boolean overrideValue) Create a drill-env.sh or distrib-env.sh file with the given environment in the recommended format.
void	createEnvFileWithCondition(File file, Map<String,String> condition, Map<String,String> env, boolean overrideValue) Creates a drill-env.sh or distrib-env.sh file with the given environment under a given condition.
void	createMockConf(File siteDir) Build a "starter" conf or site directory by creating a mock drill-override.conf file.
void	createMockDistrib()
ScriptUtils	fromDistrib(String distrib)
ScriptUtils	fromSource(String sourceDir)
void	initialSetup() Create the basic test directory.
static ScriptUtils	instance()
File	makeDummyJar(File dir, String prefix) The tests should not require jar files, but we simulate them to be a bit more realistic.
void	removeDir(File dir)
void	writeEnvFile(PrintWriter out, String key, String value, boolean overrideValue)
void	writeFile(File file, String contents) Create a simple text file with the given contents.

Methods inherited from class java.lang.Object

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

Field Detail

distribDir

public File distribDir

javaHome

public File javaHome

testDir

public File testDir

testDrillHome

public File testDrillHome

testSiteDir

public File testSiteDir

testLogDir

public File testLogDir

externalLogDir

public boolean externalLogDir

sqlLineArgs

public static String[] sqlLineArgs

Out-of-the-box command-line arguments when launching sqlline. Order is not important here (though it is to Java.)

USE_SOURCE

public static final boolean USE_SOURCE

See Also:

Constant Field Values

TEMP_DIR

public static final String TEMP_DIR

See Also:

Constant Field Values

useSource

public static boolean useSource

stdArgs

public static String[] stdArgs

Out-of-the-box command-line arguments when launching Drill. Order is not important here (though it is to Java.)

scripts

public static String[] scripts

Scripts we must copy from the source tree to create a simulated Drill bin directory.

Method Detail

instance

public static ScriptUtils instance()

fromSource

public ScriptUtils fromSource(String sourceDir)

fromDistrib

public ScriptUtils fromDistrib(String distrib)

initialSetup

public void initialSetup()
                  throws IOException

Create the basic test directory. Tests add or remove details.

Throws:

IOException

createMockDistrib

public void createMockDistrib()
                       throws IOException

Throws:

IOException

makeDummyJar

public File makeDummyJar(File dir,
                         String prefix)
                  throws IOException

The tests should not require jar files, but we simulate them to be a bit more realistic. Since we dont' run Java, the jar files can be simulated.

Throws:

IOException

writeFile

public void writeFile(File file,
                      String contents)
               throws IOException

Create a simple text file with the given contents.

Throws:

IOException

writeEnvFile

public void writeEnvFile(PrintWriter out,
                         String key,
                         String value,
                         boolean overrideValue)

createEnvFile

public void createEnvFile(File file,
                          Map<String,String> env,
                          boolean overrideValue)
                   throws IOException

Create a drill-env.sh or distrib-env.sh file with the given environment in the recommended format. different formats based on overrideValue flag

Parameters:

file - - File instance to set environment variables in

env - - Environment to be placed inside File

overrideValue - - true - Set environment value such that it overrides previously set value - false - Set environment value in recommended format.

Throws:

IOException

createEnvFileWithCondition

public void createEnvFileWithCondition(File file,
                                       Map<String,String> condition,
                                       Map<String,String> env,
                                       boolean overrideValue)
                                throws IOException

Creates a drill-env.sh or distrib-env.sh file with the given environment under a given condition. If size of env map is smaller than condition map then last env entry is repeated for rest of conditions.

Parameters:

file - - File instance to set environment and condition in

condition - - Conditions to guard environment variable

env - - Environment to be placed inside File

overrideValue - - true - Set environment value such that it overrides previously set value - false - Set environment value in recommended format.

Throws:

IOException

createMockConf

public void createMockConf(File siteDir)
                    throws IOException

Build a "starter" conf or site directory by creating a mock drill-override.conf file.

Throws:

IOException

removeDir

public void removeDir(File dir)
               throws IOException

Throws:

IOException

createDir

public File createDir(File dir)
               throws IOException

Remove, then create a directory.

Throws:

IOException

copyFile

public void copyFile(File source,
                     File dest)
              throws IOException

Throws:

IOException