Files
opensquilla--opensquilla/tests/test_secrets.py
T
2026-07-13 13:12:33 +08:00

20 lines
670 B
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
from __future__ import annotations
import pytest
from opensquilla.secrets import clean_header_secret
def test_clean_header_secret_strips_trailing_paste_punctuation() -> None:
assert clean_header_secret(" sk-test、\n", label="API key") == "sk-test"
def test_clean_header_secret_strips_leading_paste_punctuation() -> None:
assert clean_header_secret("sk-test", label="API key") == "sk-test"
assert clean_header_secret("、sk-test", label="API key") == "sk-test"
def test_clean_header_secret_rejects_non_ascii_inside_value() -> None:
with pytest.raises(ValueError, match="non-ASCII"):
clean_header_secret("sk-测-test", label="API key")