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)