diff options
| -rw-r--r-- | tests/test_sftp.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/tests/test_sftp.py b/tests/test_sftp.py index e4e18e5a..31a0eb2d 100644 --- a/tests/test_sftp.py +++ b/tests/test_sftp.py @@ -554,6 +554,32 @@ class TestSFTP(object): os.unlink(localname) sftp.unlink(sftp.FOLDER + "/bunny.txt") + def test_get_without_prefetch(self, sftp): + """ + Create a 4MB file. Verify that pull works without prefetching + using a lager file. + """ + + fd, localname = mkstemp() + os.close(fd) + + with open(localname, 'wb') as f: + num_chars = 1024 * 1024 * 4 + f.write(b'0' * num_chars) + + sftp.put(localname, sftp.FOLDER + "/dummy_file") + + os.unlink(localname) + fd, localname = mkstemp() + os.close(fd) + + sftp.get(sftp.FOLDER + "/dummy_file", localname, prefetch=False) + + assert os.stat(localname).st_size == 4194304 + + os.unlink(localname) + sftp.unlink(sftp.FOLDER + "/dummy_file") + def test_check(self, sftp): """ verify that file.check() works against our own server. |