Add missing unit tests

Resolve discrepancies in unit tests between sync and async

Assisted-by: Codex

Author: datalogics-cgreen

tests/test_convert_to_jpeg.py

@@ -244,8 +244,16 @@ def handler(_: httpx.Request) -> httpx.Response:
 
 
 @pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "color_model",
+    [
+        pytest.param("rgb", id="rgb"),
+        pytest.param("cmyk", id="cmyk"),
+        pytest.param("gray", id="gray"),
+    ],
+)
 async def test_async_convert_to_jpeg_success(
-    monkeypatch: pytest.MonkeyPatch,
+    monkeypatch: pytest.MonkeyPatch, color_model: str
 ) -> None:
     monkeypatch.delenv("PDFREST_API_KEY", raising=False)
     input_file = make_pdf_file(PdfRestFileID.generate(1))
@@ -257,7 +265,7 @@ async def test_async_convert_to_jpeg_success(
             "output_prefix": "async-output",
             "page_range": "1-2",
             "resolution": 500,
-            "color_model": "gray",
+            "color_model": color_model,
             "jpeg_quality": 85,
             "smoothing": ["all"],
         }
@@ -295,7 +303,7 @@ def handler(request: httpx.Request) -> httpx.Response:
             output_prefix="async-output",
             page_range="1-2",
             resolution=500,
-            color_model="gray",
+            color_model=color_model,
             smoothing=["all"],
             jpeg_quality=85,
         )

tests/test_convert_to_pdfa.py

@@ -90,11 +90,11 @@ def handler(request: httpx.Request) -> httpx.Response:
 @pytest.mark.parametrize(
     "output_type",
     [
-        pytest.param("PDF/A-1b", id="async-pdfa-1b"),
-        pytest.param("PDF/A-2b", id="async-pdfa-2b"),
-        pytest.param("PDF/A-2u", id="async-pdfa-2u"),
-        pytest.param("PDF/A-3b", id="async-pdfa-3b"),
-        pytest.param("PDF/A-3u", id="async-pdfa-3u"),
+        pytest.param("PDF/A-1b", id="pdfa-1b"),
+        pytest.param("PDF/A-2b", id="pdfa-2b"),
+        pytest.param("PDF/A-2u", id="pdfa-2u"),
+        pytest.param("PDF/A-3b", id="pdfa-3b"),
+        pytest.param("PDF/A-3u", id="pdfa-3u"),
     ],
 )
 async def test_async_convert_to_pdfa_success(

tests/test_summarize_pdf_text.py

@@ -245,6 +245,40 @@ def handler(request: httpx.Request) -> httpx.Response:
         assert timeout_value == pytest.approx(0.25)
 
 
+def test_summarize_text_success(monkeypatch: pytest.MonkeyPatch) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+    input_file = make_pdf_file(PdfRestFileID.generate(2))
+    payload_dump = SummarizePdfTextPayload.model_validate(
+        {"files": [input_file], "output_type": "json"}
+    ).model_dump(mode="json", by_alias=True, exclude_none=True, exclude_unset=True)
+
+    seen: dict[str, int] = {"post": 0}
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        if request.method == "POST" and request.url.path == "/summarized-pdf-text":
+            seen["post"] += 1
+            payload = json.loads(request.content.decode("utf-8"))
+            for key, value in payload_dump.items():
+                assert payload[key] == value
+            return httpx.Response(
+                200,
+                json={
+                    "summary": "Sync summary",
+                    "inputId": str(input_file.id),
+                },
+            )
+        msg = f"Unexpected request {request.method} {request.url}"
+        raise AssertionError(msg)
+
+    transport = httpx.MockTransport(handler)
+    with PdfRestClient(api_key=VALID_API_KEY, transport=transport) as client:
+        response = client.summarize_text(input_file)
+
+    assert seen == {"post": 1}
+    assert isinstance(response, SummarizePdfTextResponse)
+    assert response.summary == "Sync summary"
+
+
 @pytest.mark.asyncio
 async def test_async_summarize_text_success(
     monkeypatch: pytest.MonkeyPatch,

tests/test_translate_pdf_text.py

@@ -237,6 +237,46 @@ def handler(request: httpx.Request) -> httpx.Response:
         assert timeout_value == pytest.approx(0.3)
 
 
+def test_translate_pdf_text_success(monkeypatch: pytest.MonkeyPatch) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+    input_file = make_pdf_file(PdfRestFileID.generate(2))
+    payload_dump = TranslatePdfTextPayload.model_validate(
+        {"files": [input_file], "output_language": "de", "output_type": "json"}
+    ).model_dump(mode="json", by_alias=True, exclude_none=True, exclude_unset=True)
+
+    seen: dict[str, int] = {"post": 0}
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        if request.method == "POST" and request.url.path == "/translated-pdf-text":
+            seen["post"] += 1
+            payload = json.loads(request.content.decode("utf-8"))
+            for key, value in payload_dump.items():
+                assert payload[key] == value
+            return httpx.Response(
+                200,
+                json={
+                    "translated_text": "Hallo",
+                    "inputId": str(input_file.id),
+                    "source_languages": ["en"],
+                    "output_language": "de",
+                },
+            )
+        msg = f"Unexpected request {request.method} {request.url}"
+        raise AssertionError(msg)
+
+    transport = httpx.MockTransport(handler)
+    with PdfRestClient(api_key=VALID_API_KEY, transport=transport) as client:
+        response = client.translate_pdf_text(
+            input_file,
+            output_language="de",
+        )
+
+    assert seen == {"post": 1}
+    assert isinstance(response, TranslatePdfTextResponse)
+    assert response.translated_text == "Hallo"
+    assert response.source_languages == ["en"]
+
+
 @pytest.mark.asyncio
 async def test_async_translate_pdf_text_success(
     monkeypatch: pytest.MonkeyPatch,