diff --git a/tests/conftest.py b/tests/conftest.py index 829732dbd16df6c39f4466027bb20ee65b4e1aee..9b5fda3e766f6d7fd5694ae04b4beb04809b5ca5 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,5 +1,7 @@ from unittest.mock import MagicMock +import pytest + class NewMessageTestEvent(object): def __init__(self, message: str, forwarded=False): @@ -28,3 +30,13 @@ class NewMessageTestEvent(object): message = MagicMock() message.message = self._msg return message + + async def get_chat(self): + return MagicMock() + + +@pytest.fixture +def patch_mark_dialogue_unread(): + import telethon + + telethon.functions.messages.MarkDialogUnreadRequest = MagicMock(returning=True) diff --git a/tests/test_fun.py b/tests/test_fun.py index 1d3acd2e584cc77b0cfd5108a8185b429a58c333..f97eba375bcda880452163f1aa8459511eada230 100644 --- a/tests/test_fun.py +++ b/tests/test_fun.py @@ -5,14 +5,14 @@ from tests.conftest import NewMessageTestEvent @pytest.mark.asyncio -async def test_bl_trailing_comma(): +async def test_bl_trailing_comma(patch_mark_dialogue_unread): event_mock = NewMessageTestEvent("test,") await fun.blyaficate(event_mock) assert not event_mock.result.endswith(",") @pytest.mark.asyncio -async def test_bl_starting_comma(): +async def test_bl_starting_comma(patch_mark_dialogue_unread): event_mock = NewMessageTestEvent(",test.") await fun.blyaficate(event_mock) assert not event_mock.result.startswith(",")