def load_user(user_id):
logger.info("Loading a user: ID=%s" % user_id)
db = AndrewDB()
res = db.get_user_by_id(user_id)
if not res:
logger.warning("User not found: ID=%s" % user_id)
return None
return User(res['user_id'], res['email'], res['password'], res['role'])
def test_get_user_by_id(mock_connect):
with allure.step('Get user by id'):
with app.app_context():
db = AndrewDB()
expected = {1: 'user1'}
mock_connect().cursor.return_value.fetchone.return_value = expected
result = db.get_user_by_id(1)
assert result == expected
