make temp dir more robust

apache · Feb 15, 2024 · 297d5cf · 297d5cf
1 parent 79ff354
commit 297d5cf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/spark/src/test/scala/org/apache/comet/CometCastSuite.scala b/spark/src/test/scala/org/apache/comet/CometCastSuite.scala
@@ -31,7 +31,13 @@ import org.apache.spark.sql.types.{DataType, DataTypes}
 class CometCastSuite extends CometTestBase with AdaptiveSparkPlanHelper {
   import testImplicits._
 
-  private val tempDir = Files.createTempDirectory("CometCastSuite")
+  private lazy val tempDir = {
+    val tmp = Files.createTempDirectory("CometCastSuite")
+    if (!tmp.toFile.exists()) {
+      assert(tmp.toFile.mkdirs())
+    }
+    tmp
+  }
 
   ignore("cast long to short") {
     castTest(generateLongs, DataTypes.ShortType)