Function TestTwoVarComprehensionsResidualAST

ext/comprehensions_test.go:599–793

(t *testing.T)

}

func TestTwoVarComprehensionsResidualAST(t *testing.T) {
	tests := []struct {
		name     string
		in       map[string]any
		varOpts  []cel.EnvOption
		unks     []*interpreter.AttributePattern
		expr     string
		residual string
	}{
		{
			name: "transform map entry residual compare",
			varOpts: []cel.EnvOption{
				cel.Variable("x", cel.ListType(cel.DynType)),
				cel.Variable("y", cel.IntType),
			},
			in: map[string]any{
				"x": []any{0, uint(1)},
			},
			unks:     []*interpreter.AttributePattern{cel.AttributePattern("y")},
			expr:     `x.transformMapEntry(i, v, {v: i}).size() < y`,
			residual: `2 < y`,
		},
		{
			name: "transform map entry residual transform",
			varOpts: []cel.EnvOption{
				cel.Variable("x", cel.ListType(cel.DynType)),
				cel.Variable("y", cel.IntType),
			},
			in: map[string]any{
				"x": []any{0, uint(1)},
			},
			unks:     []*interpreter.AttributePattern{cel.AttributePattern("y")},
			expr:     `x.transformMapEntry(i, v, i < y, {v: i})`,
			residual: `[0, 1u].transformMapEntry(i, v, i < y, {v: i})`,
		},
		{
			name: "nested exists unknown inner range",
			varOpts: []cel.EnvOption{
				cel.Variable("x", cel.ListType(cel.IntType)),
				cel.Variable("y", cel.MapType(cel.IntType, cel.DynType)),
			},
			in: map[string]any{
				"x": []any{1, 2, 3},
			},
			unks:     []*interpreter.AttributePattern{cel.AttributePattern("y")},
			expr:     `x.exists(val, y.exists(key, _, key == val))`,
			residual: `[1, 2, 3].exists(val, y.exists(key, _, key == val))`,
		},
		{
			name: "nested exists unknown inner range",
			varOpts: []cel.EnvOption{
				cel.Variable("x", cel.ListType(cel.IntType)),
				cel.Variable("y", cel.MapType(cel.IntType, cel.DynType)),
			},
			in: map[string]any{
				"y": map[int]string{1: "hi", 2: "hello", 3: "howdy"},
			},
			unks:     []*interpreter.AttributePattern{cel.AttributePattern("x")},