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
111 lines
3.7 KiB
Python
111 lines
3.7 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, _reverse_proxy_trace_options, 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_gemini_registered_in_client_configs() -> None:
|
|
cfg = CLIENT_CONFIGS["gemini"]
|
|
|
|
assert cfg.cmd == "gemini"
|
|
assert cfg.label == "Gemini CLI"
|
|
assert cfg.default_target == "https://generativelanguage.googleapis.com"
|
|
assert cfg.base_url_env == "GOOGLE_GEMINI_BASE_URL"
|
|
assert cfg.extra_base_url_envs == ("GOOGLE_VERTEX_BASE_URL",)
|
|
assert cfg.base_url_suffix == ""
|
|
assert cfg.default_proxy_mode == "forward"
|
|
|
|
|
|
def test_parse_args_gemini_defaults_to_forward_mode() -> None:
|
|
args = parse_args(["--tap-client", "gemini"])
|
|
|
|
assert args.client == "gemini"
|
|
assert args.target == "https://generativelanguage.googleapis.com"
|
|
assert args.proxy_mode == "forward"
|
|
|
|
|
|
def test_parse_args_gemini_explicit_reverse_overrides_default() -> None:
|
|
args = parse_args(["--tap-client", "gemini", "--tap-proxy-mode", "reverse"])
|
|
|
|
assert args.proxy_mode == "reverse"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_client_gemini_forward_sets_proxy_ca_and_skips_base_url_envs(monkeypatch) -> None:
|
|
captured: dict[str, object] = {}
|
|
ca_path = Path("/tmp/test-ca.pem")
|
|
|
|
async def fake_create_subprocess_exec(*cmd, **kwargs):
|
|
captured["cmd"] = cmd
|
|
captured["env"] = kwargs["env"]
|
|
return _DummyProc()
|
|
|
|
monkeypatch.delenv("GOOGLE_GEMINI_BASE_URL", raising=False)
|
|
monkeypatch.delenv("GOOGLE_VERTEX_BASE_URL", raising=False)
|
|
monkeypatch.setattr("claude_tap.cli.shutil.which", lambda _: "/tmp/gemini")
|
|
monkeypatch.setattr(asyncio, "create_subprocess_exec", fake_create_subprocess_exec)
|
|
monkeypatch.setattr("sys.stdin.isatty", lambda: False)
|
|
|
|
code = await run_client(43123, ["-p", "hello"], client="gemini", proxy_mode="forward", ca_cert_path=ca_path)
|
|
|
|
assert code == 0
|
|
assert captured["cmd"] == ("/tmp/gemini", "-p", "hello")
|
|
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)
|
|
assert "GOOGLE_GEMINI_BASE_URL" not in env
|
|
assert "GOOGLE_VERTEX_BASE_URL" not in env
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_client_gemini_reverse_sets_both_base_url_envs(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/gemini")
|
|
monkeypatch.setattr(asyncio, "create_subprocess_exec", fake_create_subprocess_exec)
|
|
monkeypatch.setattr("sys.stdin.isatty", lambda: False)
|
|
|
|
code = await run_client(43123, ["-p", "hello"], client="gemini", proxy_mode="reverse")
|
|
|
|
assert code == 0
|
|
assert captured["cmd"] == ("/tmp/gemini", "-p", "hello")
|
|
env = captured["env"]
|
|
assert env["GOOGLE_GEMINI_BASE_URL"] == "http://127.0.0.1:43123"
|
|
assert env["GOOGLE_VERTEX_BASE_URL"] == "http://127.0.0.1:43123"
|
|
|
|
|
|
def test_gemini_reverse_trace_options_do_not_strip_path_prefix() -> None:
|
|
options = _reverse_proxy_trace_options("gemini", "https://generativelanguage.googleapis.com")
|
|
|
|
assert options == {
|
|
"strip_path_prefix": "",
|
|
"force_http": False,
|
|
}
|