Apply suggestions from code review

ddelange · web-flow · commit b43e93e38e0c · 2024-07-19T08:27:29.000+02:00
diff --git a/tests/system/test_fileio.py b/tests/system/test_fileio.py
@@ -16,6 +16,7 @@
 
 import pytest
 
+from google.cloud.storage.fileio import CHUNK_SIZE_MULTIPLE
 from .test_blob import _check_blob_hash
 
 
@@ -98,9 +99,9 @@ def test_blobwriter_exit(
 
     # unhandled exceptions should cancel the upload
     with pytest.raises(ValueError, match="SIGTERM received"):
-        with blob.open("wb") as writer:
+        with blob.open("wb", chunksize=CHUNK_SIZE_MULTIPLE) as writer:
             writer.write(b"first chunk")  # not yet uploaded
-            writer.write(b"big chunk" * 1024 ** 8)  # uploaded
+            writer.write(bytes(CHUNK_SIZE_MULTIPLE))  # uploaded
             raise ValueError("SIGTERM received")  # upload is cancelled in __exit__
     # blob should not exist
     assert not blob.exists()
