wehub-resource-sync
f3d80b4628
Auto Release / auto-release (push) Failing after 1s
CI / lint (push) Failing after 0s
CI / screenshot-quality (push) Failing after 3s
CI / pr-policy (push) Has been skipped
CI / test (3.11) (push) Failing after 0s
CI / test (3.12) (push) Failing after 0s
CI / test (3.13) (push) Failing after 3s
CI / coverage (push) Failing after 1s
Legibility / legibility (push) Failing after 3s
132 lines
4.6 KiB
Python
132 lines
4.6 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from claude_tap import parse_args
|
|
from claude_tap.cli import CLIENT_CONFIGS, run_client
|
|
|
|
|
|
class _DummyProc:
|
|
def __init__(self) -> None:
|
|
self.pid = 12345
|
|
self.returncode: int | None = None
|
|
|
|
async def wait(self) -> int:
|
|
self.returncode = 0
|
|
return 0
|
|
|
|
def terminate(self) -> None:
|
|
self.returncode = 0
|
|
|
|
def kill(self) -> None:
|
|
self.returncode = -9
|
|
|
|
|
|
def test_opencode_registered_in_client_configs() -> None:
|
|
cfg = CLIENT_CONFIGS["opencode"]
|
|
assert cfg.cmd == "opencode"
|
|
assert cfg.default_target == "https://api.anthropic.com"
|
|
assert cfg.base_url_env == "ANTHROPIC_BASE_URL"
|
|
# opencode is multi-provider; forward proxy is the natural default
|
|
assert cfg.default_proxy_mode == "forward"
|
|
|
|
|
|
def test_parse_args_opencode_defaults_to_forward_mode() -> None:
|
|
args = parse_args(["--tap-client", "opencode"])
|
|
assert args.client == "opencode"
|
|
assert args.proxy_mode == "forward"
|
|
|
|
|
|
def test_parse_args_opencode_explicit_reverse_overrides_default() -> None:
|
|
args = parse_args(["--tap-client", "opencode", "--tap-proxy-mode", "reverse"])
|
|
assert args.proxy_mode == "reverse"
|
|
|
|
|
|
def test_parse_args_claude_default_proxy_mode_unchanged() -> None:
|
|
# Regression: changing the default-mode plumbing must not affect claude
|
|
args = parse_args([])
|
|
assert args.client == "claude"
|
|
assert args.proxy_mode == "reverse"
|
|
|
|
|
|
def test_parse_args_codex_default_proxy_mode_unchanged() -> None:
|
|
# Regression: changing the default-mode plumbing must not affect codex
|
|
args = parse_args(["--tap-client", "codex"])
|
|
assert args.client == "codex"
|
|
assert args.proxy_mode == "reverse"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_client_opencode_forward_sets_node_ca_env(monkeypatch) -> None:
|
|
captured: dict[str, object] = {}
|
|
ca_path = Path("/tmp/test-ca.pem")
|
|
|
|
async def fake_create_subprocess_exec(*cmd, **kwargs):
|
|
captured["env"] = kwargs["env"]
|
|
return _DummyProc()
|
|
|
|
monkeypatch.setattr("claude_tap.cli.shutil.which", lambda _: "/tmp/opencode")
|
|
monkeypatch.setattr(asyncio, "create_subprocess_exec", fake_create_subprocess_exec)
|
|
monkeypatch.setattr("sys.stdin.isatty", lambda: False)
|
|
|
|
code = await run_client(43123, ["run", "hello"], client="opencode", proxy_mode="forward", ca_cert_path=ca_path)
|
|
|
|
assert code == 0
|
|
env = captured["env"]
|
|
assert env["HTTPS_PROXY"] == "http://127.0.0.1:43123"
|
|
assert env["NODE_EXTRA_CA_CERTS"] == str(ca_path)
|
|
assert env["SSL_CERT_FILE"] == str(ca_path)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_client_opencode_reverse_sets_anthropic_base_url(monkeypatch) -> None:
|
|
captured: dict[str, object] = {}
|
|
for key in ("OPENAI_BASE_URL", "GOOGLE_GEMINI_BASE_URL"):
|
|
monkeypatch.delenv(key, raising=False)
|
|
|
|
async def fake_create_subprocess_exec(*cmd, **kwargs):
|
|
captured["cmd"] = cmd
|
|
captured["env"] = kwargs["env"]
|
|
return _DummyProc()
|
|
|
|
monkeypatch.setattr("claude_tap.cli.shutil.which", lambda _: "/tmp/opencode")
|
|
monkeypatch.setattr(asyncio, "create_subprocess_exec", fake_create_subprocess_exec)
|
|
monkeypatch.setattr("sys.stdin.isatty", lambda: False)
|
|
|
|
code = await run_client(43123, ["run", "hello"], client="opencode", proxy_mode="reverse")
|
|
|
|
assert code == 0
|
|
env = captured["env"]
|
|
assert env["ANTHROPIC_BASE_URL"] == "http://127.0.0.1:43123"
|
|
assert "OPENAI_BASE_URL" not in env
|
|
assert "GOOGLE_GEMINI_BASE_URL" not in env
|
|
# Reverse mode for opencode must not inject a codex-only -c flag
|
|
assert captured["cmd"] == ("/tmp/opencode", "run", "hello")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_client_opencode_capture_only_reverse_sets_multi_provider_urls(monkeypatch) -> None:
|
|
captured: dict[str, object] = {}
|
|
|
|
async def fake_create_subprocess_exec(*cmd, **kwargs):
|
|
captured["cmd"] = cmd
|
|
captured["env"] = kwargs["env"]
|
|
return _DummyProc()
|
|
|
|
monkeypatch.setattr("claude_tap.cli.shutil.which", lambda _: "/tmp/opencode")
|
|
monkeypatch.setattr(asyncio, "create_subprocess_exec", fake_create_subprocess_exec)
|
|
monkeypatch.setattr("sys.stdin.isatty", lambda: False)
|
|
|
|
code = await run_client(43123, ["run", "hello"], client="opencode", proxy_mode="reverse", capture_only=True)
|
|
|
|
assert code == 0
|
|
env = captured["env"]
|
|
assert env["ANTHROPIC_BASE_URL"] == "http://127.0.0.1:43123"
|
|
assert env["OPENAI_BASE_URL"] == "http://127.0.0.1:43123/v1"
|
|
assert env["GOOGLE_GEMINI_BASE_URL"] == "http://127.0.0.1:43123"
|
|
# Reverse mode for opencode must not inject a codex-only -c flag
|
|
assert captured["cmd"] == ("/tmp/opencode", "run", "hello")
|