mirror of
https://github.com/open-policy-agent/opa.git
synced 2026-08-12 19:32:48 -06:00
Ville Vesilehto
f77322b3fb
Go 1.23 is no longer supported as per Go release policy. Changes: - Use Go v1.24.6 as the project SDK requirement - Apply lint fixes for Go 1.24 - Fix "non-constant format string in call" issues as seen in CI. Signed-off-by: Ville Vesilehto <ville@vesilehto.fi>
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
// Copyright 2020 The OPA Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache2
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package topdown
|
|
|
|
import (
|
|
"errors"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/open-policy-agent/opa/v1/ast"
|
|
)
|
|
|
|
func TestUUIDRFC4122SeedingAndCaching(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `uuid.rfc4122("x",x); uuid.rfc4122("y", y); uuid.rfc4122("x",x2)`
|
|
|
|
q := NewQuery(ast.MustParseBody(query)).WithSeed(rand.New(rand.NewSource(0))).WithCompiler(ast.NewCompiler())
|
|
|
|
ctx := t.Context()
|
|
|
|
qrs, err := q.Run(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if len(qrs) != 1 {
|
|
t.Fatal("expected exactly one result but got:", qrs)
|
|
}
|
|
|
|
exp := ast.MustParseTerm(`
|
|
{
|
|
{
|
|
x: "0194fdc2-fa2f-4cc0-81d3-ff12045b73c8",
|
|
x2: "0194fdc2-fa2f-4cc0-81d3-ff12045b73c8",
|
|
y: "6e4ff95f-f662-45ee-a82a-bdf44a2d0b75",
|
|
}
|
|
}
|
|
`)
|
|
|
|
result := queryResultSetToTerm(qrs)
|
|
|
|
if !result.Equal(exp) {
|
|
t.Fatalf("expected %v but got %v", exp, result)
|
|
}
|
|
}
|
|
|
|
type fakeSeedErrorReader struct{}
|
|
|
|
func (fakeSeedErrorReader) Read([]byte) (int, error) {
|
|
return 0, errors.New("xxx")
|
|
}
|
|
|
|
func TestUUIDRFC4122SeedError(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `uuid.rfc4122("x",x)`
|
|
|
|
q := NewQuery(ast.MustParseBody(query)).WithSeed(fakeSeedErrorReader{}).WithCompiler(ast.NewCompiler()).WithStrictBuiltinErrors(true)
|
|
|
|
_, err := q.Run(t.Context())
|
|
|
|
if topdownErr, ok := err.(*Error); !ok || topdownErr.Code != BuiltinErr {
|
|
t.Fatal("unexpected error (or lack of error):", err)
|
|
}
|
|
}
|
|
|
|
func TestUUIDRFC4122SavingDuringPartialEval(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := `foo = "x"; uuid.rfc4122(foo,x)`
|
|
c := ast.NewCompiler().
|
|
WithCapabilities(&ast.Capabilities{Builtins: []*ast.Builtin{ast.UUIDRFC4122}})
|
|
// Must compile to initialize type environment after WithCapabilities
|
|
c.Compile(nil)
|
|
|
|
q := NewQuery(ast.MustParseBody(query)).WithSeed(rand.New(rand.NewSource(0))).WithCompiler(c)
|
|
|
|
queries, modules, err := q.PartialRun(t.Context())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if len(modules) > 0 {
|
|
t.Fatal("expected no support")
|
|
}
|
|
|
|
exp := ast.MustParseBody(`uuid.rfc4122("x", x); foo = "x"`)
|
|
|
|
if len(queries) != 1 || !queries[0].Equal(exp) {
|
|
t.Fatalf("expected %v but got: %v", exp, queries)
|
|
}
|
|
}
|
|
|
|
func queryResultSetToTerm(qrs QueryResultSet) *ast.Term {
|
|
s := ast.NewSet()
|
|
for i := range qrs {
|
|
bindings := ast.NewObject()
|
|
for k := range qrs[i] {
|
|
bindings.Insert(ast.NewTerm(k), qrs[i][k])
|
|
}
|
|
s.Add(ast.NewTerm(bindings))
|
|
}
|
|
return ast.NewTerm(s)
|
|
}
|