add forms, shipments tests

backend/src/users/service.py

@@ -1,5 +1,9 @@
 from sqlmodel import Session, select
+
 import src.models as models
+import src.messages as messages
+
+import src.users.exceptions as exceptions
 
 def get_all(
     session: Session,
@@ -49,6 +53,8 @@ def get_roles(session: Session):
     return session.exec(statement.order_by(models.ContractType.name)).all()
 
 def create_one(session: Session, user: models.UserCreate) -> models.UserPublic:
+    if user is None:
+        raise exceptions.UserCreateError(messages.Messages.invalid_input('user', 'input cannot be None'))
     new_user = models.User(
         name=user.name,
         email=user.email
@@ -63,12 +69,13 @@ def create_one(session: Session, user: models.UserCreate) -> models.UserPublic:
     return new_user
 
 def update_one(session: Session, id: int, user: models.UserCreate) -> models.UserPublic:
+    if user is None:
+        raise exceptions.UserCreateError(messages.s.invalid_input('user', 'input cannot be None'))
     statement = select(models.User).where(models.User.id == id)
     result = session.exec(statement)
     new_user = result.first()
     if not new_user:
-        return None
-
+        raise exceptions.UserNotFoundError(f'User {id} not found')
     new_user.email = user.email
     new_user.name = user.name
 
@@ -84,7 +91,7 @@ def delete_one(session: Session, id: int) -> models.UserPublic:
     result = session.exec(statement)
     user = result.first()
     if not user:
-        return None
+        raise exceptions.UserNotFoundError(f'User {id} not found')
     result = models.UserPublic.model_validate(user)
     session.delete(user)
     session.commit()