88
99from pdfrest import AsyncPdfRestClient , PdfRestClient
1010from pdfrest .models import PdfRestFileBasedResponse , PdfRestFileID
11- from pdfrest .models ._internal import ConvertUrlsToPdfPayload
11+ from pdfrest .models ._internal import ConvertUrlToPdfPayload
1212
1313from .graphics_test_helpers import ASYNC_API_KEY , VALID_API_KEY , build_file_info_payload
1414
1515
1616def test_convert_urls_to_pdf_success (monkeypatch : pytest .MonkeyPatch ) -> None :
1717 monkeypatch .delenv ("PDFREST_API_KEY" , raising = False )
18- urls = [ "https://example.com/page" ]
18+ url = "https://example.com/page"
1919 output_id = str (PdfRestFileID .generate ())
20- payload_dump = ConvertUrlsToPdfPayload .model_validate (
20+ payload_dump = ConvertUrlToPdfPayload .model_validate (
2121 {
22- "url" : urls ,
22+ "url" : url ,
2323 "output" : "url-out" ,
2424 "page_size" : "letter" ,
2525 "page_margin" : "2.5in" ,
@@ -52,8 +52,8 @@ def handler(request: httpx.Request) -> httpx.Response:
5252
5353 transport = httpx .MockTransport (handler )
5454 with PdfRestClient (api_key = VALID_API_KEY , transport = transport ) as client :
55- response = client .convert_urls_to_pdf (
56- urls ,
55+ response = client .convert_url_to_pdf (
56+ url ,
5757 output = "url-out" ,
5858 page_size = "letter" ,
5959 page_margin = "2.5in" ,
@@ -68,11 +68,11 @@ def handler(request: httpx.Request) -> httpx.Response:
6868
6969def test_convert_urls_to_pdf_validation_errors () -> None :
7070 with pytest .raises (ValidationError , match = "Input should be a valid URL" ):
71- ConvertUrlsToPdfPayload .model_validate ({"url" : "not-a-url" })
71+ ConvertUrlToPdfPayload .model_validate ({"url" : "not-a-url" })
7272
7373 with pytest .raises (ValidationError , match = "page_margin must be a number" ):
74- ConvertUrlsToPdfPayload .model_validate (
75- {"url" : [ "https://example.com" ] , "page_margin" : "mm" }
74+ ConvertUrlToPdfPayload .model_validate (
75+ {"url" : "https://example.com" , "page_margin" : "mm" }
7676 )
7777
7878
@@ -81,7 +81,7 @@ async def test_async_convert_urls_to_pdf_request_customization(
8181 monkeypatch : pytest .MonkeyPatch ,
8282) -> None :
8383 monkeypatch .delenv ("PDFREST_API_KEY" , raising = False )
84- urls = [ "https://example.com/page" , "https://example.com/other" ]
84+ url = "https://example.com/page"
8585 output_id = str (PdfRestFileID .generate ())
8686 captured_timeout : dict [str , float | dict [str , float ] | None ] = {}
8787
@@ -91,7 +91,7 @@ def handler(request: httpx.Request) -> httpx.Response:
9191 assert request .headers ["X-Debug" ] == "async"
9292 captured_timeout ["value" ] = request .extensions .get ("timeout" )
9393 payload = json .loads (request .content .decode ("utf-8" ))
94- assert payload ["url" ] == urls
94+ assert payload ["url" ] == url
9595 assert payload ["output" ] == "async-url"
9696 assert payload ["page_orientation" ] == "portrait"
9797 assert payload ["debug" ] == "yes"
@@ -116,8 +116,8 @@ def handler(request: httpx.Request) -> httpx.Response:
116116
117117 transport = httpx .MockTransport (handler )
118118 async with AsyncPdfRestClient (api_key = ASYNC_API_KEY , transport = transport ) as client :
119- response = await client .convert_urls_to_pdf (
120- urls ,
119+ response = await client .convert_url_to_pdf (
120+ url ,
121121 output = "async-url" ,
122122 page_orientation = "portrait" ,
123123 extra_query = {"trace" : "async" },
0 commit comments