From df90b01a2b26f2a155795081711d670cda60b9f8 Mon Sep 17 00:00:00 2001 From: Chen Dai Date: Tue, 11 Jul 2023 17:09:54 -0700 Subject: [PATCH] Fix IT failure Signed-off-by: Chen Dai --- .../flint/spark/FlintSparkSkippingIndexITSuite.scala | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/flint/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkSkippingIndexITSuite.scala b/flint/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkSkippingIndexITSuite.scala index 30179891a2..4fb6f46f5e 100644 --- a/flint/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkSkippingIndexITSuite.scala +++ b/flint/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkSkippingIndexITSuite.scala @@ -181,7 +181,12 @@ class FlintSparkSkippingIndexITSuite .onTable(testTable) .addPartitions("year", "month") .create() - flint.refreshIndex(testIndex, INCREMENTAL) + + val jobId = flint.refreshIndex(testIndex, INCREMENTAL) + val job = spark.streams.get(jobId.get) + failAfter(streamingTimeout) { + job.processAllAvailable() + } assertThrows[IllegalStateException] { flint.refreshIndex(testIndex, FULL)