mockito 如何模拟SFTP连接的连接

2skhul33  于 2022-11-08  发布在  其他
关注(0)|答案(1)|浏览(350)

我的程序打开sftp连接,并连接到服务器,以获得一个文件,然后处理。我正在写一个测试用例,这个方法,试图模拟连接。
我的实际类是:

public void getFile() {
    Session session = null;
    Channel channel = null;
    ChannelSftp channelSftp = null;
    String path = ftpurl;
    try {
        JSch jsch = new JSch();
        session = jsch.getSession(username, host, port);
        session.setConfig("StrictHostKeyChecking", "no");session.setPassword(JaspytPasswordEncryptor.getDecryptedString(jaspytEncryptionKey, jaspytEncryptionAlgorithm, password));

        session.connect();
        channel = session.openChannel("sftp");
        channel.connect();
        channelSftp = (ChannelSftp) channel;

        channelSftp.cd(path);

        Vector<ChannelSftp.LsEntry> list = channelSftp.ls("*.csv");
        for (ChannelSftp.LsEntry entry : list) {
            if (entry.getFilename().startsWith("A...")) {
                findByFileName(entry.getFilename());
            }
        }
        channelSftp.exit();
        session.disconnect();
    } catch (JSchException e) {
        LOGGER.error("JSch exception"+e);
    } catch (SftpException e) {
        LOGGER.error("Sftp Exception"+e);
    }
}

测试类迄今为止:

@Test
public void getNamesTestValid() throws IOException, JSchException { 
    JSch jsch = new JSch();

    Hashtable config = new Hashtable();
    config.put("StrictHostKeyChecking", "no");
    JSch.setConfig(config);

    Session session = jsch.getSession( "remote-username", "localhost", 22999);
    session.setPassword("remote-password");

    session.connect();

    Channel channel = session.openChannel( "sftp" );
    channel.connect();

    ChannelSftp sftpChannel = (ChannelSftp) channel;
        Mockito.when(fileRepository.findByFileName(fileName)).thenReturn(fileDetail);
    scheduler.getCSVFileNames();    
}

当尝试模拟连接时,它会搜索实际端口,错误是端口无效,连接被拒绝。我只想模拟连接。我的另一个疑问是在模拟连接后,我应该从哪里读取文件细节。

wmomyfyw

wmomyfyw1#

这是因为你创建的是一个与new Jsch()的实际连接,你需要用一个库来模拟这个连接,比如Mockito

@RunWith(MockitoJUnitRunner.class)
class Test {

    @Mock(answer = Answers.RETURNS_DEEP_STUBS)
    private JSch mockJsch;

    ..

    void test() {
        Session mockSession = mockJsch.getSession("username", "localhost", 22999);
        ..
    }
}

相关问题