Markdown/powerpoint: Add missing async tests

Assisted-by: Codex

Author: datalogics-cgreen

tests/test_convert_to_markdown.py

@@ -66,6 +66,44 @@ def test_convert_to_markdown_payload_invalid_page_break_comments() -> None:
         )
 
 
+@pytest.mark.asyncio
+async def test_async_convert_to_markdown_payload_rejects_non_pdf() -> None:
+    file_id = str(PdfRestFileID.generate())
+    text_file = PdfRestFile.model_validate(
+        {
+            "id": file_id,
+            "name": "notes.txt",
+            "url": f"https://api.pdfrest.com/resource/{file_id}",
+            "type": "text/plain",
+            "size": 64,
+            "modified": "2024-01-01T00:00:00Z",
+            "scheduledDeletionTimeUtc": None,
+        }
+    )
+    with pytest.raises(ValidationError, match="Must be a PDF file"):
+        ConvertToMarkdownPayload.model_validate({"files": [text_file]})
+
+
+@pytest.mark.asyncio
+async def test_async_convert_to_markdown_payload_invalid_page_range() -> None:
+    file_repr = make_pdf_file(PdfRestFileID.generate(1))
+    with pytest.raises(
+        ValidationError, match="The start page must be less than or equal to the end"
+    ):
+        ConvertToMarkdownPayload.model_validate(
+            {"files": [file_repr], "pages": ["5-2"]}
+        )
+
+
+@pytest.mark.asyncio
+async def test_async_convert_to_markdown_payload_invalid_page_break_comments() -> None:
+    file_repr = make_pdf_file(PdfRestFileID.generate(1))
+    with pytest.raises(ValidationError, match="Input should be 'on' or 'off'"):
+        ConvertToMarkdownPayload.model_validate(
+            {"files": [file_repr], "page_break_comments": "maybe"}
+        )
+
+
 def test_convert_to_markdown_success(monkeypatch: pytest.MonkeyPatch) -> None:
     monkeypatch.delenv("PDFREST_API_KEY", raising=False)
     input_file = make_pdf_file(PdfRestFileID.generate(1))
@@ -198,6 +236,74 @@ def handler(request: httpx.Request) -> httpx.Response:
         assert get_timeout == pytest.approx(0.4)
 
 
+@pytest.mark.asyncio
+async def test_async_convert_to_markdown_request_customization(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+    input_file = make_pdf_file(PdfRestFileID.generate(2))
+    payload_dump = ConvertToMarkdownPayload.model_validate(
+        {
+            "files": [input_file],
+            "output_type": "file",
+            "page_break_comments": "off",
+        }
+    ).model_dump(mode="json", by_alias=True, exclude_none=True, exclude_unset=True)
+    output_id = str(PdfRestFileID.generate())
+    captured_timeout: dict[str, float | dict[str, float] | None] = {}
+
+    def handler(request: httpx.Request) -> httpx.Response:
+        if request.method == "POST" and request.url.path == "/markdown":
+            assert request.url.params["trace"] == "true"
+            assert request.headers["X-Debug"] == "async"
+            captured_timeout["value"] = request.extensions.get("timeout")
+            payload = json.loads(request.content.decode("utf-8"))
+            for key, value in payload_dump.items():
+                assert payload[key] == value
+            assert payload["debug"] is True
+            return httpx.Response(
+                200,
+                json={
+                    "inputId": [str(input_file.id)],
+                    "outputId": [output_id],
+                },
+            )
+        if request.method == "GET" and request.url.path == f"/resource/{output_id}":
+            assert request.url.params["format"] == "info"
+            assert request.url.params["trace"] == "true"
+            assert request.headers["X-Debug"] == "async"
+            return httpx.Response(
+                200,
+                json=_make_markdown_file(output_id, "debug-async.md").model_dump(
+                    mode="json", by_alias=True
+                ),
+            )
+        msg = f"Unexpected request {request.method} {request.url}"
+        raise AssertionError(msg)
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        response = await client.convert_to_markdown(
+            input_file,
+            extra_query={"trace": "true"},
+            extra_headers={"X-Debug": "async"},
+            extra_body={"debug": True},
+            timeout=0.4,
+            page_break_comments="off",
+        )
+
+    assert isinstance(response, PdfRestFileBasedResponse)
+    assert len(response.output_files) == 1
+    timeout_value = captured_timeout["value"]
+    assert timeout_value is not None
+    if isinstance(timeout_value, dict):
+        assert all(
+            component == pytest.approx(0.4) for component in timeout_value.values()
+        )
+    else:
+        assert timeout_value == pytest.approx(0.4)
+
+
 @pytest.mark.asyncio
 async def test_async_convert_to_markdown_success(
     monkeypatch: pytest.MonkeyPatch,

tests/test_convert_to_powerpoint.py

@@ -275,3 +275,30 @@ def test_convert_to_powerpoint_validation(monkeypatch: pytest.MonkeyPatch) -> No
         client.convert_to_powerpoint(
             [pdf_file, make_pdf_file(PdfRestFileID.generate())]
         )
+
+
+@pytest.mark.asyncio
+async def test_async_convert_to_powerpoint_validation(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+    pdf_file = make_pdf_file(PdfRestFileID.generate(1))
+    png_file = PdfRestFile.model_validate(
+        build_file_info_payload(
+            PdfRestFileID.generate(),
+            "example.png",
+            "image/png",
+        )
+    )
+    transport = httpx.MockTransport(lambda request: (_ for _ in ()).throw(RuntimeError))
+
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="Must be a PDF file"):
+            await client.convert_to_powerpoint(png_file)
+
+        with pytest.raises(
+            ValidationError, match="List should have at most 1 item after validation"
+        ):
+            await client.convert_to_powerpoint(
+                [pdf_file, make_pdf_file(PdfRestFileID.generate())]
+            )