add even more coverage idk

Author: sebastianMindee

mindee/parsing/v2/error_response.py

@@ -1,11 +1,17 @@
-class ErrorResponse(RuntimeError):
+from mindee.parsing.common.string_dict import StringDict
+
+
+class ErrorResponse:
     """Error response info."""
 
     detail: str
     """Detail relevant to the error."""
-
     status: int
     """Http error code."""
 
+    def __init__(self, raw_response: StringDict):
+        self.detail = raw_response["detail"]
+        self.status = raw_response["status"]
+
     def __str__(self):
         return f"HTTP Status: {self.status} - {self.detail}"

mindee/parsing/v2/job.py

@@ -1,3 +1,6 @@
+from datetime import datetime
+from typing import List, Optional
+
 from mindee.parsing.common.string_dict import StringDict
 from mindee.parsing.v2.error_response import ErrorResponse
 
@@ -7,7 +10,7 @@ class Job:
 
     id: str
     """Job ID."""
-    error: ErrorResponse
+    error: Optional[ErrorResponse]
     """Error response if any."""
     model_id: str
     """ID of the model."""
@@ -17,11 +20,25 @@ class Job:
     """Optional alias for the file."""
     status: str
     """Status of the job."""
+    polling_url: str
+    """URL to poll for the job status."""
+    result_url: Optional[str]
+    """URL to poll for the job result, redirects to the result if available."""
+    webhooks: List[str]
+    """ID of webhooks associated with the job."""
 
     def __init__(self, raw_response: StringDict) -> None:
         self.id = raw_response["id"]
         self.status = raw_response["status"]
-        self.error = ErrorResponse(raw_response["error"])
+        self.error = (
+            ErrorResponse(raw_response["error"]) if raw_response["error"] else None
+        )
+        self.created_at = datetime.fromisoformat(
+            raw_response["created_at"].replace("Z", "+00:00")
+        )
         self.model_id = raw_response["model_id"]
+        self.polling_url = raw_response["polling_url"]
         self.filename = raw_response["filename"]
+        self.result_url = raw_response["result_url"]
         self.alias = raw_response["alias"]
+        self.webhooks = raw_response["webhooks"]

mindee/parsing/v2/webhook.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from typing import Optional
 
 from mindee.parsing.common.string_dict import StringDict
 from mindee.parsing.v2.error_response import ErrorResponse
@@ -9,7 +10,7 @@ class Webhook:
 
     id: str
     """ID of the webhook."""
-    error: ErrorResponse
+    error: Optional[ErrorResponse]
     """Error response if any."""
     created_at: datetime
     """Date and time the webhook was sent at."""
@@ -18,7 +19,9 @@ class Webhook:
 
     def __init__(self, raw_response: StringDict) -> None:
         self.id = raw_response["id"]
-        self.error = ErrorResponse(raw_response["error"])
+        self.error = (
+            ErrorResponse(raw_response["error"]) if raw_response["error"] else None
+        )
         self.created_at = self.parse_date(raw_response["created_at"])
         self.status = raw_response["status"]
 

tests/test_client_v2.py

@@ -1,10 +1,13 @@
+import json
+
 import pytest
 
 from mindee import ClientV2, InferencePredictOptions, LocalResponse
 from mindee.error.mindee_error import MindeeApiV2Error
 from mindee.error.mindee_http_error_v2 import MindeeHTTPErrorV2
 from mindee.input import LocalInputSource, PathInput
 from mindee.mindee_http.base_settings import USER_AGENT
+from mindee.parsing.v2 import Job, PollingResponse
 from tests.test_inputs import FILE_TYPES_DIR, V2_DATA_DIR
 from tests.utils import dummy_envvars
 
@@ -17,28 +20,58 @@ def env_client(monkeypatch) -> ClientV2:
 
 @pytest.fixture
 def custom_base_url_client(monkeypatch) -> ClientV2:
-    class _FakeResp:
+    class _FakePostResp:
         status_code = 400  # any non-2xx will do
         ok = False
 
         def json(self):
             # Shape must match what handle_error_v2 expects
             return {"status": -1, "detail": "forced failure from test"}
 
+    class _FakeGetResp:
+        status_code = 200
+        ok = True
+
+        def json(self):
+            return {
+                "job": {
+                    "id": "12345678-1234-1234-1234-123456789ABC",
+                    "model_id": "87654321-4321-4321-4321-CBA987654321",
+                    "filename": "default_sample.jpg",
+                    "alias": "dummy-alias.jpg",
+                    "created_at": "2025-07-03T14:27:58.974451",
+                    "status": "Processing",
+                    "polling_url": "https://api-v2.mindee.net/v2/jobs/12345678-1234-1234-1234-123456789ABC",
+                    "result_url": None,
+                    "webhooks": [],
+                    "error": None,
+                }
+            }
+
+        @property
+        def content(self) -> bytes:
+            """
+            Raw (bytes) payload, mimicking `requests.Response.content`.
+            """
+            return json.dumps(self.json()).encode("utf-8")
+
     monkeypatch.setenv("MINDEE_V2_BASE_URL", "https://dummy-url")
 
-    def _fake_response(*args, **kwargs):
-        return _FakeResp()
+    def _fake_post_error(*args, **kwargs):
+        return _FakePostResp()
+
+    def _fake_get_error(*args, **kwargs):
+        return _FakeGetResp()
 
     monkeypatch.setattr(
         "mindee.mindee_http.mindee_api_v2.requests.post",
-        _fake_response,
+        _fake_post_error,
         raising=True,
     )
 
     monkeypatch.setattr(
         "mindee.mindee_http.mindee_api_v2.requests.get",
-        _fake_response,
+        _fake_get_error,
         raising=True,
     )
 
@@ -78,12 +111,6 @@ def test_enqueue_and_parse_path_with_env_token(custom_base_url_client):
         )
 
 
-@pytest.mark.v2
-def test_parse_queued6_and_parse_path_with_env_token(custom_base_url_client):
-    with pytest.raises(MindeeHTTPErrorV2):
-        custom_base_url_client.parse_queued("dummy-queue")
-
-
 @pytest.mark.v2
 def test_loads_from_prediction(env_client):
     input_inference = LocalResponse(
@@ -104,3 +131,24 @@ def test_error_handling(custom_base_url_client):
         )
         assert e.status_code == -1
         assert e.detail == "forced failure from test"
+
+
+def test_enqueue(custom_base_url_client):
+    response = custom_base_url_client.parse_queued(
+        "12345678-1234-1234-1234-123456789ABC"
+    )
+    assert isinstance(response, PollingResponse)
+    assert isinstance(response.job, Job)
+    assert response.job.id == "12345678-1234-1234-1234-123456789ABC"
+    assert response.job.model_id == "87654321-4321-4321-4321-CBA987654321"
+    assert response.job.filename == "default_sample.jpg"
+    assert response.job.alias == "dummy-alias.jpg"
+    assert str(response.job.created_at) == "2025-07-03 14:27:58.974451"
+    assert response.job.status == "Processing"
+    assert (
+        response.job.polling_url
+        == "https://api-v2.mindee.net/v2/jobs/12345678-1234-1234-1234-123456789ABC"
+    )
+    assert not response.job.result_url
+    assert len(response.job.webhooks) == 0
+    assert not response.job.error