leo/Stan-Ansible
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FEATURE 14: autoreaction

Browse Source 
- если фулл_нейм пустой то будет сообщение о нарушении правила, само сообщение юзера будет удалено
- добавлена правка по спаму
- немного рефакторинга
- если сообщение содержит только слово из констант привет(полное совпадение) то выдаем правило непривет
This commit is contained in:
kotolex
2023-12-26 16:58:27 +05:00
parent 77e34681ec
commit 92bb50dd49
7 changed files with 99 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_bot.py
+21 -6
View File
@@ -17,7 +17,7 @@ from src import helpers
# нужно не дать делать реальные запросы в тестах, конкретно тут -за Правилами чата
helpers.is_url_reachable = lambda a: True
from src.commands import bot
from src.constants import BDMTSS_ID, LUTZ_ID, ADMIN_ID
from src.constants import BDMTSS_ID, LUTZ_ID, ADMIN_ID, HELLO_EXAMPLES
from src.filters import is_white_id
# отключаем логи бота, в тестах они не нужны
@@ -45,11 +45,11 @@ def _all_nones():
shipping_query, pre_checkout_query, poll, poll_answer, my_chat_member, chat_member, chat_join_request)
def get_update(text, reply_to=None, user_id=10):
def get_update(text, reply_to=None, user_id=10, first_name='Some User'):
"""Хелпер, генерирующий событие обновления для бота"""
params = {'text': text}
chat = types.Chat(id=11, type='group')
user = types.User(id=user_id, is_bot=False, first_name='Some User')
user = types.User(id=user_id, is_bot=False, first_name=first_name, full_name=first_name)
if reply_to:
params["reply_to_message"] = types.Message(message_id=2, from_user=user, date=None, chat=chat,
content_type='text', options={'text': reply_to}, json_string="")
@@ -155,6 +155,15 @@ class TestBot(IsolatedAsyncioTestCase):
await self.bot.process_new_updates([get_update('/neprivet', reply_to="привет всем")])
self.assertEqual(RESULTS[0][0]['text'], "Пожалуйста, не пишите просто «Привет» в чате.")
self.assertTrue("https://neprivet.com/" in RESULTS[0][0]['reply_markup'])
self.assertEqual(RESULTS[1][1], "deleteMessage")
async def test_neprivet_auto(self):
for hello in HELLO_EXAMPLES:
with self.subTest(f"neprivet auto {hello}"):
await self.bot.process_new_updates([get_update(hello)])
self.assertEqual(RESULTS[0][0]['text'], "Пожалуйста, не пишите просто «Привет» в чате.")
self.assertTrue("https://neprivet.com/" in RESULTS[0][0]['reply_markup'])
RESULTS.clear()
async def test_lutz(self):
await self.bot.process_new_updates([get_update('/lutz', reply_to="привет всем")])
@@ -441,8 +450,8 @@ class TestBot(IsolatedAsyncioTestCase):
async def test_get_quotes(self):
params = (
("Цитаты отсутствуют. Подробнее: /get_group_info", []),
("Количество цитат: 1\n\\Последние добавленные\n\n· one", ['one']),
("Количество цитат: 2\n\\Последние добавленные\n\n· one\n· two", ['one', 'two']),
("Количество цитат: 1\n\nПоследние добавленные:\n\n· one", ['one']),
("Количество цитат: 2\n\nПоследние добавленные:\n\n· one\n· two", ['one', 'two']),
)
for expected, quotes in params:
with self.subTest(f"get_quotes {quotes}"):
@@ -453,7 +462,7 @@ class TestBot(IsolatedAsyncioTestCase):
RESULTS.clear()
async def test_get_group_info(self):
exp = f"Группа: Title\nID группы: 11 \n\nКоличество цитат: 0\nПоследние добавленные: /get_quotes\n\n" \
exp = f"Группа: Title\nID группы: 11\n\nКоличество цитат: 0\nПоследние добавленные: /get_quotes\n\n" \
f"Текущие настройки:\n Антиспам: 1\n Ежедневные отчёты: 1\n Праздники: 1"
params = (
("Группа не включена. Включить: /enable_stan", None),
@@ -467,6 +476,12 @@ class TestBot(IsolatedAsyncioTestCase):
self.assertEqual(expected, RESULTS[0][0]['text'])
RESULTS.clear()
async def test_empty_name_user(self):
await self.bot.process_new_updates([get_update('новый текст', first_name='')])
self.assertEqual(RESULTS[0], ({'chat_id': 11, 'message_id': 1}, 'deleteMessage'))
self.assertEqual(RESULTS[1][1], 'sendMessage')
self.assertEqual(RESULTS[1][0]['text'], '@None\nНапоминаем правило: Твой ник должен быть читаемым и понятным')
if __name__ == '__main__':
main()
tests/test_helpers.py
+24 -3
View File
@@ -1,12 +1,14 @@
from unittest import TestCase, main
from unittest.mock import MagicMock
from src.constants import SPAM, BAN_WORDS, ADMIN_ID, RULES_TEXT
from src.helpers import is_spam, is_mixed, is_ban_words_in_caption, is_in_not_allowed, is_nongrata, is_admin, fetch_rule
from src.helpers import is_spam, is_mixed, is_ban_words_in_caption, is_in_not_allowed, is_nongrata, is_admin, \
fetch_rule, cleaned_text, remove_spaces
CASES = (
"Оформим резидeнтствo ОАЭ без предoплаты за 5000 дирхам. Пoдробнoсти в лc",
"Всем привет Я в поисках людей в kpиптo кoмaнду У нас вы пройдете 6ecплaтноe обученue 1-3 дня Можно совмещать с основной paбoтoй",
"Здравствуйте! Нужны 2 человека от 200 $/день. Можно без опыта Пишите в личные сообщения +",
"Здравствуйте!Нужны 2 человека от 200 $/день.Пишите в личные сообщение +"
)
@@ -44,7 +46,6 @@ class TestSpam(TestCase):
self.assertFalse(is_mixed("text='слово'"))
class TestOthers(TestCase):
def test_ban(self):
for word in BAN_WORDS:
@@ -86,7 +87,7 @@ class TestOthers(TestCase):
self.assertEqual(is_admin(m), expected)
def test_fetch_rule(self):
for index, text in enumerate(RULES_TEXT,1):
for index, text in enumerate(RULES_TEXT, 1):
with self.subTest(f"fetch rule {index}"):
self.assertEqual(fetch_rule(index), text)
@@ -94,6 +95,26 @@ class TestOthers(TestCase):
self.assertEqual(fetch_rule(0), 'Пока не придумали')
self.assertEqual(fetch_rule(100), 'Пока не придумали')
def test_cleaned_text(self):
params = (
('f1 b2 print text ', 'f1!, b2# print("text")'),
('text', 'text'),
(' ', '!@#$'),
)
for expected, text in params:
with self.subTest(f"cleaned_text {text}"):
self.assertEqual(expected, cleaned_text(text))
def test_remove_spaces(self):
params = (
('f1 b2 print text', 'f1 b2 print text '),
('text', 'text'),
('!@#$', '!@#$'),
)
for expected, text in params:
with self.subTest(f"remove_spaces {text}"):
self.assertEqual(expected, remove_spaces(text))
if __name__ == '__main__':
main()