Add tests for markdown

tests/parser/test_html.py

@@ -100,6 +100,19 @@ def test_html_unparse_pre():
     assert HTML.unparse(text=text, entities=entities) == expected
 
 
+def test_html_unparse_blockquote():
+    expected = """<blockquote>Quote text</blockquote>
+    from pyrogram"""
+
+    text = """Quote text
+    from pyrogram"""
+
+    entities = pyrogram.types.List([pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.BLOCKQUOTE, offset=0,
+                                                                 length=10)])
+
+    assert HTML.unparse(text=text, entities=entities) == expected
+
+
 def test_html_unparse_mixed():
     expected = "<b>aaaaaaa<i>aaa<u>bbbb</u></i></b><u><i>bbbbbbccc</i></u><u>ccccccc<s>ddd</s></u><s>ddddd<spoiler>dd" \
                "eee</spoiler></s><spoiler>eeeeeeefff</spoiler>ffff<code>fffggggggg</code>ggghhhhhhhhhh"

tests/parser/test_markdown.py

@@ -0,0 +1,115 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+import pyrogram
+from pyrogram.parser.markdown import Markdown
+
+
+# expected: the expected unparsed Markdown
+# text: original text without entities
+# entities: message entities coming from the server
+
+def test_markdown_unparse_bold():
+    expected = "**bold**"
+    text = "bold"
+    entities = pyrogram.types.List(
+        [pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.BOLD, offset=0, length=4)])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_italic():
+    expected = "__italic__"
+    text = "italic"
+    entities = pyrogram.types.List(
+        [pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.ITALIC, offset=0, length=6)])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_strike():
+    expected = "~~strike~~"
+    text = "strike"
+    entities = pyrogram.types.List(
+        [pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.STRIKETHROUGH, offset=0, length=6)])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_spoiler():
+    expected = "||spoiler||"
+    text = "spoiler"
+    entities = pyrogram.types.List(
+        [pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.SPOILER, offset=0, length=7)])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_url():
+    expected = '[URL](https://pyrogram.org/)'
+    text = "URL"
+    entities = pyrogram.types.List([pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.TEXT_LINK,
+                                                                 offset=0, length=3, url='https://pyrogram.org/')])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_code():
+    expected = '`code`'
+    text = "code"
+    entities = pyrogram.types.List(
+        [pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.CODE, offset=0, length=4)])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_pre():
+    expected = """```python
+for i in range(10):
+    print(i)
+```"""
+
+    text = """for i in range(10):
+    print(i)"""
+
+    entities = pyrogram.types.List([pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.PRE, offset=0,
+                                                                 length=32, language='python')])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_mixed():
+    expected = "**aaaaaaa__aaabbb**__~~dddddddd||ddeee~~||||eeeeeeefff||ffff`fffggggggg`ggghhhhhhhhhh"
+    text = "aaaaaaaaaabbbddddddddddeeeeeeeeeeffffffffffgggggggggghhhhhhhhhh"
+    entities = pyrogram.types.List(
+        [pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.BOLD, offset=0, length=13),
+         pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.ITALIC, offset=7, length=6),
+         pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.STRIKETHROUGH, offset=13, length=13),
+         pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.SPOILER, offset=21, length=5),
+         pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.SPOILER, offset=26, length=10),
+         pyrogram.types.MessageEntity(type=pyrogram.enums.MessageEntityType.CODE, offset=40, length=10)])
+
+    assert Markdown.unparse(text=text, entities=entities) == expected
+
+
+def test_markdown_unparse_no_entities():
+    expected = "text"
+    text = "text"
+    entities = []
+
+    assert Markdown.unparse(text=text, entities=entities) == expected