Add tests

potaotototo · Oct 29, 2024 · 90c28ea · 90c28ea
1 parent 8bb31e9
commit 90c28ea
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 0 deletions.
diff --git a/src/test/java/tuteez/logic/commands/AddCommandTest.java b/src/test/java/tuteez/logic/commands/AddCommandTest.java
@@ -191,6 +191,11 @@ public void updateLastViewedPerson(Person personOnDisplay) {
             throw new AssertionError("This method should not be called.");
         }
 
+        @Override
+        public void removeLastViewedPerson() {
+            throw new AssertionError("This method should not be called.");
+        }
+
         @Override
         public Person findPersonByName(Name name) {
             throw new AssertionError("This method should not be called.");

diff --git a/src/test/java/tuteez/logic/commands/DeleteCommandTest.java b/src/test/java/tuteez/logic/commands/DeleteCommandTest.java
@@ -107,6 +107,52 @@ public void execute_invalidName_throwsCommandException() {
         assertCommandFailure(deleteCommand, model, expectedMessage);
     }
 
+    @Test
+    public void execute_deleteLastViewedPerson_removesLastViewedPerson() {
+        Person personToDelete = model.getFilteredPersonList().get(INDEX_FIRST_PERSON.getZeroBased());
+        Model expectedModel = new ModelManager(model.getAddressBook(), new UserPrefs());
+
+        model.updateLastViewedPerson(personToDelete);
+        expectedModel.updateLastViewedPerson(personToDelete);
+
+        DeleteCommand deleteCommand = new DeleteCommand(INDEX_FIRST_PERSON);
+        String expectedMessage = String.format(DeleteCommand.MESSAGE_DELETE_PERSON_SUCCESS,
+                Messages.format(personToDelete));
+
+        expectedModel.deletePerson(personToDelete);
+        expectedModel.removeLastViewedPerson();
+
+        assertCommandSuccess(deleteCommand, model, expectedMessage, expectedModel);
+
+        // Verify last viewed person was removed
+        assertFalse(model.getLastViewedPerson().get().isPresent(),
+                "Expected last viewed person to be removed after deleting the viewed person");
+    }
+
+    @Test
+    public void execute_deleteNonLastViewedPerson_keepsLastViewedPerson() {
+        Person personToDelete = model.getFilteredPersonList().get(INDEX_FIRST_PERSON.getZeroBased());
+        Person lastViewedPerson = model.getFilteredPersonList().get(INDEX_SECOND_PERSON.getZeroBased());
+        Model expectedModel = new ModelManager(model.getAddressBook(), new UserPrefs());
+
+        model.updateLastViewedPerson(lastViewedPerson);
+        expectedModel.updateLastViewedPerson(lastViewedPerson);
+
+        DeleteCommand deleteCommand = new DeleteCommand(INDEX_FIRST_PERSON);
+        String expectedMessage = String.format(DeleteCommand.MESSAGE_DELETE_PERSON_SUCCESS,
+                Messages.format(personToDelete));
+
+        expectedModel.deletePerson(personToDelete);
+
+        assertCommandSuccess(deleteCommand, model, expectedMessage, expectedModel);
+
+        // Verify last viewed person was not removed
+        assertTrue(model.getLastViewedPerson().get().isPresent(),
+                "Expected last viewed person to be retained when deleting a different person");
+        assertEquals(lastViewedPerson, model.getLastViewedPerson().get().get(),
+                "Expected last viewed person to remain unchanged");
+    }
+
     @Test
     public void equals() {
         DeleteCommand deleteFirstCommand = new DeleteCommand(INDEX_FIRST_PERSON);