From b2103cb653c89f6d211354358687fceb423712b8 Mon Sep 17 00:00:00 2001 From: Mehdi Ben Hamida Date: Tue, 15 Sep 2026 14:57:38 +0200 Subject: [PATCH 1/4] :test_tube: add webhook test to Python integration tests --- tests/v2/test_client_integration.py | 56 +++++++++++++++++++++++++++++ 1 file changed, 56 insertions(+) diff --git a/tests/v2/test_client_integration.py b/tests/v2/test_client_integration.py index 1d9cd18b..14f55fe4 100644 --- a/tests/v2/test_client_integration.py +++ b/tests/v2/test_client_integration.py @@ -1,5 +1,6 @@ import os from pathlib import Path +from datetime import datetime import httpx import pytest @@ -13,6 +14,8 @@ ) from mindee.v2.parsing import InferenceActiveOptions from mindee.v2.product.extraction.extraction_response import ExtractionResponse +from mindee.v2.product.split.params.split_parameters import SplitParameters +from mindee.v2.product.split.split_response import SplitResponse from tests.utils import FILE_TYPES_DIR, V2_PRODUCT_DATA_DIR @@ -166,6 +169,59 @@ def test_parse_file_filled_single_page_must_succeed( assert len(supplier_name.locations) == 0 +def _assert_webhook_job_success(response, webhook_ids: list[str]) -> None: + assert response.inference is not None + assert response.inference.result is not None + + assert response.job is not None + assert response.job.status == "Processed" + assert isinstance(response.job.completed_at, datetime) + assert response.job.error is None + assert len(response.job.webhooks) == 2 + assert all(webhook.status in {"Completed", "Failed"} for webhook in response.job.webhooks) + assert {webhook.id for webhook in response.job.webhooks} == set(webhook_ids) + + +@pytest.mark.integration +@pytest.mark.v2 +def test_extraction_with_two_webhooks_must_complete_and_succeed( + v2_client: Client, findoc_model_id: str +) -> None: + webhook_ids = [ + "-aa11-aa11-bdc5-2f8496c5641aa2286ed9", + "b2286ed9-aa11-aa11-bdc5-2f8496c5641a", + ] + + input_source = PathInput( + V2_PRODUCT_DATA_DIR / "extraction" / "financial_document" / "default_sample.jpg" + ) + params = ExtractionParameters(model_id=findoc_model_id, webhook_ids=webhook_ids) + + response = v2_client.enqueue_and_get_result(ExtractionResponse, input_source, params) + + _assert_webhook_job_success(response, webhook_ids) + assert response.inference.result.fields["supplier_name"].value == "John Smith" + + +@pytest.mark.integration +@pytest.mark.v2 +def test_split_with_two_webhooks_must_complete_and_succeed( + v2_client: Client, split_model_id: str +) -> None: + webhook_ids = [ + "a2286ed9-aa11-aa11-bdc5-2f8496c5641a", + "b2286ed9-aa11-aa11-bdc5-2f8496c5641a", + ] + + input_source = PathInput(V2_PRODUCT_DATA_DIR / "split" / "default_sample.pdf") + params = SplitParameters(model_id=split_model_id, webhook_ids=webhook_ids) + + response = v2_client.enqueue_and_get_result(SplitResponse, input_source, params) + + _assert_webhook_job_success(response, webhook_ids) + assert len(response.inference.result.splits) == 2 + + @pytest.mark.integration @pytest.mark.v2 def test_invalid_uuid_must_throw_error(v2_client: Client) -> None: From af117a7003861fbee4465ecd70763b524224f245 Mon Sep 17 00:00:00 2001 From: Mehdi Ben Hamida Date: Wed, 16 Sep 2026 10:38:24 +0200 Subject: [PATCH 2/4] :art: refactor code style in integration tests for readability --- tests/v2/test_client_integration.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/v2/test_client_integration.py b/tests/v2/test_client_integration.py index 14f55fe4..3bd24621 100644 --- a/tests/v2/test_client_integration.py +++ b/tests/v2/test_client_integration.py @@ -1,6 +1,6 @@ import os -from pathlib import Path from datetime import datetime +from pathlib import Path import httpx import pytest @@ -178,7 +178,9 @@ def _assert_webhook_job_success(response, webhook_ids: list[str]) -> None: assert isinstance(response.job.completed_at, datetime) assert response.job.error is None assert len(response.job.webhooks) == 2 - assert all(webhook.status in {"Completed", "Failed"} for webhook in response.job.webhooks) + assert all( + webhook.status in {"Completed", "Failed"} for webhook in response.job.webhooks + ) assert {webhook.id for webhook in response.job.webhooks} == set(webhook_ids) @@ -197,7 +199,9 @@ def test_extraction_with_two_webhooks_must_complete_and_succeed( ) params = ExtractionParameters(model_id=findoc_model_id, webhook_ids=webhook_ids) - response = v2_client.enqueue_and_get_result(ExtractionResponse, input_source, params) + response = v2_client.enqueue_and_get_result( + ExtractionResponse, input_source, params + ) _assert_webhook_job_success(response, webhook_ids) assert response.inference.result.fields["supplier_name"].value == "John Smith" From baa48bf74aae6df6f613ab57ec0eb19a5ed02689 Mon Sep 17 00:00:00 2001 From: Mehdi Ben Hamida Date: Wed, 16 Sep 2026 15:59:21 +0200 Subject: [PATCH 3/4] :test_tube: add split model ID fixture and update webhook test data --- tests/conftest.py | 6 ++++++ tests/v2/test_client_integration.py | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index b5cc6b4a..0a67507f 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -14,3 +14,9 @@ def force_gc(): def findoc_model_id() -> str: """Identifier of the Financial Document model, supplied through an env var.""" return os.getenv("MINDEE_V2_SE_TESTS_FINDOC_MODEL_ID", "") + + +@pytest.fixture(scope="session") +def split_model_id() -> str: + """Identifier of the Split model, supplied through an env var.""" + return os.getenv("MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID", "") \ No newline at end of file diff --git a/tests/v2/test_client_integration.py b/tests/v2/test_client_integration.py index 3bd24621..726b75f7 100644 --- a/tests/v2/test_client_integration.py +++ b/tests/v2/test_client_integration.py @@ -190,7 +190,7 @@ def test_extraction_with_two_webhooks_must_complete_and_succeed( v2_client: Client, findoc_model_id: str ) -> None: webhook_ids = [ - "-aa11-aa11-bdc5-2f8496c5641aa2286ed9", + "a2286ed9-aa11-aa11-bdc5-2f8496c5641a", "b2286ed9-aa11-aa11-bdc5-2f8496c5641a", ] From 142166ea3058021cf0d9ad53f2e8ca54293ef91e Mon Sep 17 00:00:00 2001 From: Mehdi Ben Hamida Date: Wed, 16 Sep 2026 16:06:25 +0200 Subject: [PATCH 4/4] :test_tube: add split model ID fixture and update webhook test data --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 0a67507f..c44074e4 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -19,4 +19,4 @@ def findoc_model_id() -> str: @pytest.fixture(scope="session") def split_model_id() -> str: """Identifier of the Split model, supplied through an env var.""" - return os.getenv("MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID", "") \ No newline at end of file + return os.getenv("MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID", "")