File size: 2,690 Bytes
6172a47 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | from unittest.mock import MagicMock
import pytest
from messaging.rendering.telegram_markdown import (
escape_md_v2,
escape_md_v2_code,
mdv2_bold,
mdv2_code_inline,
render_markdown_to_mdv2,
)
from messaging.transcript import RenderCtx, TranscriptBuffer
@pytest.fixture
def handler():
platform = MagicMock()
cli = MagicMock()
store = MagicMock()
return (platform, cli, store)
def _ctx() -> RenderCtx:
return RenderCtx(
bold=mdv2_bold,
code_inline=mdv2_code_inline,
escape_code=escape_md_v2_code,
escape_text=escape_md_v2,
render_markdown=render_markdown_to_mdv2,
)
def test_truncation_closes_code_blocks(handler):
"""Verify that truncation correctly closes open code blocks."""
t = TranscriptBuffer()
t.apply(
{
"type": "thinking_chunk",
"text": "Starting some long thinking process that will definitely cause truncation later on...",
}
)
t.apply(
{
"type": "text_chunk",
"text": "```python\ndef very_long_function():\n # " + ("A" * 4000),
}
)
msg = t.render(_ctx(), limit_chars=3900, status="✅ *Complete*")
# The backtick count must be even to be a valid block.
assert msg.count("```") % 2 == 0
assert msg.endswith("```") or "✅ *Complete*" in msg.split("```")[-1]
def test_truncation_preserves_status(handler):
"""Verify that status is still appended after truncation."""
status = "READY_STATUS"
t = TranscriptBuffer()
t.apply({"type": "thinking_chunk", "text": "Thinking..."})
t.apply({"type": "text_chunk", "text": "A" * 5000})
msg = t.render(_ctx(), limit_chars=3900, status=status)
assert status in msg
def test_empty_components_with_status(handler):
"""Verify message building with just a status."""
status = "Simple Status"
t = TranscriptBuffer()
msg = t.render(_ctx(), limit_chars=3900, status=status)
assert msg == "\n\nSimple Status"
def test_render_markdown_unclosed_markdown():
"""Malformed markdown (e.g. unclosed *) does not crash and produces acceptable output."""
from messaging.rendering.telegram_markdown import render_markdown_to_mdv2
md = "*bold without close"
out = render_markdown_to_mdv2(md)
assert out is not None
assert "bold" in out
def test_escape_md_v2_unicode_emoji():
"""Unicode and emoji pass through correctly (no special char escaping needed)."""
from messaging.rendering.telegram_markdown import escape_md_v2, escape_md_v2_code
text = "Hello 世界 🎉 café"
assert escape_md_v2(text) == text
assert escape_md_v2_code(text) == text
|