@@ -41,9 +41,15 @@ private[spark] class KubernetesSuite extends FunSuite with BeforeAndAfterAll wit
4141 private var sparkAppConf : SparkAppConf = _
4242 private var remoteExamplesJarUri : URI = _
4343
44- private val driverImage = System .getProperty(" spark.docker.test.driverImage" , " spark-driver:latest" )
45- private val executorImage = System .getProperty(" spark.docker.test.executorImage" , " spark-executor:latest" )
46- private val initContainerImage = System .getProperty(" spark.docker.test.initContainerImage" , " spark-init:latest" )
44+ private val driverImage = System .getProperty(
45+ " spark.docker.test.driverImage" ,
46+ " spark-driver:latest" )
47+ private val executorImage = System .getProperty(
48+ " spark.docker.test.executorImage" ,
49+ " spark-executor:latest" )
50+ private val initContainerImage = System .getProperty(
51+ " spark.docker.test.initContainerImage" ,
52+ " spark-init:latest" )
4753
4854 override def beforeAll (): Unit = {
4955 testBackend.initialize()
@@ -142,17 +148,30 @@ private[spark] class KubernetesSuite extends FunSuite with BeforeAndAfterAll wit
142148 })
143149 }
144150
145- test(" Run SparkPi using the remote example jar with a test secret mounted into the driver and " +
146- " executor pods" ) {
151+ test(" Run PageRank using remote data file" ) {
152+ sparkAppConf
153+ .set(" spark.kubernetes.mountDependencies.filesDownloadDir" ,
154+ CONTAINER_LOCAL_FILE_DOWNLOAD_PATH )
155+ .set(" spark.files" , REMOTE_PAGE_RANK_DATA_FILE )
156+ .set(" spark.kubernetes.initContainer.image" , initContainerImage)
157+ runSparkPageRankAndVerifyCompletion(
158+ appArgs = Array (CONTAINER_LOCAL_DOWNLOADED_PAGE_RANK_DATA_FILE ))
159+ }
160+
161+ test(" Run PageRank using remote data file with test secret mounted into the driver and " +
162+ " executors" ) {
147163 sparkAppConf
164+ .set(" spark.kubernetes.mountDependencies.filesDownloadDir" ,
165+ CONTAINER_LOCAL_FILE_DOWNLOAD_PATH )
166+ .set(" spark.files" , REMOTE_PAGE_RANK_DATA_FILE )
148167 .set(s " spark.kubernetes.driver.secrets. $TEST_SECRET_NAME" , TEST_SECRET_MOUNT_PATH )
149168 .set(s " spark.kubernetes.executor.secrets. $TEST_SECRET_NAME" , TEST_SECRET_MOUNT_PATH )
150- sparkAppConf .set(" spark.kubernetes.initContainer.image" , initContainerImage)
169+ .set(" spark.kubernetes.initContainer.image" , initContainerImage)
151170
152171 createTestSecret()
153172
154- runSparkPiAndVerifyCompletion (
155- appResource = remoteExamplesJarUri.toString ,
173+ runSparkPageRankAndVerifyCompletion (
174+ appArgs = Array ( CONTAINER_LOCAL_DOWNLOADED_PAGE_RANK_DATA_FILE ) ,
156175 driverPodChecker = (driverPod : Pod ) => {
157176 doBasicDriverPodCheck(driverPod)
158177 checkTestSecret(driverPod, withInitContainer = true )
@@ -163,14 +182,6 @@ private[spark] class KubernetesSuite extends FunSuite with BeforeAndAfterAll wit
163182 })
164183 }
165184
166- test(" Run PageRank using remote data file" ) {
167- sparkAppConf
168- .set(" spark.kubernetes.mountDependencies.filesDownloadDir" , CONTAINER_LOCAL_FILE_DOWNLOAD_PATH )
169- .set(" spark.files" , REMOTE_PAGE_RANK_DATA_FILE )
170- .set(" spark.kubernetes.initContainer.image" , initContainerImage)
171- runSparkPageRankAndVerifyCompletion(appArgs = Array (CONTAINER_LOCAL_DOWNLOADED_PAGE_RANK_DATA_FILE ))
172- }
173-
174185 private def runSparkPiAndVerifyCompletion (
175186 appResource : String = CONTAINER_LOCAL_SPARK_DISTRO_EXAMPLES_JAR ,
176187 driverPodChecker : Pod => Unit = doBasicDriverPodCheck,
0 commit comments