Added docker build arguments

Author: liyinan926

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

@@ -17,9 +17,10 @@
 package org.apache.spark.deploy.k8s.integrationtest.docker
 
 import java.net.URI
+import java.net.URLEncoder
 import java.nio.file.Paths
 
-import com.spotify.docker.client.{DefaultDockerClient, DockerCertificates, LoggingBuildHandler}
+import com.spotify.docker.client.{DockerClient, DefaultDockerClient, DockerCertificates, LoggingBuildHandler}
 import org.apache.http.client.utils.URIBuilder
 import org.scalatest.concurrent.{Eventually, PatienceConfiguration}
 import org.scalatest.time.{Minutes, Seconds, Span}
@@ -67,11 +68,16 @@ private[spark] class SparkDockerImageBuilder
   }
 
   private def buildImage(name: String, dockerFile: String): Unit = {
+    val buildArgs = "{" +
+      "\"spark_jars\": \"jars\", " +
+      "\"img_path\": \"kubernetes/dockerfiles\", " +
+      "\"base_image\": \"spark-base\"}"
     dockerClient.build(
       DOCKER_BUILD_PATH,
       name,
       dockerFile,
-      new LoggingBuildHandler())
+      new LoggingBuildHandler(),
+      DockerClient.BuildParam.create("buildargs", URLEncoder.encode(buildArgs, "UTF-8")))
     logInfo(s"Built $name docker image")
   }
 }