test_consensus_schema.py•822 B
"""Schema-related tests for ConsensusTool."""
from types import MethodType
from tools.consensus import ConsensusTool
def test_consensus_models_field_includes_available_models(monkeypatch):
"""Consensus schema should surface available model guidance like single-model tools."""
tool = ConsensusTool()
monkeypatch.setattr(
tool,
"_get_ranked_model_summaries",
MethodType(lambda self, limit=5: (["gemini-2.5-pro (score 100, 1.0M ctx, thinking)"], 1, False), tool),
)
monkeypatch.setattr(tool, "_get_restriction_note", MethodType(lambda self: None, tool))
schema = tool.get_input_schema()
models_field_description = schema["properties"]["models"]["description"]
assert "listmodels" in models_field_description
assert "Top models" in models_field_description
