mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +08:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from src.core.api_format.enums import ApiFamily, EndpointKind
|
||
|
|
from src.core.api_format.signature import (
|
||
|
|
EndpointSignature,
|
||
|
|
make_signature_key,
|
||
|
|
normalize_endpoint_signature,
|
||
|
|
normalize_signature_key,
|
||
|
|
parse_signature_key,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_make_signature_key_accepts_enums_and_strings() -> None:
|
||
|
|
assert make_signature_key(ApiFamily.OPENAI, EndpointKind.CHAT) == "openai:chat"
|
||
|
|
assert make_signature_key("OpenAI", "CHAT") == "openai:chat"
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_signature_key_roundtrip() -> None:
|
||
|
|
sig = parse_signature_key("OpenAI:CHAT")
|
||
|
|
assert sig.api_family == ApiFamily.OPENAI
|
||
|
|
assert sig.endpoint_kind == EndpointKind.CHAT
|
||
|
|
assert sig.key == "openai:chat"
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_signature_key_invalid_raises() -> None:
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
parse_signature_key("OPENAI") # missing ':'
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_signature_key() -> None:
|
||
|
|
assert normalize_signature_key(" OpenAI:CHAT ") == "openai:chat"
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_endpoint_signature_tuple_input() -> None:
|
||
|
|
sig = normalize_endpoint_signature((ApiFamily.GEMINI, EndpointKind.VIDEO))
|
||
|
|
assert sig == EndpointSignature(api_family=ApiFamily.GEMINI, endpoint_kind=EndpointKind.VIDEO)
|
||
|
|
|
||
|
|
|
||
|
|
def test_normalize_endpoint_signature_invalid_string_returns_default() -> None:
|
||
|
|
default = EndpointSignature(api_family=ApiFamily.CLAUDE, endpoint_kind=EndpointKind.CHAT)
|
||
|
|
assert normalize_endpoint_signature("not-a-signature", default=default) == default
|