🧪 add webhook test to Python integration tests

Author: MehdiBenHamida

tests/v2/test_client_integration.py

@@ -14,6 +14,7 @@
 )
 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
 
@@ -170,40 +171,7 @@ def test_parse_file_filled_single_page_must_succeed(
 
 @pytest.mark.integration
 @pytest.mark.v2
-@pytest.mark.parametrize(
-    ("response_cls", "input_path", "model_id"),
-    [
-        (
-            ExtractionResponse,
-            V2_PRODUCT_DATA_DIR / "extraction" / "financial_document" / "default_sample.jpg",
-            "extraction",
-        ),
-        (
-            SplitResponse,
-            V2_PRODUCT_DATA_DIR / "split" / "default_sample.pdf",
-            "split",
-        ),
-    ],
-)
-def test_enqueue_with_two_webhooks_must_complete_and_succeed(
-    v2_client: Client,
-    findoc_model_id: str,
-    split_model_id: str,
-    response_cls,
-    input_path: Path,
-    model_id: str,
-) -> None:
-    webhook_ids = [
-        "a2286ed9-aa11-aa11-bdc5-2f8496c5641a",
-        "b2286ed9-aa11-aa11-bdc5-2f8496c5641a",
-    ]
-    resolved_model_id = findoc_model_id if model_id == "extraction" else split_model_id
-
-    input_source = PathInput(input_path)
-    params = ExtractionParameters(model_id=resolved_model_id, webhook_ids=webhook_ids)
-
-    response = v2_client.enqueue_and_get_result(response_cls, input_source, params)
-
+def _assert_webhook_job_success(response, webhook_ids: list[str]) -> None:
     assert response.inference is not None
     assert response.inference.result is not None
 
@@ -215,11 +183,45 @@ def test_enqueue_with_two_webhooks_must_complete_and_succeed(
     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)
 
-    assert response.inference.result is not None
-    if response_cls is ExtractionResponse:
-        assert response.inference.result.fields["supplier_name"].value == "John Smith"
-    else:
-        assert len(response.inference.result.splits) == 2
+
+@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