add suggested test

snowflakedb · Sep 18, 2024 · 79af9d7 · 79af9d7
1 parent df3263c
commit 79af9d7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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
+            }