add forms, shipments tests

backend/src/users/users.py

@@ -5,6 +5,7 @@ from src.database import get_session
 from sqlmodel import Session
 import src.users.service as service
 from src.auth.auth import get_current_user
+import src.users.exceptions as exceptions
 
 router = APIRouter(prefix='/users')
 
@@ -36,7 +37,7 @@ def get_users(
 ):
     result = service.get_one(session, id)
     if result is None:
-        raise HTTPException(status_code=404, detail=messages.notfound)
+        raise HTTPException(status_code=404, detail=messages.Messages.not_found('user'))
     return result
 
 @router.post('', response_model=models.UserPublic)
@@ -45,7 +46,11 @@ def create_user(
     logged_user: models.User = Depends(get_current_user),
     session: Session = Depends(get_session)
 ):
-    return service.create_one(session, user)
+    try:
+        user = service.create_one(session, user)
+    except exceptions.UserCreateError as error:
+        raise HTTPException(status_code=400, detail=str(error))
+    return user
 
 @router.put('/{id}', response_model=models.UserPublic)
 def update_user(
@@ -54,9 +59,10 @@ def update_user(
     logged_user: models.User = Depends(get_current_user),
     session: Session = Depends(get_session)
 ):
-    result = service.update_one(session, id, user)
-    if result is None:
-        raise HTTPException(status_code=404, detail=messages.notfound)
+    try:
+        result = service.update_one(session, id, user)
+    except exceptions.UserNotFoundError as error:
+        raise HTTPException(status_code=404, detail=messages.Messages.not_found('user'))
     return result
 
 @router.delete('/{id}', response_model=models.UserPublic)
@@ -65,7 +71,8 @@ def delete_user(
     user: models.User = Depends(get_current_user),
     session: Session = Depends(get_session)
 ):
-    result = service.delete_one(session, id)
-    if result is None:
-        raise HTTPException(status_code=404, detail=messages.notfound)
+    try:
+        result = service.delete_one(session, id)
+    except exceptions.UserNotFoundError as error:
+        raise HTTPException(status_code=404, detail=messages.Messages.not_found('user'))
     return result