def test_save_multiple_contact(self):
'''
test_save_multiple_contact to check if we can save multiple contact
objects to our contact_list
'''
self.new_contact.save_contact()
test_contact = User("*****@*****.**","kingdom","jquery2") # new contact
test_contact.save_contact()
self.assertEqual(len(User.contact_list),7)
def test_delete_contact(self):
'''
test_delete_contact to test if we can remove a contact from our contact list
'''
self.new_contact.save_contact()
test_contact = User("*****@*****.**","kingdom","jquery2") # new contact
test_contact.save_contact()
self.new_contact.delete_contact()# Deleting a contact object
self.assertEqual(len(User.contact_list),2)
def test_find_contact_by_username(self):
'''
test to check if we can find a contact by username and display information
'''
self.new_contact.save_contact()
test_contact = User("*****@*****.**","kingdom","jquery2") # new contact
test_contact.save_contact()
found_contact = User.find_by_username("kingdom")
self.assertEqual(found_contact.email,test_contact.email)
class TestUser(unittest.TestCase):
'''
Test class that defines test cases for the user class behaviours.
Args:
unittest.TestCase: TestCase class that helps in creating test cases
'''
# Items up here .......
def setUp(self):
'''
Set up method to run before each test cases.
'''
self.new_contact = User("*****@*****.**","kingdom","jquery2") # create contact object
def test_init(self):
'''
test_init test case to test if the object is initialized properly
'''
self.assertEqual(self.new_contact.email,"*****@*****.**")
self.assertEqual(self.new_contact.username,"kingdom")
self.assertEqual(self.new_contact.password,"jquery2")
def test_save_contact(self):
'''
test_save_contact test case to test if the contact object is saved into
the contact list
'''
self.new_contact.save_contact() # saving the new contact
self.assertEqual(len(User.contact_list),5)
# Items up here...
# setup and class creation up here
def tearDown(self):
'''
tearDown method that does clean up after each test case has run.
'''
User.contact_list = []
# other test cases here
def test_save_multiple_contact(self):
'''
test_save_multiple_contact to check if we can save multiple contact
objects to our contact_list
'''
self.new_contact.save_contact()
test_contact = User("*****@*****.**","kingdom","jquery2") # new contact
test_contact.save_contact()
self.assertEqual(len(User.contact_list),7)
# More tests above
def test_delete_contact(self):
'''
test_delete_contact to test if we can remove a contact from our contact list
'''
self.new_contact.save_contact()
test_contact = User("*****@*****.**","kingdom","jquery2") # new contact
test_contact.save_contact()
self.new_contact.delete_contact()# Deleting a contact object
self.assertEqual(len(User.contact_list),2)
def delete_contact(self):
'''
delete_contact method deletes a saved contact from the contact_list
'''
User.contact_list.remove(self)
def test_find_contact_by_username(self):
'''
test to check if we can find a contact by username and display information
'''
self.new_contact.save_contact()
test_contact = User("*****@*****.**","kingdom","jquery2") # new contact
test_contact.save_contact()
found_contact = User.find_by_username("kingdom")
self.assertEqual(found_contact.email,test_contact.email)
def test_display_all_contacts(self):
'''
method that returns a list of all contacts saved
'''
self.assertEqual(User.display_contacts(),User.contact_list)
def test_copy_email(self):
'''
Test to confirm that we are copying the email address from a found contact
'''
self.new_contact.save_contact()
User.copy_email("kingdom")
self.assertEqual(self.new_contact.email,pyperclip.paste())