Fixed datalake tests

Author: weirongw23-msft

sdk/storage/azure-storage-file-datalake/tests/test_file.py

@@ -1759,7 +1759,9 @@ def test_download_file_no_decompress_chunks(self, **kwargs):
             self.account_url(datalake_storage_account_name, 'dfs'),
             self.file_system_name,
             file_name,
-            credential=datalake_storage_account_key
+            credential=datalake_storage_account_key,
+            max_chunk_get_size=4,
+            max_single_get_size=4,
         )
         file_client.create_file()
 
@@ -1774,17 +1776,7 @@ def test_download_file_no_decompress_chunks(self, **kwargs):
             content_settings=content_settings
         )
 
-        stream = file_client.download_file(decompress=False)
-        result = bytearray()
-        read_size = 4
-        num_chunks = int(ceil(len(compressed_data) / read_size))
-        for i in range(num_chunks):
-            content = stream.read(read_size)
-            start = i * read_size
-            end = start + read_size
-            assert compressed_data[start:end] == content
-            result.extend(content)
-
+        result = file_client.download_file(decompress=False).readall()
         assert result == compressed_data
 
 # ------------------------------------------------------------------------------

sdk/storage/azure-storage-file-datalake/tests/test_file_async.py

@@ -1663,7 +1663,9 @@ async def test_download_file_no_decompress_chunks(self, **kwargs):
             self.account_url(datalake_storage_account_name, 'dfs'),
             self.file_system_name,
             file_name,
-            credential=datalake_storage_account_key
+            credential=datalake_storage_account_key,
+            max_chunk_get_size=4,
+            max_single_get_size=4,
         )
         await file_client.create_file()
 
@@ -1678,17 +1680,7 @@ async def test_download_file_no_decompress_chunks(self, **kwargs):
             content_settings=content_settings
         )
 
-        stream = await file_client.download_file(decompress=False)
-        result = bytearray()
-        read_size = 4
-        num_chunks = int(ceil(len(compressed_data) / read_size))
-        for i in range(num_chunks):
-            content = await stream.read(read_size)
-            start = i * read_size
-            end = start + read_size
-            assert compressed_data[start:end] == content
-            result.extend(content)
-
+        result = await (await file_client.download_file(decompress=False)).readall()
         assert result == compressed_data
 
 # ------------------------------------------------------------------------------