from __future__ import annotations


from yuxi.channels.adapters.wechat.format import (
    map_wecom_msg_type,
    map_mp_msg_type,
    map_bridge_msg_type,
    truncate_text,
    TRUNCATION_MARKER,
)
from yuxi.channels.models import MessageType


class TestMessageTypeMapping:
    def test_map_wecom_text(self):
        assert map_wecom_msg_type("text") == MessageType.TEXT

    def test_map_wecom_image(self):
        assert map_wecom_msg_type("image") == MessageType.IMAGE

    def test_map_wecom_voice(self):
        assert map_wecom_msg_type("voice") == MessageType.AUDIO

    def test_map_wecom_video(self):
        assert map_wecom_msg_type("video") == MessageType.VIDEO

    def test_map_wecom_file(self):
        assert map_wecom_msg_type("file") == MessageType.FILE

    def test_map_wecom_location(self):
        assert map_wecom_msg_type("location") == MessageType.LOCATION

    def test_map_wecom_unknown(self):
        assert map_wecom_msg_type("unknown") == MessageType.TEXT

    def test_map_mp_text(self):
        assert map_mp_msg_type("text") == MessageType.TEXT

    def test_map_mp_image(self):
        assert map_mp_msg_type("image") == MessageType.IMAGE

    def test_map_mp_voice(self):
        assert map_mp_msg_type("voice") == MessageType.AUDIO

    def test_map_bridge_text(self):
        assert map_bridge_msg_type(1) == MessageType.TEXT

    def test_map_bridge_image(self):
        assert map_bridge_msg_type(3) == MessageType.IMAGE

    def test_map_bridge_voice(self):
        assert map_bridge_msg_type(34) == MessageType.AUDIO

    def test_map_bridge_video(self):
        assert map_bridge_msg_type(43) == MessageType.VIDEO

    def test_map_bridge_file(self):
        assert map_bridge_msg_type(49) == MessageType.FILE

    def test_map_bridge_unknown(self):
        assert map_bridge_msg_type(99) == MessageType.TEXT


class TestTextTruncation:
    def test_no_truncation_needed(self):
        text = "短文本"
        result = truncate_text(text, 100)
        assert result == text
        assert len(result) <= 100

    def test_truncation_applied(self):
        text = "A" * 3000
        result = truncate_text(text, 100)
        assert len(result) <= 100
        assert TRUNCATION_MARKER in result

    def test_truncation_exact_limit(self):
        text = "A" * 100
        result = truncate_text(text, 100)
        assert result == text

    def test_truncation_at_boundary(self):
        text = "A" * 101
        result = truncate_text(text, 100)
        assert len(result) <= 100
        assert TRUNCATION_MARKER in result