From 70696fe80e0f9131fbf5c1f529c60cbff9600151 Mon Sep 17 00:00:00 2001 From: Tim Stolarski Date: Wed, 28 Apr 2021 13:29:38 +0200 Subject: Add Test fpr get() without using prefetch --- tests/test_sftp.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'tests') 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. -- cgit v1.2.3