mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-01 02:04:21 +00:00
b9caae1e50
* feat(service): bind network services to all interfaces * fix(service): keep network listeners local by default * fix(service): make remote access explicit
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
"""Startup logo metadata tests."""
|
|
|
|
from io import StringIO
|
|
from types import SimpleNamespace
|
|
|
|
from rich.console import Console
|
|
|
|
from reme import application as application_module
|
|
from reme.application import Application
|
|
from reme.constants import REME_DEFAULT_HOST, REME_DEFAULT_PORT
|
|
from reme.schema import ApplicationConfig, ComponentConfig
|
|
from reme.utils import logo_utils
|
|
|
|
|
|
def _render_logo(monkeypatch, config: ApplicationConfig, runtime_service=None) -> str:
|
|
output = StringIO()
|
|
console = Console(file=output, force_terminal=False, width=120)
|
|
monkeypatch.setattr(logo_utils, "Console", lambda: console)
|
|
|
|
logo_utils.print_logo(config, runtime_service)
|
|
|
|
return output.getvalue()
|
|
|
|
|
|
def test_logo_uses_runtime_http_address(monkeypatch) -> None:
|
|
"""Display the address resolved by the instantiated service."""
|
|
config = ApplicationConfig(service=ComponentConfig(backend="http"))
|
|
runtime_service = SimpleNamespace(host="0.0.0.0", port=8123)
|
|
|
|
output = _render_logo(monkeypatch, config, runtime_service)
|
|
|
|
assert "http://127.0.0.1:8123" in output
|
|
assert "http://127.0.0.1:8123/mcp" in output
|
|
assert "http://0.0.0.0:8123" not in output
|
|
|
|
|
|
def test_logo_fallback_matches_service_defaults(monkeypatch) -> None:
|
|
"""Keep direct print_logo callers aligned with service defaults."""
|
|
config = ApplicationConfig(service=ComponentConfig(backend="http"))
|
|
|
|
output = _render_logo(monkeypatch, config)
|
|
|
|
assert f"http://{REME_DEFAULT_HOST}:{REME_DEFAULT_PORT}" in output
|
|
assert f"http://{REME_DEFAULT_HOST}:{REME_DEFAULT_PORT}/mcp" in output
|
|
|
|
|
|
def test_logo_hides_disabled_http_mcp_endpoint(monkeypatch) -> None:
|
|
"""Do not advertise MCP when it is explicitly disabled on the HTTP service."""
|
|
config = ApplicationConfig(
|
|
service=ComponentConfig(backend="http", mcp_enabled=False),
|
|
)
|
|
|
|
output = _render_logo(monkeypatch, config)
|
|
|
|
assert f"http://{REME_DEFAULT_HOST}:{REME_DEFAULT_PORT}/mcp" not in output
|
|
|
|
|
|
def test_logo_uses_runtime_mcp_transport_and_address(monkeypatch) -> None:
|
|
"""Display the transport resolved by the instantiated MCP service."""
|
|
config = ApplicationConfig(service=ComponentConfig(backend="mcp"))
|
|
runtime_service = SimpleNamespace(transport="sse", host="0.0.0.0", port=8123)
|
|
|
|
output = _render_logo(monkeypatch, config, runtime_service)
|
|
|
|
assert "Transport: sse" in output
|
|
assert "http://127.0.0.1:8123/sse" in output
|
|
assert "http://0.0.0.0:8123/sse" not in output
|
|
|
|
|
|
def test_logo_mcp_fallback_matches_service_defaults(monkeypatch) -> None:
|
|
"""Keep direct print_logo callers aligned with MCP service defaults."""
|
|
config = ApplicationConfig(service=ComponentConfig(backend="mcp"))
|
|
|
|
output = _render_logo(monkeypatch, config)
|
|
|
|
assert "Transport: sse" in output
|
|
assert f"http://{REME_DEFAULT_HOST}:{REME_DEFAULT_PORT}/sse" in output
|
|
|
|
|
|
def test_application_passes_instantiated_service_to_logo(monkeypatch, tmp_path) -> None:
|
|
"""Render the service address after backend defaults and overrides resolve."""
|
|
captured = {}
|
|
monkeypatch.setattr(Application, "_init_components", lambda self: None)
|
|
monkeypatch.setattr(Application, "_init_jobs", lambda self: None)
|
|
monkeypatch.setattr(
|
|
application_module,
|
|
"print_logo",
|
|
lambda config, runtime_service: captured.update(
|
|
host=runtime_service.host,
|
|
port=runtime_service.port,
|
|
),
|
|
)
|
|
|
|
Application(
|
|
workspace_dir=str(tmp_path),
|
|
service={"backend": "http", "host": "0.0.0.0", "port": 8123},
|
|
log_to_console=False,
|
|
log_to_file=False,
|
|
)
|
|
|
|
assert captured == {"host": "0.0.0.0", "port": 8123}
|