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

tests/gateway/test_telegram_voice_duration.py

@@ -0,0 +1,116 @@
+"""Telegram voice/audio messages must carry an explicit duration.
+
+Telegram only auto-derives a clip's duration from container metadata for short
+recordings; longer ones (~5 min+) are sent with duration 0 and render as
+``0:00`` in the player. ``TelegramAdapter.send_voice`` now probes the length
+locally and passes ``duration=`` to ``sendVoice`` / ``sendAudio`` so the
+bubble shows the real time.
+
+These tests confirm:
+  1. ``_probe_voice_duration_seconds`` rounds to whole seconds (real WAV via
+     stdlib ``wave``), reads OGG/MP3 lengths via mutagen, and degrades to
+     ``None`` (omit duration) when nothing can read the file.
+  2. The ``.ogg`` voice path forwards the probed duration to ``bot.send_voice``.
+  3. The ``.mp3`` audio path forwards the probed duration to ``bot.send_audio``.
+
+Hermetic: no real mutagen / ffprobe required. The WAV path uses stdlib only;
+the mutagen path is exercised with a fake module injected into ``sys.modules``.
+"""
+import sys
+import wave
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, MagicMock
+
+import pytest
+
+from gateway.config import PlatformConfig
+from plugins.platforms.telegram import adapter as telegram_mod  # noqa: E402
+from plugins.platforms.telegram.adapter import (  # noqa: E402
+    TelegramAdapter,
+    _coerce_duration_seconds,
+    _probe_voice_duration_seconds,
+)
+from tools.send_message_tool import _send_telegram  # noqa: E402
+
+
+def _write_wav(path, *, rate, frames):
+    """Write a silent mono WAV of ``frames`` samples at ``rate`` Hz."""
+    with wave.open(str(path), "wb") as wf:
+        wf.setnchannels(1)
+        wf.setsampwidth(1)
+        wf.setframerate(rate)
+        wf.writeframes(b"\x00" * frames)
+
+
+# ---------------------------------------------------------------------------
+# 1a. WAV path — real stdlib read, rounding to whole seconds
+# ---------------------------------------------------------------------------
+
+@pytest.mark.parametrize(
+    "rate,frames,expected",
+    [
+        (10, 2900, 290),   # 290.0 -> 290
+        (10, 2904, 290),   # 290.4 -> 290
+        (10, 2906, 291),   # 290.6 -> 291
+        (10, 12, 1),       # 1.2   -> 1
+    ],
+)
+def test_probe_wav_rounds_to_whole_seconds(tmp_path, rate, frames, expected):
+    f = tmp_path / "clip.wav"
+    _write_wav(f, rate=rate, frames=frames)
+    assert _probe_voice_duration_seconds(str(f)) == expected
+
+
+# ---------------------------------------------------------------------------
+# 1b. mutagen path — fake module so ogg/mp3 work without the real dependency
+# ---------------------------------------------------------------------------
+
+def _inject_fake_mutagen(monkeypatch, length):
+    fake = SimpleNamespace(
+        File=lambda _p: SimpleNamespace(info=SimpleNamespace(length=length))
+    )
+    monkeypatch.setitem(sys.modules, "mutagen", fake)
+
+
+# ---------------------------------------------------------------------------
+# 1c. _coerce_duration_seconds rounding/guard contract
+# ---------------------------------------------------------------------------
+
+@pytest.mark.parametrize(
+    "value,expected",
+    [(290.0, 290), (290.6, 291), ("12.7", 13), (0, None), (-3, None),
+     (None, None), ("", None), ("x", None)],
+)
+def test_coerce_duration_seconds(value, expected):
+    assert _coerce_duration_seconds(value) == expected
+
+
+# ---------------------------------------------------------------------------
+# 2 + 3. send_voice forwards the probed duration to the Bot API
+# ---------------------------------------------------------------------------
+
+def _make_adapter() -> TelegramAdapter:
+    adapter = TelegramAdapter(PlatformConfig(enabled=True, token="***"))
+    adapter._bot = MagicMock()
+    adapter._bot.send_voice = AsyncMock(return_value=MagicMock(message_id=1))
+    adapter._bot.send_audio = AsyncMock(return_value=MagicMock(message_id=2))
+    return adapter
+
+
+@pytest.mark.asyncio
+async def test_voice_send_forwards_duration(monkeypatch, tmp_path):
+    monkeypatch.setattr(
+        telegram_mod, "_probe_voice_duration_seconds", lambda _p: 314
+    )
+    audio = tmp_path / "reply.ogg"
+    audio.write_bytes(b"\x00" * 16)
+
+    adapter = _make_adapter()
+    result = await adapter.send_voice("123", str(audio))
+
+    assert result.success is True
+    adapter._bot.send_voice.assert_awaited_once()
+    assert adapter._bot.send_voice.await_args.kwargs["duration"] == 314
+    adapter._bot.send_audio.assert_not_awaited()
+
+