Teknium
6b81590c55
Systematic prune per AGENTS.md test policy, one pass over every major test tree (gateway, hermes_cli, tools, agent, run_agent, plugins, cli, cron, tui_gateway, honcho/openviking, root-level): - DELETE: source-reading tests (read_text/getsource on prod files), change-detector tests (exact catalog counts, model-name snapshots, config version literals), mock-echo tests (assert a mock returns what it was told), assertion-free/trivial tests, near-duplicate parametrizations (boundaries + one representative kept), async/sync twin duplicates, cosmetic within-file variations. - KEEP (mandatory): security/redaction/approval guards, message-role alternation invariants, prompt-caching/deterministic-call-id invariants, issue-number regression tests (deduped), E2E tests. - 6 test files deleted outright (script-style/no-assert or fully redundant); conftest.py, fakes/, fixtures/ untouched. - tests/acp/conftest.py added: autouse fixture stubs the live models.dev/GitHub/Copilot/Anthropic inventory fetches that ACP server tests performed on every session create — test_server.py 147s → 3.4s, and the tests are now genuinely hermetic. - Sleep-based slowness shrunk where safe (codex_ttfb_watchdog, compression_concurrent_fork, etc.); no wall-clock assertion tightened. Verification: full hermetic suite via scripts/run_tests.sh — 2439 files, 31,130 tests passed, 0 failed, 0 flaky retries, 315s wall (baseline: 583s wall, 13,564s subprocess CPU).
89 lines
3.6 KiB
Python
89 lines
3.6 KiB
Python
"""Tests for live+curated merge in the generic profile-based provider path.
|
|
|
|
Guards two contracts:
|
|
|
|
* #46850 — when a provider's live /v1/models endpoint returns a stale or
|
|
incomplete list, the static curated models from ``_PROVIDER_MODELS`` must
|
|
still appear in the merged result (nothing is dropped).
|
|
* #46309 / #49129 — merge *order* is per-provider. Single providers
|
|
(kimi, zai) stay **curated-first** so a deliberately surfaced newest model
|
|
leads even when the live API lags. ``_LIVE_FIRST_PICKER_PROVIDERS``
|
|
(OpenCode Zen / Go) flip to **live-first** because their live API is the
|
|
authoritative catalog and stale curated entries must not lead the picker.
|
|
"""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from hermes_cli.models import (
|
|
_LIVE_FIRST_PICKER_PROVIDERS,
|
|
provider_model_ids,
|
|
)
|
|
|
|
|
|
class TestGenericProviderLiveCuratedMerge:
|
|
"""provider_model_ids merges live + curated for generic api_key providers."""
|
|
|
|
def _make_profile(self, models=None):
|
|
"""Create a minimal mock provider profile."""
|
|
p = MagicMock()
|
|
p.auth_type = "api_key"
|
|
p.base_url = "https://api.example.com/v1"
|
|
p.fetch_models.return_value = models
|
|
p.fallback_models = None
|
|
return p
|
|
|
|
def test_curated_first_for_single_provider(self):
|
|
"""Single providers (zai) stay curated-first; live-only appended."""
|
|
assert "zai" not in _LIVE_FIRST_PICKER_PROVIDERS
|
|
curated = ["glm-5.2", "glm-5.1", "glm-5"] # authoritative-intent order
|
|
# Live API lags AND surfaces a brand-new model not yet curated.
|
|
live = ["glm-5", "glm-6-preview"]
|
|
profile = self._make_profile(live)
|
|
|
|
with (
|
|
patch("providers.get_provider_profile", return_value=profile),
|
|
patch(
|
|
"hermes_cli.auth.resolve_api_key_provider_credentials",
|
|
return_value={"api_key": "k", "base_url": ""},
|
|
),
|
|
patch.dict("hermes_cli.models._PROVIDER_MODELS", {"zai": curated}),
|
|
):
|
|
result = provider_model_ids("zai")
|
|
|
|
# Curated entries lead (commit 658ac1d86, #46309).
|
|
assert result[: len(curated)] == curated
|
|
# Live-only entries (glm-6-preview) still surface, appended afterwards.
|
|
assert "glm-6-preview" in result
|
|
assert result.index("glm-6-preview") >= len(curated)
|
|
# No duplicates for models present in both.
|
|
assert result.count("glm-5") == 1
|
|
|
|
|
|
def test_no_models_dropped_either_direction(self):
|
|
"""Every live AND curated model survives the merge for both modes."""
|
|
live = ["a", "b"]
|
|
# zai = curated-first
|
|
with (
|
|
patch("providers.get_provider_profile", return_value=self._make_profile(live)),
|
|
patch(
|
|
"hermes_cli.auth.resolve_api_key_provider_credentials",
|
|
return_value={"api_key": "k", "base_url": ""},
|
|
),
|
|
patch.dict("hermes_cli.models._PROVIDER_MODELS", {"zai": ["c", "b"]}),
|
|
):
|
|
zai_result = set(provider_model_ids("zai"))
|
|
assert {"a", "b", "c"} <= zai_result
|
|
|
|
# opencode-zen = live-first
|
|
with (
|
|
patch("providers.get_provider_profile", return_value=self._make_profile(live)),
|
|
patch(
|
|
"hermes_cli.auth.resolve_api_key_provider_credentials",
|
|
return_value={"api_key": "k", "base_url": ""},
|
|
),
|
|
patch.dict("hermes_cli.models._PROVIDER_MODELS", {"opencode-zen": ["c", "b"]}),
|
|
):
|
|
zen_result = set(provider_model_ids("opencode-zen"))
|
|
assert {"a", "b", "c"} <= zen_result
|
|
|