From b38541f330754c2aabe02a1307aa684d1b7c2f9a Mon Sep 17 00:00:00 2001 From: Takeshi Yamamuro Date: Wed, 23 Dec 2020 10:40:18 +0900 Subject: [PATCH] Fix --- .../BaseScriptTransformationSuite.scala | 36 +++++++++---------- 1 file changed, 17 insertions(+), 19 deletions(-) diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/BaseScriptTransformationSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/BaseScriptTransformationSuite.scala index 9a103c1be6f9c..863657a7862a6 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/execution/BaseScriptTransformationSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/execution/BaseScriptTransformationSuite.scala @@ -422,25 +422,23 @@ abstract class BaseScriptTransformationSuite extends SparkPlanTest with SQLTestU } test("SPARK-32106: TRANSFORM with non-existent command/file") { - val e1 = intercept[SparkException] { - sql( - """ - |SELECT TRANSFORM(a) - |USING 'some_non_existent_command' AS (a) - |FROM VALUES (1) t(a) - """.stripMargin).collect() - }.getMessage - assert(e1.contains("Subprocess exited")) - - val e2 = intercept[SparkException] { - sql( - """ - |SELECT TRANSFORM(a) - |USING 'python some_non_existent_file' AS (a) - |FROM VALUES (1) t(a) - """.stripMargin).collect() - }.getMessage - assert(e2.contains("Subprocess exited")) + Seq( + s""" + |SELECT TRANSFORM(a) + |USING 'some_non_existent_command' AS (a) + |FROM VALUES (1) t(a) + """.stripMargin, + s""" + |SELECT TRANSFORM(a) + |USING 'python some_non_existent_file' AS (a) + |FROM VALUES (1) t(a) + """.stripMargin).foreach { query => + intercept[SparkException] { + // Since an error message is shell-dependent, this test just checks + // if the expected exception will be thrown. + sql(query).collect() + } + } } }