From 6ad3447bda3c5cefac1c45ba057c9b8f75751abb Mon Sep 17 00:00:00 2001 From: Luciano Resende Date: Thu, 26 Jan 2017 15:55:49 -0800 Subject: Improve test coverage --- .../apache/toree/gateway/ToreeGatewaySpec.scala | 41 ++++++++++++++++------ 1 file changed, 31 insertions(+), 10 deletions(-) (limited to 'src/test') diff --git a/src/test/scala/org/apache/toree/gateway/ToreeGatewaySpec.scala b/src/test/scala/org/apache/toree/gateway/ToreeGatewaySpec.scala index b2b9c03..5bb01c5 100644 --- a/src/test/scala/org/apache/toree/gateway/ToreeGatewaySpec.scala +++ b/src/test/scala/org/apache/toree/gateway/ToreeGatewaySpec.scala @@ -23,7 +23,6 @@ import org.apache.toree.kernel.protocol.v5.client.boot.layers.{StandardHandlerIn import org.scalatest.{FlatSpec, Ignore} import org.slf4j.LoggerFactory -@Ignore class ToreeGatewaySpec extends FlatSpec { final val log = LoggerFactory.getLogger(this.getClass.getName.stripSuffix("$")) @@ -31,15 +30,17 @@ class ToreeGatewaySpec extends FlatSpec { val profileJSON: String = """ { - "stdin_port": 48701, - "control_port": 48702, - "hb_port": 48705, - "shell_port": 48703, - "iopub_port": 48704, - "ip": "9.125.72.72", - "transport": "tcp", - "signature_scheme": "hmac-sha256", - "key": "" + "stdin_port": 48701, + "control_port": 48702, + "hb_port": 48705, + "shell_port": 48703, + "iopub_port": 48704, + "ip": "9.30.137.220", + "transport": "tcp", + "signature_scheme": "hmac-sha256", + "key": "", + "py4j_java": 25433, + "py4j_python": 25434 } """.stripMargin @@ -90,6 +91,16 @@ class ToreeGatewaySpec extends FlatSpec { assert(result.contains("only showing top 1 row")) } + "gateway" should "evaluate simple math statements" in { + val result = toreeGateway.eval( + """ + 1 + 1 + """.stripMargin + ).toString.stripMargin + + assert(result.contains("2")) + } + "gateway" should "receive error messages when exception is thrown" in { val result = toreeGateway.eval( """ @@ -129,4 +140,14 @@ class ToreeGatewaySpec extends FlatSpec { assert(result.contains("Hi")) } + + "gateway" should "import statement sucessfully" in { + val result = toreeGateway.eval( + """ + import org.apache.spark.sql.functions._ + """.stripMargin + ).toString.stripMargin + + assert(result.contains("done")) + } } -- cgit v1.2.3