wehub-resource-sync
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
from opensquilla.provider.openrouter_attribution import (
|
|
OPENROUTER_APP_REFERER,
|
|
OPENROUTER_APP_TITLE,
|
|
is_openrouter_url,
|
|
openrouter_app_headers,
|
|
)
|
|
|
|
|
|
def test_openrouter_app_headers_match_app_attribution_contract() -> None:
|
|
# OpenRouter's documented app-attribution contract: HTTP-Referer + X-Title.
|
|
assert openrouter_app_headers("https://openrouter.ai/api/v1") == {
|
|
"HTTP-Referer": "https://opensquilla.ai",
|
|
"X-Title": "OpenSquilla",
|
|
}
|
|
assert OPENROUTER_APP_REFERER == "https://opensquilla.ai"
|
|
assert OPENROUTER_APP_TITLE == "OpenSquilla"
|
|
|
|
|
|
def test_openrouter_url_detection_accepts_openrouter_hosts_only() -> None:
|
|
assert is_openrouter_url("https://openrouter.ai/api/v1")
|
|
assert is_openrouter_url("https://api.openrouter.ai/v1")
|
|
assert not is_openrouter_url("https://openrouter.example/api/v1")
|
|
assert not is_openrouter_url("http://localhost:4000/v1")
|
|
|
|
|
|
def test_openrouter_app_headers_skip_non_openrouter_urls() -> None:
|
|
assert openrouter_app_headers("https://api.openai.com/v1") == {}
|
|
assert openrouter_app_headers("http://localhost:4000/v1") == {}
|
|
|
|
|
|
def test_openrouter_compat_headers_remain_openrouter_only() -> None:
|
|
url = "https://tokenrhythm.studio/v1"
|
|
assert not is_openrouter_url(url)
|
|
assert openrouter_app_headers(url) == {}
|