Apache
/
jclouds
mirror of https://github.com/apache/jclouds.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Issue 760:sshj driver throws exception disconnecting when client is already disconnected

This commit is contained in:
Adrian Cole 2011-11-28 12:11:55 -05:00
parent 4981ca6efb
commit 09678a1df0
2 changed files with 31 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
drivers/sshj/src/main/java/org/jclouds/sshj/SshjSshClient.java
View File

@ -39,6 +39,7 @@ import javax.annotation.PreDestroy;
import javax.annotation.Resource;
import javax.inject.Named;
import net.schmizz.sshj.SSHClient;
import net.schmizz.sshj.common.IOUtils;
import net.schmizz.sshj.connection.ConnectionException;
import net.schmizz.sshj.connection.channel.direct.Session;
@ -126,7 +127,8 @@ public class SshjSshClient implements SshClient {
@Named("jclouds.ssh")
protected Logger logger = Logger.NULL;
private net.schmizz.sshj.SSHClient ssh;
@VisibleForTesting
SSHClient ssh;
private final byte[] privateKey;
final byte[] emptyPassPhrase = new byte[0];
private final int timeoutMillis;
@ -176,7 +178,7 @@ public class SshjSshClient implements SshClient {
try {
ssh.disconnect();
} catch (IOException e) {
Throwables.propagate(e);
logger.warn(e, "<< exception disconnecting from %s: %s", e, e.getMessage());
}
ssh = null;
}

26
drivers/sshj/src/test/java/org/jclouds/sshj/SshjSshClientTest.java
View File

@ -19,12 +19,18 @@
package org.jclouds.sshj;
import static com.google.inject.name.Names.bindProperties;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.expectLastCall;
import static org.easymock.classextension.EasyMock.createMock;
import static org.easymock.classextension.EasyMock.replay;
import static org.easymock.classextension.EasyMock.verify;
import java.io.IOException;
import java.net.ConnectException;
import java.net.SocketTimeoutException;
import java.util.Properties;
import net.schmizz.sshj.SSHClient;
import net.schmizz.sshj.common.SSHException;
import net.schmizz.sshj.connection.ConnectionException;
import net.schmizz.sshj.transport.TransportException;
@ -123,8 +129,7 @@ public class SshjSshClientTest {
new SSHException("Session.connect: java.io.IOException: End of IO Stream Read")).apply(
" End of IO Stream Read");
assert ssh.causalChainHasMessageContaining(
new SSHException("Session.connect: java.net.SocketException: Connection reset"))
.apply("java.net.Socket");
new SSHException("Session.connect: java.net.SocketException: Connection reset")).apply("java.net.Socket");
assert !ssh.causalChainHasMessageContaining(new NullPointerException()).apply(" End of IO Stream Read");
}
@ -140,7 +145,10 @@ public class SshjSshClientTest {
private static class ExceptionWithStrangeToString extends RuntimeException {
private static final long serialVersionUID = 1L;
private static final String MESSAGE = "foo-bar-exception-tostring";
public String toString() { return MESSAGE; }
public String toString() {
return MESSAGE;
}
}
public void testRetryOnToStringCustom() {
@ -151,6 +159,18 @@ public class SshjSshClientTest {
assert ssh1.shouldRetry(new RuntimeException(nex));
}
public void testDontThrowIOExceptionOnClear() throws Exception {
SshjSshClient ssh1 = createClient();
SSHClient ssh = createMock(SSHClient.class);
expect(ssh.isConnected()).andReturn(true);
ssh.disconnect();
expectLastCall().andThrow(new ConnectionException("disconnected"));
replay(ssh);
ssh1.ssh = ssh;
ssh1.sshConnection.clear();
verify(ssh);
}
public void testRetryNotOnToStringCustomMismatch() {
Exception nex = new ExceptionWithStrangeToString();
Properties props = new Properties();