scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

test: fix response ID security tests to mock encrypt helper

Browse source
This commit is contained in:
Yuta Saito 2026-01-13 14:42:43 +09:00
parent a1bba8c99b
commit 76e7a5437b
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/test_responses_id_security.py
View file

@ -127,7 +127,7 @@ class TestEncryptResponseId:
"litellm.proxy.hooks.responses_id_security.encrypt_value_helper"
) as mock_encrypt:
mock_encrypt.return_value = "encrypted_base64_value"
with patch.object(
responses_id_security, "_get_signing_key", return_value="test-key"
):
@ -148,9 +148,10 @@ class TestEncryptResponseId:
)
with patch(
"litellm.proxy.common_utils.encrypt_decrypt_utils._get_salt_key",
return_value="test-salt-key"
):
"litellm.proxy.hooks.responses_id_security.encrypt_value_helper"
) as mock_encrypt:
mock_encrypt.return_value = "encrypted_value_456"
with patch.object(
responses_id_security, "_get_signing_key", return_value="test-key"
):
@ -158,9 +159,9 @@ class TestEncryptResponseId:
mock_response, mock_user_api_key_dict
)
assert result.id == "resp_encrypted_value_456"
assert result.id.startswith("resp_")
# The encrypted ID should be different from the original
assert result.id != "resp_456"
mock_encrypt.assert_called_once()
class TestCheckUserAccessToResponseId: