Fix #20: When calling System.exit() multiple times in a test, repor…

…t the first exit code only (#21)

CHANGELOG.md

@@ -1,7 +1,7 @@
 # Change Log for `junit5-system-exit`
 
 ### 2.0.1
-- TBD
+- Bugfix: [[#20]](https://github.com/tginsberg/junit5-system-exit/issues/20): Multiple calls to `System.exit()` do not always report the first exit status code.
 
 ### 2.0.0
 - Remove terminally deprecated `SecurityManager` approach for preventing `System.exit()` calls.

src/main/java/com/ginsberg/junit/exit/agent/AgentSystemExitHandlerStrategy.java

@@ -39,7 +39,7 @@ public static void handleExit(final int status) {
             if (firstExitStatusCode == null) {
                 firstExitStatusCode = status;
             }
-            throw new SystemExitPreventedException(status);
+            throw new SystemExitPreventedException(firstExitStatusCode);
         } else {
             System.exit(status);
         }

src/test/java/com/ginsberg/junit/exit/assertions/SystemExitAssertionTest.java

@@ -102,6 +102,18 @@ void failsWhenNoExit() {
         }
     }
 
+    @Test
+    void multipleCallsToExit() {
+        assertThatCallsSystemExit(() -> {
+            try {
+                System.exit(42);
+                System.exit(1);
+            } catch (final Exception e) {
+                System.exit(2);
+            }
+        }).withExitCode(42);
+    }
+
     private void justExit() {
         System.exit(42);
     }