add tests forn forms, products, productors

backend/src/forms/forms.py

@@ -4,6 +4,7 @@ import src.models as models
 from src.database import get_session
 from sqlmodel import Session
 import src.forms.service as service
+import src.forms.exceptions as exceptions
 from src.auth.auth import get_current_user
 
 router = APIRouter(prefix='/forms')
@@ -40,7 +41,15 @@ async def create_form(
     user: models.User = Depends(get_current_user),
     session: Session = Depends(get_session)
 ):
-    return service.create_one(session, form)
+    try:
+        form = service.create_one(session, form)
+    except exceptions.ProductorNotFoundError:
+        raise HTTPException(status_code=404, detail=messages.productornotfound)
+    except exceptions.UserNotFoundError:
+        raise HTTPException(status_code=404, detail=messages.usernotfound)
+    except exceptions.FormCreateError:
+        raise HTTPException(status_code=400, detail=messages.forminputinvalid)
+    return form
 
 @router.put('/{id}', response_model=models.FormPublic)
 async def update_form(
@@ -48,9 +57,14 @@ async def update_form(
     user: models.User = Depends(get_current_user),
     session: Session = Depends(get_session)
 ):
-    result = service.update_one(session, id, form)
-    if result is None:
+    try:
+        result = service.update_one(session, id, form)
+    except exceptions.FormNotFoundError:
         raise HTTPException(status_code=404, detail=messages.notfound)
+    except exceptions.ProductorNotFoundError:
+        raise HTTPException(status_code=404, detail=messages.productornotfound)
+    except exceptions.UserNotFoundError:
+        raise HTTPException(status_code=404, detail=messages.usernotfound)
     return result
 
 @router.delete('/{id}', response_model=models.FormPublic)
@@ -59,7 +73,8 @@ async def delete_form(
     user: models.User = Depends(get_current_user),
     session: Session = Depends(get_session)
 ):
-    result = service.delete_one(session, id)
-    if result is None:
+    try:
+        result = service.delete_one(session, id)
+    except exceptions.FormNotFoundError:
         raise HTTPException(status_code=404, detail=messages.notfound)
     return result