wehub-resource-sync
103 lines
3.4 KiB
Python
103 lines
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
|