diff options
Diffstat (limited to 'tests/test_sftp_big.py')
| -rw-r--r-- | tests/test_sftp_big.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_sftp_big.py b/tests/test_sftp_big.py index 1cafe931..c1827624 100644 --- a/tests/test_sftp_big.py +++ b/tests/test_sftp_big.py @@ -376,8 +376,9 @@ class BigSFTPTest (unittest.TestCase): # try to read it too. f = sftp.open('%s/hongry.txt' % FOLDER, 'r', 128 * 1024) f.prefetch() - for i in xrange(32): - f.read(32 * 1024) + total = 0 + while total < 1024 * 1024: + total += len(f.read(32 * 1024)) f.close() finally: sftp.remove('%s/hongry.txt' % FOLDER) |