Write most backend tests

src/crud/courseprogress.py

@@ -1,6 +1,7 @@
 from sqlalchemy.orm import Session
+from fastapi import HTTPException
 
-from src.enums import CourseEnum, course_enum_list
+from src.enums import CourseEnum
 from src.models import CourseProgress, User
 from src.schemas.courseprogress import CourseProgressBase
 
@@ -41,13 +42,19 @@ def get_course_progress(db: Session, user: User, course: CourseEnum):
 
 def initialize_user(db: Session, user: User):
     """Create CourseProgress records with a value of 0 for a new user"""
-    for course in course_enum_list:
-        db.add(CourseProgress(progress_value=0.0, course=course, owner_id=user.user_id))
-        db.commit()
+    for course in CourseEnum:
+        if course != CourseEnum.All:
+            db.add(
+                CourseProgress(progress_value=0.0, course=course, owner_id=user.user_id)
+            )
+            db.commit()
 
 
 def patch_course_progress(db: Session, user: User, course_progress: CourseProgressBase):
     """Change the progress value for a given course"""
+    if course_progress.progress_value > 1 or course_progress.progress_value < 0:
+        raise HTTPException(status_code=400, detail="Invalid progress value")
+
     db_course_progress_list = []
     if course_progress.course != CourseEnum.All:
         db_course_progress_list = (
@@ -64,7 +71,12 @@ def patch_course_progress(db: Session, user: User, course_progress: CourseProgre
             .filter(CourseProgress.owner_id == user.user_id)
             .all()
         )
-
+    print(f"LENGTH OF LIST OF {course_progress.course}: {len(db_course_progress_list)}")
     for db_course_progress in db_course_progress_list:
         db_course_progress.progress_value = course_progress.progress_value
         db.commit()
+
+    return [
+        CourseProgressBase(course=db_cp.course, progress_value=db_cp.progress_value)
+        for db_cp in db_course_progress_list
+    ]

src/enums.py

@@ -34,15 +34,3 @@ class CourseEnum(StrEnum):
     Colors = "Colors"
     FruitsVegetables = "FruitsVegetables"
     All = "All"
-
-
-# This is needed because for some reason iterating over an enum doesn't work properly...
-course_enum_list = [
-    CourseEnum.Fingerspelling,
-    # CourseEnum.Basics,
-    CourseEnum.Hobbies,
-    CourseEnum.Animals,
-    CourseEnum.Colors,
-    CourseEnum.FruitsVegetables,
-    CourseEnum.All,
-]

src/main.py

@@ -92,7 +92,7 @@ async def get_course_progress(
     return crud_courseprogress.get_course_progress(db, current_user, course)
 
 
-@app.patch("/courseprogress")
+@app.patch("/courseprogress", response_model=List[courseprogress.CourseProgressBase])
 async def patch_course_progress(
     course_progress: courseprogress.CourseProgressBase,
     current_user_name: str = Depends(crud_authentication.get_current_user_name),