Skip to content

Commit 4eef021

Browse files
committed
Merge branch 'use-git-rev-parse-for-root' into remote-jars-tests
2 parents 1dd81f7 + 90af481 commit 4eef021

File tree

1 file changed

+27
-16
lines changed

1 file changed

+27
-16
lines changed

integration-test/src/test/scala/org/apache/spark/deploy/k8s/integrationtest/KubernetesSuite.scala

Lines changed: 27 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)