1 files changed, 139 insertions, 0 deletions
diff --git a/src/test/java/org/onap/a1pesimulator/service/fileready/FtpServerServiceTest.java b/src/test/java/org/onap/a1pesimulator/service/fileready/FtpServerServiceTest.java
new file mode 100644
index 0000000..e8b77ef
--- /dev/null
+++ b/src/test/java/org/onap/a1pesimulator/service/fileready/FtpServerServiceTest.java
@@ -0,0 +1,139 @@
+package org.onap.a1pesimulator.service.fileready;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.onap.a1pesimulator.service.fileready.FtpServerService.deletePMBulkFile;
+import static org.onap.a1pesimulator.util.Constants.TEMP_DIR;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.jupiter.api.Test;
+import org.mockito.Mock;
+import org.mockito.Spy;
+import org.onap.a1pesimulator.data.fileready.FileData;
+import org.slf4j.LoggerFactory;
+import org.springframework.test.util.ReflectionTestUtils;
+
+import ch.qos.logback.classic.Logger;
+import ch.qos.logback.classic.spi.ILoggingEvent;
+import ch.qos.logback.core.read.ListAppender;
+import net.schmizz.sshj.SSHClient;
+import net.schmizz.sshj.sftp.SFTPClient;
+import reactor.test.StepVerifier;
+
+
+class FtpServerServiceTest extends CommonFileReady {
+
+    @Spy
+    FtpServerService ftpServerService;
+
+    @Mock
+    SSHClient sshClient;
+
+    @Mock
+    SFTPClient sftpClient;
+
+    /**
+     * Test to save archived PM Bulk File into specify directory
+     */
+    @Test
+    void saveFileToFtp() {
+        ReflectionTestUtils.setField(ftpServerService, "xmlPmLocation", TEMP_DIR);
+        File archivedPmBulkFile = createTempFile(ARCHIVED_PM_BULK_FILE);
+        FileData testFileData = getTestFileData();
+        FileData expectedFileData = FileData.builder().archivedPmBulkFile(archivedPmBulkFile).pmBulkFile(testFileData.getPmBulkFile()).build();
+        expectedFileData.setArchivedPmBulkFile(archivedPmBulkFile);
+
+        StepVerifier.create(ftpServerService.uploadFileToFtp(testFileData))
+                .expectNext(expectedFileData)
+                .verifyComplete();
+    }
+
+    /**
+     * Test successful FTP upload
+     */
+    @Test
+    void uploadFileToFtp() {
+        ReflectionTestUtils.setField(ftpServerService, "ftpServerUpload", true);
+        doReturn(sshClient).when(ftpServerService).getSSHClient();
+        try {
+            doReturn(sftpClient).when(sshClient).newSFTPClient();
+            doNothing().when(sftpClient).put(anyString(), anyString());
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+
+        File archivedPmBulkFile = createTempFile(ARCHIVED_PM_BULK_FILE);
+        FileData testFileData = getTestFileData();
+        FileData expectedFileData = FileData.builder().archivedPmBulkFile(archivedPmBulkFile).pmBulkFile(testFileData.getPmBulkFile()).build();
+        expectedFileData.setArchivedPmBulkFile(archivedPmBulkFile);
+
+        StepVerifier.create(ftpServerService.uploadFileToFtp(testFileData))
+                .expectNext(expectedFileData).verifyComplete();
+    }
+
+    /**
+     * Test error while trying to upload archived PM Bulk File
+     */
+    @Test
+    void errorWhileUploadingFileToFtp() {
+        ReflectionTestUtils.setField(ftpServerService, "ftpServerUpload", true);
+        ReflectionTestUtils.setField(ftpServerService, "ftpServerPort", "22");
+        FileData testFileData = getTestFileData();
+        StepVerifier.create(ftpServerService.uploadFileToFtp(testFileData))
+                .verifyComplete();
+        verify(ftpServerService, times(1)).resumeError(any(), any());
+    }
+
+    /**
+     * Test error while trying to delete not existing file
+     */
+    @Test
+    void errorWhileDeletingFile() {
+        Logger testLog = (Logger) LoggerFactory.getLogger(FtpServerService.class);
+        ListAppender<ILoggingEvent> appender = new ListAppender<>();
+        appender.start();
+        testLog.addAppender(appender);
+
+        deletePMBulkFile(new File("test.txt"));
+        assertThat(appender.list).extracting(ILoggingEvent::getFormattedMessage).containsExactly("Could not delete file: test.txt");
+    }
+
+    /**
+     * Test if path to FTP is created correctly
+     */
+    @Test
+    void getFtpPath() {
+        List<String> ftpPathVars = new ArrayList<>();
+        ftpPathVars.add("ftpServerProtocol");
+        ftpPathVars.add("ftpServerUsername");
+        ftpPathVars.add("ftpServerPassword");
+        ftpPathVars.add("ftpServerUrl");
+        ftpPathVars.add("ftpServerPort");
+        ftpPathVars.add("ftpServerFilepath");
+
+        ftpPathVars.forEach(var -> ReflectionTestUtils.setField(ftpServerService, var, var));
+        String ftpPath = ftpServerService.getFtpPath();
+
+        assertTrue(ftpPathVars.stream().allMatch(ftpPath::contains));
+    }
+
+    /**
+     * Creates FileData object for test cases
+     *
+     * @return test FileData object
+     */
+    private FileData getTestFileData() {
+        return FileData.builder().pmBulkFile(createTempFile(PM_BULK_FILE)).build();
+    }
+
+}
+\ No newline at end of file