Skip to content

Commit 916372c

Browse files
committed
Switch some constant uses of username/password to the object ref and do a more complete env check
1 parent 3ccddbf commit 916372c

File tree

1 file changed

+12
-6
lines changed
  • resource-managers/kubernetes/integration-tests/src/test/scala/org/apache/spark/deploy/k8s/integrationtest

1 file changed

+12
-6
lines changed

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

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,17 @@ private[spark] trait SecretsTestsSuite { k8sSuite: KubernetesSuite =>
5757
createTestSecret()
5858
sparkAppConf
5959
.set(s"spark.kubernetes.driver.secrets.$ENV_SECRET_NAME", SECRET_MOUNT_PATH)
60-
.set(s"spark.kubernetes.driver.secretKeyRef.USERNAME", s"$ENV_SECRET_NAME:username")
61-
.set(s"spark.kubernetes.driver.secretKeyRef.PASSWORD", s"$ENV_SECRET_NAME:password")
60+
.set(
61+
s"spark.kubernetes.driver.secretKeyRef.${ENV_SECRET_KEY_1.toUpperCase}",
62+
s"$ENV_SECRET_NAME:${ENV_SECRET_KEY_1}")
63+
.set(
64+
s"spark.kubernetes.driver.secretKeyRef.${ENV_SECRET_KEY_2.toUpperCase}",
65+
s"$ENV_SECRET_NAME:${ENV_SECRET_KEY_2}")
6266
.set(s"spark.kubernetes.executor.secrets.$ENV_SECRET_NAME", SECRET_MOUNT_PATH)
63-
.set(s"spark.kubernetes.executor.secretKeyRef.USERNAME", s"$ENV_SECRET_NAME:username")
64-
.set(s"spark.kubernetes.executor.secretKeyRef.PASSWORD", s"$ENV_SECRET_NAME:password")
67+
.set(s"spark.kubernetes.executor.secretKeyRef.USERNAME",
68+
s"${ENV_SECRET_NAME}:$ENV_SECRET_KEY_1")
69+
.set(s"spark.kubernetes.executor.secretKeyRef.PASSWORD",
70+
s"${ENV_SECRET_NAME}:$ENV_SECRET_KEY_2")
6571
try {
6672
runSparkPiAndVerifyCompletion(
6773
driverPodChecker = (driverPod: Pod) => {
@@ -91,8 +97,8 @@ private[spark] trait SecretsTestsSuite { k8sSuite: KubernetesSuite =>
9197
assert(!env.isEmpty)
9298
env
9399
}
94-
env.toString should include (ENV_SECRET_VALUE_1)
95-
env.toString should include (ENV_SECRET_VALUE_2)
100+
env.toString should include (s"${ENV_SECRET_KEY_1.toUpperCase}=$ENV_SECRET_VALUE_1")
101+
env.toString should include (s"${ENV_SECRET_KEY_2.toUpperCase}=$ENV_SECRET_VALUE_2")
96102

97103
// Make sure our secret files are mounted correctly
98104
val files = Utils.executeCommand("ls", s"$SECRET_MOUNT_PATH")

0 commit comments

Comments
 (0)