added additional unit test

codeatri · codeatri · commit 652663077e38 · 2018-08-08T13:01:22.000-07:00
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameFunctionsSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameFunctionsSuite.scala
@@ -2266,6 +2266,13 @@ class DataFrameFunctionsSuite extends QueryTest with SharedSQLContext {
       }
       assert(ex2.getMessage.contains(
       "The number of lambda function arguments '3' does not match"))
+
+      val ex3 = intercept[RuntimeException] {
+        dfExample3.selectExpr("transform_keys(x, (k, v) -> v)").show()
+      }
+      assert(ex3.getMessage.contains("Cannot use null as map key!"))
+
+
     }
 
     testInvalidLambdaFunctions()

Original file line number	Diff line number	Diff line change
`@@ -2266,6 +2266,13 @@ class DataFrameFunctionsSuite extends QueryTest with SharedSQLContext {`
`2266`	`2266`	`}`
`2267`	`2267`	`assert(ex2.getMessage.contains(`
`2268`	`2268`	`"The number of lambda function arguments '3' does not match"))`
	`2269`	`+`
	`2270`	`+ val ex3 = intercept[RuntimeException] {`
	`2271`	`+ dfExample3.selectExpr("transform_keys(x, (k, v) -> v)").show()`
	`2272`	`+ }`
	`2273`	`+ assert(ex3.getMessage.contains("Cannot use null as map key!"))`
	`2274`	`+`
	`2275`	`+`
`2269`	`2276`	`}`
`2270`	`2277`
`2271`	`2278`	`testInvalidLambdaFunctions()`