Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,53 @@
|
||||
"""Tests for provider-agnostic billing recovery links (agent/billing_links.py).
|
||||
|
||||
Behavior/invariant tests — no snapshotting of the exact URL strings beyond the
|
||||
few that are the whole point of the mapping (the host they must land on).
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from agent.billing_links import (
|
||||
BillingBlock,
|
||||
build_billing_block,
|
||||
is_nous_inference_route,
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def test_is_nous_inference_route_helper():
|
||||
assert is_nous_inference_route("nous", "") is True
|
||||
assert is_nous_inference_route("", "https://inference-api.nousresearch.com/v1") is True
|
||||
assert is_nous_inference_route("openai", "https://api.openai.com/v1") is False
|
||||
|
||||
|
||||
def test_known_provider_by_slug_resolves_label_and_url():
|
||||
block = build_billing_block(provider="openai", base_url="", model="gpt-5")
|
||||
assert block.is_nous is False
|
||||
assert block.provider_label == "OpenAI"
|
||||
assert block.billing_url is not None
|
||||
assert "openai.com" in block.billing_url
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def test_to_dict_round_trips_all_fields():
|
||||
block = build_billing_block(provider="openai", base_url="", model="gpt-5")
|
||||
data = block.to_dict()
|
||||
assert set(data) == {
|
||||
"provider",
|
||||
"provider_label",
|
||||
"model",
|
||||
"billing_url",
|
||||
"is_nous",
|
||||
"message",
|
||||
}
|
||||
assert isinstance(block, BillingBlock)
|
||||
Reference in New Issue
Block a user