add tests forn forms, products, productors

backend/src/forms/service.py

@@ -1,7 +1,9 @@
 from sqlmodel import Session, select
-import src.models as models
 from sqlalchemy import func
 
+import src.models as models
+import src.forms.exceptions as exceptions
+
 def get_all(
     session: Session, 
     seasons: list[str],
@@ -46,6 +48,12 @@ def get_one(session: Session, form_id: int) -> models.FormPublic:
     return session.get(models.Form, form_id)
 
 def create_one(session: Session, form: models.FormCreate) -> models.FormPublic:
+    if not form:
+        raise exceptions.FormCreateError('FormCreate input cannot be None')
+    if not session.get(models.Productor, form.productor_id):
+        raise exceptions.ProductorNotFoundError(f'Productor {form.productor_id} not found')
+    if not session.get(models.User, form.referer_id):
+        raise exceptions.UserNotFoundError(f'User {form.referer_id} not found')
     form_create = form.model_dump(exclude_unset=True)
     new_form = models.Form(**form_create)
     session.add(new_form)
@@ -58,7 +66,11 @@ def update_one(session: Session, id: int, form: models.FormUpdate) -> models.For
     result = session.exec(statement)
     new_form = result.first()
     if not new_form:
-        return None
+        raise exceptions.FormNotFoundError(f'Form {id} not found')
+    if form.productor_id and not session.get(models.Productor, form.productor_id):
+        raise exceptions.ProductorNotFoundError(f'Productor {form.productor_id} not found')
+    if form.referer_id and not session.get(models.User, form.referer_id):
+        raise exceptions.UserNotFoundError(f'User {form.referer_id} not found')
     form_updates = form.model_dump(exclude_unset=True)
     for key, value in form_updates.items():
         setattr(new_form, key, value)
@@ -72,7 +84,7 @@ def delete_one(session: Session, id: int) -> models.FormPublic:
     result = session.exec(statement)
     form = result.first()
     if not form:
-        return None
+        raise exceptions.FormNotFoundError(f'Form {id} not found')
     result = models.FormPublic.model_validate(form)
     session.delete(form)
     session.commit()