diff --git a/src/main/java/it/aboutbits/springboot/testing/validation/core/BaseValidationAssert.java b/src/main/java/it/aboutbits/springboot/testing/validation/core/BaseValidationAssert.java index af80537..5817814 100644 --- a/src/main/java/it/aboutbits/springboot/testing/validation/core/BaseValidationAssert.java +++ b/src/main/java/it/aboutbits/springboot/testing/validation/core/BaseValidationAssert.java @@ -6,6 +6,7 @@ import lombok.NonNull; import lombok.RequiredArgsConstructor; import lombok.Setter; +import org.apache.logging.log4j.util.TriConsumer; import java.util.HashSet; import java.util.Set; @@ -70,6 +71,18 @@ public R calling( ); return parent.ruleBuilder; } + + @SuppressWarnings({"unused", "unchecked"}) + public R calling( + @NonNull TriConsumer functionToCallWithParameter, + @NonNull A a, + @NonNull B b + ) { + parent.setFunctionToCallWithParameter( + p -> functionToCallWithParameter.accept(a, b, (P) p) + ); + return parent.ruleBuilder; + } } private void assertValidation() {