This is an automated email from the ASF dual-hosted git repository.

lgoldstein pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/mina-sshd.git

commit 16083e60b88469994f3d40303fddb3c83855e2b8
Author: Lyor Goldstein <[email protected]>
AuthorDate: Tue Jan 29 13:46:32 2019 +0200

    Slightly stricter code in AsyncAuth related tests
---
 .../sshd/server/auth/AsyncAuthInteractiveTest.java | 32 ++++++++++---------
 .../org/apache/sshd/server/auth/AsyncAuthTest.java | 36 ++++++++++++----------
 .../apache/sshd/server/auth/AsyncAuthTestBase.java |  5 ++-
 3 files changed, 39 insertions(+), 34 deletions(-)

diff --git 
a/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthInteractiveTest.java
 
b/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthInteractiveTest.java
index 40e86c8..c5b31ff 100644
--- 
a/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthInteractiveTest.java
+++ 
b/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthInteractiveTest.java
@@ -18,6 +18,7 @@
  */
 package org.apache.sshd.server.auth;
 
+import org.apache.sshd.common.channel.Channel;
 import org.junit.FixMethodOrder;
 import org.junit.runners.MethodSorters;
 
@@ -39,10 +40,7 @@ public class AsyncAuthInteractiveTest extends 
AsyncAuthTestBase {
     @Override
     protected boolean authenticate() throws Exception {
         JSch jsch = new JSch();
-        Session session;
-        ChannelShell channel;
-
-        session = jsch.getSession("whatever", "localhost", port);
+        Session session = jsch.getSession("whatever", "localhost", port);
         session.setUserInfo(new UserInfo() {
             @Override
             public String getPassphrase() {
@@ -77,7 +75,8 @@ public class AsyncAuthInteractiveTest extends 
AsyncAuthTestBase {
         try {
             session.connect();
         } catch (JSchException e) {
-            switch (e.getMessage()) {
+            String reason = e.getMessage();
+            switch (reason) {
                 case "Auth cancel":
                 case "Auth fail":
                     return false;
@@ -85,19 +84,22 @@ public class AsyncAuthInteractiveTest extends 
AsyncAuthTestBase {
                     throw e;
             }
         }
-        channel = (ChannelShell) session.openChannel("shell");
-        channel.connect();
 
         try {
-            channel.disconnect();
-        } catch (Exception ignore) {
-            // ignore
-        }
+            ChannelShell channel = (ChannelShell) 
session.openChannel(Channel.CHANNEL_SHELL);
+            channel.connect();
 
-        try {
-            session.disconnect();
-        } catch (Exception ignore) {
-            // ignore
+            try {
+                channel.disconnect();
+            } catch (Exception ignore) {
+                // ignore
+            }
+        } finally {
+            try {
+                session.disconnect();
+            } catch (Exception ignore) {
+                // ignore
+            }
         }
 
         return true;
diff --git 
a/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTest.java 
b/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTest.java
index 70530db..2ca0000 100644
--- a/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTest.java
+++ b/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTest.java
@@ -18,6 +18,9 @@
  */
 package org.apache.sshd.server.auth;
 
+import java.util.Objects;
+
+import org.apache.sshd.common.channel.Channel;
 import org.junit.FixMethodOrder;
 import org.junit.runners.MethodSorters;
 
@@ -32,7 +35,6 @@ import com.jcraft.jsch.UserInfo;
  */
 @FixMethodOrder(MethodSorters.NAME_ASCENDING)
 public class AsyncAuthTest extends AsyncAuthTestBase {
-
     public AsyncAuthTest() {
         super();
     }
@@ -40,10 +42,7 @@ public class AsyncAuthTest extends AsyncAuthTestBase {
     @Override
     protected boolean authenticate() throws Exception {
         JSch jsch = new JSch();
-        Session session;
-        ChannelShell channel;
-
-        session = jsch.getSession("whatever", "localhost", port);
+        Session session = jsch.getSession("whatever", "localhost", port);
         session.setPassword("whocares");
         session.setUserInfo(new UserInfo() {
             @Override
@@ -76,28 +75,33 @@ public class AsyncAuthTest extends AsyncAuthTestBase {
                 // Do nothing
             }
         });
+
         try {
             session.connect();
         } catch (JSchException e) {
-            if (e.getMessage().equals("Auth cancel")) {
+            String reason = e.getMessage();
+            if (Objects.equals(reason, "Auth cancel")) {
                 return false;
             } else {
                 throw e;
             }
         }
-        channel = (ChannelShell) session.openChannel("shell");
-        channel.connect();
 
         try {
-            channel.disconnect();
-        } catch (Exception ignore) {
-            // ignored
-        }
+            ChannelShell channel = (ChannelShell) 
session.openChannel(Channel.CHANNEL_SHELL);
+            channel.connect();
 
-        try {
-            session.disconnect();
-        } catch (Exception ignore) {
-            // ignored
+            try {
+                channel.disconnect();
+            } catch (Exception ignore) {
+                // ignored
+            }
+        } finally {
+            try {
+                session.disconnect();
+            } catch (Exception ignore) {
+                // ignored
+            }
         }
 
         return true;
diff --git 
a/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTestBase.java 
b/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTestBase.java
index ba903f9..9f035aa 100644
--- a/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTestBase.java
+++ b/sshd-core/src/test/java/org/apache/sshd/server/auth/AsyncAuthTestBase.java
@@ -37,9 +37,8 @@ import com.jcraft.jsch.JSchException;
  * @author <a href="mailto:[email protected]";>Apache MINA SSHD Project</a>
  */
 public abstract class AsyncAuthTestBase extends BaseTestSupport {
-
-    SshServer server;
-    int port;
+    protected SshServer server;
+    protected int port;
 
     private PasswordAuthenticator authenticator;
 

Reply via email to