feat: terminate Until when test function returns error.

Author: ghosind

until.go

@@ -37,12 +37,16 @@ func until(parent context.Context, testFn, fn AsyncFn) ([]any, error) {
 	ctx := getContext(parent)
 
 	for {
-		out, err := invokeAsyncFn(fn, ctx, nil)
+		out, _ := invokeAsyncFn(fn, ctx, nil)
+
+		testOut, testErr := invokeAsyncFn(testFn, ctx, out)
+		if testErr != nil {
+			return out, testErr
+		}
 
-		testOut, _ := invokeAsyncFn(testFn, ctx, out)
 		isDone := testOut[0].(bool)
 		if isDone {
-			return out, err
+			return out, nil
 		}
 	}
 }

until_test.go

@@ -2,6 +2,7 @@ package async
 
 import (
 	"context"
+	"errors"
 	"testing"
 	"time"
 
@@ -54,6 +55,35 @@ func TestUntilInvalidParameters(t *testing.T) {
 	}, ErrInvalidTestFunc)
 }
 
+func TestUntilWithFunctionError(t *testing.T) {
+	a := assert.New(t)
+	count := 0
+	unexpectedErr := errors.New("unexpected error")
+
+	out, err := Until(func(c int, err error) bool {
+		return c == 5
+	}, func() (int, error) {
+		count++
+		return count, unexpectedErr
+	})
+	a.NilNow(err)
+	a.EqualNow(out, []any{5, unexpectedErr})
+}
+
+func TestUntilWithTestFunctionError(t *testing.T) {
+	a := assert.New(t)
+	expectedErr := errors.New("expected error")
+
+	out, err := Until(func(n int) bool {
+		panic(expectedErr)
+	}, func() int {
+		return 0
+	})
+	a.NotNilNow(err)
+	a.EqualNow(err, expectedErr)
+	a.EqualNow(out, []any{0})
+}
+
 func TestUntilWithContext(t *testing.T) {
 	a := assert.New(t)
 	ctx, canFunc := context.WithTimeout(context.Background(), 100*time.Millisecond)