test: simplify unit tests for starlette >= 0.24

AntaresSimulatorTeam · Sep 8, 2023 · 9bf5c8f · 9bf5c8f
1 parent 6e7f881
commit 9bf5c8f
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/tests/matrixstore/test_service.py b/tests/matrixstore/test_service.py
@@ -348,14 +348,10 @@ def test_dataset_lifecycle():
 
 
 def _create_upload_file(filename: str, file: t.IO = None, content_type: str = "") -> UploadFile:
-    if hasattr(UploadFile, "content_type"):
-        # `content_type` attribute was replace by a read-ony property in starlette-v0.24.
-        headers = Headers(headers={"content-type": content_type})
-        # noinspection PyTypeChecker,PyArgumentList
-        return UploadFile(filename=filename, file=file, headers=headers)
-    else:
-        # noinspection PyTypeChecker,PyArgumentList
-        return UploadFile(filename=filename, file=file, content_type=content_type)
+    # `content_type` attribute was replace by a read-ony property in starlette-v0.24.
+    headers = Headers(headers={"content-type": content_type})
+    # noinspection PyTypeChecker,PyArgumentList
+    return UploadFile(filename=filename, file=file, headers=headers)
 
 
 def test_import():