gerrit.onap Code Review - dcaegen2/collectors/datafile.git/blob

   1 /*
   2  * ============LICENSE_START======================================================================
   3  * Copyright (C) 2018 Nordix Foundation. All rights reserved.
   4  * ===============================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   6  * in compliance with the License. You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software distributed under the License
  11  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  12  * or implied. See the License for the specific language governing permissions and limitations under
  13  * the License.
  14  * ============LICENSE_END========================================================================
  15  */
  16
  17 package org.onap.dcaegen2.collectors.datafile.tasks;
  18
  19 import static org.mockito.Mockito.mock;
  20 import static org.mockito.Mockito.times;
  21 import static org.mockito.Mockito.verify;
  22 import static org.mockito.Mockito.verifyNoMoreInteractions;
  23 import static org.mockito.Mockito.when;
  24
  25 import java.io.File;
  26
  27 import org.junit.jupiter.api.Test;
  28 import org.onap.dcaegen2.collectors.datafile.ftp.FileServerData;
  29 import org.onap.dcaegen2.collectors.datafile.ftp.FtpsClient;
  30 import org.onap.dcaegen2.collectors.datafile.ftp.ImmutableFileServerData;
  31 import org.onap.dcaegen2.collectors.datafile.ftp.SftpClient;
  32 import org.onap.dcaegen2.collectors.datafile.model.ConsumerDmaapModel;
  33 import org.onap.dcaegen2.collectors.datafile.model.FileData;
  34 import org.onap.dcaegen2.collectors.datafile.model.ImmutableConsumerDmaapModel;
  35 import org.onap.dcaegen2.collectors.datafile.model.ImmutableFileData;
  36
  37 import reactor.test.StepVerifier;
  38
  39 /**
  40  * @author <a href="mailto:henrik.b.andersson@est.tech">Henrik Andersson</a>
  41  *
  42  */
  43 public class XnfCollectorTaskImplTest {
  44
  45     private static final String PM_MEAS_CHANGE_IDINTIFIER = "PM_MEAS_FILES";
  46     private static final String FILE_READY_CHANGE_TYPE = "FileReady";
  47     private static final String FTPES_SCHEME = "ftpes://";
  48     private static final String SFTP_SCHEME = "sftp://";
  49     private static final String SERVER_ADDRESS = "192.168.0.101";
  50     private static final int PORT_22 = 22;
  51     private static final String PM_FILE_NAME = "A20161224.1030-1045.bin.gz";
  52     private static final String REMOTE_FILE_LOCATION = "/ftp/rop/" + PM_FILE_NAME;
  53     private static final String LOCAL_FILE_LOCATION = "target" + File.separator + PM_FILE_NAME;
  54     private static final String USER = "usr";
  55     private static final String PWD = "pwd";
  56     private static final String FTPES_LOCATION =
  57             FTPES_SCHEME + USER + ":" + PWD + "@" + SERVER_ADDRESS + ":" + PORT_22 + REMOTE_FILE_LOCATION;
  58     private static final String SFTP_LOCATION = SFTP_SCHEME + SERVER_ADDRESS + ":" + PORT_22 + REMOTE_FILE_LOCATION;
  59     private static final String GZIP_COMPRESSION = "gzip";
  60     private static final String MEAS_COLLECT_FILE_FORMAT_TYPE = "org.3GPP.32.435#measCollec";
  61     private static final String FILE_FORMAT_VERSION = "V10";
  62
  63     private FtpsClient ftpsClientMock = mock(FtpsClient.class);
  64
  65     private SftpClient sftpClientMock = mock(SftpClient.class);
  66
  67     private XnfCollectorTask collectorUndetTest = new XnfCollectorTaskImpl(ftpsClientMock, sftpClientMock);
  68
  69     @Test
  70     public void whenSingleFtpesFile_returnCorrectResponse() {
  71         FileData fileData = ImmutableFileData.builder().changeIdentifier(PM_MEAS_CHANGE_IDINTIFIER)
  72                 .changeType(FILE_READY_CHANGE_TYPE).name(PM_FILE_NAME).location(FTPES_LOCATION)
  73                 .compression(GZIP_COMPRESSION).fileFormatType(MEAS_COLLECT_FILE_FORMAT_TYPE)
  74                 .fileFormatVersion(FILE_FORMAT_VERSION).build();
  75
  76         FileServerData fileServerData = ImmutableFileServerData.builder().serverAddress(SERVER_ADDRESS).userId(USER)
  77                 .password(PWD).port(PORT_22).build();
  78         when(ftpsClientMock.collectFile(fileServerData, REMOTE_FILE_LOCATION, LOCAL_FILE_LOCATION))
  79                 .thenReturn(Boolean.TRUE);
  80
  81         ConsumerDmaapModel expectedConsumerDmaapModel = ImmutableConsumerDmaapModel.builder().name(PM_FILE_NAME)
  82                 .location(LOCAL_FILE_LOCATION).compression(GZIP_COMPRESSION)
  83                 .fileFormatType(MEAS_COLLECT_FILE_FORMAT_TYPE).fileFormatVersion(FILE_FORMAT_VERSION).build();
  84
  85         StepVerifier.create(collectorUndetTest.execute(fileData)).expectNext(expectedConsumerDmaapModel)
  86                 .verifyComplete();
  87
  88         verify(ftpsClientMock, times(1)).collectFile(fileServerData, REMOTE_FILE_LOCATION, LOCAL_FILE_LOCATION);
  89         verifyNoMoreInteractions(ftpsClientMock);
  90     }
  91
  92     @Test
  93     public void whenSingleSftpFile_returnCorrectResponse() {
  94         FileData fileData = ImmutableFileData.builder().changeIdentifier(PM_MEAS_CHANGE_IDINTIFIER)
  95                 .changeType(FILE_READY_CHANGE_TYPE).name(PM_FILE_NAME).location(SFTP_LOCATION)
  96                 .compression(GZIP_COMPRESSION).fileFormatType(MEAS_COLLECT_FILE_FORMAT_TYPE)
  97                 .fileFormatVersion(FILE_FORMAT_VERSION).build();
  98
  99         FileServerData fileServerData = ImmutableFileServerData.builder().serverAddress(SERVER_ADDRESS).userId("")
 100                 .password("").port(PORT_22).build();
 101         when(sftpClientMock.collectFile(fileServerData, REMOTE_FILE_LOCATION, LOCAL_FILE_LOCATION))
 102                 .thenReturn(Boolean.TRUE);
 103
 104         ConsumerDmaapModel expectedConsumerDmaapModel = ImmutableConsumerDmaapModel.builder().name(PM_FILE_NAME)
 105                 .location(LOCAL_FILE_LOCATION).compression(GZIP_COMPRESSION)
 106                 .fileFormatType(MEAS_COLLECT_FILE_FORMAT_TYPE).fileFormatVersion(FILE_FORMAT_VERSION).build();
 107
 108         StepVerifier.create(collectorUndetTest.execute(fileData)).expectNext(expectedConsumerDmaapModel)
 109                 .verifyComplete();
 110
 111         verify(sftpClientMock, times(1)).collectFile(fileServerData, REMOTE_FILE_LOCATION, LOCAL_FILE_LOCATION);
 112         verifyNoMoreInteractions(ftpsClientMock);
 113     }
 114
 115     @Test
 116     public void whenWrongScheme_returnEmpty() {
 117         FileData fileData = ImmutableFileData.builder().changeIdentifier(PM_MEAS_CHANGE_IDINTIFIER)
 118                 .changeType(FILE_READY_CHANGE_TYPE).name(PM_FILE_NAME).location("http://host.com/file.zip")
 119                 .compression(GZIP_COMPRESSION).fileFormatType(MEAS_COLLECT_FILE_FORMAT_TYPE)
 120                 .fileFormatVersion(FILE_FORMAT_VERSION).build();
 121
 122         FileServerData fileServerData = ImmutableFileServerData.builder().serverAddress(SERVER_ADDRESS).userId("")
 123                 .password("").port(PORT_22).build();
 124         when(sftpClientMock.collectFile(fileServerData, REMOTE_FILE_LOCATION, LOCAL_FILE_LOCATION))
 125                 .thenReturn(Boolean.TRUE);
 126
 127         StepVerifier.create(collectorUndetTest.execute(fileData)).expectNextCount(0).verifyComplete();
 128
 129         verifyNoMoreInteractions(ftpsClientMock);
 130     }
 131 }