mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-23 16:36:23 +00:00
35e0f56fbf
Raise the Matrix adapter default chunk size from 4,000 to 16,000 characters and allow overrides via config.yaml or MATRIX_MAX_MESSAGE_LENGTH. Fixes #53026
85 lines
3 KiB
Python
85 lines
3 KiB
Python
"""Tests for Matrix outbound message length configuration (#53026)."""
|
|
import asyncio
|
|
import os
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from gateway.config import PlatformConfig
|
|
|
|
|
|
def _make_adapter(**extra):
|
|
from plugins.platforms.matrix.adapter import MatrixAdapter
|
|
|
|
config = PlatformConfig(
|
|
enabled=True,
|
|
token="syt_test_token",
|
|
extra={
|
|
"homeserver": "https://matrix.example.org",
|
|
"user_id": "@bot:example.org",
|
|
**extra,
|
|
},
|
|
)
|
|
return MatrixAdapter(config)
|
|
|
|
|
|
class TestMatrixMaxMessageLength:
|
|
def test_default_limit_is_16000(self):
|
|
adapter = _make_adapter()
|
|
assert adapter.max_message_length == 16000
|
|
assert adapter._split_threshold == 15900
|
|
|
|
def test_extra_override(self):
|
|
adapter = _make_adapter(max_message_length=12000)
|
|
assert adapter.max_message_length == 12000
|
|
assert adapter._split_threshold == 11900
|
|
|
|
def test_env_override(self, monkeypatch):
|
|
monkeypatch.setenv("MATRIX_MAX_MESSAGE_LENGTH", "20000")
|
|
adapter = _make_adapter()
|
|
assert adapter.max_message_length == 20000
|
|
|
|
def test_extra_beats_env(self, monkeypatch):
|
|
monkeypatch.setenv("MATRIX_MAX_MESSAGE_LENGTH", "20000")
|
|
adapter = _make_adapter(max_message_length=10000)
|
|
assert adapter.max_message_length == 10000
|
|
|
|
def test_invalid_values_fall_back_to_default(self, monkeypatch):
|
|
monkeypatch.setenv("MATRIX_MAX_MESSAGE_LENGTH", "not-a-number")
|
|
adapter = _make_adapter()
|
|
assert adapter.max_message_length == 16000
|
|
|
|
def test_values_are_clamped(self):
|
|
adapter = _make_adapter(max_message_length=100)
|
|
assert adapter.max_message_length == 500
|
|
adapter = _make_adapter(max_message_length=999999)
|
|
assert adapter.max_message_length == 65535
|
|
|
|
def test_apply_yaml_config_sets_env(self, monkeypatch):
|
|
from plugins.platforms.matrix.adapter import _apply_yaml_config
|
|
|
|
monkeypatch.delenv("MATRIX_MAX_MESSAGE_LENGTH", raising=False)
|
|
_apply_yaml_config({}, {"max_message_length": 12000})
|
|
assert os.getenv("MATRIX_MAX_MESSAGE_LENGTH") == "12000"
|
|
|
|
def test_register_uses_default_limit(self):
|
|
from plugins.platforms.matrix.adapter import DEFAULT_MAX_MESSAGE_LENGTH, register
|
|
|
|
ctx = MagicMock()
|
|
register(ctx)
|
|
kwargs = ctx.register_platform.call_args[1]
|
|
assert kwargs["max_message_length"] == DEFAULT_MAX_MESSAGE_LENGTH
|
|
|
|
def test_send_uses_configured_limit(self):
|
|
adapter = _make_adapter(max_message_length=5000)
|
|
adapter._client = MagicMock()
|
|
adapter._client.send_message_event = AsyncMock(return_value="evt")
|
|
long_text = "x" * 12000
|
|
|
|
async def _run():
|
|
with patch.object(adapter, "truncate_message", wraps=adapter.truncate_message) as trunc:
|
|
await adapter.send("!room:example.org", long_text)
|
|
trunc.assert_called_once()
|
|
assert trunc.call_args[0][1] == 5000
|
|
|
|
asyncio.run(_run())
|