SPARK-24348 + error message check

wajda · wajda · commit 045307bc4088 · 2018-05-22T19:11:26.000+02: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
@@ -757,9 +757,11 @@ class DataFrameFunctionsSuite extends QueryTest with SharedSQLContext {
       Seq(Row("3"), Row(""), Row(null))
     )
 
-    intercept[AnalysisException] {
+    val e = intercept[AnalysisException] {
       Seq(("a string element", 1)).toDF().selectExpr("element_at(_1, _2)")
     }
+    assert(e.message.contains(
+      "argument 1 requires (array or map) type, however, '`_1`' is of string type"))
   }
 
   test("concat function - arrays") {

Original file line number	Diff line number	Diff line change
`@@ -757,9 +757,11 @@ class DataFrameFunctionsSuite extends QueryTest with SharedSQLContext {`
`757`	`757`	`Seq(Row("3"), Row(""), Row(null))`
`758`	`758`	`)`
`759`	`759`
`760`		`- intercept[AnalysisException] {`
	`760`	`+ val e = intercept[AnalysisException] {`
`761`	`761`	`Seq(("a string element", 1)).toDF().selectExpr("element_at(_1, _2)")`
`762`	`762`	`}`
	`763`	`+ assert(e.message.contains(`
	`764`	+ "argument 1 requires (array or map) type, however, '`_1`' is of string type"))
`763`	`765`	`}`
`764`	`766`
`765`	`767`	`test("concat function - arrays") {`