Function TestConcurrentEvalRecover

cel/program_async_test.go:816–916

(t *testing.T)

}

func TestConcurrentEvalRecover(t *testing.T) {
	env, err := cel.NewEnv(
		cel.Function("panic",
			cel.Overload("global_panic", []*cel.Type{}, cel.BoolType,
				cel.FunctionBinding(func(args ...ref.Val) ref.Val {
					panic("watch me recover")
				}),
			),
		),
		cel.Function("cancel_panic",
			cel.Overload("global_cancel_panic", []*cel.Type{}, cel.BoolType,
				cel.FunctionBinding(func(args ...ref.Val) ref.Val {
					panic(interpreter.EvalCancelledError{Message: "eval cancelled", Cause: interpreter.ContextCancelled})
				}),
			),
		),
		cel.Function("sleep_func",
			cel.Overload("global_sleep_func", []*cel.Type{}, cel.BoolType,
				cel.AsyncBinding(func(ctx context.Context, args ...ref.Val) ref.Val {
					time.Sleep(1 * time.Second)
					return types.True
				}),
			),
		),
	)
	if err != nil {
		t.Fatalf("cel.NewEnv() failed: %v", err)
	}

	tests := []struct {
		name    string
		expr    string
		prgOpts []cel.ProgramOption
		getCtx  func() (context.Context, context.CancelFunc)
		wantErr any
	}{
		{
			name:    "panic",
			expr:    "panic()",
			wantErr: "internal error: watch me recover",
		},
		{
			name:    "panic_tracked_state",
			expr:    "panic()",
			prgOpts: []cel.ProgramOption{cel.EvalOptions(cel.OptTrackState)},
			wantErr: "internal error: watch me recover",
		},
		{
			name:    "eval_cancelled_error",
			expr:    "cancel_panic()",
			wantErr: &interpreter.EvalCancelledError{},
		},
		{
			name: "context_timeout",
			expr: "sleep_func()",
			getCtx: func() (context.Context, context.CancelFunc) {
				return context.WithTimeout(context.Background(), 10*time.Millisecond)
			},