From 79af9d78291c23ca74ace885f3cf60bc30dabf6c Mon Sep 17 00:00:00 2001 From: Afroz Alam Date: Wed, 18 Sep 2024 15:41:14 -0700 Subject: [PATCH] add suggested test --- tests/integ/test_multithreading.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/integ/test_multithreading.py b/tests/integ/test_multithreading.py index 10fcc6ef70..b704fe5e81 100644 --- a/tests/integ/test_multithreading.py +++ b/tests/integ/test_multithreading.py @@ -189,3 +189,9 @@ def put_and_get_file(upload_file_path, download_dir): with ThreadPoolExecutor(max_workers=10) as executor: for file_path in resources_files: executor.submit(put_and_get_file, file_path, download_dir) + + if not use_stream: + # assert all files are downloaded + assert set(os.listdir(download_dir)) == { + os.path.basename(file_path) for file_path in resources_files + }