def get_user_info_by_email(self, email):
"""
Get user information by eamil for reset password
"""
user = User.get_useid_by_email(email)
if not user:
return False
return user
class TestUserDB(unittest.TestCase):
def setUp(self):
print("setUp test env")
# init test env
self.app = create_app()
db.create_all()
self.user = User()
# init user data
self.test_user = User(user_id=1,
user_name="test",
user_mail="*****@*****.**",
user_password="******")
db.session.add(self.test_user)
db.session.commit()
def tearDown(self):
print("tearDown")
db.session.remove()
db.drop_all()
def test_filter_register_user_email_and_name(self):
# Valid data
res = self.user.find_user_email_and_name(name="test",
email="*****@*****.**")
self.assertFalse(res)
# exist name in db
res = self.user.find_user_email_and_name(name="noexist",
email="*****@*****.**")
self.assertFalse(res)
# exist email in db
res = self.user.find_user_email_and_name(name="test",
email="*****@*****.**")
self.assertFalse(res)
# non exist user name and email in db
res = self.user.find_user_email_and_name(name="noexist",
email="*****@*****.**")
self.assertTrue(res)
def test_find_login_user_data(self):
# user put correct user info
user = self.user.find_user_info(name="test",
mail="*****@*****.**",
password="******")
self.assertEqual(user, self.test_user)
# user put wrong name
res = self.user.find_user_info(name="aaaa",
mail="*****@*****.**",
password="******")
self.assertFalse(res)
# user put wrong email
res = self.user.find_user_info(name="test",
mail="*****@*****.**",
password="******")
self.assertFalse(res)
# user put wrong password
res = self.user.find_user_info(name="test",
mail="*****@*****.**",
password="******")
self.assertFalse(res)
def test_find_user_by_id(self):
user = self.user.update_user_info(user_id=1)
self.assertEqual(user, self.test_user)
user = self.user.delete_user_info(user_id=1)
self.assertEqual(user, self.test_user)
def test_get_user_by_email(self):
user = self.user.get_useid_by_email(email="*****@*****.**")
self.assertEqual(user, self.test_user)
# non exist email in db
res = self.user.get_useid_by_email(email="*****@*****.**")
self.assertFalse(res)