forked from BerriAI/litellm
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_url_encoding.py
More file actions
31 lines (26 loc) · 1.15 KB
/
test_url_encoding.py
File metadata and controls
31 lines (26 loc) · 1.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# Test URL encoding handling for emails with + characters
import re
from urllib.parse import unquote
def test_user_id_parsing():
# Simulate the raw query string that would come from the URL
# When user calls: http://0.0.0.0:4000/user/info?user_id=machine-user+alp-air-admin-b58-b@tempus.com
# The query string would be: user_id=machine-user+alp-air-admin-b58-b@tempus.com
test_cases = [
"user_id=machine-user+alp-air-admin-b58-b@tempus.com",
"user_id=machine-user%2Balp-air-admin-b58-b@tempus.com", # URL encoded +
"user_id=regular@email.com",
"user_id=test-user@domain.com&other_param=value"
]
for query_string in test_cases:
print(f"\nTesting query string: {query_string}")
if 'user_id=' in query_string:
match = re.search(r'user_id=([^&]*)', query_string)
if match:
raw_user_id = unquote(match.group(1))
print(f"Extracted user_id: {raw_user_id}")
else:
print("No match found")
else:
print("user_id not found in query string")
if __name__ == "__main__":
test_user_id_parsing()