From ca9f7ceed6e75f24bd926b8cf8038db02a2f0462 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 22 Jan 2025 18:00:51 -0800 Subject: [PATCH] test(test_auth_utils.py): ensure get end user id from request body always returns a string --- tests/local_testing/test_auth_utils.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/local_testing/test_auth_utils.py b/tests/local_testing/test_auth_utils.py index 1118b8a63b5..73abedb3f04 100644 --- a/tests/local_testing/test_auth_utils.py +++ b/tests/local_testing/test_auth_utils.py @@ -68,3 +68,12 @@ def test_configurable_clientside_parameters( ) print(resp) assert resp == should_return_true + + +def test_get_end_user_id_from_request_body_always_returns_str(): + from litellm.proxy.auth.auth_utils import get_end_user_id_from_request_body + + request_body = {"user": 123} + end_user_id = get_end_user_id_from_request_body(request_body) + assert end_user_id == "123" + assert isinstance(end_user_id, str)