Fix sporadic unit test failure

ronf · ronf · commit b77bf85ac8f3 · 2024-10-12T19:24:34.000-07:00
diff --git a/tests/test_connection.py b/tests/test_connection.py
@@ -1002,7 +1002,7 @@ def send_newkeys(self, k, h):
 
         with patch('asyncssh.connection.SSHClientConnection.send_newkeys',
                    send_newkeys):
-            with self.assertRaises(asyncssh.ProtocolError):
+            with self.assertRaises((ConnectionError, asyncssh.ProtocolError)):
                 await self.connect()
 
     @asynctest
