Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/tools/test_media_caption_split.py

@@ -0,0 +1,47 @@
+"""Guard test for the MEDIA:<path> caption chokepoint (_media_caption_split).
+
+`hermes send` strips the MEDIA: tag and leaves the remaining prose as the
+accompanying text. Historically every standalone sender posted that text as a
+*separate* message before an uncaptioned media bubble, splitting
+``hermes send --to whatsapp "MEDIA:/x.png This Caption"`` into two parts.
+
+`_media_caption_split` is the single enforced decision point that all standalone
+senders (WhatsApp, Telegram, Discord) consult to decide whether the text should
+ride on the media bubble as a native caption. This test pins that contract so
+the platforms can't diverge.
+"""
+
+from tools.send_message_tool import (
+    _DEFAULT_CAPTION_LIMIT,
+    _TELEGRAM_CAPTION_LIMIT,
+    _media_caption_split,
+)
+
+
+def test_single_image_short_text_becomes_caption():
+    caption, body = _media_caption_split(
+        "This Caption", [("/tmp/F22.png", False)], max_caption_len=_DEFAULT_CAPTION_LIMIT
+    )
+    assert caption == "This Caption"
+    assert body == ""
+
+
+def test_multi_file_keeps_separate_body():
+    text = "two photos"
+    caption, body = _media_caption_split(
+        text,
+        [("/tmp/a.png", False), ("/tmp/b.png", False)],
+        max_caption_len=_DEFAULT_CAPTION_LIMIT,
+    )
+    assert caption is None
+    assert body == text
+
+
+def test_unknown_extension_keeps_separate_body():
+    # A non-captionable kind (e.g. an audio note that isn't flagged voice)
+    text = "some audio"
+    caption, body = _media_caption_split(
+        text, [("/tmp/song.mp3", False)], max_caption_len=_DEFAULT_CAPTION_LIMIT
+    )
+    assert caption is None
+    assert body == text