added test mode.

ParallelAI · Feb 20, 2014 · e068f01 · e068f01
1 parent 3f2642d
commit e068f01
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExample.scala b/src/main/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExample.scala
@@ -6,6 +6,8 @@ import org.apache.log4j.{Level, Logger}
 import parallelai.spyglass.hbase.{HBasePipeConversions, HBaseSource}
 import parallelai.spyglass.hbase.HBaseConstants.SourceMode
 import cascading.tuple.Fields
+import cascading.property.AppProps
+import java.util.Properties
 
 /**
   * Simple example of HBaseSource usage
@@ -17,7 +19,10 @@ class SimpleHBaseSourceExample(args: Args) extends JobBase(args) with HBasePipeC
    if (isDebug) Logger.getRootLogger.setLevel(Level.DEBUG)
 
    val output = args("output")
-
+
+   val properties = new Properties()
+   AppProps.setApplicationJarClass( properties, classOf[SimpleHBaseSourceExample] );
+
    val hbs = new HBaseSource(
      "table_name",
      "quorum_name:2181",

diff --git a/src/test/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExampleTest.scala b/src/test/scala/parallelai/spyglass/hbase/example/SimpleHBaseSourceExampleTest.scala
@@ -30,7 +30,7 @@ class SimpleHBaseSourceExampleTest extends FunSpec with TupleConversions {
   )
 
   JobTest("parallelai.spyglass.hbase.example.SimpleHBaseSourceExample")
-    .arg("local", "")
+    .arg("test", "")
     .arg("app.conf.path", "app.conf")
     .arg("output", output)
     .arg("debug", "true")