Added unit tests for raw record check

Author: inv-jishnu

data-loader/core/src/test/java/com/scalar/db/dataloader/core/dataimport/log/SingleFileImportLoggerTest.java

@@ -2,6 +2,8 @@
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import com.fasterxml.jackson.core.type.TypeReference;
@@ -160,6 +162,60 @@ private void assertTransactionBatchResults(
     }
   }
 
+  @Test
+  void onTransactionBatchCompleted_HasErrors_ShouldWriteToFailureLogFileWithoutRawRecord()
+      throws IOException {
+    testTransactionBatchCompletedForRawRecords(false);
+  }
+
+  @Test
+  void onTransactionBatchCompleted_HasErrors_ShouldWriteToFailureLogFileWithRawRecord()
+      throws IOException {
+    testTransactionBatchCompletedForRawRecords(true);
+  }
+
+  private void testTransactionBatchCompletedForRawRecords(boolean logRawRecords)
+      throws IOException {
+    // Arrange
+    ImportLoggerConfig config =
+        ImportLoggerConfig.builder()
+            .logDirectoryPath(tempDir.toString() + "/")
+            .isLogRawSourceRecordsEnabled(logRawRecords)
+            .isLogSuccessRecordsEnabled(false)
+            .build();
+    SingleFileImportLogger importLogger = new SingleFileImportLogger(config, logWriterFactory);
+
+    List<ImportTransactionBatchResult> batchResults = createBatchResults(1, false);
+
+    // Act
+    for (ImportTransactionBatchResult batchResult : batchResults) {
+      importLogger.onTransactionBatchCompleted(batchResult);
+      importLogger.onDataChunkCompleted(
+          ImportDataChunkStatus.builder().dataChunkId(batchResult.getDataChunkId()).build());
+    }
+    importLogger.onAllDataChunksCompleted();
+
+    // Assert
+    assertTransactionBatchResultsForRawRecords(logRawRecords);
+  }
+
+  private void assertTransactionBatchResultsForRawRecords(boolean logRawRecord) throws IOException {
+    DataLoaderObjectMapper objectMapper = new DataLoaderObjectMapper();
+
+    Path logFileName = tempDir.resolve(SingleFileImportLogger.FAILURE_LOG_FILE_NAME);
+    assertTrue(Files.exists(logFileName), "Log file should exist");
+    String logContent = new String(Files.readAllBytes(logFileName), StandardCharsets.UTF_8);
+    List<ImportTransactionBatchResult> logEntries =
+        objectMapper.readValue(
+            logContent, new TypeReference<List<ImportTransactionBatchResult>>() {});
+    ImportTaskResult importTaskResult = logEntries.get(0).getRecords().get(0);
+    if (logRawRecord) {
+      assertEquals(OBJECT_MAPPER.createObjectNode(), importTaskResult.getRawRecord());
+    } else {
+      assertNull(importTaskResult.getRawRecord());
+    }
+  }
+
   private void assertTransactionBatchResult(
       ImportTransactionBatchResult expected, ImportTransactionBatchResult actual) {
     assertEquals(expected.getDataChunkId(), actual.getDataChunkId(), "Data chunk ID should match");