def test_putfo_relative_path(self):
conn_params = {"hostname": "dummy.host.org", "username": "******"}
client = ParamikoSSHClient(**conn_params)
mock_client = Mock()
mock_sftp_client = Mock()
mock_transport = Mock()
mock_client.get_transport.return_value = mock_transport
mock_sftp_client.getcwd.return_value = "/mock/cwd"
client.client = mock_client
client.sftp_client = mock_sftp_client
mock_fo = StringIO("mock stream data 2")
result = client.putfo(path="path2.txt", fo=mock_fo, chmod=466)
self.assertEqual(result, "/mock/cwd/path2.txt")
calls = [call(".")]
mock_sftp_client.chdir.assert_has_calls(calls, any_order=False)
mock_sftp_client.putfo.assert_called_once_with(mock_fo, "path2.txt")
calls = [call("path2.txt"), call().chmod(466), call().close()]
mock_sftp_client.file.assert_has_calls(calls, any_order=False)
def test_putfo_relative_path(self):
conn_params = {'hostname': 'dummy.host.org', 'username': '******'}
client = ParamikoSSHClient(**conn_params)
mock_client = Mock()
mock_sftp_client = Mock()
mock_transport = Mock()
mock_client.get_transport.return_value = mock_transport
mock_sftp_client.getcwd.return_value = '/mock/cwd'
client.client = mock_client
client.sftp_client = mock_sftp_client
mock_fo = StringIO('mock stream data 2')
result = client.putfo(path='path2.txt', fo=mock_fo, chmod=466)
self.assertEqual(result, '/mock/cwd/path2.txt')
calls = [call('.')]
mock_sftp_client.chdir.assert_has_calls(calls, any_order=False)
mock_sftp_client.putfo.assert_called_once_with(mock_fo, "path2.txt")
calls = [call('path2.txt'), call().chmod(466), call().close()]
mock_sftp_client.file.assert_has_calls(calls, any_order=False)