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.springframework.test.util.ReflectionTestUtils;

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() {
        ListAppender<ILoggingEvent> appender = createCommonLog(FtpServerService.class);

        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();
    }

}