mirror of
https://github.com/headroomlabs-ai/headroom.git
synced 2026-08-27 14:17:10 -04:00
30 lines
975 B
Python
30 lines
975 B
Python
from __future__ import annotations
|
|
|
|
from headroom.providers.cursor import build_proxy_targets, render_setup_lines
|
|
from headroom.providers.cursor.install import build_install_env
|
|
|
|
|
|
def test_cursor_proxy_targets_use_local_headroom_proxy() -> None:
|
|
targets = build_proxy_targets(9999)
|
|
|
|
assert targets.openai_base_url == "http://127.0.0.1:9999/v1"
|
|
assert targets.anthropic_base_url == "http://127.0.0.1:9999"
|
|
|
|
|
|
def test_cursor_setup_lines_include_both_provider_urls() -> None:
|
|
lines = render_setup_lines(8787)
|
|
joined = "\n".join(lines)
|
|
|
|
assert "http://127.0.0.1:8787/v1" in joined
|
|
assert "http://127.0.0.1:8787" in joined
|
|
|
|
|
|
def test_cursor_build_install_env_returns_both_proxy_urls() -> None:
|
|
# Arrange / Act
|
|
env = build_install_env(port=7654, backend="ignored")
|
|
|
|
# Assert
|
|
assert env == {
|
|
"OPENAI_BASE_URL": "http://127.0.0.1:7654/v1",
|
|
"ANTHROPIC_BASE_URL": "http://127.0.0.1:7654",
|
|
}
|