bug fix in tests

CSC207-2022F-UofT · Dec 5, 2022 · 7eb5f53 · 7eb5f53
1 parent ca0c833
commit 7eb5f53
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/src/test/java/course_tracker_use_cases/GradeCalculatorTest.java b/src/test/java/course_tracker_use_cases/GradeCalculatorTest.java
@@ -52,9 +52,12 @@ void setup() {
         TaskMap.setTaskMap(tasks);
 
         Course course1 = new Course("testCourse", "", new ArrayList<>());
-        HashMap<String, Course> courses = new HashMap<>();
-        courses.put(course1.getCourseID(), course1);
-        this.courseAccess = new InMemoryCourse(courses);
+        HashMap<String, Course> courseMap = new HashMap<>();
+        courseMap.put(course1.getCourseID(), course1);
+        this.courseAccess = new InMemoryCourse(courseMap);
+        ArrayList<String> courseList = new ArrayList<>();
+        courseList.add("testCourse");
+        ((StudentUser) CurrentUser.getCurrentUser()).setCourses(courseList);
 
     }
 
@@ -131,7 +134,7 @@ public GradeCalculatorResponseModel failView(String error) {
         ArrayList<String> ungradedTasks = new ArrayList<>();
         ungradedTasks.add("testAssignment2");
         ungradedTasks.add("testTest1");
-        ungradedTasks.add("testTest2");
+            ungradedTasks.add("testTest2");
         GradeCalculatorRequestModel request = new GradeCalculatorRequestModel("testCourse",
                 "77,89,90", ungradedTasks);
 

diff --git a/src/test/java/course_tracker_use_cases/ProgressTrackerTest.java b/src/test/java/course_tracker_use_cases/ProgressTrackerTest.java
@@ -9,6 +9,7 @@
 import use_cases.course_tracker.progress_tracker_use_case.ProgressTrackerRequestModel;
 import use_cases.course_tracker.progress_tracker_use_case.ProgressTrackerResponseModel;
 
+import java.lang.reflect.Array;
 import java.util.ArrayList;
 import java.util.HashMap;
 
@@ -54,9 +55,12 @@ void setup() {
         TaskMap.setTaskMap(tasks);
 
         Course course1 = new Course("testCourse", "", new ArrayList<>());
-        HashMap<String, Course> courses = new HashMap<>();
-        courses.put(course1.getCourseID(), course1);
-        this.courseAccess = new InMemoryCourse(courses);
+        HashMap<String, Course> courseMap = new HashMap<>();
+        courseMap.put(course1.getCourseID(), course1);
+        this.courseAccess = new InMemoryCourse(courseMap);
+        ArrayList<String> courseList = new ArrayList<>();
+        courseList.add("testCourse");
+        ((StudentUser) CurrentUser.getCurrentUser()).setCourses(courseList);
     }
 
     /**