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)
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)
def save_user(user): ''' function to save a new user ''' return User.save_user(user)