def test_find_user_by_account_name(self): ''' test to check if we can find user by account name and display information ''' self.new_user.save_user() test_user = User("Roman", "Facebook", "@roman.com", "reigns18") test_user.save_user() found_user = User.find_by_account_name("Facebook") self.assertEqual(found_user.password, test_user.password)
def find_user(account_name): ''' Function for returning user ''' return User.find_by_account_name(account_name)