Add fluent interface to test stages

Signed-off-by: l-1sqared <[email protected]>
TNG · Feb 9, 2022 · 3d2fc1c · 3d2fc1c
1 parent bc0a12b
commit 3d2fc1c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/jgiven-tests/src/main/java/com/tngtech/jgiven/tests/GivenTestStage.java b/jgiven-tests/src/main/java/com/tngtech/jgiven/tests/GivenTestStage.java
@@ -7,13 +7,15 @@ public GivenTestStage an_exception_is_thrown() {
         throw new RuntimeException("Some Exception");
     }
 
-    public void nothing() {
+    public GivenTestStage nothing() {
+        return this;
     }
 
-    public void a_failed_step(boolean fail) {
+    public GivenTestStage a_failed_step(boolean fail) {
         if (fail) {
             throw new IllegalArgumentException();
         }
+        return this;
     }
 
 

diff --git a/jgiven-tests/src/main/java/com/tngtech/jgiven/tests/WhenTestStage.java b/jgiven-tests/src/main/java/com/tngtech/jgiven/tests/WhenTestStage.java
@@ -8,7 +8,7 @@ public WhenTestStage something_happens() {
         return self();
     }
 
-    public void a_step_fails() {
+    public WhenTestStage a_step_fails() {
         throw new AssertionError( "assertion failed in test step" );
     }