class TestContact(unittest.TestCase):
def setUp(self):
self.new_contact = Contact("James", "Muriuki", "0712345678",
"*****@*****.**")
def test_init(self):
self.assertEqual(self.new_contact.first_name, "James")
self.assertEqual(self.new_contact.last_name, "Muriuki")
self.assertEqual(self.new_contact.phone_number, "0712345678")
self.assertEqual(self.new_contact.email, "*****@*****.**")
def test_save_contact(self):
self.new_contact.save_contact()
self.assertEqual(len(Contact.contact_list), 1)
def test_save_multiple_contact(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0712345678", "*****@*****.**")
test_contact.save_contact()
self.assertEqual(len(Contact.contact_list), 2)
def tearDown(self):
Contact.contact_list = []
def test_delete_contact(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0712345678", "*****@*****.**")
test_contact.save_contact()
self.new_contact.delete_contact()
self.assertEqual(len(Contact.contact_list), 1)
def test_find_contact_by_number(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0711223344", "*****@*****.**")
test_contact.save_contact()
found_contact = Contact.find_by_number("0711223344")
self.assertEqual(found_contact.email, test_contact.email)
def test_contact_exists(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0711223344", "*****@*****.**")
test_contact.save_contact()
contact_exists = Contact.contact_exist("0711223344")
self.assertTrue(contact_exists)
def test_display_all_contacts(self):
self.assertEqual(Contact.display_contacts(), Contact.contact_list)
def test_copy_email(self):
self.new_contact.save_contact()
Contact.copy_email("0712345678")
self.assertEqual(self.new_contact.email, pyperclip.paste())
def test_find_contact_by_number(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0711223344", "*****@*****.**")
test_contact.save_contact()
found_contact = Contact.find_by_number("0711223344")
self.assertEqual(found_contact.email, test_contact.email)
def test_contact_exists(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0711223344", "*****@*****.**")
test_contact.save_contact()
contact_exists = Contact.contact_exist("0711223344")
self.assertTrue(contact_exists)
def test_delete_contact(self):
self.new_contact.save_contact()
test_contact = Contact("Test", "user", "0712345678", "*****@*****.**")
test_contact.save_contact()
self.new_contact.delete_contact()
self.assertEqual(len(Contact.contact_list), 1)