Function TestFunctionDeclIncludeOverloads

cel/decls_test.go:876–963

(t *testing.T)

}

func TestFunctionDeclIncludeOverloads(t *testing.T) {
	funcs := []*decls.FunctionDecl{}
	for _, fn := range stdlib.Functions() {
		if fn.Name() == operators.Add {
			fn = fn.Subset(IncludeOverloads(overloads.AddInt64, overloads.AddDouble))
		}
		funcs = append(funcs, fn)
	}
	env, err := NewCustomEnv(FunctionDecls(funcs...))
	if err != nil {
		t.Fatalf("NewCustomEnv() failed: %v", err)
	}

	successTests := []struct {
		name string
		expr string
		want ref.Val
	}{
		{
			name: "ints",
			expr: "1 + 1",
			want: types.Int(2),
		},
		{
			name: "doubles",
			expr: "1.5 + 1.5",
			want: types.Double(3.0),
		},
	}
	for _, tst := range successTests {
		tc := tst
		t.Run(tc.name, func(t *testing.T) {
			ast, iss := env.Compile(tc.expr)
			if iss.Err() != nil {
				t.Fatalf("env.Compile() failed: %v", iss.Err())
			}
			prg, err := env.Program(ast)
			if err != nil {
				t.Fatalf("env.Program() failed: %v", err)
			}
			out, _, err := prg.Eval(NoVars())
			if err != nil {
				t.Fatalf("prg.Eval() errored: %v", err)
			}
			if out.Equal(tc.want) != types.True {
				t.Errorf("Eval() got %v, wanted %v", out, tc.want)
			}
		})
	}
	failureTests := []struct {
		name string
		expr string
	}{
		{
			name: "strings",
			expr: "'a' + 'b'",
		},
		{