Write most backend tests

This commit is contained in:
lvrossem
2023-04-10 14:07:30 -06:00
parent 8f3c303a2b
commit e7145369b5
6 changed files with 348 additions and 18 deletions

View File

@@ -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
]