class TestUser(unittest.TestCase):
'''
Test class that defines test cases for the user class behaviours.
'''
def setUp(self):
'''
Set up method to run before each test cases.
'''
self.new_user = User("Jadiel", "Muriithi", "0727867684",
"*****@*****.**", "sirmwas", "101010")
# create user object
def test_init(self):
'''
test_init to test if the object is initialized properly
'''
self.assertEqual(self.new_user.first_name, "Jadiel")
self.assertEqual(self.new_user.second_name, "Muriithi")
self.assertEqual(self.new_user.phone_number, "0727867684")
self.assertEqual(self.new_user.email, "*****@*****.**")
self.assertEqual(self.new_user.user_name, "sirmwas")
self.assertEqual(self.new_user.password, "101010")
def test_save_user(self):
'''
To test if the user object is saved into the user list,
'''
self.new_user.save_user() #saving the new user
self.assertEqual(len(User.user_list), 1)
def tearDown(self):
'''
tearDown method that does clean up after each test case has run
'''
User.user_list = []
def test_delete_user(self):
'''
Method to test whether we can delete a user from our user's list
'''
self.new_user.save_user()
test_user = User("Jadiel", "Muriithi", "0727867684",
"*****@*****.**", "sirmwas",
"101010") #new user
test_user.save_user()
self.new_user.delete_user() #Deleting a user
self.assertEqual(len(User.user_list), 1)
