Skip to content

Commit 8d6d28b

Browse files
fix(client): preserve hardcoded query params when merging with user params
1 parent afe4e29 commit 8d6d28b

File tree

2 files changed

+52
-0
lines changed

2 files changed

+52
-0
lines changed

src/openlayer/_base_client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -540,6 +540,10 @@ def _build_request(
540540
files = cast(HttpxRequestFiles, ForceMultipartDict())
541541

542542
prepared_url = self._prepare_url(options.url)
543+
# preserve hard-coded query params from the url
544+
if params and prepared_url.query:
545+
params = {**dict(prepared_url.params.items()), **params}
546+
prepared_url = prepared_url.copy_with(raw_path=prepared_url.raw_path.split(b"?", 1)[0])
543547
if "_" in prepared_url.host:
544548
# work around https://github.com/encode/httpx/discussions/2880
545549
kwargs["extensions"] = {"sni_hostname": prepared_url.host.replace("_", "-")}

tests/test_client.py

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -438,6 +438,30 @@ def test_default_query_option(self) -> None:
438438

439439
client.close()
440440

441+
def test_hardcoded_query_params_in_url(self, client: Openlayer) -> None:
442+
request = client._build_request(FinalRequestOptions(method="get", url="/foo?beta=true"))
443+
url = httpx.URL(request.url)
444+
assert dict(url.params) == {"beta": "true"}
445+
446+
request = client._build_request(
447+
FinalRequestOptions(
448+
method="get",
449+
url="/foo?beta=true",
450+
params={"limit": "10", "page": "abc"},
451+
)
452+
)
453+
url = httpx.URL(request.url)
454+
assert dict(url.params) == {"beta": "true", "limit": "10", "page": "abc"}
455+
456+
request = client._build_request(
457+
FinalRequestOptions(
458+
method="get",
459+
url="/files/a%2Fb?beta=true",
460+
params={"limit": "10"},
461+
)
462+
)
463+
assert request.url.raw_path == b"/files/a%2Fb?beta=true&limit=10"
464+
441465
def test_request_extra_json(self, client: Openlayer) -> None:
442466
request = client._build_request(
443467
FinalRequestOptions(
@@ -1410,6 +1434,30 @@ async def test_default_query_option(self) -> None:
14101434

14111435
await client.close()
14121436

1437+
async def test_hardcoded_query_params_in_url(self, async_client: AsyncOpenlayer) -> None:
1438+
request = async_client._build_request(FinalRequestOptions(method="get", url="/foo?beta=true"))
1439+
url = httpx.URL(request.url)
1440+
assert dict(url.params) == {"beta": "true"}
1441+
1442+
request = async_client._build_request(
1443+
FinalRequestOptions(
1444+
method="get",
1445+
url="/foo?beta=true",
1446+
params={"limit": "10", "page": "abc"},
1447+
)
1448+
)
1449+
url = httpx.URL(request.url)
1450+
assert dict(url.params) == {"beta": "true", "limit": "10", "page": "abc"}
1451+
1452+
request = async_client._build_request(
1453+
FinalRequestOptions(
1454+
method="get",
1455+
url="/files/a%2Fb?beta=true",
1456+
params={"limit": "10"},
1457+
)
1458+
)
1459+
assert request.url.raw_path == b"/files/a%2Fb?beta=true&limit=10"
1460+
14131461
def test_request_extra_json(self, client: Openlayer) -> None:
14141462
request = client._build_request(
14151463
FinalRequestOptions(

0 commit comments

Comments
 (0)