Fix Log message level and minor PR comments

Signed-off-by: Dinu John <[email protected]>
dinujoh · May 9, 2024 · 50b940b · 50b940b
1 parent 28d500a
commit 50b940b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/...s/mongodb/src/main/java/org/opensearch/dataprepper/plugins/mongo/stream/StreamWorker.java b/...s/mongodb/src/main/java/org/opensearch/dataprepper/plugins/mongo/stream/StreamWorker.java
@@ -354,7 +354,7 @@ private void bufferWriteAndCheckpointStream() {
                         LOG.debug("Perform regular checkpoint for resume token {} at record count {}", lastLocalCheckpoint, lastLocalRecordCount);
                         partitionCheckpoint.checkpoint(lastLocalCheckpoint, lastLocalRecordCount);
                     } catch (Exception e) {
-                        LOG.error("Exception checkpointing the current state. New thread should start the stream from previous checkpoint.", e);
+                        LOG.warn("Exception checkpointing the current state. New thread should start the stream from previous checkpoint.", e);
                         stop();
                     } finally {
                         lock.unlock();;

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
@@ -310,9 +310,8 @@ void test_processStream_checkPointIntervalSuccess() {
         verify(cursor).close();
         verify(cursor, times(4)).hasNext();
         verify(mockPartitionCheckpoint).getGlobalS3FolderCreationStatus(collection);
-        //verify(mockPartitionCheckpoint, atLeast(1)).checkpoint(resumeToken3, 3);
+        verify(mockPartitionCheckpoint, atLeast(1)).checkpoint(resumeToken3, 3);
         verify(successItemsCounter).increment(1);
-        //verify(mockPartitionCheckpoint, atLeast(1)).checkpoint(resumeToken2, 2);
         verify(mockRecordBufferWriter, times(2)).writeToBuffer(eq(null), any());
         verify(successItemsCounter).increment(2);
         verify(failureItemsCounter, never()).increment();