From b3f90c7d2d7c2e618fd685dacf918751ce2ebc87 Mon Sep 17 00:00:00 2001 From: Jakub Jelen Date: Tue, 19 Nov 2024 11:30:49 +0100 Subject: [PATCH] tests: Adjust the SFTP tests to new chunk size Signed-off-by: Jakub Jelen --- tests/unit/sftp_test.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/unit/sftp_test.py b/tests/unit/sftp_test.py index 1822522d..7a179e10 100644 --- a/tests/unit/sftp_test.py +++ b/tests/unit/sftp_test.py @@ -97,17 +97,19 @@ def test_put_existing(dst_exists_path, src_path, sftp_session, transmit_payload) @pytest.fixture def large_payload(): - """Generate a large 1025 byte (1024 + 1B) test payload.""" - payload_len = 1024 + 1 - random_bytes = [ord(random.choice(string.printable)) for _ in range(payload_len)] - return bytes(random_bytes) + """Generate a large 32769 byte (32kB + 1B) test payload.""" + random_char_kilobyte = [ord(random.choice(string.printable)) for _ in range(1024)] + full_bytes_number = 32 + a_32kB_chunk = bytes(random_char_kilobyte * full_bytes_number) + the_last_byte = random.choice(random_char_kilobyte).to_bytes(length=1, byteorder='big') + return a_32kB_chunk + the_last_byte @pytest.fixture def src_path_large(tmp_path, large_payload): - """Return a remote path to a 1025 byte-sized file. + """Return a remote path to a 32769 byte-sized file. - The pylibssh chunk size is 1024 so the test needs a file that would + The pylibssh chunk size is 32769 so the test needs a file that would execute at least two loops. """ path = tmp_path / 'large.txt'