fixed download

Author: NimishaShrivastava-dev

src/pytfe/resources/state_versions.py

@@ -272,12 +272,9 @@ def download(self, state_version_id: str) -> bytes:
             raise NotFound("download url not available for this state version")
 
         # Download the bytes from the signed Archivist URL (follow redirects).
-        # Avoid API default headers here; Accept */* is fine.
+        # Avoid JSON:API headers here; Accept */* is fine.
         resp = self.t.request(
-            "GET",
-            url,
-            allow_redirects=True,
-            headers={"Accept": "*/*"},
+            "GET", url, allow_redirects=True, headers={"Accept": "application/json"}
         )
         return resp.content
 
@@ -293,10 +290,7 @@ def download_current(self, workspace_id: str) -> bytes:
 
             raise NotFound("download url not available for current state")
         resp = self.t.request(
-            "GET",
-            url,
-            allow_redirects=True,
-            headers={"Accept": "*/*"},
+            "GET", url, allow_redirects=True, headers={"Accept": "*/*"}
         )
         return resp.content
 

tests/units/test_state_version.py

@@ -365,7 +365,7 @@ def test_download_state_version_success(
             "GET",
             "https://example.com/signed-download",
             allow_redirects=True,
-            headers={"Accept": "*/*"},
+            headers={"Accept": "application/json"},
         )
         assert result == b"{}"