tests: add async validation parity for add-to-pdf endpoints

datalogics-cgreen · datalogics-cgreen · commit 935272e5afb6 · 2026-02-11T16:04:11.000-06:00
Assisted-by: Codex
diff --git a/tests/test_add_attachment_to_pdf.py b/tests/test_add_attachment_to_pdf.py
@@ -341,3 +341,66 @@ def handler(_: httpx.Request) -> httpx.Response:
                 make_attachment_file(str(PdfRestFileID.generate(2)), "more.txt"),
             ],
         )
+
+
+@pytest.mark.asyncio
+async def test_async_add_attachment_to_pdf_requires_pdf_file(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+    not_pdf = make_attachment_file(
+        str(PdfRestFileID.generate()),
+        "image.png",
+        "image/png",
+    )
+    attachment = make_attachment_file(str(PdfRestFileID.generate()), "note.txt")
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="Must be a PDF file"):
+            await client.add_attachment_to_pdf(not_pdf, attachment=attachment)
+
+
+@pytest.mark.asyncio
+async def test_async_add_attachment_to_pdf_rejects_multiple_input_files(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="at most 1 item"):
+            await client.add_attachment_to_pdf(
+                [
+                    make_pdf_file(PdfRestFileID.generate(1)),
+                    make_pdf_file(PdfRestFileID.generate(2)),
+                ],
+                attachment=make_attachment_file(str(PdfRestFileID.generate())),
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_attachment_to_pdf_rejects_multiple_attachments(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="at most 1 item"):
+            await client.add_attachment_to_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                attachment=[
+                    make_attachment_file(str(PdfRestFileID.generate(2))),
+                    make_attachment_file(str(PdfRestFileID.generate(2)), "more.txt"),
+                ],
+            )
diff --git a/tests/test_add_image_to_pdf.py b/tests/test_add_image_to_pdf.py
@@ -401,3 +401,93 @@ def handler(_: httpx.Request) -> httpx.Response:
                 y=1,
                 page=1,
             )
+
+
+@pytest.mark.asyncio
+async def test_async_add_image_to_pdf_invalid_pdf_type(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="Must be a PDF file"):
+            await client.add_image_to_pdf(
+                make_image_file(PdfRestFileID.generate(1)),
+                image=make_image_file(PdfRestFileID.generate(2)),
+                x=1,
+                y=2,
+                page=1,
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_image_to_pdf_page_minimum(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="greater than or equal to 1"):
+            await client.add_image_to_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                image=make_image_file(PdfRestFileID.generate(2)),
+                x=0,
+                y=0,
+                page=0,
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_image_to_pdf_rejects_multiple_input_files(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="at most 1 item"):
+            await client.add_image_to_pdf(
+                [
+                    make_pdf_file(PdfRestFileID.generate(1)),
+                    make_pdf_file(PdfRestFileID.generate(2)),
+                ],
+                image=make_image_file(PdfRestFileID.generate(2)),
+                x=1,
+                y=1,
+                page=1,
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_image_to_pdf_rejects_multiple_images(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="at most 1 item"):
+            await client.add_image_to_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                image=[
+                    make_image_file(PdfRestFileID.generate(2)),
+                    make_image_file(PdfRestFileID.generate(2), name="secondary.png"),
+                ],
+                x=1,
+                y=1,
+                page=1,
+            )
diff --git a/tests/test_add_text_to_pdf.py b/tests/test_add_text_to_pdf.py
@@ -410,3 +410,89 @@ def handler(_: httpx.Request) -> httpx.Response:
                     )
                 ],
             )
+
+
+@pytest.mark.asyncio
+async def test_async_add_text_to_pdf_requires_color(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    text_object = make_text_object()
+    text_object.pop("text_color_rgb")
+
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(
+            ValidationError,
+            match=re.escape(
+                "Either text_color_rgb or text_color_cmyk must be provided."
+            ),
+        ):
+            await client.add_text_to_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                text_objects=[text_object],
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_text_to_pdf_rgb_bounds(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(
+            ValidationError,
+            match=re.escape("text_color_rgb values must be between 0 and 255."),
+        ):
+            await client.add_text_to_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                text_objects=[make_text_object(text_color_rgb=(0, 0, 256))],
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_text_to_pdf_text_size_bounds(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="greater than or equal to 5"):
+            await client.add_text_to_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                text_objects=[make_text_object(text_size=4)],
+            )
+
+
+@pytest.mark.asyncio
+async def test_async_add_text_to_pdf_rejects_multiple_input_files(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    monkeypatch.delenv("PDFREST_API_KEY", raising=False)
+
+    def handler(_: httpx.Request) -> httpx.Response:
+        pytest.fail("Request should not be sent when validation fails.")
+
+    transport = httpx.MockTransport(handler)
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match="at most 1 item"):
+            await client.add_text_to_pdf(
+                [
+                    make_pdf_file(PdfRestFileID.generate(1)),
+                    make_pdf_file(PdfRestFileID.generate(2)),
+                ],
+                text_objects=[make_text_object()],
+            )
