mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-12 21:04:10 +00:00
user
29 lines
951 B
Python
29 lines
951 B
Python
import pytest
|
|
|
|
from litellm.llms.custom_httpx.container_handler import _build_url
|
|
|
|
|
|
def test_build_url_encodes_path_params_and_preserves_query():
|
|
url = _build_url(
|
|
api_base="https://example.com/v1/containers?api-version=v1",
|
|
path_template="/containers/{container_id}/files/{file_id}/content",
|
|
path_params={
|
|
"container_id": "../../containers/other",
|
|
"file_id": "file?download=1#frag",
|
|
},
|
|
)
|
|
|
|
assert (
|
|
url
|
|
== "https://example.com/v1/containers/..%2F..%2Fcontainers%2Fother/files/file%3Fdownload%3D1%23frag/content?api-version=v1"
|
|
)
|
|
|
|
|
|
def test_build_url_rejects_dot_segment_path_param():
|
|
with pytest.raises(ValueError, match="container_id cannot be a dot path segment"):
|
|
_build_url(
|
|
api_base="https://example.com/v1/containers",
|
|
path_template="/containers/{container_id}",
|
|
path_params={"container_id": ".."},
|
|
)
|