Unit test fix

Signed-off-by: Dinu John <[email protected]>
dinujoh · Apr 1, 2024 · bd4c251 · bd4c251
1 parent f77c08a
commit bd4c251
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...ngodb/src/test/java/org/opensearch/dataprepper/plugins/mongo/stream/StreamWorkerTest.java b/...ngodb/src/test/java/org/opensearch/dataprepper/plugins/mongo/stream/StreamWorkerTest.java
@@ -175,9 +175,7 @@ void test_processStream_checkPointIntervalSuccess() {
             when(streamDoc2.getResumeToken()).thenReturn(bsonDoc2);
             when(streamDoc3.getResumeToken()).thenReturn(bsonDoc3);
             when(cursor.next())
-                .thenReturn(streamDoc1)
-                .thenReturn(streamDoc2)
-                .thenReturn(streamDoc3);
+                .thenReturn(streamDoc1, streamDoc2, streamDoc3);
             when(doc1.toJson(any(JsonWriterSettings.class))).thenReturn(UUID.randomUUID().toString());
             when(doc2.toJson(any(JsonWriterSettings.class))).thenReturn(UUID.randomUUID().toString());
             when(doc3.toJson(any(JsonWriterSettings.class))).thenReturn(UUID.randomUUID().toString());