zwangsheng commented on code in PR #3869:
URL: https://github.com/apache/incubator-kyuubi/pull/3869#discussion_r1039424073


##########
integration-tests/kyuubi-kubernetes-it/src/test/scala/org/apache/kyuubi/kubernetes/test/WithKyuubiServerOnKubernetes.scala:
##########
@@ -25,26 +26,30 @@ trait WithKyuubiServerOnKubernetes extends KyuubiFunSuite {
   protected def connectionConf: Map[String, String] = Map.empty
   private val miniKubernetesClient: DefaultKubernetesClient = 
MiniKube.getKubernetesClient
 
-  protected def getJdbcUrl: String = {
-    val kyuubiServers =
-      miniKubernetesClient.pods().list().getItems
-    assert(kyuubiServers.size() == 1)
-    val kyuubiServer = kyuubiServers.get(0)
+  protected def getJdbcUrl(connectionConf: Map[String, String]): String = {
+    val kyuuubiPod = kyuubiPod
     // Kyuubi server state should be running since mvn compile is quite 
slowly..
-    if (!"running".equalsIgnoreCase(kyuubiServer.getStatus.getPhase)) {
+    if (!"running".equalsIgnoreCase(kyuuubiPod.getStatus.getPhase)) {
       val log =
         miniKubernetesClient
           .pods()
-          .withName(kyuubiServer.getMetadata.getName)
+          .withName(kyuuubiPod.getMetadata.getName)
           .getLog
       throw new IllegalStateException(
-        s"Kyuubi server pod state error: ${kyuubiServer.getStatus.getPhase}, 
log:\n$log")
+        s"Kyuubi server pod state error: ${kyuuubiPod.getStatus.getPhase}, 
log:\n$log")
     }
-    val kyuubiServerIp = MiniKube.getIp
+    val kyuubiServerIp = miniKubeIp
     val kyuubiServerPort =
-      kyuubiServer.getSpec.getContainers.get(0).getPorts.get(0).getHostPort
-    s"jdbc:hive2://$kyuubiServerIp:$kyuubiServerPort/;"
+      kyuuubiPod.getSpec.getContainers.get(0).getPorts.get(0).getHostPort
+    val connectStr = connectionConf.map(kv => kv._1 + "=" + 
kv._2).mkString("#", ";", "")
+    s"jdbc:hive2://$kyuubiServerIp:$kyuubiServerPort/;$connectStr"
   }
 
-  def getMiniKubeApiMaster: String = miniKubernetesClient.getMasterUrl.toString
+  def miniKubeApiMaster: String = miniKubernetesClient.getMasterUrl.toString
+
+  def kyuubiPod: Pod = 
miniKubernetesClient.pods().withName("kyuubi-test").get()

Review Comment:
   done



##########
integration-tests/kyuubi-kubernetes-it/src/test/scala/org/apache/kyuubi/kubernetes/test/WithKyuubiServerOnKubernetes.scala:
##########
@@ -25,26 +26,30 @@ trait WithKyuubiServerOnKubernetes extends KyuubiFunSuite {
   protected def connectionConf: Map[String, String] = Map.empty
   private val miniKubernetesClient: DefaultKubernetesClient = 
MiniKube.getKubernetesClient
 
-  protected def getJdbcUrl: String = {
-    val kyuubiServers =
-      miniKubernetesClient.pods().list().getItems
-    assert(kyuubiServers.size() == 1)
-    val kyuubiServer = kyuubiServers.get(0)
+  protected def getJdbcUrl(connectionConf: Map[String, String]): String = {
+    val kyuuubiPod = kyuubiPod

Review Comment:
   done



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to