Stan-Ansible/tests/test_helpers.py

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

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 $/день.Пишите в личные сообщение +"
)


class TestSpam(TestCase):
    def test_simple(self):
        for spam_word in SPAM:
            with self.subTest(f"check spam word({spam_word})"):
                self.assertTrue(is_spam(spam_word))

    def test_case_first(self):
        for text in CASES:
            with self.subTest(f"check case({text})"):
                self.assertTrue(is_spam(text))

    def test_one(self):
        self.assertTrue(is_spam("привeт"))  # e английская

    def test_is_mixed(self):
        self.assertTrue(is_mixed("привeт"))  # e английская
        self.assertFalse(is_mixed("привет ребята!"))
        self.assertFalse(is_mixed("hi guys"))
        self.assertFalse(is_mixed("hi guys! #%^&%^$ @Вася привет g! п@"))
        self.assertFalse(is_mixed("hi guys! #%^&%^$ @Вася привет g! п@"))
        self.assertFalse(is_mixed("тут *подробнее* https://hack.comp.com"))
        self.assertTrue(is_mixed("hi guys! #%^&%^$ ghb привет g! пQ"))
        text = """
        В целом очень интересный чат, где 4 человека
 считают себя выше мира всего и нападают/насмехаются на всех/над всеми, кто что-то "недо". 
Один из которых вообще выдаёт какие-то фразы через бота) интересно, потому что сам не может сказать, или потому что раздвоение личности дело такое)
«Я пуп земли», – твердит гордец надменно,
Забыв, что он – песчинка во вселенной."""
        self.assertFalse(is_mixed(text))
        self.assertFalse(is_mixed('print("привет")'))
        self.assertFalse(is_mixed("print('привет')"))
        self.assertFalse(is_mixed("text='слово'"))


class TestOthers(TestCase):
    def test_ban(self):
        for word in BAN_WORDS:
            with self.subTest(f"check ban word({word})"):
                self.assertTrue(is_ban_words_in_caption(word))

    def test_allowed(self):
        params = (
            (True, [], 'any'),
            (True, ['one'], 'two'),
            (False, ['one', 'two'], 'two'),
            (False, ['one', 'two'], 'Two'),
            (False, ['one', 'two'], 'TWO'),
        )
        for expected, words, mess in params:
            with self.subTest(f"check allowed({words, mess})"):
                self.assertEqual(is_in_not_allowed(words, mess), expected)

    def test_non_grata(self):
        params = (
            (False, 'any'),
            (True, 'дудар'),
            (True, 'хауди'),
            (True, 'dudar'),
        )
        for expected, text in params:
            with self.subTest(f"check non-grata({text})"):
                m = MagicMock(text=text)
                self.assertEqual(is_nongrata(m), expected)

    def test_is_admin(self):
        params = (
            (False, 'any'),
            (True, ADMIN_ID),
        )
        for expected, text in params:
            with self.subTest(f"check non-grata({text})"):
                m = MagicMock(from_user=MagicMock(id=text))
                self.assertEqual(is_admin(m), expected)

    def test_fetch_rule(self):
        for index, text in enumerate(RULES_TEXT, 1):
            with self.subTest(f"fetch rule {index}"):
                self.assertEqual(fetch_rule(index), text)

    def test_fetch_non_existent_rule(self):
        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))

    def test_has_no_letters(self):
        params = (
            (False, 'f1   b2  print  text  '),
            (True, ''),
            (True, ' '),
            (True, '12 @# &*'),
            (False, '12 @# &* f'),
            (False, '12 @# &* Z'),
            (False, '12 @# &* я'),
            (False, '12 @# &* Я'),
        )
        for expected, text in params:
            with self.subTest(f"has_no_letters {text}"):
                self.assertEqual(expected, has_no_letters(text))


if __name__ == '__main__':
    main()