add md5 test

Author: blink1073

flask_pymongo/__init__.py

@@ -27,6 +27,7 @@
 __all__ = ("PyMongo", "ASCENDING", "DESCENDING", "BSONObjectIdConverter", "BSONProvider")
 
 import hashlib
+import warnings
 from mimetypes import guess_type
 from typing import Any
 
@@ -191,7 +192,9 @@ def get_upload(filename):
         try:
             etag = fileobj.sha1
         except AttributeError:
-            etag = fileobj.md5
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore")
+                etag = fileobj.md5
             if etag is None:
                 pos = fileobj.tell()
                 raw_data = fileobj.read()

tests/test_gridfs.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from hashlib import sha1
+import warnings
+from hashlib import md5, sha1
 from io import BytesIO
 
 import pytest
@@ -95,6 +96,26 @@ def test_it_sets_supports_conditional_gets(self):
             resp = self.mongo.send_file("myfile.txt")
             assert resp.status_code == 304
 
+    def test_it_sets_supports_conditional_gets_md5(self):
+        # a basic conditional GET
+        md5_hash = md5(self.myfile.getvalue()).hexdigest()
+        environ_args = {
+            "method": "GET",
+            "headers": {
+                "If-None-Match": md5_hash,
+            },
+        }
+        storage = storage = GridFS(self.mongo.db)
+        with storage.new_file(filename="myfile.txt") as grid_file:
+            grid_file.write(self.myfile.getvalue())
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore")
+                grid_file.set("md5", md5_hash)
+
+        with self.app.test_request_context(**environ_args):
+            resp = self.mongo.send_file("myfile.txt")
+            assert resp.status_code == 304
+
     def test_it_sets_cache_headers(self):
         resp = self.mongo.send_file("myfile.txt", cache_for=60)
         assert resp.cache_control.max_age == 60