mirror of
https://github.com/headroomlabs-ai/headroom.git
synced 2026-08-27 14:17:10 -04:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from headroom.proxy.beta_header_merge import (
|
||
|
|
merge_anthropic_beta,
|
||
|
|
merge_beta_tokens,
|
||
|
|
merge_openai_beta,
|
||
|
|
split_beta_tokens,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_split_beta_tokens_drops_empty_entries() -> None:
|
||
|
|
assert split_beta_tokens(None) == []
|
||
|
|
assert split_beta_tokens("") == []
|
||
|
|
assert split_beta_tokens(" alpha, , beta ,, ") == ["alpha", "beta"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_merge_beta_tokens_preserves_client_order_then_appends_required() -> None:
|
||
|
|
assert merge_beta_tokens("client-1,client-2", ["required-1", "required-2"]) == (
|
||
|
|
"client-1,client-2,required-1,required-2"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_merge_beta_tokens_dedupes_case_insensitively_with_first_casing() -> None:
|
||
|
|
assert merge_beta_tokens("Foo,foo", ["FOO", "bar"]) == "Foo,bar"
|
||
|
|
|
||
|
|
|
||
|
|
def test_merge_beta_tokens_skips_empty_required_values() -> None:
|
||
|
|
assert merge_beta_tokens("alpha", ["", " beta ", " "]) == "alpha,beta"
|
||
|
|
|
||
|
|
|
||
|
|
def test_provider_wrappers_share_merge_semantics() -> None:
|
||
|
|
assert merge_anthropic_beta("a", ["b"]) == "a,b"
|
||
|
|
assert merge_openai_beta("a", ["b"]) == "a,b"
|