Fix

apache · Dec 23, 2020 · b38541f · b38541f
1 parent d205a13
commit b38541f
Showing 1 changed file with 17 additions and 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
@@ -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()
+      }
+    }
   }
 }