Skip to content

Commit 0559593

Browse files
tests: Refactor and relocate live PNG conversion tests
- Moved `test_live_convert_to_png` and `test_live_async_convert_to_png` from `tests/test_convert_to_png.py` to a new file `tests/live/test_live_convert_to_png.py`. - Simplified `test_convert_to_png.py` by removing live test cases. Assisted-by: Codex
1 parent fa9e513 commit 0559593

2 files changed

Lines changed: 49 additions & 35 deletions

File tree

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
from __future__ import annotations
2+
3+
import pytest
4+
5+
from pdfrest import AsyncPdfRestClient, PdfRestClient
6+
from pdfrest.models import PdfRestFileBasedResponse
7+
8+
from ..resources import get_test_resource_path
9+
10+
11+
def test_live_convert_to_png(
12+
pdfrest_api_key: str,
13+
pdfrest_live_base_url: str,
14+
) -> None:
15+
resource = get_test_resource_path("report.pdf")
16+
with PdfRestClient(
17+
api_key=pdfrest_api_key,
18+
base_url=pdfrest_live_base_url,
19+
) as client:
20+
uploaded = client.files.create_from_paths([resource])[0]
21+
response = client.convert_to_png(
22+
uploaded,
23+
output_prefix="live-convert",
24+
page_range="1",
25+
)
26+
27+
assert isinstance(response, PdfRestFileBasedResponse)
28+
assert response.output_files
29+
30+
31+
@pytest.mark.asyncio
32+
async def test_live_async_convert_to_png(
33+
pdfrest_api_key: str,
34+
pdfrest_live_base_url: str,
35+
) -> None:
36+
resource = get_test_resource_path("report.pdf")
37+
async with AsyncPdfRestClient(
38+
api_key=pdfrest_api_key,
39+
base_url=pdfrest_live_base_url,
40+
) as client:
41+
uploaded = (await client.files.create_from_paths([resource]))[0]
42+
response = await client.convert_to_png(
43+
uploaded,
44+
output_prefix="live-async-convert",
45+
page_range="1",
46+
)
47+
48+
assert isinstance(response, PdfRestFileBasedResponse)
49+
assert response.output_files

tests/test_convert_to_png.py

Lines changed: 0 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
build_file_info_payload,
1919
make_pdf_file,
2020
)
21-
from .resources import get_test_resource_path
2221

2322

2423
@pytest.mark.parametrize("color_model", ["rgb", "rgba", "gray"])
@@ -577,37 +576,3 @@ def handler(_: httpx.Request) -> httpx.Response:
577576
make_pdf_file(PdfRestFileID.generate(1)),
578577
page_range=[],
579578
)
580-
581-
582-
def test_live_convert_to_png(pdfrest_api_key: str, pdfrest_live_base_url: str) -> None:
583-
resource = get_test_resource_path("report.pdf")
584-
with PdfRestClient(
585-
api_key=pdfrest_api_key, base_url=pdfrest_live_base_url
586-
) as client:
587-
uploaded = client.files.create_from_paths([resource])
588-
response = client.convert_to_png(
589-
uploaded[0],
590-
output_prefix="live-convert",
591-
page_range="1",
592-
)
593-
assert isinstance(response, PdfRestFileBasedResponse)
594-
assert response.output_files
595-
596-
597-
@pytest.mark.asyncio
598-
async def test_live_async_convert_to_png(
599-
pdfrest_api_key: str,
600-
pdfrest_live_base_url: str,
601-
) -> None:
602-
resource = get_test_resource_path("report.pdf")
603-
async with AsyncPdfRestClient(
604-
api_key=pdfrest_api_key, base_url=pdfrest_live_base_url
605-
) as client:
606-
uploaded = await client.files.create_from_paths([resource])
607-
response = await client.convert_to_png(
608-
uploaded[0],
609-
output_prefix="live-async-convert",
610-
page_range="1",
611-
)
612-
assert isinstance(response, PdfRestFileBasedResponse)
613-
assert response.output_files

0 commit comments

Comments
 (0)