Index: content/test/gpu/gpu_tests/cloud_storage_test_base.py |
diff --git a/content/test/gpu/gpu_tests/cloud_storage_test_base.py b/content/test/gpu/gpu_tests/cloud_storage_test_base.py |
index 2a7438c0733b3e83460acefd1e977d32457b9f84..f22ce3fb17878fb4d04389fb11c3482e364f98f1 100644 |
--- a/content/test/gpu/gpu_tests/cloud_storage_test_base.py |
+++ b/content/test/gpu/gpu_tests/cloud_storage_test_base.py |
@@ -143,7 +143,7 @@ class ValidatorBase(page_test.PageTest): |
# PNG to disk, following the pattern in bitmap_unittest.py. The key to |
# avoiding PermissionErrors seems to be to not actually try to write to |
# the temporary file object, but to re-open its name for all operations. |
- temp_file = tempfile.NamedTemporaryFile().name |
+ temp_file = tempfile.NamedTemporaryFile(suffix='.png').name |
image_util.WritePngFile(bitmap, temp_file) |
cloud_storage.Insert(bucket, name, temp_file, publicly_readable=public) |
@@ -169,7 +169,7 @@ class ValidatorBase(page_test.PageTest): |
# temporary file and gsutil's overwriting it. |
if not self.options.refimg_cloud_storage_bucket: |
raise Exception('--refimg-cloud-storage-bucket argument is required') |
- temp_file = tempfile.NamedTemporaryFile().name |
+ temp_file = tempfile.NamedTemporaryFile(suffix='.png').name |
cloud_storage.Get(self.options.refimg_cloud_storage_bucket, |
self._FormatReferenceImageName(img_name, page, tab), |
temp_file) |