Function TestConcurrentEvalProgramThreadSafety

cel/program_async_test.go:603–638

(t *testing.T)

}

func TestConcurrentEvalProgramThreadSafety(t *testing.T) {
	prg := mustProgram(t, `async_func(x) + 1`,
		cel.Variable("x", cel.IntType),
		cel.Function("async_func",
			cel.Overload("async_func_int", []*cel.Type{cel.IntType}, cel.IntType,
				cel.AsyncBinding(func(ctx context.Context, args ...ref.Val) ref.Val {
					time.Sleep(5 * time.Millisecond)
					return args[0]
				}),
			),
		),
	)

	const numGoroutines = 10
	errCh := make(chan error, numGoroutines)
	for i := range numGoroutines {
		go func(val int64) {
			res := awaitEval(t, prg, context.Background(), map[string]any{"x": val})
			if res.Err != nil {
				errCh <- res.Err
				return
			}
			if res.Val.Equal(types.Int(val+1)) != types.True {
				errCh <- errors.New("unexpected eval result")
				return
			}
			errCh <- nil
		}(int64(i * 10))
	}

	for range numGoroutines {
		if err := <-errCh; err != nil {
			t.Errorf("Concurrent thread safety evaluation failed: %v", err)
		}
	}
}

func TestConcurrentEvalPreCanceledContext(t *testing.T) {
	prg := mustProgram(t, `async_func(42)`,