feat: allow the test function of Until does not have parameters.

Author: ghosind

until.go

@@ -32,14 +32,18 @@ func UntilWithContext(ctx context.Context, testFn, fn AsyncFn) ([]any, error) {
 
 // until repeatedly calls the function until the test function returns true.
 func until(parent context.Context, testFn, fn AsyncFn) ([]any, error) {
-	validateUntilFuncs(testFn, fn)
+	isNoParam := validateUntilFuncs(testFn, fn)
 
 	ctx := getContext(parent)
 
 	for {
 		out, _ := invokeAsyncFn(fn, ctx, nil)
 
-		testOut, testErr := invokeAsyncFn(testFn, ctx, out)
+		params := out
+		if isNoParam {
+			params = nil
+		}
+		testOut, testErr := invokeAsyncFn(testFn, ctx, params)
 		if testErr != nil {
 			return out, testErr
 		}
@@ -52,7 +56,7 @@ func until(parent context.Context, testFn, fn AsyncFn) ([]any, error) {
 }
 
 // validateUntilFuncs validates the test function and the execution function.
-func validateUntilFuncs(testFn, fn AsyncFn) {
+func validateUntilFuncs(testFn, fn AsyncFn) (isNoParam bool) {
 	if testFn == nil || fn == nil {
 		panic(ErrNotFunction)
 	}
@@ -74,9 +78,12 @@ func validateUntilFuncs(testFn, fn AsyncFn) {
 		numIn--
 		ii++
 	}
-	if numIn != ft.NumOut() {
+	if numIn != 0 && numIn != ft.NumOut() {
 		panic(ErrInvalidTestFunc)
 	}
+	if numIn == 0 {
+		return true
+	}
 
 	for oi < numIn {
 		it := tft.In(ii) // type of the value in the test function input parameters list
@@ -89,4 +96,6 @@ func validateUntilFuncs(testFn, fn AsyncFn) {
 		ii++
 		oi++
 	}
+
+	return false
 }

until_test.go

@@ -47,6 +47,9 @@ func TestUntilInvalidParameters(t *testing.T) {
 	a.NotPanicNow(func() {
 		Until(func(ctx context.Context, err error) bool { return true }, func() error { return nil })
 	})
+	a.NotPanicNow(func() {
+		Until(func(ctx context.Context) bool { return true }, func() error { return nil })
+	})
 	a.PanicOfNow(func() {
 		Until(func(ctx context.Context, i int) bool { return true }, func() error { return nil })
 	}, ErrInvalidTestFunc)