[incubator-kyuubi] branch master updated: [KYUUBI #2299] Fix flaky test: support engine alive probe to fast fail on engine broken

[fe...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

feiwang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-kyuubi.git


The following commit(s) were added to refs/heads/master by this push:
     new 9c4af55fe [KYUUBI #2299] Fix flaky test: support engine alive probe to fast fail on engine broken
9c4af55fe is described below

commit 9c4af55fe7ddd0db5900f20b012ad664bf1af8b8
Author: wforget <64...@qq.com>
AuthorDate: Fri Apr 8 20:59:28 2022 +0800

    [KYUUBI #2299] Fix flaky test: support engine alive probe to fast fail on engine broken
    
    ### _Why are the changes needed?_
    
    close #2299
    
    ### _How was this patch tested?_
    - [ ] Add some test cases that check the changes thoroughly including negative and positive cases if possible
    
    - [ ] Add screenshots for manual tests if appropriate
    
    - [X] [Run test](https://kyuubi.apache.org/docs/latest/develop_tools/testing.html#running-tests) locally before make a pull request
    
    Closes #2302 from wForget/KYUUBI-2299.
    
    Closes #2299
    
    1bfa83a5 [wforget] [KYUUBI-2299] Fix flaky test
    
    Authored-by: wforget <64...@qq.com>
    Signed-off-by: Fei Wang <fw...@ebay.com>
---
 .../apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala b/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
index f675466c0..c59c5b033 100644
--- a/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
+++ b/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
@@ -212,6 +212,12 @@ class KyuubiOperationPerConnectionSuite extends WithKyuubiServer with HiveJDBCTe
       KyuubiConf.OPERATION_THRIFT_CLIENT_REQUEST_MAX_ATTEMPTS.key -> "10000",
       KyuubiConf.ENGINE_REQUEST_TIMEOUT.key -> "100"))(Map.empty)(Map.empty) {
       withSessionHandle { (client, handle) =>
+        val preReq = new TExecuteStatementReq()
+        preReq.setStatement("select engine_name()")
+        preReq.setSessionHandle(handle)
+        preReq.setRunAsync(false)
+        client.ExecuteStatement(preReq)
+
         val executeStmtReq = new TExecuteStatementReq()
         executeStmtReq.setStatement("select java_method('java.lang.System', 'exit', 1)")
         executeStmtReq.setSessionHandle(handle)