Fix broken UT

Signed-off-by: Chen Dai <[email protected]>
dai-chen · Sep 30, 2024 · ed0d3d9 · ed0d3d9
1 parent 28d8af3
commit ed0d3d9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...gration/src/test/scala/org/opensearch/flint/spark/FlintSparkTransactionSupportSuite.scala b/...gration/src/test/scala/org/opensearch/flint/spark/FlintSparkTransactionSupportSuite.scala
@@ -32,8 +32,11 @@ class FlintSparkTransactionSupportSuite extends FlintSuite with Matchers {
 
   override protected def beforeEach(): Unit = {
     super.beforeEach()
+
+    val logEntry = mock[FlintMetadataLog[FlintMetadataLogEntry]]
+    when(logEntry.getLatest).thenReturn(Optional.of(mock[FlintMetadataLogEntry]))
     when(mockFlintMetadataLogService.getIndexMetadataLog(testIndex))
-      .thenReturn(Optional.of(mock[FlintMetadataLog[FlintMetadataLogEntry]]))
+      .thenReturn(Optional.of(logEntry))
   }
 
   override protected def afterEach(): Unit = {