apache/kyuubi
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

kyuubi sql exception test

Browse Source
This commit is contained in:
Kent Yao 2020-09-21 20:38:11 +08:00
parent cab07c3001
commit 69ee4c18e8
2 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kyuubi-common/src/main/scala/org/apache/kyuubi/KyuubiSQLException.scala
View File

@ -23,7 +23,7 @@ import scala.collection.JavaConverters._
import org.apache.hive.service.rpc.thrift.{TStatus, TStatusCode}
case class KyuubiSQLException(msg: String, cause: Throwable) extends SQLException(msg, cause) {
class KyuubiSQLException(msg: String, cause: Throwable) extends SQLException(msg, cause) {
/**
* Converts current object to a [[TStatus]] object
*
@ -44,6 +44,9 @@ object KyuubiSQLException {
private final val HEAD_MARK: String = "*"
private final val SEPARATOR: Char = ':'
def apply(msg: String, throwable: Throwable): KyuubiSQLException = {
new KyuubiSQLException(msg, throwable)
}
def apply(cause: Throwable): KyuubiSQLException = {
new KyuubiSQLException(cause.getMessage, cause)
}

17
kyuubi-common/src/test/scala/org/apache/kyuubi/KyuubiSQLExceptionSuite.scala
View File

@ -17,6 +17,8 @@
package org.apache.kyuubi
import org.apache.hive.service.rpc.thrift.TStatusCode
class KyuubiSQLExceptionSuite extends KyuubiFunSuite {
test("KyuubiSQLException") {
@ -28,11 +30,24 @@ class KyuubiSQLExceptionSuite extends KyuubiFunSuite {
val e1 = new KyuubiException(msg1, e0)
val e2 = new KyuubiSQLException(msg2, e1)
assert(e2.toTStatus === KyuubiSQLException.toTStatus(e2))
val e3 = KyuubiSQLException(e2.toTStatus)
assert(e3.getMessage === e2.getMessage)
assert(e3.getStackTrace === e2.getStackTrace)
assert(e3.getCause.getMessage === e1.getMessage)
assert(e3.getCause.getCause.getMessage === e0.getMessage)
}
val ts0 = KyuubiSQLException.toTStatus(e0)
assert(ts0.getStatusCode === TStatusCode.ERROR_STATUS)
assert(ts0.getErrorMessage === msg0)
assert(ts0.getInfoMessages.get(0).startsWith("*"))
val e4 = KyuubiSQLException(ts0)
assert(e4.getMessage === msg0)
assert(e4.getCause.getStackTrace === e0.getStackTrace)
val e5 = KyuubiSQLException(e0)
assert(e5.getMessage === msg0)
assert(e5.getCause === e0)
}
}