from aiogram.types import MessageEntity, MessageEntityType
from aiogram.utils import text_decorations


class TestTextDecorations:
    @staticmethod
    def test_unparse_entities_normal_text():
        if (
            text_decorations.markdown_decoration.unparse(
                "hi i'm bold and italic and still bold",
                entities=[
                    MessageEntity(offset=3, length=34,
                                  type=MessageEntityType.BOLD),
                    MessageEntity(offset=12, length=10,
                                  type=MessageEntityType.ITALIC),
                ],
            )
            != "hi *i'm bold _\rand italic_\r and still bold*"
        ):
            raise AssertionError

    @staticmethod
    def test_unparse_entities_emoji_text():
        """
        emoji is encoded as two chars in json
        """
        if (
            text_decorations.markdown_decoration.unparse(
                "🚀 i'm bold and italic and still bold",
                entities=[
                    MessageEntity(offset=3, length=34,
                                  type=MessageEntityType.BOLD),
                    MessageEntity(offset=12, length=10,
                                  type=MessageEntityType.ITALIC),
                ],
            )
            != "🚀 *i'm bold _\rand italic_\r and still bold*"
        ):
            raise AssertionError