ut

kyuubi-main/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationSuite.scala

@@ -0,0 +1,52 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kyuubi.operation
+
+import org.apache.hive.service.rpc.thrift.{TExecuteStatementReq, TGetOperationStatusReq, TOperationState, TStatusCode}
+import org.scalatest.time.SpanSugar.convertIntToGrainOfTime
+
+import org.apache.kyuubi.WithKyuubiServer
+import org.apache.kyuubi.config.KyuubiConf
+
+class KyuubiOperationSuite extends WithKyuubiServer with JDBCTestUtils {
+  override protected val conf: KyuubiConf = KyuubiConf()
+
+  test("KYUUBI #647 - remote thrift client crash") {
+    withSessionHandle { (client, handle) =>
+      val executeStmtReq = new TExecuteStatementReq()
+      executeStmtReq.setStatement(
+        """
+          | select
+          |   java_method('java.lang.Thread', 'sleep', 1000L),
+          |   java_method('java.lang.System', 'exit', 1)
+          |""".stripMargin)
+      executeStmtReq.setSessionHandle(handle)
+      executeStmtReq.setRunAsync(true)
+      val executeStmtResp = client.ExecuteStatement(executeStmtReq)
+
+      eventually(timeout(10.seconds), interval(500.milliseconds)) {
+        val getOpStatusReq = new TGetOperationStatusReq(executeStmtResp.getOperationHandle)
+        val getOpStatusResp = client.GetOperationStatus(getOpStatusReq)
+        assert(getOpStatusResp.getStatus.getStatusCode === TStatusCode.SUCCESS_STATUS)
+        assert(getOpStatusResp.getOperationState === TOperationState.ERROR_STATE)
+      }
+    }
+  }
+
+  override protected def jdbcUrl: String = getJdbcUrl
+}