136 lines
4.0 KiB
Python
136 lines
4.0 KiB
Python
"""Unit tests for gateway.runtime_footer — the opt-in runtime-metadata footer
|
|
appended to final gateway replies."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from gateway.runtime_footer import (
|
|
_home_relative_cwd,
|
|
_model_short,
|
|
build_footer_line,
|
|
format_runtime_footer,
|
|
resolve_footer_config,
|
|
)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _model_short + _home_relative_cwd
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@pytest.mark.parametrize(
|
|
"model,expected",
|
|
[
|
|
("openai/gpt-5.4", "gpt-5.4"),
|
|
("anthropic/claude-sonnet-4.6", "claude-sonnet-4.6"),
|
|
("gpt-5.4", "gpt-5.4"),
|
|
("", ""),
|
|
(None, ""),
|
|
],
|
|
)
|
|
def test_model_short_drops_vendor_prefix(model, expected):
|
|
assert _model_short(model) == expected
|
|
|
|
|
|
def test_home_relative_cwd_collapses_home(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("HOME", str(tmp_path))
|
|
sub = tmp_path / "projects" / "hermes"
|
|
sub.mkdir(parents=True)
|
|
result = _home_relative_cwd(str(sub))
|
|
assert result == "~/projects/hermes"
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# format_runtime_footer
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_format_footer_all_fields(monkeypatch, tmp_path):
|
|
monkeypatch.setenv("HOME", str(tmp_path))
|
|
monkeypatch.setenv("TERMINAL_CWD", str(tmp_path / "projects" / "hermes"))
|
|
(tmp_path / "projects" / "hermes").mkdir(parents=True)
|
|
out = format_runtime_footer(
|
|
model="openrouter/openai/gpt-5.4",
|
|
context_tokens=68000,
|
|
context_length=100000,
|
|
cwd=None, # falls back to TERMINAL_CWD env var
|
|
fields=("model", "context_pct", "cwd"),
|
|
)
|
|
assert out == "gpt-5.4 · 68% · ~/projects/hermes"
|
|
|
|
|
|
def test_format_footer_skips_missing_context_length():
|
|
out = format_runtime_footer(
|
|
model="openai/gpt-5.4",
|
|
context_tokens=500,
|
|
context_length=None,
|
|
cwd="/tmp/wd",
|
|
fields=("model", "context_pct", "cwd"),
|
|
)
|
|
# context_pct dropped silently; no "?%" artifact
|
|
assert "%" not in out
|
|
assert "gpt-5.4" in out
|
|
assert "/tmp/wd" in out
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# resolve_footer_config
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_resolve_platform_override_wins():
|
|
user = {
|
|
"display": {
|
|
"runtime_footer": {"enabled": True, "fields": ["model"]},
|
|
"platforms": {
|
|
"slack": {"runtime_footer": {"enabled": False}},
|
|
},
|
|
},
|
|
}
|
|
# Telegram picks up the global enable
|
|
assert resolve_footer_config(user, "telegram")["enabled"] is True
|
|
# Slack overrides to off
|
|
assert resolve_footer_config(user, "slack")["enabled"] is False
|
|
|
|
|
|
def test_resolve_platform_can_add_fields_only():
|
|
user = {
|
|
"display": {
|
|
"runtime_footer": {"enabled": True},
|
|
"platforms": {
|
|
"discord": {"runtime_footer": {"fields": ["context_pct"]}},
|
|
},
|
|
},
|
|
}
|
|
tg = resolve_footer_config(user, "telegram")
|
|
assert tg["enabled"] is True
|
|
assert tg["fields"] == ["model", "context_pct", "cwd"]
|
|
dc = resolve_footer_config(user, "discord")
|
|
assert dc["enabled"] is True
|
|
assert dc["fields"] == ["context_pct"]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# build_footer_line — top-level entry point used by gateway/run.py
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_build_footer_per_platform_off_suppresses():
|
|
user = {
|
|
"display": {
|
|
"runtime_footer": {"enabled": True},
|
|
"platforms": {"slack": {"runtime_footer": {"enabled": False}}},
|
|
},
|
|
}
|
|
out = build_footer_line(
|
|
user_config=user,
|
|
platform_key="slack",
|
|
model="openai/gpt-5.4",
|
|
context_tokens=10, context_length=100,
|
|
cwd="/tmp",
|
|
)
|
|
assert out == ""
|
|
|
|
|