scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2025-07-27 09:52:22 -07:00
parent e8a374fc47
commit ff0b40a22b
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py
View file

@ -21,30 +21,30 @@ from litellm.proxy.common_utils.encrypt_decrypt_utils import (
def test_encrypt_decrypt_with_master_key():
setattr(proxy_server, "master_key", "sk-1234")
assert decrypt_value_helper(encrypt_value_helper("test", key="test_key")) == "test"
assert decrypt_value_helper(encrypt_value_helper(10, key="test_key")) == 10
assert decrypt_value_helper(encrypt_value_helper(True, key="test_key")) is True
assert decrypt_value_helper(encrypt_value_helper(None, key="test_key")) is None
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10}, key="test_key")) == {
assert decrypt_value_helper(encrypt_value_helper("test"), key="test_key") == "test"
assert decrypt_value_helper(encrypt_value_helper(10), key="test_key") == 10
assert decrypt_value_helper(encrypt_value_helper(True), key="test_key") is True
assert decrypt_value_helper(encrypt_value_helper(None), key="test_key") is None
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10}), key="test_key") == {
"rpm": 10
}
# encryption should actually occur for strings
assert encrypt_value_helper("test", key="test_key") != "test"
assert encrypt_value_helper("test") != "test"
def test_encrypt_decrypt_with_salt_key():
os.environ["LITELLM_SALT_KEY"] = "sk-salt-key2222"
print(f"LITELLM_SALT_KEY: {os.environ['LITELLM_SALT_KEY']}")
assert decrypt_value_helper(encrypt_value_helper("test", key="test_key")) == "test"
assert decrypt_value_helper(encrypt_value_helper(10, key="test_key")) == 10
assert decrypt_value_helper(encrypt_value_helper(True, key="test_key")) is True
assert decrypt_value_helper(encrypt_value_helper(None, key="test_key")) is None
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10}, key="test_key")) == {
assert decrypt_value_helper(encrypt_value_helper("test"), key="test_key") == "test"
assert decrypt_value_helper(encrypt_value_helper(10), key="test_key") == 10
assert decrypt_value_helper(encrypt_value_helper(True), key="test_key") is True
assert decrypt_value_helper(encrypt_value_helper(None), key="test_key") is None
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10}), key="test_key") == {
"rpm": 10
}
# encryption should actually occur for strings
assert encrypt_value_helper("test", key="test_key") != "test"
assert encrypt_value_helper("test") != "test"
os.environ.pop("LITELLM_SALT_KEY", None)