Skip to content

Commit 3857258

Browse files
fix tests, again
1 parent 6f8c8f0 commit 3857258

1 file changed

Lines changed: 9 additions & 7 deletions

File tree

tests/v2/test_client_integration.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import os
2+
import re
23
from pathlib import Path
34

45
import httpx
@@ -9,8 +10,10 @@
910
from mindee.input.path_input import PathInput
1011
from mindee.input.url_input_source import URLInputSource
1112
from mindee.v2.client import Client
12-
from mindee.v2.error import MindeeAPIV2Error
13-
from mindee.v2.error.mindee_http_error_v2 import MindeeHTTPErrorV2
13+
from mindee.v2.error.mindee_http_error_v2 import (
14+
MindeeHTTPErrorV2,
15+
MindeeHTTPUnknownErrorV2,
16+
)
1417
from mindee.v2.parsing import InferenceActiveOptions
1518
from mindee.v2.product.extraction.extraction_response import ExtractionResponse
1619
from tests.utils import FILE_TYPES_DIR, V2_PRODUCT_DATA_DIR
@@ -355,16 +358,15 @@ def test_explicit_timeout_failure(findoc_model_id) -> None:
355358
@pytest.mark.v2
356359
@pytest.mark.integration
357360
@respx.mock
358-
def test_explicit_500_server_error(findoc_model_id) -> None:
359-
respx.post("https://api-v2.mindee.net/v2/inferences/enqueue").mock(
361+
def test_explicit_500_server_error(findoc_model_id: str) -> None:
362+
respx.post(re.compile(r"https://api-v2\.mindee\.net/v2/.+/enqueue")).mock(
360363
return_value=httpx.Response(500, json={"message": "Internal Server Error"})
361364
)
362365

363366
client = Client(api_key="dummy")
364367
input_source = PathInput(FILE_TYPES_DIR / "pdf" / "blank_1.pdf")
365368
params = ExtractionParameters(model_id=findoc_model_id)
366-
367-
with pytest.raises(MindeeAPIV2Error) as exc_info:
369+
with pytest.raises(MindeeHTTPUnknownErrorV2) as exc_info:
368370
client.enqueue(input_source, params)
369371

370-
assert exc_info.value.status_code == 500
372+
assert "Couldn't deserialize server error" in str(exc_info.value)

0 commit comments

Comments
 (0)