模擬 sftp 連接中斷

Question

我試圖在不使用 sftp.disconnect() 方法的情況下突然斷開 Java 中的 SFTP 連接。 使用它來構建一個集成測試，檢查是否每次都進行了清理。 請參閱下面的測試：

public void checkDisruptedConnections() throws JSchException, InterruptedException {
    ChannelSftp sftp = setupSftp(null);
    sftp.connect();

    try {
        //disrupt connection OVER HERE
    } catch (Exception e) {
        assertEquals("1", jedis.get(SESSION_KEY));
    }

    waitForConnectionClose();
    assertEquals("0", jedis.get(SESSION_KEY));
}

Answer 1

最終使用一個單獨的線程連接並被中斷。

@Test
public void testSftpInterupt() throws InterruptedException, JSchException, SftpException {
    Connect thread = new Connect();
    thread.start();

    while (thread.isAlive()) {
        waitForConnectionClose();
    }

    waitForConnectionClose();
    assertEquals("0", jedis.get(SESSION_KEY));
}

連接看起來如下：

private class Connect extends Thread {
    @Override
    public void run() {
        ChannelSftp sftp = null;
        Thread thread = this;

        SftpProgressMonitor monitor = new SftpProgressMonitor() {
            @Override
            public void init(int op, String src, String dest, long max) {
            }

            @Override
            public boolean count(long count) {
                currentThread().interrupt();
                return false;
            }

            @Override
            public void end() {
            }
        };

        try {
            sftp.connect();
            sftp.put(LOCAL_FILE_LARGE, remoteFile, monitor);
        } catch (JSchException | SftpException e) {
            assertEquals("java.io.InterruptedIOException", e.getMessage());
        }
    }
}