Fix tests.

Author: drieseng

src/Renci.SshNet.Tests/Classes/SessionTest_Connected_ConnectionReset.cs

@@ -64,14 +64,8 @@ public void ErrorOccurredIsRaisedOnce()
 
             var connectionException = (SshConnectionException) exception;
             Assert.AreEqual(DisconnectReason.ConnectionLost, connectionException.DisconnectReason);
-
-            var innerException = exception.InnerException;
-            Assert.IsNotNull(innerException);
-            Assert.AreEqual(typeof(SocketException), innerException.GetType());
-
-            var socketException = (SocketException) innerException;
-            Assert.AreSame(connectionException.Message, socketException.Message);
-            Assert.AreEqual(SocketError.ConnectionReset, socketException.SocketErrorCode);
+            Assert.IsNull(connectionException.InnerException);
+            Assert.AreEqual("An established connection was aborted by the server.", connectionException.Message);
         }
 
         [TestMethod]
@@ -140,7 +134,7 @@ public void ISession_TrySendMessageShouldReturnFalse()
         }
 
         [TestMethod]
-        public void ISession_WaitOnHandle_WaitHandle_ShouldThrowSshConnectionExceptionDetailingConnectionReset()
+        public void ISession_WaitOnHandle_WaitHandle_ShouldThrowSshConnectionException()
         {
             var session = (ISession) Session;
             var waitHandle = new ManualResetEvent(false);
@@ -152,22 +146,14 @@ public void ISession_WaitOnHandle_WaitHandle_ShouldThrowSshConnectionExceptionDe
             }
             catch (SshConnectionException ex)
             {
+                Assert.AreEqual("An established connection was aborted by the server.", ex.Message);
+                Assert.IsNull(ex.InnerException);
                 Assert.AreEqual(DisconnectReason.ConnectionLost, ex.DisconnectReason);
-
-                var innerException = ex.InnerException;
-                Assert.IsNotNull(innerException);
-                Assert.AreEqual(typeof(SocketException), innerException.GetType());
-
-                var socketException = (SocketException) ex.InnerException;
-                Assert.IsNotNull(socketException);
-                Assert.IsNull(socketException.InnerException);
-                Assert.AreSame(innerException.Message, ex.Message);
-                Assert.AreEqual(SocketError.ConnectionReset, socketException.SocketErrorCode);
             }
         }
 
         [TestMethod]
-        public void ISession_WaitOnHandle_WaitHandleAndTimeout_ShouldThrowSshConnectionExceptionDetailingConnectionReset()
+        public void ISession_WaitOnHandle_WaitHandleAndTimeout_ShouldThrowSshConnectionException()
         {
             var session = (ISession) Session;
             var waitHandle = new ManualResetEvent(false);
@@ -180,16 +166,8 @@ public void ISession_WaitOnHandle_WaitHandleAndTimeout_ShouldThrowSshConnectionE
             catch (SshConnectionException ex)
             {
                 Assert.AreEqual(DisconnectReason.ConnectionLost, ex.DisconnectReason);
-
-                var innerException = ex.InnerException;
-                Assert.IsNotNull(innerException);
-                Assert.AreEqual(typeof(SocketException), innerException.GetType());
-
-                var socketException = (SocketException) ex.InnerException;
-                Assert.IsNotNull(socketException);
-                Assert.IsNull(socketException.InnerException);
-                Assert.AreSame(innerException.Message, socketException.Message);
-                Assert.AreEqual(SocketError.ConnectionReset, socketException.SocketErrorCode);
+                Assert.IsNull(ex.InnerException);
+                Assert.AreEqual("An established connection was aborted by the server.", ex.Message);
             }
         }
 

src/Renci.SshNet/Session.cs

@@ -1942,8 +1942,7 @@ private void MessageListener()
                         var connectionClosedOrDataAvailable = socket.Poll(-1, SelectMode.SelectRead);
                         if (connectionClosedOrDataAvailable && socket.Available == 0)
                         {
-                            // connection with SSH server was closed or socket was disposed;
-                            // break out of the message loop
+                            // connection with SSH server was closed or connection was reset
                             break;
                         }
     #elif FEATURE_SOCKET_SELECT
@@ -1993,6 +1992,8 @@ private void MessageListener()
                         // * a call to Disconnect()
                         // * a call to Dispose()
                         // * a SSH_MSG_DISCONNECT received from server
+
+                        Console.WriteLine("B");
                         break;
                     }
 
@@ -2001,13 +2002,16 @@ private void MessageListener()
                     {
                         // connection with SSH server was closed;
                         // break out of the message loop
+                        Console.WriteLine("C");
                         break;
                     }
 
                     // process message
                     message.Process(this);
                 }
 
+                Console.WriteLine("D");
+
                 // connection with SSH server was closed or socket was disposed
                 RaiseError(CreateConnectionAbortedByServerException());
             }