FEATURE: bam for url in bio

- добавил бан за наличие ссылки в био
- добавил тест на такой бан
This commit is contained in:
kotolex
2024-01-15 08:54:49 +05:00
parent fbdd7ca244
commit 6f8d81f41b
4 changed files with 58 additions and 26 deletions
+15 -2
View File
@@ -1,8 +1,10 @@
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, cleaned_text, remove_spaces, has_no_letters
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, has_no_letters, has_links
)
REAL_CASES = (
"Оформим резидeнтствo ОАЭ без предoплаты за 5000 дирхам. Пoдробнoсти в лc",
@@ -180,6 +182,17 @@ class TestOthers(TestCase):
with self.subTest(f"has_no_letters {text}"):
self.assertEqual(expected, has_no_letters(text))
def test_has_links(self):
params = (
(True, 'My bot https://t.me/bot.some'),
(True, 'My bot http://t.me/bot.some'),
(False, 'My bot here'),
(False, 'I can use http'),
)
for expected, text in params:
with self.subTest(f"has_links {text}"):
self.assertEqual(expected, has_links(text))
if __name__ == '__main__':
main()