Function TestSetsMembershipRewriter

ext/sets_test.go:303–469

(t *testing.T)

}

func TestSetsMembershipRewriter(t *testing.T) {
	tests := []struct {
		expr      string
		optimized string
		opts      []cel.EnvOption
		in        map[string]any
		out       ref.Val
	}{
		{
			expr:      `a in [1, 2, 3, 4]`,
			optimized: `a in {1: true, 2: true, 3: true, 4: true}`,
			opts: []cel.EnvOption{
				cel.Variable("a", cel.IntType),
			},
			in: map[string]any{
				"a": 3,
			},
			out: types.True,
		},
		{
			expr:      `a in ['1', '2', '3', 4]`,
			optimized: `a in {"1": true, "2": true, "3": true, 4: true}`,
			opts: []cel.EnvOption{
				cel.Variable("a", cel.IntType),
			},
			in: map[string]any{
				"a": 3,
			},
			out: types.False,
		},
		{
			expr:      `a in [1u, '2', '3', 4]`,
			optimized: `a in {1u: true, "2": true, "3": true, 4: true}`,
			opts: []cel.EnvOption{
				cel.Variable("a", cel.IntType),
			},
			in: map[string]any{
				"a": 4,
			},
			out: types.True,
		},
		{
			expr:      `a in [1u, 2.0, '3', 4]`,
			optimized: `a in [1u, 2.0, "3", 4]`,
			opts: []cel.EnvOption{
				cel.Variable("a", cel.IntType),
			},
			in: map[string]any{
				"a": 4,
			},
			out: types.True,
		},
		{
			expr:      `a in [b, 32]`,
			optimized: `a in [b, 32]`,
			opts: []cel.EnvOption{
				cel.Variable("a", cel.IntType),
				cel.Variable("b", cel.IntType),