forked from phoenix/litellm-mirror
refactor: move all testing to top-level of repo
Closes https://github.com/BerriAI/litellm/issues/486
This commit is contained in:
parent
5403c5828c
commit
3560f0ef2c
213 changed files with 74 additions and 217 deletions
45
tests/local_testing/test_proxy_encrypt_decrypt.py
Normal file
45
tests/local_testing/test_proxy_encrypt_decrypt.py
Normal file
|
@ -0,0 +1,45 @@
|
|||
import os
|
||||
import sys
|
||||
|
||||
import pytest
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
import io
|
||||
import os
|
||||
|
||||
sys.path.insert(
|
||||
0, os.path.abspath("../..")
|
||||
) # Adds-the parent directory to the system path
|
||||
|
||||
from litellm.proxy import proxy_server
|
||||
from litellm.proxy.common_utils.encrypt_decrypt_utils import (
|
||||
decrypt_value_helper,
|
||||
encrypt_value_helper,
|
||||
)
|
||||
|
||||
|
||||
def test_encrypt_decrypt_with_master_key():
|
||||
setattr(proxy_server, "master_key", "sk-1234")
|
||||
assert decrypt_value_helper(encrypt_value_helper("test")) == "test"
|
||||
assert decrypt_value_helper(encrypt_value_helper(10)) == 10
|
||||
assert decrypt_value_helper(encrypt_value_helper(True)) is True
|
||||
assert decrypt_value_helper(encrypt_value_helper(None)) is None
|
||||
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10})) == {"rpm": 10}
|
||||
|
||||
# encryption should actually occur for strings
|
||||
assert encrypt_value_helper("test") != "test"
|
||||
|
||||
|
||||
def test_encrypt_decrypt_with_salt_key():
|
||||
os.environ["LITELLM_SALT_KEY"] = "sk-salt-key2222"
|
||||
assert decrypt_value_helper(encrypt_value_helper("test")) == "test"
|
||||
assert decrypt_value_helper(encrypt_value_helper(10)) == 10
|
||||
assert decrypt_value_helper(encrypt_value_helper(True)) is True
|
||||
assert decrypt_value_helper(encrypt_value_helper(None)) is None
|
||||
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10})) == {"rpm": 10}
|
||||
|
||||
# encryption should actually occur for strings
|
||||
assert encrypt_value_helper("test") != "test"
|
||||
|
||||
os.environ.pop("LITELLM_SALT_KEY", None)
|
Loading…
Add table
Add a link
Reference in a new issue