update FrontendServiceSuite to adapt KyuubiSQLException.featureNotSup…

…ported
apache · Jul 4, 2021 · 396e70d · 396e70d
1 parent c06dbfd
commit 396e70d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/kyuubi-common/src/test/scala/org/apache/kyuubi/service/FrontendServiceSuite.scala b/kyuubi-common/src/test/scala/org/apache/kyuubi/service/FrontendServiceSuite.scala
@@ -326,8 +326,8 @@ class FrontendServiceSuite extends KyuubiFunSuite {
       val resp = client.GetPrimaryKeys(req)
       assert(resp.getOperationHandle === null)
       assert(resp.getStatus.getStatusCode === TStatusCode.ERROR_STATUS)
-      assert(resp.getStatus.getSqlState === null)
-      assert(resp.getStatus.getErrorMessage startsWith "Feature is not available")
+      assert(resp.getStatus.getSqlState === "0A000")
+      assert(resp.getStatus.getErrorMessage startsWith "feature not supported")
     }
   }
 
@@ -338,8 +338,8 @@ class FrontendServiceSuite extends KyuubiFunSuite {
       val resp = client.GetCrossReference(req)
       assert(resp.getOperationHandle === null)
       assert(resp.getStatus.getStatusCode === TStatusCode.ERROR_STATUS)
-      assert(resp.getStatus.getSqlState === null)
-      assert(resp.getStatus.getErrorMessage startsWith "Feature is not available")
+      assert(resp.getStatus.getSqlState === "0A000")
+      assert(resp.getStatus.getErrorMessage startsWith "feature not supported")
     }
   }