Skip to content
This repository was archived by the owner on Jan 9, 2020. It is now read-only.

Commit 5d3e4c2

Browse files
committed
Add test case for driver pod name
1 parent 85a7c53 commit 5d3e4c2

File tree

1 file changed

+16
-0
lines changed
  • resource-managers/kubernetes/integration-tests/src/test/scala/org/apache/spark/deploy/kubernetes/integrationtest

1 file changed

+16
-0
lines changed

resource-managers/kubernetes/integration-tests/src/test/scala/org/apache/spark/deploy/kubernetes/integrationtest/KubernetesV1Suite.scala

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,22 @@ private[spark] class KubernetesV1Suite(testBackend: IntegrationTestBackend)
171171
"Unexpected value for annotation2")
172172
}
173173

174+
test("Run with driver pod name") {
175+
sparkConf.set(KUBERNETES_DRIVER_POD_NAME, "spark-pi")
176+
new Client(
177+
sparkConf = sparkConf,
178+
mainClass = KubernetesSuite.SPARK_PI_MAIN_CLASS,
179+
mainAppResource = KubernetesSuite.SUBMITTER_LOCAL_MAIN_APP_RESOURCE,
180+
appArgs = Array.empty[String]).run()
181+
val driverPodMetadata = kubernetesTestComponents.kubernetesClient
182+
.pods()
183+
.withName("spark-pi")
184+
.get()
185+
.getMetadata()
186+
val driverName = driverPodMetadata.getName
187+
assert(driverName === "spark-pi", "Unexpected driver pod name.")
188+
}
189+
174190
test("Enable SSL on the driver submit server") {
175191
assume(testBackend.name == MINIKUBE_TEST_BACKEND)
176192

0 commit comments

Comments
 (0)