26 lines
922 B
Python
26 lines
922 B
Python
|
import unittest
|
||
|
|
from guardrails.telemetry.common import ismatchingkey
|
||
|
|
|
||
|
|
|
||
|
|
class TestIsMatchingKey(unittest.TestCase):
|
||
|
|
def test_key_matches_with_default_keys(self):
|
||
|
|
self.assertTrue(ismatchingkey("api_key"))
|
||
|
|
self.assertTrue(ismatchingkey("user_token"))
|
||
|
|
self.assertFalse(ismatchingkey("username"))
|
||
|
|
self.assertTrue(ismatchingkey("password"))
|
||
|
|
|
||
|
|
def test_key_matches_with_custom_keys(self):
|
||
|
|
self.assertTrue(ismatchingkey("api_secret", keys_to_match=("secret",)))
|
||
|
|
self.assertTrue(ismatchingkey("client_id", keys_to_match=("id",)))
|
||
|
|
self.assertFalse(ismatchingkey("session", keys_to_match=("key", "token")))
|
||
|
|
|
||
|
|
def test_empty_key(self):
|
||
|
|
self.assertFalse(ismatchingkey("", keys_to_match=("key", "token")))
|
||
|
|
|
||
|
|
def test_empty_keys_to_match(self):
|
||
|
|
self.assertFalse(ismatchingkey("key", keys_to_match=()))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|