Fix test

apache-spark-on-k8s · Sep 15, 2017 · 6273b74 · 6273b74
1 parent a9375c5
commit 6273b74
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...rc/test/scala/org/apache/spark/scheduler/cluster/kubernetes/ExecutorPodFactorySuite.scala b/...rc/test/scala/org/apache/spark/scheduler/cluster/kubernetes/ExecutorPodFactorySuite.scala
@@ -143,6 +143,7 @@ class ExecutorPodFactoryImplSuite extends SparkFunSuite with BeforeAndAfter {
     val conf = baseConf.clone()
     conf.set(KUBERNETES_SHUFFLE_LABELS, "label=value")
     conf.set(KUBERNETES_SHUFFLE_NAMESPACE, "default")
+    conf.set(KUBERNETES_SHUFFLE_DIR, "/tmp")
 
     val kubernetesExternalShuffleClient = new KubernetesExternalShuffleClientImpl(
       SparkTransportConf.fromSparkConf(conf, "shuffle"),
@@ -185,7 +186,7 @@ class ExecutorPodFactoryImplSuite extends SparkFunSuite with BeforeAndAfter {
     assert(executor.getSpec.getVolumes.get(0).getSecret.getSecretName == "secret1")
 
     checkOwnerReferences(executor, driverPodUid)
-    checkEnv(executor, Set())
+    checkEnv(executor, Set("SPARK_MOUNTED_FILES_FROM_SECRET_DIR"))
   }
 
   test("classpath and extra java options get translated into environment variables") {