mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-16 09:31:37 +00:00
112 lines
4 KiB
Python
112 lines
4 KiB
Python
"""Tests for `hermes photon setup`'s access auto-configuration.
|
|
|
|
`_autoconfigure_access` allowlists the operator and points the cron home
|
|
channel at their DM, writing to the per-test ~/.hermes/.env (the hermetic
|
|
HERMES_HOME fixture isolates this). It must fill only unset keys so a re-run
|
|
never clobbers a hand-tuned allowlist.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
|
|
import pytest
|
|
|
|
from hermes_cli.config import get_env_value, save_env_value
|
|
from plugins.platforms.photon.adapter import _env_enablement
|
|
from plugins.platforms.photon import cli
|
|
|
|
|
|
def test_autoconfigure_access_fills_unset(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.delenv("PHOTON_ALLOWED_USERS", raising=False)
|
|
monkeypatch.delenv("PHOTON_HOME_CHANNEL", raising=False)
|
|
|
|
cli._autoconfigure_access("+15551234567")
|
|
|
|
assert get_env_value("PHOTON_ALLOWED_USERS") == "+15551234567"
|
|
assert get_env_value("PHOTON_HOME_CHANNEL") == "+15551234567"
|
|
|
|
|
|
def test_autoconfigure_access_preserves_existing_allowlist(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.delenv("PHOTON_ALLOWED_USERS", raising=False)
|
|
monkeypatch.delenv("PHOTON_HOME_CHANNEL", raising=False)
|
|
# A hand-tuned allowlist already in place must survive a setup re-run.
|
|
save_env_value("PHOTON_ALLOWED_USERS", "+19998887777,+15551112222")
|
|
|
|
cli._autoconfigure_access("+15551234567")
|
|
|
|
assert get_env_value("PHOTON_ALLOWED_USERS") == "+19998887777,+15551112222"
|
|
# The still-unset home channel is filled.
|
|
assert get_env_value("PHOTON_HOME_CHANNEL") == "+15551234567"
|
|
|
|
|
|
def test_env_enablement_seeds_home_channel(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("PHOTON_PROJECT_ID", "project_123")
|
|
monkeypatch.setenv("PHOTON_PROJECT_SECRET", "secret_123")
|
|
monkeypatch.setenv("PHOTON_HOME_CHANNEL", "+15551234567")
|
|
monkeypatch.setenv("PHOTON_HOME_CHANNEL_NAME", "Primary DM")
|
|
|
|
seed = _env_enablement()
|
|
|
|
assert seed is not None
|
|
assert seed["home_channel"] == {
|
|
"chat_id": "+15551234567",
|
|
"name": "Primary DM",
|
|
}
|
|
|
|
|
|
def test_env_enablement_home_channel_defaults_name(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("PHOTON_PROJECT_ID", "project_123")
|
|
monkeypatch.setenv("PHOTON_PROJECT_SECRET", "secret_123")
|
|
monkeypatch.setenv("PHOTON_HOME_CHANNEL", "+15551234567")
|
|
monkeypatch.delenv("PHOTON_HOME_CHANNEL_NAME", raising=False)
|
|
|
|
seed = _env_enablement()
|
|
|
|
assert seed is not None
|
|
assert seed["home_channel"] == {
|
|
"chat_id": "+15551234567",
|
|
"name": "Home",
|
|
}
|
|
|
|
|
|
def test_setup_hint_uses_gateway_service_command(monkeypatch: pytest.MonkeyPatch, capsys) -> None:
|
|
monkeypatch.setattr(cli.photon_auth, "load_photon_token", lambda: "token")
|
|
monkeypatch.setattr(cli.photon_auth, "load_dashboard_project_id", lambda: "dashboard")
|
|
monkeypatch.setattr(
|
|
cli.photon_auth,
|
|
"ensure_spectrum_enabled",
|
|
lambda token, dashboard_id: {"spectrumProjectId": "project_123"},
|
|
)
|
|
monkeypatch.setattr(
|
|
cli.photon_auth,
|
|
"regenerate_project_secret",
|
|
lambda token, dashboard_id: "secret_123",
|
|
)
|
|
monkeypatch.setattr(cli.photon_auth, "store_project_credentials", lambda **kwargs: None)
|
|
monkeypatch.setattr(
|
|
cli.photon_auth,
|
|
"register_user_if_absent",
|
|
lambda *args, **kwargs: ({"id": "user_123", "phoneNumber": "+15551234567"}, True),
|
|
)
|
|
monkeypatch.setattr(cli.photon_auth, "user_assigned_line", lambda user: "+15557654321")
|
|
monkeypatch.setattr(cli.photon_auth, "store_user_numbers", lambda **kwargs: None)
|
|
monkeypatch.setattr(cli, "_install_sidecar", lambda: 0)
|
|
|
|
rc = cli._cmd_setup(
|
|
argparse.Namespace(
|
|
project_name=None,
|
|
phone="+15551234567",
|
|
first_name=None,
|
|
last_name=None,
|
|
email=None,
|
|
no_browser=True,
|
|
skip_sidecar_install=False,
|
|
)
|
|
)
|
|
|
|
assert rc == 0
|
|
out = capsys.readouterr().out
|
|
assert "Start the gateway: hermes gateway start" in out
|
|
assert "--platform photon" not in out
|