Function TestRetryBindingCancellation

cel/async/async_test.go:101–130

(t *testing.T)

}

func TestRetryBindingCancellation(t *testing.T) {
	var attempts atomic.Int32
	op := buildZeroArgAsync(t, async.RetryBinding(func(ctx context.Context, args ...ref.Val) ref.Val {
		attempts.Add(1)
		return types.WrapErr(retryableTestErr{})
	}, async.RetryAttempts(5), async.RetryBackoff(500*time.Millisecond)))

	ctx, cancel := context.WithCancel(context.Background())
	go func() {
		time.Sleep(40 * time.Millisecond)
		cancel()
	}()

	start := time.Now()
	var res ref.Val
	select {
	case res = <-op(ctx):
	case <-time.After(2 * time.Second):
		t.Fatal("retry op did not return after cancellation")
	}
	elapsed := time.Since(start)

	if !types.IsError(res) || !strings.Contains(res.(*types.Err).Error(), "cancelled") {
		t.Errorf("result = %v, want a cancellation error", res)
	}
	// Cancellation must interrupt the backoff wait rather than running it to completion.
	if elapsed >= 500*time.Millisecond {
		t.Errorf("retry waited the full backoff (%v); cancellation did not interrupt it", elapsed)
	}
}

func TestRetryNonRetryableError(t *testing.T) {
	var attempts atomic.Int32