odysseusphone / tests /cli /test_webhook_cli_mask.py
Earring9405's picture
Upload folder using huggingface_hub (part 2)
b594234 verified
Raw
History Blame Contribute Delete
471 Bytes
from tests.helpers.cli_loader import load_script
from tests.helpers.db_stubs import make_core_db_stub
def test_mask_token_handles_short_values(monkeypatch):
make_core_db_stub(monkeypatch, models=["ScheduledTask"])
cli = load_script("odysseus-webhook")
assert cli._mask_token("") == ""
assert cli._mask_token("short") == "***"
assert cli._mask_token("abcdef1234567890") == "abcdef…7890"
assert cli._mask_token("short", reveal=True) == "short"