mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-24 22:29:56 +02:00
Alexander Myasoedov
16 lines
511 B
Python
16 lines
511 B
Python
from agentic_security.dependencies import InMemorySecrets, get_in_memory_secrets
|
|
|
|
|
|
def test_in_memory_secrets():
|
|
secrets = InMemorySecrets()
|
|
secrets.set_secret("api_key", "12345")
|
|
assert secrets.get_secret("api_key") == "12345"
|
|
assert secrets.get_secret("non_existent_key") is None
|
|
|
|
|
|
def test_get_in_memory_secrets():
|
|
secrets = get_in_memory_secrets()
|
|
assert isinstance(secrets, InMemorySecrets)
|
|
secrets.set_secret("token", "abcde")
|
|
assert secrets.get_secret("token") == "abcde"
|