def setup_method(self): app = RestApp() self._app_client = app.test_client() self.endpoint = "v1/sign-up" self.content_type = "application/json"
def setup_method(self): app = RestApp() self._app_client = app.test_client() @app.login_manager.user_loader def load_user(user_id): try: user = UserModel.get(user_id, is_deleted=False) except UserModel.DoesNotExist: return None else: return user self.endpoint = "v1/users" self.content_type = "application/json"
import random import string from flask import g, request from app import RestApp from app.v1.user.models import UserModel app = RestApp() @app.login_manager.user_loader def load_user(user_id): try: user = UserModel.get(user_id, is_deleted=False) except UserModel.DoesNotExist: return None else: return user @app.before_request def log_request_info(): log_id = "".join( random.choice(string.ascii_letters + string.digits) for _ in range(5)) g.log_id = log_id # TODO: mask password app.logger.info(f"LOG ID: {g.log_id} HEADERS: {request.headers}\n" f"LOG ID: {g.log_id} REQUEST METHOD: {request.method}\n" f"LOG ID: {g.log_id} REQUEST PARAMS: {request.args}\n" f"LOG ID: {g.log_id} REQUEST BODY: {request.data}")