Function TestConstantFoldingOptimizerTwoVar

cel/optimizer_test.go:341–443

(t *testing.T)

}

func TestConstantFoldingOptimizerTwoVar(t *testing.T) {
	tests := []struct {
		expr        string
		folded      string
		knownValues map[string]any
	}{
		{
			expr:   `[1 + 1, 1 + 2].all(i, v, i < 10 && v < 10)`,
			folded: `true`,
		},
		{
			expr:   `[1 + 1, 1 + 2].all(i, v, i < 10 && v < x)`,
			folded: `[2, 3].all(i, v, i < 10 && v < x)`,
		},
		{
			expr:   `[1, 2, 3].transformList(i, v, i + v)`,
			folded: `[1, 3, 5]`,
		},
		{
			expr:   `{'a': 1, 'b': 2}.all(k, v, v > 0)`,
			folded: `true`,
		},
		{
			expr:   `v > 0 && [1 + 1, 1 + 2].all(i, v, i < 10 && v < 10)`,
			folded: `v > 0`,
		},
		{
			expr:   `[1 + 1, 1 + 2].all(i, v, i < 10 && v < 10) && v > 0`,
			folded: `v > 0`,
		},
		{
			expr:   `[1, 2, 3].transformList(i, v, i + v + k)`,
			folded: `[1, 2, 3].transformList(i, v, i + v + k)`,
		},
		{
			expr:   `l.transformList(i, v, i + v)`,
			folded: `l.transformList(i, v, i + v)`,
		},
		{
			expr:   `v > 0 && [1 + 1, 1 + 2].all(i, v, i < 10 && v < 10)`,
			folded: `true`,
			knownValues: map[string]any{
				"v": 5,
			},
		},
		{
			expr:   `[1 + 1, 1 + 2].all(i, v, i < 10 && v < .v)`,
			folded: `true`,
			knownValues: map[string]any{
				"v": 10,
			},
		},
	}
	e, err := cel.NewEnv(
		cel.OptionalTypes(),
		cel.EnableMacroCallTracking(),
		ext.TwoVarComprehensions(),
		cel.Variable("x", cel.IntType),