Add filters.command tests

tests/filters/__init__.py

@@ -0,0 +1,34 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-2021 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/>.
+
+class Client:
+    @staticmethod
+    async def get_me():
+        return User("username")
+
+
+class User:
+    def __init__(self, username: str = None):
+        self.username = username
+
+
+class Message:
+    def __init__(self, text: str = None, caption: str = None):
+        self.text = text
+        self.caption = caption
+        self.command = None

tests/filters/test_command.py

@@ -0,0 +1,132 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-2021 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 pytest
+
+from pyrogram import filters
+from tests.filters import Client, Message
+
+c = Client()
+
+
+@pytest.mark.asyncio
+async def test_single():
+    f = filters.command("start")
+
+    m = Message("/start")
+    assert await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_multiple():
+    f = filters.command(["start", "help"])
+
+    m = Message("/start")
+    assert await f(c, m)
+
+    m = Message("/help")
+    assert await f(c, m)
+
+    m = Message("/settings")
+    assert not await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_prefixes():
+    f = filters.command("start", prefixes=list(".!#"))
+
+    m = Message(".start")
+    assert await f(c, m)
+
+    m = Message("!start")
+    assert await f(c, m)
+
+    m = Message("#start")
+    assert await f(c, m)
+
+    m = Message("/start")
+    assert not await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_case_sensitive():
+    f = filters.command("start", case_sensitive=True)
+
+    m = Message("/start")
+    assert await f(c, m)
+
+    m = Message("/StArT")
+    assert not await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_case_insensitive():
+    f = filters.command("start", case_sensitive=False)
+
+    m = Message("/start")
+    assert await f(c, m)
+
+    m = Message("/StArT")
+    assert await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_with_mention():
+    f = filters.command("start")
+
+    m = Message("/start@username")
+    assert await f(c, m)
+
+    m = Message("/start@UserName")
+    assert await f(c, m)
+
+    m = Message("/start@another")
+    assert not await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_with_args():
+    f = filters.command("start")
+
+    m = Message("/start a b c")
+    await f(c, m)
+    assert m.command == ["start"] + list("abc")
+
+    m = Message("/start 'a b' c")
+    await f(c, m)
+    assert m.command == ["start", "a b", "c"]
+
+    m = Message('/start     a     b     "c     d"')
+    await f(c, m)
+    assert m.command == ["start"] + list("ab") + ["c     d"]
+
+
+@pytest.mark.asyncio
+async def test_caption():
+    f = filters.command("start")
+
+    m = Message(caption="/start")
+    assert await f(c, m)
+
+
+@pytest.mark.asyncio
+async def test_no_text():
+    f = filters.command("start")
+
+    m = Message()
+    assert not await f(c, m)