Browse Source

[livy] Tests should wait for the sessions to close

Erick Tryzelaar 10 years ago
parent
commit
77a7a21fdc

+ 1 - 1
apps/spark/java/livy-repl/src/test/scala/com/cloudera/hue/livy/repl/PythonSessionSpec.scala

@@ -19,7 +19,7 @@ class PythonSessionSpec extends FunSpec with Matchers with BeforeAndAfter {
   }
 
   after {
-    session.close()
+    Await.result(session.close(), Duration.Inf)
   }
 
   describe("A python session") {

+ 1 - 1
apps/spark/java/livy-repl/src/test/scala/com/cloudera/hue/livy/repl/SparkSessionSpec.scala

@@ -19,7 +19,7 @@ class SparkSessionSpec extends FunSpec with Matchers with BeforeAndAfter {
   }
 
   after {
-    session.close()
+    Await.result(session.close(), Duration.Inf)
   }
 
   describe("A spark session") {