Add a test for the quit button

Author: Arun Prasaad

src/main/java/ihm/Editor.java

@@ -229,21 +229,22 @@ private void createQuitButton(String name) {
         JButton quit = new JButton("Quit");
         quit.setName(Component.QUIT_BUTTON.name());
         quit.setBounds(windowWidth + 20, 250, 110, 30);
-
-        quit.addActionListener(new ActionListener() {
-            @Override
-            public void actionPerformed(ActionEvent e) {
-                try {
-                    File level_drop = new File(new File(".").getCanonicalPath() + "/levels/" + name + ".txt");
-                    level_drop.delete();
-                } catch (IOException e1) {
-                    LOGGER.log(Level.SEVERE, "Error while deleting level file: " + name, e1);
-                }
-                System.exit(0);
+        quit.addActionListener(_ -> {
+            try {
+                File level_drop = new File(new File(".").getCanonicalPath() + "/levels/" + name + ".txt");
+                level_drop.delete();
+            } catch (IOException e) {
+                LOGGER.log(Level.SEVERE, "Error while deleting level file: " + name, e);
             }
+            defaultExitHandler().exit(ExitHandler.SUCCESS);
         });
         this.add(quit);
     }
+    
+    @VisibleForTesting
+    ExitHandler defaultExitHandler() {
+        return System::exit;
+    }
 
     private void createPlayerButton() {
         JButton button = new JButton(new ImageIcon("img/Joueur.jpg"));

src/test/java/ihm/EditorTest.java

@@ -10,11 +10,14 @@
 import java.awt.event.MouseListener;
 import java.awt.event.MouseMotionListener;
 import java.io.IOException;
+import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.Arrays;
 
 import static java.nio.file.Files.deleteIfExists;
 import static org.assertj.core.api.BDDAssertions.then;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
 import static swing.ComponentFinder.findComponentByNameAsType;
 
 @NullMarked
@@ -31,6 +34,7 @@ class EditorTest {
     void setUp() throws Exception {
         // Create a new Editor instance for testing
         editor = new Editor(TEST_ROWS, TEST_COLUMNS, TEST_LEVEL_NAME);
+        deleteIfExists(TEST_LEVEL_PATH);
     }
 
     @AfterEach
@@ -162,10 +166,38 @@ void target_button_sets_content_to_storage_area() {
     void empty_button_sets_content_to_floor() {
         // Given
         JButton button = findComponentByNameAsType(editor, Editor.Component.EMPTY_BUTTON.name(), JButton.class);
-        
-        // When - Simulate button click
+
+        // When
         button.doClick();
-        
+
+        // Then
         then(editor.getContent()).isEqualTo(TileType.FLOOR);
     }
+
+    @Test
+    void quit_button_triggers_exit_handler_and_deletes_file() throws IOException {
+        // Given
+        ExitHandler mockExitHandler = mock(ExitHandler.class);
+
+        // Create a test level file that should be deleted
+        Files.createDirectories(TEST_LEVEL_PATH.getParent());
+        Files.createFile(TEST_LEVEL_PATH);
+        then(Files.exists(TEST_LEVEL_PATH)).isTrue();
+
+        editor = new Editor(TEST_ROWS, TEST_COLUMNS, TEST_LEVEL_NAME) {
+            @Override
+            ExitHandler defaultExitHandler() {
+                return mockExitHandler;
+            }
+        };
+
+        JButton quitButton = findComponentByNameAsType(editor, Editor.Component.QUIT_BUTTON.name(), JButton.class);
+
+        // When
+        quitButton.doClick();
+
+        // Then
+        then(Files.exists(TEST_LEVEL_PATH)).isFalse();
+        verify(mockExitHandler).exit(ExitHandler.SUCCESS);
+    }
 }