From bfa134bd89ae0d706a3d8e473b98536d9952f0c1 Mon Sep 17 00:00:00 2001 From: Kent Yao Date: Thu, 21 Jun 2018 16:20:49 +0800 Subject: [PATCH] fix ut --- .../kyuubi/server/BackendServiceSuite.scala | 22 ++++++++++++------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/src/test/scala/yaooqinn/kyuubi/server/BackendServiceSuite.scala b/src/test/scala/yaooqinn/kyuubi/server/BackendServiceSuite.scala index fdf0ab9ad..918fdad96 100644 --- a/src/test/scala/yaooqinn/kyuubi/server/BackendServiceSuite.scala +++ b/src/test/scala/yaooqinn/kyuubi/server/BackendServiceSuite.scala @@ -26,16 +26,23 @@ import yaooqinn.kyuubi.operation.{CANCELED, RUNNING} class BackendServiceSuite extends SparkFunSuite { - test("verify default cli service protocol") { - assert(BackendService.SERVER_VERSION === TProtocolVersion.HIVE_CLI_SERVICE_PROTOCOL_V8) + var backendService: BackendService = _ + val user = KyuubiSparkUtil.getCurrentUserName() + val conf = new SparkConf(loadDefaults = true).setAppName("be test") + KyuubiServer.setupCommonConfig(conf) + conf.remove(KyuubiSparkUtil.CATALOG_IMPL) + conf.setMaster("local") + + override protected def beforeAll(): Unit = { + backendService = new BackendService() + } + + protected override def afterAll(): Unit = { + backendService.stop() + backendService = null } test("backend service function tests") { - val backendService = new BackendService() - val conf = new SparkConf(loadDefaults = true).setAppName("spark session test") - conf.remove(KyuubiSparkUtil.CATALOG_IMPL) - conf.setMaster("local") - KyuubiServer.setupCommonConfig(conf) assert(backendService.getName === classOf[BackendService].getSimpleName) // before init assert(backendService.getSessionManager === null) @@ -47,7 +54,6 @@ class BackendServiceSuite extends SparkFunSuite { assert(backendService.getConf !== null) backendService.start() - val user = KyuubiSparkUtil.getCurrentUserName() val session = backendService.openSession( TProtocolVersion.HIVE_CLI_SERVICE_PROTOCOL_V8, user,