Add ExtraLateDay tests and LateDaysForUser.get() tests [no ci][wip]

eecs-autograder · Nov 21, 2024 · d983971 · d983971
1 parent e2fd933
commit d983971
Show file tree

Hide file tree

Showing 7 changed files with 532 additions and 22 deletions.
diff --git a/autograder/core/models/user_late_days.py b/autograder/core/models/user_late_days.py
@@ -12,6 +12,7 @@
 
 from autograder.rest_api.serialize_user import serialize_user
 
+
 class ExtraLateDays(AutograderModel):
     objects = AutograderModelManager['ExtraLateDays']()
 
@@ -59,7 +60,6 @@ def _days_late(group: Group, submission_timestamp: models.DateTimeField) -> int:
 
         return delta.days + 1 if delta > timedelta() else 0
 
-
     def to_dict(self):
         return {
             'user': serialize_user(self.user),
@@ -109,7 +109,11 @@ def get_many(users_queryset: models.QuerySet, course: Course) -> list["LateDaysF
 
         results = []
         for user in users_with_groups:
-            extra = user.late_days_for_course[0].extra_late_days if user.late_days_for_course else 0
+            if user.late_days_for_course:
+                extra = user.late_days_for_course[0].extra_late_days
+            else:
+                extra = 0
+
             used = 0
 
             for group in user.groups_with_submissions:

diff --git a/autograder/core/tests/test_models/test_course.py b/autograder/core/tests/test_models/test_course.py
@@ -4,7 +4,7 @@
 
 import autograder.core.utils as core_ut
 import autograder.utils.testing.model_obj_builders as obj_build
-from autograder.core.models import Course, LateDaysRemaining, Semester
+from autograder.core.models import Course, Semester
 from autograder.core.models.course import clear_cached_user_roles
 from autograder.utils.testing import UnitTestBase