Flatten annots/transparencies/linearize/rasterize: Fix async test parity

Assisted-by: Codex

Author: datalogics-cgreen

tests/test_flatten_annotations.py

@@ -279,3 +279,30 @@ def test_flatten_annotations_validation(monkeypatch: pytest.MonkeyPatch) -> None
         ),
     ):
         client.flatten_annotations([pdf_file, make_pdf_file(PdfRestFileID.generate())])
+
+
+@pytest.mark.asyncio
+async def test_async_flatten_annotations_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.flatten_annotations(png_file)
+
+        with pytest.raises(
+            ValidationError, match="List should have at most 1 item after validation"
+        ):
+            await client.flatten_annotations(
+                [pdf_file, make_pdf_file(PdfRestFileID.generate())]
+            )

tests/test_flatten_transparencies.py

@@ -295,3 +295,38 @@ def test_flatten_transparencies_validation(monkeypatch: pytest.MonkeyPatch) -> N
         ),
     ):
         client.flatten_transparencies(pdf_file, quality="ultra")  # type: ignore[arg-type]
+
+
+@pytest.mark.asyncio
+async def test_async_flatten_transparencies_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.flatten_transparencies(png_file)
+
+        with pytest.raises(
+            ValidationError, match="List should have at most 1 item after validation"
+        ):
+            await client.flatten_transparencies(
+                [pdf_file, make_pdf_file(PdfRestFileID.generate())]
+            )
+
+        with pytest.raises(
+            ValidationError, match="Input should be 'low', 'medium' or 'high'"
+        ):
+            await client.flatten_transparencies(
+                pdf_file,
+                quality="ultra",  # type: ignore[arg-type]
+            )

tests/test_linearize_pdf.py

@@ -280,3 +280,45 @@ def test_linearize_pdf_validation(
         pytest.raises(ValidationError, match=match),
     ):
         client.linearize_pdf(files_argument)
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    ("files", "match"),
+    [
+        pytest.param(
+            "png",
+            "Must be a PDF file",
+            id="non-pdf-file",
+        ),
+        pytest.param(
+            "multiple",
+            "List should have at most 1 item after validation",
+            id="multiple-files",
+        ),
+    ],
+)
+async def test_async_linearize_pdf_validation(
+    monkeypatch: pytest.MonkeyPatch,
+    files: str,
+    match: str,
+) -> 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))
+    files_argument = (
+        png_file
+        if files == "png"
+        else [pdf_file, make_pdf_file(PdfRestFileID.generate())]
+    )
+
+    async with AsyncPdfRestClient(api_key=ASYNC_API_KEY, transport=transport) as client:
+        with pytest.raises(ValidationError, match=match):
+            await client.linearize_pdf(files_argument)

tests/test_rasterize_pdf.py

@@ -263,3 +263,35 @@ def test_rasterize_pdf_validation(monkeypatch: pytest.MonkeyPatch) -> None:
         ),
     ):
         client.rasterize_pdf([pdf_file, make_pdf_file(PdfRestFileID.generate())])
+
+
+@pytest.mark.asyncio
+async def test_async_rasterize_pdf_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.rasterize_pdf(png_file)
+
+        with pytest.raises(
+            ValidationError, match="List should have at most 1 item after validation"
+        ):
+            await client.rasterize_pdf(
+                [pdf_file, make_pdf_file(PdfRestFileID.generate())]
+            )
+
+        with pytest.raises(ValidationError, match="Timeout must be greater than 0"):
+            await client.rasterize_pdf(
+                make_pdf_file(PdfRestFileID.generate(1)),
+                output="output",
+                timeout=0,
+            )