kubelatte-ce

1
package opa
2

3
import (
4
	"github.com/open-policy-agent/opa/rego"
5
	"reflect"
6
	"testing"
7
)
8

9
func TestPrecompile(t *testing.T) {
10
	type args struct {
11
		module string
12
	}
13
	tests := []struct {
14
		name    string
15
		args    args
16
		want    rego.PreparedEvalQuery
17
		wantErr bool
18
	}{
19
		{
20
			name: "success",
21
			args: args{
22
				module: "import future.keywords.contains\n    import future.keywords.if\n    \n    violation contains {\"msg\": msg} if {\n      objName := \"hello\"\n      msg := sprintf(\"There is no HTTP. Just '%v'!\", [objName])\n    }",
23
			},
24
			want:    rego.PreparedEvalQuery{},
25
			wantErr: false,
26
		},
27
	}
28
	for _, tt := range tests {
29
		t.Run(tt.name, func(t *testing.T) {
30
			got, err := Precompile(tt.args.module)
31
			if (err != nil) != tt.wantErr {
32
				t.Errorf("Precompile() error = %v, wantErr %v", err, tt.wantErr)
33
				return
34
			}
35
			if reflect.DeepEqual(got, tt.want) {
36
				t.Errorf("Precompile() got = %v, want %v", got, tt.want)
37
			}
38
		})
39
	}
40
}
41