diff --git a/src/main/java/org/example/Main.java b/src/main/java/org/example/Main.java index e506a41..2154de4 100644 --- a/src/main/java/org/example/Main.java +++ b/src/main/java/org/example/Main.java @@ -85,15 +85,9 @@ public class Main { // View Grades from Student break; case 11: - // View User - break; - case 12: // Delete User break; - case 13: - // View Course List - break; - case 14: + case 12: // Exit (includes saving data in the files) exit = true; System.out.println("Exiting..."); diff --git a/src/test/java/org/example/AdministrationTest.java b/src/test/java/org/example/AdministrationTest.java index 647c024..0397e89 100644 --- a/src/test/java/org/example/AdministrationTest.java +++ b/src/test/java/org/example/AdministrationTest.java @@ -324,6 +324,33 @@ class AdministrationTest { } + + @Test + void dropProfFromCourse(){ + Professor prof = sampleProf; + Course course = sampleCourse; + List courseList = prof.getCoursesTaught(); + courseList.add(course); + prof.setCoursesTaught(courseList); + + //The "User input" to determine the professor + String profID = prof.getProfessorID(); + + //Grabs the coursesTaught List from the Professor the user wants to enroll + List courseList2 = prof.getCoursesTaught(); + //Assigning the course to the Prof by adding it the to the courseList + courseList2.remove(course); + //The Professor gets the updated courseList + prof.setCoursesTaught(courseList); + + + //Checks if the course got removed + assertEquals(0,prof.getCoursesTaught().size() ); + + } + + + @Test void registerUser() { Administration administration = new Administration();