Reset test status on success and set the exception on abort

Fixes #36250
quarkusio · Oct 25, 2023 · 38461d0 · 38461d0
1 parent a36b6a6
commit 38461d0
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/...k/junit5/src/main/java/io/quarkus/test/junit/AbstractQuarkusTestWithContextExtension.java b/...k/junit5/src/main/java/io/quarkus/test/junit/AbstractQuarkusTestWithContextExtension.java
@@ -52,6 +52,16 @@ public void testFailed(ExtensionContext context, Throwable cause) {
         markTestAsFailed(context, cause);
     }
 
+    @Override
+    public void testSuccessful(ExtensionContext context) {
+        markTestAsSuccessful(context);
+    }
+
+    @Override
+    public void testAborted(ExtensionContext context, Throwable cause) {
+        markTestAsAborted(context, cause);
+    }
+
     protected QuarkusTestExtensionState getState(ExtensionContext context) {
         ExtensionContext.Store store = getStoreFromContext(context);
         QuarkusTestExtensionState state = store.get(QuarkusTestExtensionState.class.getName(), QuarkusTestExtensionState.class);
@@ -92,4 +102,18 @@ protected void markTestAsFailed(ExtensionContext context, Throwable throwable) {
             state.setTestFailed(throwable);
         }
     }
+
+    protected void markTestAsSuccessful(ExtensionContext context) {
+        QuarkusTestExtensionState state = getState(context);
+        if (state != null) {
+            state.setTestSuccessful();
+        }
+    }
+
+    protected void markTestAsAborted(ExtensionContext context, Throwable throwable) {
+        QuarkusTestExtensionState state = getState(context);
+        if (state != null) {
+            state.setTestAborted(throwable);
+        }
+    }
 }
diff --git a/test-framework/junit5/src/main/java/io/quarkus/test/junit/QuarkusTestExtensionState.java b/test-framework/junit5/src/main/java/io/quarkus/test/junit/QuarkusTestExtensionState.java
@@ -49,6 +49,14 @@ public void close() throws IOException {
         }
     }
 
+    protected void setTestSuccessful() {
+        setTestFailed(null);
+    }
+
+    protected void setTestAborted(Throwable failure) {
+        setTestFailed(failure);
+    }
+
     protected void setTestFailed(Throwable failure) {
         try {
             this.testErrorCause = failure;