fix failed tests

jeanlyn · Apr 21, 2015 · 0565831 · 0565831
1 parent 4fb25ed
commit 0565831
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/launcher/src/test/java/org/apache/spark/launcher/SparkSubmitCommandBuilderSuite.java b/launcher/src/test/java/org/apache/spark/launcher/SparkSubmitCommandBuilderSuite.java
@@ -59,6 +59,8 @@ public void testClusterCmdBuilder() throws Exception {
   @Test
   public void testCliParser() throws Exception {
     List<String> sparkSubmitArgs = Arrays.asList(
+      parser.CLASS,
+      "org.apache.spark.sql.hive.thriftserver.SparkSQLCLIDriver",
       parser.MASTER,
       "local",
       parser.DRIVER_MEMORY,
@@ -135,6 +137,7 @@ public void testPySparkLauncher() throws Exception {
   @Test
   public void testPySparkFallback() throws Exception {
     List<String> sparkSubmitArgs = Arrays.asList(
+      "pyspark-shell-main",
       "--master=foo",
       "--deploy-mode=bar",
       "script.py",