forked from phoenix/litellm-mirror
27e18358ab
* fix(pattern_match_deployments.py): default to user input if unable to map based on wildcards * test: fix test * test: reset test name * test: update conftest to reload proxy server module between tests * ci(config.yml): move langfuse out of local_testing reduce ci/cd time * ci(config.yml): cleanup langfuse ci/cd tests * fix: update test to not use global proxy_server app module * ci: move caching to a separate test pipeline speed up ci pipeline * test: update conftest to check if proxy_server attr exists before reloading * build(conftest.py): don't block on inability to reload proxy_server * ci(config.yml): update caching unit test filter to work on 'cache' keyword as well * fix(encrypt_decrypt_utils.py): use function to get salt key * test: mark flaky test * test: handle anthropic overloaded errors * refactor: create separate ci/cd pipeline for proxy unit tests make ci/cd faster * ci(config.yml): add litellm_proxy_unit_testing to build_and_test jobs * ci(config.yml): generate prisma binaries for proxy unit tests * test: readd vertex_key.json * ci(config.yml): remove `-s` from proxy_unit_test cmd speed up test * ci: remove any 'debug' logging flag speed up ci pipeline * test: fix test * test(test_braintrust.py): rerun * test: add delay for braintrust test
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
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"
|
|
print(f"LITELLM_SALT_KEY: {os.environ['LITELLM_SALT_KEY']}")
|
|
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)
|