mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-06 02:37:05 +08:00
e89376d66f
_setup_slack() was the only platform setup function that did not prompt for a home channel. All four sibling setups (_setup_telegram, _setup_discord, _setup_mattermost, _setup_bluebubbles) close with an identical home-channel block, and setup_gateway() already checks for SLACK_HOME_CHANNEL presence at the end of the wizard — but the value was never collected, leaving cron delivery and cross-platform notifications silently broken for Slack after a fresh hermes setup run. Add the standard home-channel prompt at the end of _setup_slack(), symmetric with the Discord implementation. Add two unit tests that verify the prompt is saved when provided and skipped when left blank.
648 lines
23 KiB
Python
648 lines
23 KiB
Python
"""Tests for setup.py configuration flows."""
|
|
import json
|
|
import os
|
|
import sys
|
|
import types
|
|
|
|
import pytest
|
|
|
|
from hermes_cli.auth import get_active_provider
|
|
from hermes_cli.config import load_config, save_config
|
|
from hermes_cli import setup as setup_mod
|
|
from hermes_cli.setup import setup_model_provider
|
|
|
|
|
|
def _maybe_keep_current_tts(question, choices):
|
|
if question != "Select TTS provider:":
|
|
return None
|
|
assert choices[-1].startswith("Keep current (")
|
|
return len(choices) - 1
|
|
|
|
|
|
def _clear_provider_env(monkeypatch):
|
|
for key in (
|
|
"NOUS_API_KEY",
|
|
"OPENROUTER_API_KEY",
|
|
"OPENAI_BASE_URL",
|
|
"OPENAI_API_KEY",
|
|
"LLM_MODEL",
|
|
):
|
|
monkeypatch.delenv(key, raising=False)
|
|
|
|
|
|
def _clear_vercel_env(monkeypatch):
|
|
for key in (
|
|
"TERMINAL_VERCEL_RUNTIME",
|
|
"VERCEL_OIDC_TOKEN",
|
|
"VERCEL_TOKEN",
|
|
"VERCEL_PROJECT_ID",
|
|
"VERCEL_TEAM_ID",
|
|
):
|
|
monkeypatch.delenv(key, raising=False)
|
|
|
|
|
|
def _stub_tts(monkeypatch):
|
|
"""Stub out TTS prompts so setup_model_provider doesn't block."""
|
|
monkeypatch.setattr("hermes_cli.setup.prompt_choice", lambda q, c, d=0: (
|
|
_maybe_keep_current_tts(q, c) if _maybe_keep_current_tts(q, c) is not None
|
|
else d
|
|
))
|
|
monkeypatch.setattr("hermes_cli.setup.prompt_yes_no", lambda *a, **kw: False)
|
|
|
|
|
|
def _write_model_config(tmp_path, provider, base_url="", model_name="test-model"):
|
|
"""Simulate what a _model_flow_* function writes to disk."""
|
|
cfg = load_config()
|
|
m = cfg.get("model")
|
|
if not isinstance(m, dict):
|
|
m = {"default": m} if m else {}
|
|
cfg["model"] = m
|
|
m["provider"] = provider
|
|
if base_url:
|
|
m["base_url"] = base_url
|
|
if model_name:
|
|
m["default"] = model_name
|
|
save_config(cfg)
|
|
|
|
|
|
def test_setup_delegates_to_select_provider_and_model(tmp_path, monkeypatch):
|
|
"""setup_model_provider calls select_provider_and_model and syncs config."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
|
|
def fake_select():
|
|
_write_model_config(tmp_path, "custom", "http://localhost:11434/v1", "qwen3.5:32b")
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert isinstance(reloaded["model"], dict)
|
|
assert reloaded["model"]["provider"] == "custom"
|
|
assert reloaded["model"]["base_url"] == "http://localhost:11434/v1"
|
|
assert reloaded["model"]["default"] == "qwen3.5:32b"
|
|
|
|
|
|
def test_setup_syncs_openrouter_from_disk(tmp_path, monkeypatch):
|
|
"""When select_provider_and_model saves OpenRouter config to disk,
|
|
the wizard's config dict picks it up."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
assert isinstance(config.get("model"), str) # fresh install
|
|
|
|
def fake_select():
|
|
_write_model_config(tmp_path, "openrouter", model_name="anthropic/claude-opus-4.6")
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert isinstance(reloaded["model"], dict)
|
|
assert reloaded["model"]["provider"] == "openrouter"
|
|
|
|
|
|
def test_setup_syncs_nous_from_disk(tmp_path, monkeypatch):
|
|
"""Nous OAuth writes config to disk; wizard config dict must pick it up."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
|
|
def fake_select():
|
|
_write_model_config(tmp_path, "nous", "https://inference.example.com/v1", "gemini-3-flash")
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert isinstance(reloaded["model"], dict)
|
|
assert reloaded["model"]["provider"] == "nous"
|
|
assert reloaded["model"]["base_url"] == "https://inference.example.com/v1"
|
|
|
|
|
|
def test_setup_custom_providers_synced(tmp_path, monkeypatch):
|
|
"""custom_providers written by select_provider_and_model must survive."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
|
|
def fake_select():
|
|
_write_model_config(tmp_path, "custom", "http://localhost:8080/v1", "llama3")
|
|
cfg = load_config()
|
|
cfg["custom_providers"] = [{"name": "Local", "base_url": "http://localhost:8080/v1"}]
|
|
save_config(cfg)
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert reloaded.get("custom_providers") == [{"name": "Local", "base_url": "http://localhost:8080/v1"}]
|
|
|
|
|
|
def test_setup_gateway_skips_service_install_when_systemctl_missing(monkeypatch, capsys):
|
|
env = {
|
|
"TELEGRAM_BOT_TOKEN": "",
|
|
"TELEGRAM_HOME_CHANNEL": "",
|
|
"DISCORD_BOT_TOKEN": "",
|
|
"DISCORD_HOME_CHANNEL": "",
|
|
"SLACK_BOT_TOKEN": "",
|
|
"SLACK_HOME_CHANNEL": "",
|
|
"MATRIX_HOMESERVER": "https://matrix.example.com",
|
|
"MATRIX_USER_ID": "@alice:example.com",
|
|
"MATRIX_PASSWORD": "",
|
|
"MATRIX_ACCESS_TOKEN": "token",
|
|
"BLUEBUBBLES_SERVER_URL": "",
|
|
"BLUEBUBBLES_HOME_CHANNEL": "",
|
|
"WHATSAPP_ENABLED": "",
|
|
"WEBHOOK_ENABLED": "",
|
|
}
|
|
|
|
import hermes_cli.gateway as gateway_mod
|
|
|
|
monkeypatch.setattr(setup_mod, "get_env_value", lambda key: env.get(key, ""))
|
|
monkeypatch.setattr(gateway_mod, "get_env_value", lambda key: env.get(key, ""))
|
|
monkeypatch.setattr(setup_mod, "prompt_yes_no", lambda *args, **kwargs: False)
|
|
monkeypatch.setattr("platform.system", lambda: "Linux")
|
|
|
|
monkeypatch.setattr(gateway_mod, "supports_systemd_services", lambda: False)
|
|
monkeypatch.setattr(gateway_mod, "is_macos", lambda: False)
|
|
monkeypatch.setattr(gateway_mod, "_is_service_installed", lambda: False)
|
|
monkeypatch.setattr(gateway_mod, "_is_service_running", lambda: False)
|
|
|
|
setup_mod.setup_gateway({})
|
|
|
|
out = capsys.readouterr().out
|
|
assert "Messaging platforms configured!" in out
|
|
assert "Start the gateway to bring your bots online:" in out
|
|
assert "hermes gateway" in out
|
|
|
|
|
|
def test_setup_gateway_in_container_shows_docker_guidance(monkeypatch, capsys):
|
|
"""setup_gateway() in a Docker container shows Docker-specific restart instructions."""
|
|
env = {
|
|
"TELEGRAM_BOT_TOKEN": "",
|
|
"TELEGRAM_HOME_CHANNEL": "",
|
|
"DISCORD_BOT_TOKEN": "",
|
|
"DISCORD_HOME_CHANNEL": "",
|
|
"SLACK_BOT_TOKEN": "",
|
|
"SLACK_HOME_CHANNEL": "",
|
|
"MATRIX_HOMESERVER": "https://matrix.example.com",
|
|
"MATRIX_USER_ID": "@alice:example.com",
|
|
"MATRIX_PASSWORD": "",
|
|
"MATRIX_ACCESS_TOKEN": "token",
|
|
"BLUEBUBBLES_SERVER_URL": "",
|
|
"BLUEBUBBLES_HOME_CHANNEL": "",
|
|
"WHATSAPP_ENABLED": "",
|
|
"WEBHOOK_ENABLED": "",
|
|
}
|
|
|
|
import hermes_cli.gateway as gateway_mod
|
|
|
|
monkeypatch.setattr(setup_mod, "get_env_value", lambda key: env.get(key, ""))
|
|
monkeypatch.setattr(gateway_mod, "get_env_value", lambda key: env.get(key, ""))
|
|
monkeypatch.setattr(setup_mod, "prompt_yes_no", lambda *args, **kwargs: False)
|
|
monkeypatch.setattr("platform.system", lambda: "Linux")
|
|
|
|
monkeypatch.setattr(gateway_mod, "supports_systemd_services", lambda: False)
|
|
monkeypatch.setattr(gateway_mod, "is_macos", lambda: False)
|
|
monkeypatch.setattr(gateway_mod, "_is_service_installed", lambda: False)
|
|
monkeypatch.setattr(gateway_mod, "_is_service_running", lambda: False)
|
|
|
|
# Patch is_container at the import location in setup.py
|
|
import hermes_constants
|
|
monkeypatch.setattr(hermes_constants, "is_container", lambda: True)
|
|
|
|
setup_mod.setup_gateway({})
|
|
|
|
out = capsys.readouterr().out
|
|
assert "Messaging platforms configured!" in out
|
|
assert "docker" in out.lower() or "Docker" in out
|
|
assert "restart" in out.lower()
|
|
|
|
|
|
def test_setup_syncs_custom_provider_removal_from_disk(tmp_path, monkeypatch):
|
|
"""Removing the last custom provider in model setup should persist."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
config["custom_providers"] = [{"name": "Local", "base_url": "http://localhost:8080/v1"}]
|
|
save_config(config)
|
|
|
|
def fake_select():
|
|
cfg = load_config()
|
|
cfg["model"] = {"provider": "openrouter", "default": "anthropic/claude-opus-4.6"}
|
|
cfg["custom_providers"] = []
|
|
save_config(cfg)
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert reloaded.get("custom_providers") == []
|
|
|
|
|
|
def test_setup_cancel_preserves_existing_config(tmp_path, monkeypatch):
|
|
"""When the user cancels provider selection, existing config is preserved."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
# Pre-set a provider
|
|
_write_model_config(tmp_path, "openrouter", model_name="gpt-4o")
|
|
|
|
config = load_config()
|
|
assert config["model"]["provider"] == "openrouter"
|
|
|
|
def fake_select():
|
|
pass # user cancelled — nothing written to disk
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert isinstance(reloaded["model"], dict)
|
|
assert reloaded["model"]["provider"] == "openrouter"
|
|
assert reloaded["model"]["default"] == "gpt-4o"
|
|
|
|
|
|
def test_setup_exception_in_select_gracefully_handled(tmp_path, monkeypatch):
|
|
"""If select_provider_and_model raises, setup continues with existing config."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
|
|
def fake_select():
|
|
raise RuntimeError("something broke")
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
# Should not raise
|
|
setup_model_provider(config)
|
|
|
|
|
|
def test_setup_keyboard_interrupt_gracefully_handled(tmp_path, monkeypatch):
|
|
"""KeyboardInterrupt during provider selection is handled."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
_stub_tts(monkeypatch)
|
|
|
|
config = load_config()
|
|
|
|
def fake_select():
|
|
raise KeyboardInterrupt()
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
|
|
|
|
def test_select_provider_and_model_warns_if_named_custom_provider_disappears(
|
|
tmp_path, monkeypatch, capsys
|
|
):
|
|
"""If a saved custom provider is deleted mid-selection, show a warning instead of silently doing nothing."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
|
|
cfg = load_config()
|
|
cfg["custom_providers"] = [{"name": "Local", "base_url": "http://localhost:8080/v1"}]
|
|
save_config(cfg)
|
|
|
|
def fake_prompt_provider_choice(choices, default=0):
|
|
current = load_config()
|
|
current["custom_providers"] = []
|
|
save_config(current)
|
|
return next(i for i, label in enumerate(choices) if label.startswith("Local (localhost:8080/v1)"))
|
|
|
|
monkeypatch.setattr("hermes_cli.auth.resolve_provider", lambda provider: None)
|
|
monkeypatch.setattr("hermes_cli.main._prompt_provider_choice", fake_prompt_provider_choice)
|
|
monkeypatch.setattr(
|
|
"hermes_cli.main._model_flow_named_custom",
|
|
lambda *args, **kwargs: (_ for _ in ()).throw(AssertionError("named custom flow should not run")),
|
|
)
|
|
|
|
from hermes_cli.main import select_provider_and_model
|
|
|
|
select_provider_and_model()
|
|
|
|
out = capsys.readouterr().out
|
|
assert "selected saved custom provider is no longer available" in out
|
|
|
|
|
|
def test_select_provider_and_model_accepts_named_provider_from_providers_section(
|
|
tmp_path, monkeypatch, capsys
|
|
):
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_provider_env(monkeypatch)
|
|
|
|
cfg = load_config()
|
|
cfg["model"] = {
|
|
"provider": "volcengine-plan",
|
|
"default": "doubao-seed-2.0-code",
|
|
}
|
|
cfg["providers"] = {
|
|
"volcengine-plan": {
|
|
"name": "volcengine-plan",
|
|
"base_url": "https://ark.cn-beijing.volces.com/api/coding/v3",
|
|
"default_model": "doubao-seed-2.0-code",
|
|
"models": {"doubao-seed-2.0-code": {}},
|
|
}
|
|
}
|
|
save_config(cfg)
|
|
|
|
monkeypatch.setattr(
|
|
"hermes_cli.main._prompt_provider_choice",
|
|
lambda choices, default=0: len(choices) - 1,
|
|
)
|
|
|
|
from hermes_cli.main import select_provider_and_model
|
|
|
|
select_provider_and_model()
|
|
|
|
out = capsys.readouterr().out
|
|
assert "Warning: Unknown provider 'volcengine-plan'" not in out
|
|
assert "Active provider: volcengine-plan" in out
|
|
|
|
|
|
def test_codex_setup_uses_runtime_access_token_for_live_model_list(tmp_path, monkeypatch):
|
|
"""Codex model list fetching uses the runtime access token."""
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
monkeypatch.setenv("OPENROUTER_API_KEY", "or-test-key")
|
|
_clear_provider_env(monkeypatch)
|
|
monkeypatch.setenv("OPENROUTER_API_KEY", "or-test-key")
|
|
|
|
config = load_config()
|
|
_stub_tts(monkeypatch)
|
|
|
|
def fake_select():
|
|
_write_model_config(tmp_path, "openai-codex", "https://api.openai.com/v1", "gpt-4o")
|
|
|
|
monkeypatch.setattr("hermes_cli.main.select_provider_and_model", fake_select)
|
|
|
|
setup_model_provider(config)
|
|
save_config(config)
|
|
|
|
reloaded = load_config()
|
|
assert isinstance(reloaded["model"], dict)
|
|
assert reloaded["model"]["provider"] == "openai-codex"
|
|
|
|
|
|
def test_modal_setup_can_use_nous_subscription_without_modal_creds(tmp_path, monkeypatch, capsys):
|
|
monkeypatch.setattr("hermes_cli.setup.managed_nous_tools_enabled", lambda: True)
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
config = load_config()
|
|
|
|
def fake_prompt_choice(question, choices, default=0):
|
|
if question == "Select terminal backend:":
|
|
return 2
|
|
if question == "Select how Modal execution should be billed:":
|
|
return 0
|
|
raise AssertionError(f"Unexpected prompt_choice call: {question}")
|
|
|
|
def fake_prompt(message, *args, **kwargs):
|
|
assert "Modal Token" not in message
|
|
raise AssertionError(f"Unexpected prompt call: {message}")
|
|
|
|
monkeypatch.setattr("hermes_cli.setup.prompt_choice", fake_prompt_choice)
|
|
monkeypatch.setattr("hermes_cli.setup.prompt", fake_prompt)
|
|
monkeypatch.setattr("hermes_cli.setup._prompt_container_resources", lambda config: None)
|
|
monkeypatch.setattr(
|
|
"hermes_cli.setup.get_nous_subscription_features",
|
|
lambda config: type("Features", (), {"nous_auth_present": True})(),
|
|
)
|
|
monkeypatch.setitem(
|
|
sys.modules,
|
|
"tools.managed_tool_gateway",
|
|
types.SimpleNamespace(
|
|
is_managed_tool_gateway_ready=lambda vendor: vendor == "modal",
|
|
resolve_managed_tool_gateway=lambda vendor: None,
|
|
),
|
|
)
|
|
|
|
from hermes_cli.setup import setup_terminal_backend
|
|
|
|
setup_terminal_backend(config)
|
|
|
|
out = capsys.readouterr().out
|
|
assert config["terminal"]["backend"] == "modal"
|
|
assert config["terminal"]["modal_mode"] == "managed"
|
|
assert "bill to your subscription" in out
|
|
|
|
|
|
def test_modal_setup_persists_direct_mode_when_user_chooses_their_own_account(tmp_path, monkeypatch):
|
|
monkeypatch.setattr("hermes_cli.setup.managed_nous_tools_enabled", lambda: True)
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
monkeypatch.delenv("MODAL_TOKEN_ID", raising=False)
|
|
monkeypatch.delenv("MODAL_TOKEN_SECRET", raising=False)
|
|
config = load_config()
|
|
|
|
def fake_prompt_choice(question, choices, default=0):
|
|
if question == "Select terminal backend:":
|
|
return 2
|
|
if question == "Select how Modal execution should be billed:":
|
|
return 1
|
|
raise AssertionError(f"Unexpected prompt_choice call: {question}")
|
|
|
|
prompt_values = iter(["token-id", "token-secret", ""])
|
|
|
|
monkeypatch.setattr("hermes_cli.setup.prompt_choice", fake_prompt_choice)
|
|
monkeypatch.setattr("hermes_cli.setup.prompt", lambda *args, **kwargs: next(prompt_values))
|
|
monkeypatch.setattr("hermes_cli.setup._prompt_container_resources", lambda config: None)
|
|
monkeypatch.setattr(
|
|
"hermes_cli.setup.get_nous_subscription_features",
|
|
lambda config: type("Features", (), {"nous_auth_present": True})(),
|
|
)
|
|
monkeypatch.setitem(
|
|
sys.modules,
|
|
"tools.managed_tool_gateway",
|
|
types.SimpleNamespace(
|
|
is_managed_tool_gateway_ready=lambda vendor: vendor == "modal",
|
|
resolve_managed_tool_gateway=lambda vendor: None,
|
|
),
|
|
)
|
|
monkeypatch.setitem(sys.modules, "swe_rex", object())
|
|
|
|
from hermes_cli.setup import setup_terminal_backend
|
|
|
|
setup_terminal_backend(config)
|
|
|
|
assert config["terminal"]["backend"] == "modal"
|
|
assert config["terminal"]["modal_mode"] == "direct"
|
|
|
|
|
|
def test_vercel_setup_configures_access_token_auth(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_vercel_env(monkeypatch)
|
|
monkeypatch.setenv("VERCEL_OIDC_TOKEN", "old-oidc")
|
|
monkeypatch.setitem(sys.modules, "vercel", types.ModuleType("vercel"))
|
|
config = load_config()
|
|
|
|
def fake_prompt_choice(question, choices, default=0):
|
|
if question == "Select terminal backend:":
|
|
return 5
|
|
raise AssertionError(f"Unexpected prompt_choice call: {question}")
|
|
|
|
prompt_values = iter(["python3.13", "yes", "2", "4096", "token", "project", "team"])
|
|
|
|
monkeypatch.setattr("hermes_cli.setup.prompt_choice", fake_prompt_choice)
|
|
monkeypatch.setattr("hermes_cli.setup.prompt", lambda *args, **kwargs: next(prompt_values))
|
|
|
|
from hermes_cli.setup import setup_terminal_backend
|
|
|
|
setup_terminal_backend(config)
|
|
|
|
assert config["terminal"]["backend"] == "vercel_sandbox"
|
|
assert config["terminal"]["vercel_runtime"] == "python3.13"
|
|
assert config["terminal"]["container_disk"] == 51200
|
|
assert os.environ["TERMINAL_VERCEL_RUNTIME"] == "python3.13"
|
|
assert "VERCEL_OIDC_TOKEN" not in os.environ
|
|
assert os.environ["VERCEL_TOKEN"] == "token"
|
|
assert os.environ["VERCEL_PROJECT_ID"] == "project"
|
|
assert os.environ["VERCEL_TEAM_ID"] == "team"
|
|
|
|
|
|
def test_vercel_setup_prefills_project_and_team_from_link_file(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
_clear_vercel_env(monkeypatch)
|
|
project_root = tmp_path / "project"
|
|
nested = project_root / "app" / "src"
|
|
nested.mkdir(parents=True)
|
|
vercel_dir = project_root / ".vercel"
|
|
vercel_dir.mkdir()
|
|
(vercel_dir / "project.json").write_text(
|
|
json.dumps({"projectId": "linked-project", "orgId": "linked-team"}),
|
|
encoding="utf-8",
|
|
)
|
|
monkeypatch.chdir(nested)
|
|
monkeypatch.setitem(sys.modules, "vercel", types.ModuleType("vercel"))
|
|
config = load_config()
|
|
config["terminal"]["container_disk"] = 999
|
|
|
|
def fake_prompt_choice(question, choices, default=0):
|
|
if question == "Select terminal backend:":
|
|
return 5
|
|
raise AssertionError(f"Unexpected prompt_choice call: {question}")
|
|
|
|
prompt_values = iter(["node24", "no", "1", "5120", "token", "", ""])
|
|
defaults = {}
|
|
|
|
def fake_prompt(message, default="", **kwargs):
|
|
defaults[message] = default
|
|
value = next(prompt_values)
|
|
return value or default
|
|
|
|
monkeypatch.setattr("hermes_cli.setup.prompt_choice", fake_prompt_choice)
|
|
monkeypatch.setattr("hermes_cli.setup.prompt", fake_prompt)
|
|
|
|
from hermes_cli.setup import setup_terminal_backend
|
|
|
|
setup_terminal_backend(config)
|
|
|
|
assert config["terminal"]["backend"] == "vercel_sandbox"
|
|
assert config["terminal"]["container_persistent"] is False
|
|
assert config["terminal"]["container_disk"] == 51200
|
|
assert "VERCEL_OIDC_TOKEN" not in os.environ
|
|
assert os.environ["VERCEL_TOKEN"] == "token"
|
|
assert os.environ["VERCEL_PROJECT_ID"] == "linked-project"
|
|
assert os.environ["VERCEL_TEAM_ID"] == "linked-team"
|
|
assert defaults[" Vercel project ID"] == "linked-project"
|
|
assert defaults[" Vercel team ID"] == "linked-team"
|
|
|
|
|
|
def test_offer_launch_chat_relaunches_via_bin(monkeypatch):
|
|
from hermes_cli import setup as setup_mod
|
|
from hermes_cli import relaunch as relaunch_mod
|
|
|
|
monkeypatch.setattr(setup_mod, "prompt_yes_no", lambda *_args, **_kwargs: True)
|
|
monkeypatch.setattr(relaunch_mod, "resolve_hermes_bin", lambda: "/usr/local/bin/hermes")
|
|
|
|
exec_calls = []
|
|
|
|
def fake_execvp(path, argv):
|
|
exec_calls.append((path, argv))
|
|
raise SystemExit(0)
|
|
|
|
monkeypatch.setattr(relaunch_mod.os, "execvp", fake_execvp)
|
|
|
|
with pytest.raises(SystemExit):
|
|
setup_mod._offer_launch_chat()
|
|
|
|
assert exec_calls == [("/usr/local/bin/hermes", ["/usr/local/bin/hermes", "chat"])]
|
|
|
|
|
|
def test_offer_launch_chat_falls_back_to_module(monkeypatch):
|
|
from hermes_cli import setup as setup_mod
|
|
from hermes_cli import relaunch as relaunch_mod
|
|
|
|
monkeypatch.setattr(setup_mod, "prompt_yes_no", lambda *_args, **_kwargs: True)
|
|
monkeypatch.setattr(relaunch_mod, "resolve_hermes_bin", lambda: None)
|
|
|
|
exec_calls = []
|
|
|
|
def fake_execvp(path, argv):
|
|
exec_calls.append((path, argv))
|
|
raise SystemExit(0)
|
|
|
|
monkeypatch.setattr(relaunch_mod.os, "execvp", fake_execvp)
|
|
|
|
with pytest.raises(SystemExit):
|
|
setup_mod._offer_launch_chat()
|
|
|
|
assert exec_calls == [(sys.executable, [sys.executable, "-m", "hermes_cli.main", "chat"])]
|
|
|
|
|
|
def test_setup_slack_saves_home_channel(monkeypatch):
|
|
"""_setup_slack() saves SLACK_HOME_CHANNEL when the user provides one."""
|
|
saved = {}
|
|
prompts = iter(["xoxb-test-token", "xapp-test-token", "", "C01ABC2DE3F"])
|
|
|
|
monkeypatch.setattr(setup_mod, "get_env_value", lambda key: "")
|
|
monkeypatch.setattr(setup_mod, "save_env_value", lambda k, v: saved.update({k: v}))
|
|
monkeypatch.setattr(setup_mod, "prompt", lambda *_a, **_kw: next(prompts))
|
|
monkeypatch.setattr(setup_mod, "prompt_yes_no", lambda *_a, **_kw: False)
|
|
monkeypatch.setattr(setup_mod, "_write_slack_manifest_and_instruct", lambda: None)
|
|
|
|
setup_mod._setup_slack()
|
|
|
|
assert saved.get("SLACK_HOME_CHANNEL") == "C01ABC2DE3F"
|
|
|
|
|
|
def test_setup_slack_home_channel_empty_not_saved(monkeypatch):
|
|
"""_setup_slack() does not save SLACK_HOME_CHANNEL when left blank."""
|
|
saved = {}
|
|
prompts = iter(["xoxb-test-token", "xapp-test-token", "", ""])
|
|
|
|
monkeypatch.setattr(setup_mod, "get_env_value", lambda key: "")
|
|
monkeypatch.setattr(setup_mod, "save_env_value", lambda k, v: saved.update({k: v}))
|
|
monkeypatch.setattr(setup_mod, "prompt", lambda *_a, **_kw: next(prompts))
|
|
monkeypatch.setattr(setup_mod, "prompt_yes_no", lambda *_a, **_kw: False)
|
|
monkeypatch.setattr(setup_mod, "_write_slack_manifest_and_instruct", lambda: None)
|
|
|
|
setup_mod._setup_slack()
|
|
|
|
assert "SLACK_HOME_CHANNEL" not in saved
|