Coverage for app/utils/response_format.py: 100%

1from fastapi.responses import JSONResponse 

2from app.utils.typing import JSONObject, JSONValue 

3 

4def generate_response(status_code: int, data: JSONObject | JSONValue | None, message: str): 

5 response = JSONResponse( 

6 content={ 

7 "success": True, 

8 "data": data, 

9 "message": message, 

10 "error": None 

11 }, 

12 status_code=status_code 

13 ) 

14 return response 

15 

16def generate_error_response(status_code: int, error_code: str, error_message: str, message: str): 

17 response = JSONResponse( 

18 content={ 

19 "success": False, 

20 "data": None, 

21 "message": message, 

22 "error": { 

23 "code": error_code, 

24 "message": error_message 

25 } 

26 }, 

27 status_code=status_code 

28 ) 

29 return response