hkuds--openharness
Siaochuan
63ac368975
Prefer OPENHARNESS_<PROVIDER>_API_KEY variables over provider-native globals while resolving API-key auth. Keep provider-native variables as fallback and avoid applying unrelated native keys across active profiles. Forward the OpenHarness-scoped auth/provider environment variables to spawned teammates. Based-on: #93
103 行
3.4 KiB
Python
103 行
3.4 KiB
Python
"""Tests for teammate spawn helper behavior."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
|
|
from openharness.swarm.spawn_utils import (
|
|
TEAMMATE_COMMAND_ENV_VAR,
|
|
build_inherited_cli_flags,
|
|
build_inherited_env_vars,
|
|
get_teammate_command,
|
|
)
|
|
|
|
|
|
def test_get_teammate_command_prefers_current_interpreter(monkeypatch):
|
|
monkeypatch.delenv(TEAMMATE_COMMAND_ENV_VAR, raising=False)
|
|
monkeypatch.setattr(sys, "executable", "/tmp/current-python")
|
|
|
|
command = get_teammate_command()
|
|
|
|
assert command == "/tmp/current-python"
|
|
|
|
|
|
def test_build_inherited_env_vars_disables_coordinator_mode(monkeypatch):
|
|
monkeypatch.setenv("CLAUDE_CODE_COORDINATOR_MODE", "1")
|
|
|
|
env = build_inherited_env_vars()
|
|
|
|
assert env["CLAUDE_CODE_COORDINATOR_MODE"] == "0"
|
|
|
|
|
|
def test_build_inherited_env_vars_forwards_openharness_config_dir(monkeypatch):
|
|
monkeypatch.setenv("OPENHARNESS_CONFIG_DIR", "/opt/data/.openharness")
|
|
|
|
env = build_inherited_env_vars()
|
|
|
|
assert env["OPENHARNESS_CONFIG_DIR"] == "/opt/data/.openharness"
|
|
|
|
|
|
def test_build_inherited_env_vars_includes_openharness_auth_vars(monkeypatch):
|
|
monkeypatch.setenv("OPENHARNESS_PROVIDER", "openai")
|
|
monkeypatch.setenv("OPENHARNESS_BASE_URL", "https://relay.example.com/v1")
|
|
monkeypatch.setenv("OPENHARNESS_OPENAI_API_KEY", "sk-oh-openai")
|
|
monkeypatch.setenv("OPENHARNESS_ANTHROPIC_API_KEY", "sk-oh-anthropic")
|
|
|
|
env = build_inherited_env_vars()
|
|
|
|
assert env["OPENHARNESS_AGENT_TEAMS"] == "1"
|
|
assert env["OPENHARNESS_PROVIDER"] == "openai"
|
|
assert env["OPENHARNESS_BASE_URL"] == "https://relay.example.com/v1"
|
|
assert env["OPENHARNESS_OPENAI_API_KEY"] == "sk-oh-openai"
|
|
assert env["OPENHARNESS_ANTHROPIC_API_KEY"] == "sk-oh-anthropic"
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# build_inherited_cli_flags – model handling
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_build_inherited_cli_flags_explicit_model_included():
|
|
flags = build_inherited_cli_flags(model="claude-opus-4-5")
|
|
assert "--model" in flags
|
|
idx = flags.index("--model")
|
|
assert "claude-opus-4-5" in flags[idx + 1]
|
|
|
|
|
|
def test_build_inherited_cli_flags_inherit_model_excluded():
|
|
"""model='inherit' must NOT produce a --model flag so the subprocess
|
|
picks up the parent's model from the OPENHARNESS_MODEL env var."""
|
|
flags = build_inherited_cli_flags(model="inherit")
|
|
assert "--model" not in flags
|
|
|
|
|
|
def test_build_inherited_cli_flags_none_model_excluded():
|
|
flags = build_inherited_cli_flags(model=None)
|
|
assert "--model" not in flags
|
|
|
|
|
|
def test_build_inherited_cli_flags_empty_string_model_excluded():
|
|
flags = build_inherited_cli_flags(model="")
|
|
assert "--model" not in flags
|
|
|
|
|
|
def test_build_inherited_cli_flags_forwards_system_prompt_as_replace():
|
|
flags = build_inherited_cli_flags(system_prompt="You are a specialized worker.")
|
|
|
|
assert "--system-prompt" in flags
|
|
idx = flags.index("--system-prompt")
|
|
assert "specialized worker" in flags[idx + 1]
|
|
assert "--append-system-prompt" not in flags
|
|
|
|
|
|
def test_build_inherited_cli_flags_forwards_system_prompt_as_append():
|
|
flags = build_inherited_cli_flags(
|
|
system_prompt="Extra worker instructions.",
|
|
system_prompt_mode="append",
|
|
)
|
|
|
|
assert "--append-system-prompt" in flags
|
|
idx = flags.index("--append-system-prompt")
|
|
assert "Extra worker instructions." in flags[idx + 1]
|
|
assert "--system-prompt" not in flags
|