class TestUser(unittest.TestCase):
'''
Test class that checks the behaviours of the User class.
'''
def setUp(self):
self.new_user = User("michael", "password") # create user object
def test_init(self):
self.assertEqual(self.new_user.username, "michael")
self.assertEqual(self.new_user.password, "password")
def test_save_user(self):
self.new_user.save_user()
self.assertEqual(len(User.user_list), 1)
def tearDown(self):
'''
this will clean up after each test run.
'''
User.user_list = []
def test_save_multiple_user(self):
'''
check if we can hold multiple user accounts
'''
self.new_user.save_user()
test_user = User("andrew", "password")
test_user.save_user()
self.assertEqual(len(User.user_list), 2)
def test_delete_user(self):
'''
test_delete_user to test if we can remove a user from our user list
'''
self.new_user.save_user()
test_user = User("michael", "password")
test_user.save_user()
self.new_user.delete_user()
self.assertEqual(len(User.user_list), 1)
class TestContact(unittest.TestCase):
def setUp(self):
'''
Set up method to run before each test cases.
'''
self.new_user = User("lulu","aS2olidPas5") # create contact object
def test_init(self):
'''
test_init test case to test if the object is initialized properly
'''
self.assertEqual(self.new_user.user_name,"lulu")
print(self.new_user.user_name)
self.assertEqual(self.new_user.password,"aS2olidPas5")
print(self.new_user.password)
def test_save_users(self):
self.new_user.save_new_user() # save_new_user is defined in the user_class...
self.assertEqual(len(User.list_of_users),1)
print(len(User.list_of_users))
def tearDown(self):
'''
tearDown method that does clean up after each test case has run.
'''
User.list_of_users = []
def test_save_multiple_users(self):
'''
test_save_multiple_contact to check if we can save multiple contact
objects to our contact_list
'''
self.new_user.save_new_user()
test_user = User("juya","1234") # new user
test_user.save_new_user()
self.assertEqual(len(User.list_of_users),2)
def test_delete_user(self):
'''
test_delete_contact to test if we can remove a contact from our contact list
'''
self.new_user.save_new_user()
test_user = User("testy","Simpson") # new contact
test_user.save_new_user()
self.new_user.delete_user()# Deleting a contact object
self.assertEqual(len(User.list_of_users),1)
def test_find_user_by_name(self):
'''
test to check if we can find a user by name and display information
'''
self.new_user.save_new_user()
test_user = User("testy","Simpson")
test_user.save_new_user()
active_user = User.find_by_name("testy")
print(active_user.user_name)
self.assertEqual(active_user.user_name,test_user.user_name)
def test_user_exists(self):
self.new_user.save_new_user()
test_user = User("James","jamo") # new contact
test_user.save_new_user()
user_exists = User.user_exist("James")
self.assertTrue(user_exists)
def test_display_all_users(self):
self.assertEqual(User.display_users(),User.list_of_users)
