Add tests of Add Image tool

Assisted-by: Codex

Author: datalogics-cgreen

tests/graphics_test_helpers.py

@@ -40,6 +40,26 @@ def make_pdf_file(file_id: str, name: str = "example.pdf") -> PdfRestFile:
     )
 
 
+def make_image_file(
+    file_id: str,
+    mime_type: str = "image/png",
+    name: str = "example.png",
+) -> PdfRestFile:
+    return PdfRestFile.model_validate(
+        {
+            "id": file_id,
+            "name": name,
+            "url": f"https://api.pdfrest.com/resource/{file_id}",
+            "type": mime_type,
+            "size": 2048,
+            "modified": datetime(2024, 1, 1, tzinfo=timezone.utc)
+            .isoformat()
+            .replace("+00:00", "Z"),
+            "scheduledDeletionTimeUtc": None,
+        }
+    )
+
+
 def assert_conversion_payload(
     payload: dict[str, Any],
     expected: dict[str, Any],

tests/live/test_live_add_image_to_pdf.py

@@ -0,0 +1,110 @@
+from __future__ import annotations
+
+import pytest
+
+from pdfrest import AsyncPdfRestClient, PdfRestApiError, PdfRestClient
+from pdfrest.models import PdfRestFile
+
+from ..resources import get_test_resource_path
+
+
+@pytest.fixture(scope="module")
+def uploaded_pdf_for_image_addition(
+    pdfrest_api_key: str,
+    pdfrest_live_base_url: str,
+) -> PdfRestFile:
+    resource = get_test_resource_path("report.pdf")
+    with PdfRestClient(
+        api_key=pdfrest_api_key,
+        base_url=pdfrest_live_base_url,
+    ) as client:
+        return client.files.create_from_paths([resource])[0]
+
+
+@pytest.fixture(scope="module")
+def uploaded_image(
+    pdfrest_api_key: str,
+    pdfrest_live_base_url: str,
+) -> PdfRestFile:
+    resource = get_test_resource_path("logo.png")
+    with PdfRestClient(
+        api_key=pdfrest_api_key,
+        base_url=pdfrest_live_base_url,
+    ) as client:
+        return client.files.create_from_paths([resource])[0]
+
+
+def test_live_add_image_to_pdf(
+    pdfrest_api_key: str,
+    pdfrest_live_base_url: str,
+    uploaded_pdf_for_image_addition: PdfRestFile,
+    uploaded_image: PdfRestFile,
+) -> None:
+    with PdfRestClient(
+        api_key=pdfrest_api_key,
+        base_url=pdfrest_live_base_url,
+    ) as client:
+        response = client.add_image_to_pdf(
+            uploaded_pdf_for_image_addition,
+            image=uploaded_image,
+            x=25,
+            y=50,
+            page=1,
+            output="live-added-image",
+        )
+
+    assert response.output_files
+    output_file = response.output_file
+    assert output_file.type == "application/pdf"
+    assert output_file.name.startswith("live-added-image")
+    assert uploaded_pdf_for_image_addition.id in response.input_ids
+    assert uploaded_image.id in response.input_ids
+
+
+@pytest.mark.asyncio
+async def test_live_async_add_image_to_pdf(
+    pdfrest_api_key: str,
+    pdfrest_live_base_url: str,
+    uploaded_pdf_for_image_addition: PdfRestFile,
+    uploaded_image: PdfRestFile,
+) -> None:
+    async with AsyncPdfRestClient(
+        api_key=pdfrest_api_key,
+        base_url=pdfrest_live_base_url,
+    ) as client:
+        response = await client.add_image_to_pdf(
+            uploaded_pdf_for_image_addition,
+            image=uploaded_image,
+            x=75,
+            y=125,
+            page=1,
+        )
+
+    assert response.output_files
+    output_file = response.output_file
+    assert output_file.type == "application/pdf"
+    assert uploaded_pdf_for_image_addition.id in response.input_ids
+    assert uploaded_image.id in response.input_ids
+
+
+def test_live_add_image_to_pdf_invalid_page(
+    pdfrest_api_key: str,
+    pdfrest_live_base_url: str,
+    uploaded_pdf_for_image_addition: PdfRestFile,
+    uploaded_image: PdfRestFile,
+) -> None:
+    with (
+        PdfRestClient(
+            api_key=pdfrest_api_key,
+            base_url=pdfrest_live_base_url,
+        ) as client,
+        pytest.raises(PdfRestApiError, match=r"(?i)page"),
+    ):
+        client.add_image_to_pdf(
+            uploaded_pdf_for_image_addition,
+            image=uploaded_image,
+            x=0,
+            y=0,
+            page=1,
+            extra_body={"page": 0},
+        )