Coverage for app/api/v1/category.py: 100%

1from fastapi import APIRouter, Depends 

2from sqlalchemy.orm import Session 

3from app.core.database import get_db 

4from app.schemas.category_schema import CategoryCrate, CategoryOut 

5from app.services import category_service 

6from app.utils.response import success_response 

7 

8router = APIRouter() 

9 

10@router.post("/category") 

11def create_category(category: CategoryCrate, db: Session = Depends(get_db)): 

12 result = category_service.create_category_service(db, category) 

13 return success_response(data=CategoryOut.model_validate(result).model_dump(), message="OK", status_code=201) 

14 

15@router.get("/category") 

16def get_all_category(db: Session = Depends(get_db)): 

17 category = category_service.get_all_category_service(db) 

18 return success_response(data=[CategoryOut.model_validate(t).model_dump() for t in category]) 

19 

20@router.get("/category/{id}") 

21def get_category_by_id(id: int, db: Session = Depends(get_db)): 

22 category = category_service.get_category_by_id_service(db, id) 

23 if not category: 

24 return success_response(message="Not found", status_code=404) 

25 return success_response(data=CategoryOut.model_validate(category).model_dump()) 

26 

27@router.put("/category/{id}") 

28def update_category(id: int, data: CategoryCrate, db: Session = Depends(get_db)): 

29 result = category_service.update_category_service(db, id, data) 

30 if not result: 

31 return success_response(message="Not found", status_code=404) 

32 return success_response(data=CategoryOut.model_validate(result).model_dump(), message="OK") 

33 

34@router.delete("/category/{id}") 

35def delete_category(id: int, db: Session = Depends(get_db)): 

36 success = category_service.delete_category_service(db, id) 

37 if not success: 

38 return success_response(message="Not found", status_code=404) 

39 return success_response(message="Deleted")