diff --git a/src/actions/fun.py b/src/actions/fun.py index 2446f693604554af6b5a154f1ef402c452d2aed7..c3d8157cd498c03b374a1150940a07f94fca7e7a 100644 --- a/src/actions/fun.py +++ b/src/actions/fun.py @@ -25,7 +25,7 @@ swearing = [ @config.telegram_client.on(events.NewMessage(pattern=r"^\.bl (.*)", forwards=False)) async def blyaficate(event: events.NewMessage.Event): - message = event.message.lstrip(".bl ").strip() + message = event.message.message.lstrip(".bl ").strip() response = "" for symbol in message: if symbol == ",": diff --git a/tests/conftest.py b/tests/conftest.py index f93d08519095372475caf3664c253e111de1ad00..829732dbd16df6c39f4466027bb20ee65b4e1aee 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -25,4 +25,6 @@ class NewMessageTestEvent(object): @property def message(self): - return self._msg + message = MagicMock() + message.message = self._msg + return message