import unittest
import BO3


class TestVulgarism(unittest.TestCase):
    def test_1(self):
        usr_input = 'W ogóle przystojny, zajebisty, kurwa, koleś'
        correct = 'W ogóle przystojny, ---, ---, koleś'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here

    def test_2(self):
        usr_input = 'Jebnąć cię czy się zamkniesz już?'
        correct = '--- cię czy się zamkniesz już?'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_3(self):
        usr_input = 'Jebać to mało, zajebać trzeba!'
        correct = '--- to mało, --- trzeba!'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_4(self):
        usr_input = 'Wyjebał się chuj'
        correct = '--- się ---'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_5(self):
        usr_input = 'Stara kurwa napierdala w telefon, a ja tu browarka sączę'
        correct = 'Stara --- --- w telefon, a ja tu browarka sączę'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_6(self):
        usr_input = 'Kurwa, jak to, kurwa, usłyszałem, to mnie, kurwa, z zawiasów wypierdoliło'
        correct = '---, jak to, ---, usłyszałem, to mnie, ---, z zawiasów ---'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_7(self):
        usr_input = 'Ale mówię ci, zajebiście było'
        correct = 'Ale mówię ci, --- było'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_8(self):
        usr_input = 'A ja się, kurwa, czuję właśnie jak w Dniu świra'
        correct = 'A ja się, ---, czuję właśnie jak w Dniu świra'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_9(self):
        usr_input = 'Mój wujek ma, kurwa, no ten, kurwa...'
        correct = 'Mój wujek ma, ---, no ten, ---...'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here
    def test_10(self):
        usr_input = 'Kurwa, choć nie na temat to dodam, że...'
        correct = '---, choć nie na temat to dodam, że...'
        self.assertEqual(BO3.replace_vulgarism(usr_input), correct)  # add assertion here


if __name__ == '__main__':
    unittest.main()