Test configurable channel timeout

1 files changed, 19 insertions, 0 deletions

diff --git a/tests/test_client.py b/tests/test_client.py
index dae5b13a..7b77ef49 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -564,6 +564,25 @@ class SSHClientTest(ClientTest):
             auth_timeout=0.5,
         )
 
+    def _mock_set_remote_channel(self, chainid, init_window, max_packet_size):
+        time.sleep(100)
+
+    @patch.object(paramiko.Channel, "_set_remote_channel", _mock_set_remote_channel)
+    def test_channel_timeout(self):
+        """
+        verify that the SSHClient has a configurable channel timeout
+        """
+        threading.Thread(target=self._run).start()
+        # Client setup
+        self.tc = SSHClient()
+        self.tc.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+        # Actual connection
+        self.tc.connect(**dict(self.connect_kwargs, password="pygmalion", channel_timeout=0.5))
+        self.event.wait(1.0)
+
+        self.assertRaises(paramiko.SSHException, self.tc.open_sftp)
+
     @requires_gss_auth
     def test_auth_trickledown_gsskex(self):
         """