it do be coveraging

Author: sebastianMindee

tests/test_client_v2.py

@@ -3,7 +3,8 @@
 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
+from mindee.input import LocalInputSource, PathInput
+from mindee.mindee_http.base_settings import USER_AGENT
 from tests.test_inputs import FILE_TYPES_DIR, V2_DATA_DIR
 from tests.utils import dummy_envvars
 
@@ -22,9 +23,7 @@ class _FakeResp:
 
         def json(self):
             # Shape must match what handle_error_v2 expects
-            return {
-                "error": {"status": -1, "detail": "forced failure from test"},
-            }
+            return {"status": -1, "detail": "forced failure from test"}
 
     monkeypatch.setenv("MINDEE_V2_BASE_URL", "https://dummy-url")
 
@@ -57,6 +56,8 @@ def test_enqueue_path_with_env_token(custom_base_url_client):
     assert custom_base_url_client.mindee_api.base_url == "https://dummy-url"
     assert custom_base_url_client.mindee_api.url_root == "https://dummy-url"
     assert custom_base_url_client.mindee_api.api_key == "dummy"
+    assert custom_base_url_client.mindee_api.base_headers["Authorization"] == "dummy"
+    assert custom_base_url_client.mindee_api.base_headers["User-Agent"] == USER_AGENT
     input_doc: LocalInputSource = custom_base_url_client.source_from_path(
         f"{FILE_TYPES_DIR}/receipt.jpg"
     )
@@ -90,3 +91,16 @@ def test_loads_from_prediction(env_client):
     )
     prediction = env_client.load_inference(input_inference)
     assert prediction.inference.model.id == "12345678-1234-1234-1234-123456789abc"
+
+
+@pytest.mark.v2
+def test_error_handling(custom_base_url_client):
+    with pytest.raises(MindeeHTTPErrorV2) as e:
+        custom_base_url_client.enqueue(
+            PathInput(
+                V2_DATA_DIR / "products" / "financial_document" / "default_sample.jpg"
+            ),
+            InferencePredictOptions("dummy-model"),
+        )
+        assert e.status_code == -1
+        assert e.detail == "forced failure from test"

tests/v2/test_inference_response.py

@@ -90,7 +90,9 @@ def inference_result_json() -> StringDict:
                         }
                     },
                 },
-                "options": None,
+                "options": {
+                    "raw_text": ["toto", "tata", "titi"],
+                },
             },
         }
     }
@@ -164,6 +166,10 @@ def test_inference_response(inference_result_json):
         == "value_9"
     )
 
+    assert inference_result.inference.result.options
+    assert len(inference_result.inference.result.options.raw_text) == 3
+    assert inference_result.inference.result.options.raw_text[0] == "toto"
+
 
 @pytest.mark.v2
 def test_full_inference_response():
@@ -188,4 +194,5 @@ def test_full_inference_response():
 
     assert isinstance(load_response.inference.file, InferenceFile)
     assert load_response.inference.file.name == "complete.jpg"
-    assert load_response.inference.file.alias == None
+    assert not load_response.inference.file.alias
+    assert not load_response.inference.result.options