Kris
69fe97a90d
1. 移除Telegram格式化测试中未使用的导入项 2. 修复Teams测试用例，添加monkeypatch参数并配置通配符开关 3. 更新钉钉适配器测试，替换弃用的流属性检查 4. 修正Twitch规范化测试，更新ROOMSTATE测试逻辑 5. 重构会话映射测试，完善数据库执行结果模拟 6. 格式化Slack块构建测试的长参数调用 7. 修复LINE适配器测试，更新能力断言和异步锁使用 8. 修正Slack会话解析测试，修复聊天类型判断错误 9. 更新能力测试，补充缺失的字段检查 10. 修复Matrix适配器测试，修正位置参数和配置校验逻辑 11. 为飞书分析模块测试添加跳过标记 12. 新增微信能力、限流、链接格式、会话路由等模块的单元测试 13. 修复Twitch适配器导入路径和测试断言 14. 新增Discord Webhook、Nextcloud Talk、Signal多账户等模块的单元测试 15. 修复Manager阶段测试的导入路径 16. 新增iMessage异常和命令处理的单元测试 17. 新增Nostr健康检查和相关模块的单元测试 18. 新增Signal守护进程和SSE重连相关测试
262 lines
9.3 KiB
Python
262 lines
9.3 KiB
Python
"""Unit tests for Signal stream buffer, coalescer, and formatting."""
|
|
|
|
import asyncio
|
|
import time
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from yuxi.channels.adapters.signal.streaming import (
|
|
StreamBuffer,
|
|
StreamCoalescer,
|
|
ReasoningLaneFormatter,
|
|
)
|
|
from yuxi.channels.adapters.signal.channel import SignalChannel
|
|
from yuxi.channels.models import DeliveryResult
|
|
|
|
|
|
class TestStreamBuffer:
|
|
def test_set_and_get(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "hello")
|
|
assert buf.get("chat1") == "hello"
|
|
|
|
def test_get_nonexistent_returns_empty(self):
|
|
buf = StreamBuffer()
|
|
assert buf.get("nonexistent") == ""
|
|
|
|
def test_pop_returns_and_removes(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "hello")
|
|
result = buf.pop("chat1")
|
|
assert result == "hello"
|
|
assert buf.get("chat1") == ""
|
|
|
|
def test_pop_nonexistent_returns_none(self):
|
|
buf = StreamBuffer()
|
|
assert buf.pop("nonexistent") is None
|
|
|
|
def test_clear_removes_all(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "a")
|
|
buf.set("chat2", "b")
|
|
buf.clear()
|
|
assert buf.get("chat1") == ""
|
|
assert buf.get("chat2") == ""
|
|
|
|
def test_idle_ms_returns_zero_when_empty(self):
|
|
buf = StreamBuffer()
|
|
assert buf.idle_ms("chat1") == 0
|
|
|
|
def test_idle_ms_returns_positive_after_delay(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "hello")
|
|
time.sleep(0.05)
|
|
ms = buf.idle_ms("chat1")
|
|
assert ms > 0
|
|
|
|
def test_prune_removes_expired_entries(self):
|
|
buf = StreamBuffer(ttl=0.01)
|
|
buf.set("chat1", "hello")
|
|
time.sleep(0.02)
|
|
buf.prune()
|
|
assert buf.get("chat1") == ""
|
|
|
|
def test_prune_removes_excess_entries(self):
|
|
buf = StreamBuffer(max_entries=2)
|
|
for i in range(5):
|
|
buf.set(f"chat{i}", f"msg{i}")
|
|
buf.prune()
|
|
count = sum(1 for i in range(5) if buf.get(f"chat{i}"))
|
|
assert count <= 2
|
|
|
|
|
|
class TestStreamCoalescer:
|
|
def test_should_flush_on_idle_below_threshold(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "hello")
|
|
coalescer = StreamCoalescer(buf, idle_ms=1000)
|
|
assert coalescer.should_flush_on_idle("chat1") is False
|
|
|
|
def test_should_flush_on_idle_empty_buffer(self):
|
|
buf = StreamBuffer()
|
|
coalescer = StreamCoalescer(buf, idle_ms=100)
|
|
assert coalescer.should_flush_on_idle("empty") is False
|
|
|
|
def test_should_flush_on_size_below_threshold(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "short")
|
|
coalescer = StreamCoalescer(buf, min_chars=1500)
|
|
assert coalescer.should_flush_on_size("chat1") is False
|
|
|
|
def test_should_flush_on_size_above_threshold(self):
|
|
buf = StreamBuffer()
|
|
buf.set("chat1", "x" * 1600)
|
|
coalescer = StreamCoalescer(buf, min_chars=1500)
|
|
assert coalescer.should_flush_on_size("chat1") is True
|
|
|
|
def test_is_reasoning_chunk_true(self):
|
|
assert StreamCoalescer.is_reasoning_chunk("lane:reasoning:hello") is True
|
|
|
|
def test_is_reasoning_chunk_false(self):
|
|
assert StreamCoalescer.is_reasoning_chunk("normal message") is False
|
|
|
|
def test_format_reasoning(self):
|
|
coalescer = StreamCoalescer(StreamBuffer(), lane_separator="---")
|
|
result = coalescer.format_reasoning("lane:reasoning:hello world")
|
|
assert "(thinking)" in result
|
|
assert "hello world" in result
|
|
assert "---" in result
|
|
|
|
|
|
class TestReasoningLaneFormatter:
|
|
def test_format(self):
|
|
fmt = ReasoningLaneFormatter("~~~")
|
|
result = fmt.format("lane:reasoning:test")
|
|
assert "(thinking)" in result
|
|
assert "test" in result
|
|
assert "~~~" in result
|
|
|
|
def test_is_reasoning_true(self):
|
|
assert ReasoningLaneFormatter.is_reasoning("lane:reasoning:x") is True
|
|
|
|
def test_is_reasoning_false(self):
|
|
assert ReasoningLaneFormatter.is_reasoning("normal") is False
|
|
|
|
|
|
class TestStreamCoalesceIntegration:
|
|
def test_coalesce_below_threshold_not_sent(self):
|
|
channel = SignalChannel({"block_streaming": True})
|
|
channel._sender = MagicMock()
|
|
channel._rpc_client = MagicMock()
|
|
channel._sender.send_text = AsyncMock(return_value=DeliveryResult(success=True, message_id="1"))
|
|
|
|
async def _test():
|
|
result = await channel.send_stream_chunk("+1234", "1", "short", False)
|
|
assert result.success is True
|
|
|
|
asyncio.run(_test())
|
|
|
|
def test_coalesce_idle_timeout_sends_buffered(self):
|
|
channel = SignalChannel(
|
|
{"block_streaming": True, "block_streaming_coalesce": {"min_chars": 5000, "idle_ms": 0}}
|
|
)
|
|
channel._sender = MagicMock()
|
|
channel._rpc_client = MagicMock()
|
|
channel._sender.send_text = AsyncMock(return_value=DeliveryResult(success=True, message_id="1"))
|
|
|
|
async def _test():
|
|
channel._stream_buffer.set("+1234", "buffered_content")
|
|
channel._stream_coalesce_idle_ms = 0
|
|
result = await channel.send_stream_chunk("+1234", "1", " new part", False)
|
|
assert result.success is True
|
|
|
|
asyncio.run(_test())
|
|
|
|
def test_coalesce_finished_sends_all(self):
|
|
channel = SignalChannel({"block_streaming": True})
|
|
channel._sender = MagicMock()
|
|
channel._rpc_client = MagicMock()
|
|
channel._sender.send_text = AsyncMock(return_value=DeliveryResult(success=True, message_id="final"))
|
|
|
|
async def _test():
|
|
result = await channel.send_stream_chunk("+1234", "1", "final message", True)
|
|
assert result.success is True
|
|
|
|
asyncio.run(_test())
|
|
|
|
def test_block_streaming_mode_sends_immediately(self):
|
|
channel = SignalChannel({"block_streaming": True})
|
|
channel._sender = MagicMock()
|
|
channel._rpc_client = MagicMock()
|
|
channel._sender.send_text = AsyncMock(return_value=DeliveryResult(success=True, message_id="1"))
|
|
|
|
async def _test():
|
|
result = await channel.send_stream_chunk("+1234", None, "immediate", False)
|
|
assert result.success is True
|
|
|
|
asyncio.run(_test())
|
|
|
|
def test_progress_streaming_sends_each_chunk(self):
|
|
channel = SignalChannel({"block_streaming": False})
|
|
channel._sender = MagicMock()
|
|
channel._rpc_client = MagicMock()
|
|
channel._sender.send_text = AsyncMock(return_value=DeliveryResult(success=True, message_id="1"))
|
|
|
|
async def _test():
|
|
result = await channel.send_stream_chunk("+1234", None, "chunk1", False)
|
|
assert result.success is True
|
|
|
|
asyncio.run(_test())
|
|
|
|
|
|
class TestMarkdownFullConversion:
|
|
def test_bold_italic_strikethrough(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("**bold** *italic* ~~strike~~")
|
|
assert result is not None
|
|
assert "bold" in result.body
|
|
assert "italic" in result.body
|
|
assert "strike" in result.body
|
|
style_names = {s.style for s in result.styles}
|
|
assert "BOLD" in style_names
|
|
assert "ITALIC" in style_names
|
|
assert "STRIKETHROUGH" in style_names
|
|
|
|
def test_code_and_spoiler(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("`code` ||spoiler||")
|
|
assert result is not None
|
|
assert "code" in result.body
|
|
assert "spoiler" in result.body
|
|
style_names = {s.style for s in result.styles}
|
|
assert "MONOSPACE" in style_names
|
|
assert "SPOILER" in style_names
|
|
|
|
def test_h1_to_bold(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("# Heading", heading_style="bold")
|
|
assert result is not None
|
|
assert "Heading" in result.body
|
|
|
|
def test_blockquote_prefix(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("> quote text", blockquote_prefix="| ")
|
|
assert result is not None
|
|
assert "| " in result.body or "quote" in result.body
|
|
|
|
def test_tables_to_bullets(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles(
|
|
"| A | B |\n| --- | --- |\n| 1 | 2 |",
|
|
table_mode="bullets",
|
|
)
|
|
assert result is not None
|
|
assert "A" in result.body or "B" in result.body
|
|
|
|
def test_link_extraction(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("[click me](https://example.com)")
|
|
assert result is not None
|
|
assert "https://example.com" in result.body or "click me" in result.body
|
|
|
|
def test_nested_styles(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("**bold *and italic***")
|
|
assert result is not None
|
|
assert "bold" in result.body
|
|
|
|
def test_overlapping_styles_merge(self):
|
|
from yuxi.channels.adapters.signal.format import markdown_to_signal_styles
|
|
|
|
result = markdown_to_signal_styles("**same** **style**")
|
|
assert result is not None
|
|
assert "same" in result.body
|
|
assert "style" in result.body |