1 files changed, 31 insertions, 6 deletions
diff --git a/src/main/scala/parallelai/spyglass/hbase/testing/HBaseSaltTesterRunner.scala b/src/main/scala/parallelai/spyglass/hbase/testing/HBaseSaltTesterRunner.scala
index e6744b7..a8de7d6 100644
--- a/src/main/scala/parallelai/spyglass/hbase/testing/HBaseSaltTesterRunner.scala
+++ b/src/main/scala/parallelai/spyglass/hbase/testing/HBaseSaltTesterRunner.scala
@@ -1,6 +1,7 @@
 package parallelai.spyglass.hbase.testing
 
 import parallelai.spyglass.base.JobRunner
+import com.twitter.scalding.Args
 
 object HBaseSaltTesterRunner extends App {
   
@@ -18,12 +19,36 @@ object HBaseSaltTesterRunner extends App {
   assert  (quorum != null, {"Environment Variable BIGDATA_QUORUM_NAMES is undefined or Null"})
   println( "Quorum is [%s]".format(quorum) )
 
+  val mArgs = Args(args) // get ("make-data")
 
-  JobRunner.main(Array(classOf[HBaseSaltTester].getName, 
-      "--hdfs", 
-      "--app.conf.path", appPath, 
+  val make = mArgs.getOrElse("make.data", "false").toBoolean
+  val test = mArgs.getOrElse("test.data", "false").toBoolean
+  val delete = mArgs.getOrElse("delete.data", "false").toBoolean
+
+  if( make ) {
+    JobRunner.main(Array(classOf[HBaseSaltTestSetup].getName,
+      "--hdfs",
+      "--app.conf.path", appPath,
+      "--job.lib.path", jobLibPath,
+      "--quorum", quorum
+    ))
+  }
+
+  if( test ) {
+    JobRunner.main(Array(classOf[HBaseSaltTester].getName,
+        "--hdfs",
+        "--app.conf.path", appPath,
+        "--job.lib.path", jobLibPath,
+        "--quorum", quorum
+    ))
+  }
+
+  if( delete ) {
+    JobRunner.main(Array(classOf[HBaseSaltTestShutdown].getName,
+      "--hdfs",
+      "--app.conf.path", appPath,
       "--job.lib.path", jobLibPath,
-      "--quorum", quorum,
-      "--debug", "true"
-  ))
+      "--quorum", quorum
+    ))
+  }
 }
 \ No newline at end of file