diff options
author | Jeff Forcier <jeff@bitprophet.org> | 2018-03-12 09:15:55 -0700 |
---|---|---|
committer | Jeff Forcier <jeff@bitprophet.org> | 2018-03-12 09:15:55 -0700 |
commit | 66f90d8ca12c377db71285d123eeac3d06c658a5 (patch) | |
tree | 26b2997ca96d658afc7185158945fcbfbb43a024 | |
parent | afbe7a4112ac843f4d364a7a4122d40ab7c83225 (diff) | |
download | paramiko-66f90d8ca12c377db71285d123eeac3d06c658a5.tar.gz |
Allow overriding test client connect kwargs in Transport test suite
-rw-r--r-- | tests/test_transport.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tests/test_transport.py b/tests/test_transport.py index 188ce623..778fed4a 100644 --- a/tests/test_transport.py +++ b/tests/test_transport.py @@ -126,7 +126,9 @@ class TransportTest(unittest.TestCase): self.socks.close() self.sockc.close() - def setup_test_server(self, client_options=None, server_options=None): + def setup_test_server( + self, client_options=None, server_options=None, connect_kwargs=None, + ): host_key = RSAKey.from_private_key_file(test_path('test_rsa.key')) public_host_key = RSAKey(data=host_key.asbytes()) self.ts.add_server_key(host_key) @@ -140,8 +142,13 @@ class TransportTest(unittest.TestCase): self.server = NullServer() self.assertTrue(not event.is_set()) self.ts.start_server(event, self.server) - self.tc.connect(hostkey=public_host_key, - username='slowdive', password='pygmalion') + if connect_kwargs is None: + connect_kwargs = dict( + hostkey=public_host_key, + username='slowdive', + password='pygmalion', + ) + self.tc.connect(**connect_kwargs) event.wait(1.0) self.assertTrue(event.is_set()) self.assertTrue(self.ts.is_active()) |