From e397f0f2bd59cbddd46c2d1409e6e8f909b5acc2 Mon Sep 17 00:00:00 2001
From: Pavel Kirilin <win10@list.ru>
Date: Sun, 28 Jun 2020 05:39:40 +0400
Subject: [PATCH] Fixed bl function.
Signed-off-by: Pavel Kirilin <win10@list.ru>
---
src/actions/fun.py | 2 +-
tests/conftest.py | 4 +++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/actions/fun.py b/src/actions/fun.py
index 2446f69..c3d8157 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 f93d085..829732d 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
--
GitLab