hermes-mcp/tests/test_argus_client.py

from hermes_mcp.argus_client import _payload_from_result


class _Text:
    def __init__(self, text: str):
        self.text = text


class _Result:
    def __init__(self, structured=None, content=None):
        self.structuredContent = structured
        self.content = content or []


def test_payload_from_result_prefers_structured_content():
    result = _Result(structured={"regime": "compression", "confidence": 0.6})
    assert _payload_from_result(result)["regime"] == "compression"


def test_payload_from_result_parses_json_text_content():
    result = _Result(content=[_Text('{"regime":"compression","confidence":0.6,"summary":"range-like"}')])
    payload = _payload_from_result(result)
    assert payload["regime"] == "compression"
    assert payload["summary"] == "range-like"