func setupBenchmark(nodes int, pods int) *topdown.QueryParams {
// policy compilation
c := ast.NewCompiler()
modules := map[string]*ast.Module{
"test": ast.MustParseModule(policy),
}
if c.Compile(modules); c.Failed() {
panic(c.Errors)
}
// storage setup
store := storage.New(storage.InMemoryConfig())
// parameter setup
ctx := context.Background()
request := ast.ObjectTerm(ast.Item(ast.StringTerm("pod"), ast.MustParseTerm(requestedPod)))
path := ast.MustParseRef("data.opa.test.scheduler.fit")
txn := storage.NewTransactionOrDie(ctx, store)
params := topdown.NewQueryParams(ctx, c, store, txn, request.Value, path)
// data setup
setupNodes(ctx, store, txn, nodes)
setupRCs(ctx, store, txn, 1)
setupPods(ctx, store, txn, pods, nodes)
return params
}
func setup(t *testing.T, filename string) *topdown.QueryParams {
// policy compilation
c := ast.NewCompiler()
modules := map[string]*ast.Module{
"test": ast.MustParseModule(policy),
}
if c.Compile(modules); c.Failed() {
t.Fatal("unexpected error:", c.Errors)
}
// storage setup
store := storage.New(storage.Config{
Builtin: loadDataStore(filename),
})
// parameter setup
ctx := context.Background()
request := ast.ObjectTerm(ast.Item(ast.StringTerm("pod"), ast.MustParseTerm(requestedPod)))
path := ast.MustParseRef("data.opa.test.scheduler.fit")
txn := storage.NewTransactionOrDie(ctx, store)
params := topdown.NewQueryParams(ctx, c, store, txn, request.Value, path)
return params
}
func parseBindings(s string) *ast.ValueMap {
t := ast.MustParseTerm(s)
obj, ok := t.Value.(ast.Object)
if !ok {
return nil
}
r := ast.NewValueMap()
for _, pair := range obj {
k, v := pair[0], pair[1]
r.Put(k.Value, v.Value)
}
return r
}
func TestMakeRequest(t *testing.T) {
tests := []struct {
note string
request [][2]string
expected interface{}
}{
{"var", [][2]string{{`hello`, `"world"`}}, `{"hello": "world"}`},
{"multiple vars", [][2]string{{`a`, `"a"`}, {`b`, `"b"`}}, `{"a": "a", "b": "b"}`},
{"multiple overlapping vars",
[][2]string{{`a.b.c`, `"c"`}, {`a.b.d`, `"d"`}, {`x.y`, `[]`}},
`{"a": {"b": {"c": "c", "d": "d"}}, "x": {"y": []}}`},
{"ref value",
[][2]string{{"foo.bar", "data.com.example.widgets[i]"}},
`{"foo": {"bar": data.com.example.widgets[i]}}`},
{"non-object", [][2]string{{"", "[1,2,3]"}}, "[1,2,3]"},
{"non-object conflict",
[][2]string{{"", "[1,2,3]"}, {"a.b", "true"}},
fmt.Errorf("conflicting request values: check request parameters")},
{"conflicting vars",
[][2]string{{`a.b`, `"c"`}, {`a.b.d`, `"d"`}},
fmt.Errorf("conflicting request value request.a.b.d: check request parameters")},
{"conflicting vars-2",
[][2]string{{`a.b`, `{"c":[]}`}, {`a.b.c`, `["d"]`}},
fmt.Errorf("conflicting request value request.a.b.c: check request parameters")},
{"conflicting vars-3",
[][2]string{{"a", "100"}, {`a.b`, `"c"`}},
fmt.Errorf("conflicting request value request.a.b: check request parameters")},
{"conflicting vars-4",
[][2]string{{`a.b`, `"c"`}, {`a`, `100`}},
fmt.Errorf("conflicting request value request.a: check request parameters")},
{"bad path",
[][2]string{{`a[1]`, `1`}},
fmt.Errorf("invalid request path: invalid path request.a[1]: path elements must be strings"),
},
}
for i, tc := range tests {
pairs := make([][2]*ast.Term, len(tc.request))
for j := range tc.request {
var k *ast.Term
if len(tc.request[j][0]) == 0 {
k = ast.NewTerm(ast.EmptyRef())
} else {
k = ast.MustParseTerm("request." + tc.request[j][0])
}
v := ast.MustParseTerm(tc.request[j][1])
pairs[j] = [...]*ast.Term{k, v}
}
request, err := MakeRequest(pairs)
switch e := tc.expected.(type) {
case error:
if err == nil {
t.Errorf("%v (#%d): Expected error %v but got: %v", tc.note, i+1, e, request)
continue
}
if err.Error() != e.Error() {
t.Errorf("%v (#%d): Expected error %v but got: %v", tc.note, i+1, e, err)
}
case string:
if err != nil {
t.Errorf("%v (#%d): Unexpected error: %v", tc.note, i+1, err)
continue
}
expected := ast.MustParseTerm(e)
if !expected.Value.Equal(request) {
t.Errorf("%v (#%d): Expected request to equal %v but got: %v", tc.note, i+1, expected, request)
}
}
}
}