def test_keep_alive_and_compression(self):
conn_params = {"hostname": "dummy.host.org", "username": "******"}
client = ParamikoSSHClient(**conn_params)
mock_transport = Mock()
client.client.get_transport = Mock(return_value=mock_transport)
transport = client._get_transport()
self.assertEqual(transport.set_keepalive.call_count, 0)
self.assertEqual(transport.use_compression.call_count, 0)
conn_params = {
"hostname": "dummy.host.org",
"username": "******",
"keep_alive": 15,
"use_compression": True,
}
client = ParamikoSSHClient(**conn_params)
mock_transport = Mock()
client.client.get_transport = Mock(return_value=mock_transport)
transport = client._get_transport()
self.assertEqual(transport.set_keepalive.call_count, 1)
self.assertEqual(transport.use_compression.call_count, 1)
def test_keep_alive_and_compression(self):
conn_params = {'hostname': 'dummy.host.org', 'username': '******'}
client = ParamikoSSHClient(**conn_params)
mock_transport = Mock()
client.client.get_transport = Mock(return_value=mock_transport)
transport = client._get_transport()
self.assertEqual(transport.set_keepalive.call_count, 0)
self.assertEqual(transport.use_compression.call_count, 0)
conn_params = {
'hostname': 'dummy.host.org',
'username': '******',
'keep_alive': 15,
'use_compression': True
}
client = ParamikoSSHClient(**conn_params)
mock_transport = Mock()
client.client.get_transport = Mock(return_value=mock_transport)
transport = client._get_transport()
self.assertEqual(transport.set_keepalive.call_count, 1)
self.assertEqual(transport.use_compression.call_count, 1)
