@@ -41,9 +41,15 @@ private[spark] class KubernetesSuite extends FunSuite with BeforeAndAfterAll wit
41
41
private var sparkAppConf : SparkAppConf = _
42
42
private var remoteExamplesJarUri : URI = _
43
43
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" )
47
53
48
54
override def beforeAll (): Unit = {
49
55
testBackend.initialize()
@@ -142,17 +148,30 @@ private[spark] class KubernetesSuite extends FunSuite with BeforeAndAfterAll wit
142
148
})
143
149
}
144
150
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" ) {
147
163
sparkAppConf
164
+ .set(" spark.kubernetes.mountDependencies.filesDownloadDir" ,
165
+ CONTAINER_LOCAL_FILE_DOWNLOAD_PATH )
166
+ .set(" spark.files" , REMOTE_PAGE_RANK_DATA_FILE )
148
167
.set(s " spark.kubernetes.driver.secrets. $TEST_SECRET_NAME" , TEST_SECRET_MOUNT_PATH )
149
168
.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)
151
170
152
171
createTestSecret()
153
172
154
- runSparkPiAndVerifyCompletion (
155
- appResource = remoteExamplesJarUri.toString ,
173
+ runSparkPageRankAndVerifyCompletion (
174
+ appArgs = Array ( CONTAINER_LOCAL_DOWNLOADED_PAGE_RANK_DATA_FILE ) ,
156
175
driverPodChecker = (driverPod : Pod ) => {
157
176
doBasicDriverPodCheck(driverPod)
158
177
checkTestSecret(driverPod, withInitContainer = true )
@@ -163,14 +182,6 @@ private[spark] class KubernetesSuite extends FunSuite with BeforeAndAfterAll wit
163
182
})
164
183
}
165
184
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
-
174
185
private def runSparkPiAndVerifyCompletion (
175
186
appResource : String = CONTAINER_LOCAL_SPARK_DISTRO_EXAMPLES_JAR ,
176
187
driverPodChecker : Pod => Unit = doBasicDriverPodCheck,
0 commit comments