class MailListTest(unittest.TestCase): """docstring for MailListTest""" def setUp(self): self.maillist = MailList(1, 'new list') def test_get_name(self): self.assertEqual('new list', self.maillist.get_name()) def test_add_user(self): self.maillist.add_user('Ivan', '*****@*****.**') expected_user = User('Ivan', '*****@*****.**') self.assertEqual(expected_user.take_name(), self.maillist.users[0].take_name()) self.assertEqual(expected_user.take_email(), self.maillist.users[0].take_email()) def test_print_with_something_in_it(self): self.maillist.add_user('Ivan', '*****@*****.**') self.maillist.add_user('Peho', '*****@*****.**') expected = '[1] Ivan - [email protected]\n[2] Peho - [email protected]' self.assertEqual(expected, self.maillist.print_()) def test_search_email_available(self): self.maillist.add_user('Ivan', '*****@*****.**') self.maillist.add_user('Peho', '*****@*****.**') check = '*****@*****.**' self.assertTrue(self.maillist.search_email(check)) def test_serach_email_unavailable(self): self.maillist.add_user('Ivan', '*****@*****.**') self.maillist.add_user('Peho', '*****@*****.**') check = '*****@*****.**' self.assertTrue(not self.maillist.search_email(check)) def test_name_available(self): self.maillist.add_user('Ivan', '*****@*****.**') self.maillist.add_user('Peho', '*****@*****.**') check = 'Peho' self.assertTrue(self.maillist.search_name(check)) def test_name_unavailable(self): self.maillist.add_user('Ivan', '*****@*****.**') self.maillist.add_user('Peho', '*****@*****.**') check = 'Georgi' self.assertFalse(self.maillist.search_name(check))