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