Fix broken IT

Signed-off-by: Chen Dai <[email protected]>
opensearch-project · Feb 7, 2024 · a8f51f0 · a8f51f0
1 parent 694d603
commit a8f51f0
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkTransactionITSuite.scala b/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkTransactionITSuite.scala
@@ -78,17 +78,23 @@ class FlintSparkTransactionITSuite extends OpenSearchTransactionSuite with Match
   }
 
   test("incremental refresh index") {
-    flint
-      .skippingIndex()
-      .onTable(testTable)
-      .addPartitions("year", "month")
-      .options(FlintSparkIndexOptions(Map("incremental_refresh" -> "true")))
-      .create()
-    flint.refreshIndex(testFlintIndex)
+    withTempDir { checkpointDir =>
+      flint
+        .skippingIndex()
+        .onTable(testTable)
+        .addPartitions("year", "month")
+        .options(
+          FlintSparkIndexOptions(
+            Map(
+              "incremental_refresh" -> "true",
+              "checkpoint_location" -> checkpointDir.getAbsolutePath)))
+        .create()
+      flint.refreshIndex(testFlintIndex)
 
-    val latest = latestLogEntry(testLatestId)
-    latest should contain("state" -> "active")
-    latest("jobStartTime").asInstanceOf[Number].longValue() should be > 0L
+      val latest = latestLogEntry(testLatestId)
+      latest should contain("state" -> "active")
+      latest("jobStartTime").asInstanceOf[Number].longValue() should be > 0L
+    }
   }
 
   test("auto refresh index") {