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

add more ut

Browse Source
This commit is contained in:
Kent Yao 2018-05-30 10:11:32 +08:00
parent 20404139b2
commit 5318d99d1a
1 changed files with 28 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/test/scala/yaooqinn/kyuubi/spark/SparkSessionWithUGISuite.scala
View File

@ -23,7 +23,6 @@ import org.apache.spark.sql.SparkSession
import org.apache.spark.sql.catalyst.analysis.NoSuchDatabaseException import org.apache.spark.sql.catalyst.analysis.NoSuchDatabaseException
import yaooqinn.kyuubi.server.KyuubiServer import yaooqinn.kyuubi.server.KyuubiServer
import yaooqinn.kyuubi.session.{KyuubiSession, SessionManager}
import yaooqinn.kyuubi.utils.ReflectUtils import yaooqinn.kyuubi.utils.ReflectUtils
class SparkSessionWithUGISuite extends SparkFunSuite { class SparkSessionWithUGISuite extends SparkFunSuite {
@ -34,8 +33,6 @@ class SparkSessionWithUGISuite extends SparkFunSuite {
conf.remove(KyuubiSparkUtil.CATALOG_IMPL) conf.remove(KyuubiSparkUtil.CATALOG_IMPL)
conf.setMaster("local") conf.setMaster("local")
val userName = user.getShortUserName val userName = user.getShortUserName
val passwd = ""
val statement = "show tables"
var spark: SparkSession = _ var spark: SparkSession = _
override protected def beforeAll(): Unit = { override protected def beforeAll(): Unit = {
@ -55,19 +52,44 @@ class SparkSessionWithUGISuite extends SparkFunSuite {
spark.stop() spark.stop()
} }
test("testInit failed with no such database") { test("test init failed with no such database") {
val sparkSessionWithUGI = new SparkSessionWithUGI(user, conf) val sparkSessionWithUGI = new SparkSessionWithUGI(user, conf)
intercept[NoSuchDatabaseException](sparkSessionWithUGI.init(Map("use:database" -> "fakedb"))) intercept[NoSuchDatabaseException](sparkSessionWithUGI.init(Map("use:database" -> "fakedb")))
} }
test("testInit success") { test("test init success with empty session conf") {
val sparkSessionWithUGI = new SparkSessionWithUGI(user, conf) val sparkSessionWithUGI = new SparkSessionWithUGI(user, conf)
sparkSessionWithUGI.init(Map.empty) sparkSessionWithUGI.init(Map.empty)
assert(sparkSessionWithUGI.sparkSession.sparkContext.sparkUser === userName) assert(sparkSessionWithUGI.sparkSession.sparkContext.sparkUser === userName)
assert(sparkSessionWithUGI.userName === userName) assert(sparkSessionWithUGI.userName === userName)
} }
test("test init success with spark properties") {
val sessionConf = Map("set:hivevar:spark.foo" -> "bar")
val sparkSessionWithUGI = new SparkSessionWithUGI(user, conf)
sparkSessionWithUGI.init(sessionConf)
assert(sparkSessionWithUGI.sparkSession.conf.get("spark.foo") === "bar")
}
test("test init success with hive/hadoop/extra properties") {
val sessionConf = Map("set:hivevar:foo" -> "bar")
val sparkSessionWithUGI = new SparkSessionWithUGI(user, conf)
sparkSessionWithUGI.init(sessionConf)
assert(sparkSessionWithUGI.sparkSession.conf.get("spark.hadoop.foo") === "bar")
}
test("test init with new spark context") {
val userName1 = "test"
val ru = UserGroupInformation.createRemoteUser(userName1)
val sessionConf = Map("set:hivevar:spark.foo" -> "bar", "set:hivevar:foo" -> "bar")
val sparkSessionWithUGI = new SparkSessionWithUGI(ru, conf)
sparkSessionWithUGI.init(sessionConf)
assert(sparkSessionWithUGI.sparkSession.conf.get("spark.foo") === "bar")
assert(sparkSessionWithUGI.sparkSession.conf.get("spark.hadoop.foo") === "bar")
assert(!sparkSessionWithUGI.sparkSession.sparkContext.getConf.contains(KyuubiSparkUtil.KEYTAB))
sparkSessionWithUGI.sparkSession.stop()
}
test("testSetPartiallyConstructed") { test("testSetPartiallyConstructed") {
SparkSessionWithUGI.setPartiallyConstructed("Kent") SparkSessionWithUGI.setPartiallyConstructed("Kent")
assert(SparkSessionWithUGI.isPartiallyConstructed("Kent")) assert(SparkSessionWithUGI.isPartiallyConstructed("Kent"))