Function TestTwoVarComprehensionsCost

ext/comprehensions_test.go:218–394

(t *testing.T)

}

func TestTwoVarComprehensionsCost(t *testing.T) {
	tests := []struct {
		name          string
		expr          string
		vars          []cel.EnvOption
		in            map[string]any
		hints         map[string]uint64
		estimatedCost checker.CostEstimate
		actualCost    uint64
	}{
		{
			name:          "all list literal",
			expr:          `[1, 2, 3, 4].all(i, v, i < 5 && v > 0)`,
			estimatedCost: checker.CostEstimate{Min: 23, Max: 39},
			actualCost:    39,
		},
		{
			name:          "all map literal - true",
			expr:          `{1: 1, 2: 2, 3: 3}.all(i, v, i < 5 && v > 0)`,
			estimatedCost: checker.CostEstimate{Min: 40, Max: 52},
			actualCost:    52,
		},
		{
			name:          "all map literal - false",
			expr:          `!{0: 0}.all(i, v, i < 5 && v > 0)`,
			estimatedCost: checker.CostEstimate{Min: 35, Max: 39},
			actualCost:    39,
		},
		{
			name: "all map(int,int) variable",
			expr: `m.all(i, v, i < 5 && v > 0)`,
			vars: []cel.EnvOption{cel.Variable("m", cel.MapType(cel.IntType, cel.IntType))},
			hints: map[string]uint64{
				"m": 3,
			},
			in: map[string]any{
				"m": map[int]int{1: 1, 2: 2},
			},
			estimatedCost: checker.CostEstimate{Min: 2, Max: 23},
			actualCost:    16,
		},
		{
			name: "all map(string,string) variable",
			expr: `m.all(k, v, k < v)`,
			vars: []cel.EnvOption{cel.Variable("m", cel.MapType(cel.StringType, cel.StringType))},
			hints: map[string]uint64{
				"m":         3,
				"m.@keys":   16,
				"m.@values": 128,
			},
			in: map[string]any{
				"m": map[string]string{"he": "hello", "go": "goodbye"},
			},
			estimatedCost: checker.CostEstimate{Min: 2, Max: 23},
			actualCost:    14,
		},
		{
			name:          "transformList empty",