Update spark/src/test/scala/org/apache/comet/CometExpressionCoverageS…

…uite.scala Co-authored-by: advancedxy <[email protected]>
apache · May 9, 2024 · 5bc9a27 · 5bc9a27
1 parent 88795f3
commit 5bc9a27
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/spark/src/test/scala/org/apache/comet/CometExpressionCoverageSuite.scala b/spark/src/test/scala/org/apache/comet/CometExpressionCoverageSuite.scala
@@ -213,8 +213,9 @@ class CometExpressionCoverageSuite extends CometTestBase with AdaptiveSparkPlanH
       sys.error("DATAFUSIONCLI_PATH env variable not set"))
 
     val tempFilePath = Files.createTempFile("temp-", ".sql")
+    try {
     Files.write(tempFilePath, sql.getBytes)
-
+   
     val command = s"""$datafusionCliPath/datafusion-cli -f $tempFilePath"""
 
     val stdout = new StringBuilder
@@ -225,8 +226,9 @@ class CometExpressionCoverageSuite extends CometTestBase with AdaptiveSparkPlanH
         out => stdout.append(out).append("\n"), // stdout
         err => stderr.append(err).append("\n") // stderr
       ))
-
-    Files.delete(tempFilePath)
+   } finally {
+      Files.delete(tempFilePath)
+   }
 
     val err = stderr.toString()
     val out = stdout.toString()