scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

test(vertex_ai/gemini): cover _is_valid_gcs_bucket_name IP, invalid chars, edge dots

Browse source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
S0ngRu1 2026-05-13 16:56:45 +08:00
parent afe5772862
commit f3e0fd11d6
1 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/test_litellm/llms/vertex_ai/test_vertex.py
View file

@ -1428,6 +1428,31 @@ def test_is_valid_gcs_bucket_name_rejects_double_dot():
assert _is_valid_gcs_bucket_name("ab..cd") is False
def test_is_valid_gcs_bucket_name_rejects_ip_address_format():
"""Bucket names must not look like IPv4 addresses (GCS DNS constraint)."""
from litellm.llms.vertex_ai.gemini.transformation import _is_valid_gcs_bucket_name
assert _is_valid_gcs_bucket_name("1.2.3.4") is False
assert _is_valid_gcs_bucket_name("192.168.0.1") is False
def test_is_valid_gcs_bucket_name_rejects_invalid_chars():
from litellm.llms.vertex_ai.gemini.transformation import _is_valid_gcs_bucket_name
assert _is_valid_gcs_bucket_name("Bucket-Upper") is False
assert _is_valid_gcs_bucket_name("bucket@name") is False
assert _is_valid_gcs_bucket_name("bucket name") is False
def test_is_valid_gcs_bucket_name_rejects_leading_trailing_special():
from litellm.llms.vertex_ai.gemini.transformation import _is_valid_gcs_bucket_name
assert _is_valid_gcs_bucket_name("-mybucket") is False
assert _is_valid_gcs_bucket_name("mybucket-") is False
assert _is_valid_gcs_bucket_name(".mybucket") is False
assert _is_valid_gcs_bucket_name("mybucket.") is False
def test_get_gcs_object_content_type_uses_shared_vertex_base_instance():
from litellm.llms.vertex_ai.gemini import transformation as gemini_transformation