mirror of
https://github.com/open-policy-agent/opa.git
synced 2026-08-12 19:32:48 -06:00
Torin Sandall
e7df89db62
Previously topdown did not expose a way for the caller to set the wall clock time. This makes it impossible to reliably replay policy queries that depend on time of day. With this change, callers can supply the wall clock time to use for time.now_ns() calls. Signed-off-by: Torin Sandall <torinsandall@gmail.com>
46 lines
827 B
Go
46 lines
827 B
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 (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/open-policy-agent/opa/ast"
|
|
)
|
|
|
|
func TestTimeSeeding(t *testing.T) {
|
|
|
|
query := `time.now_ns(x)`
|
|
clock := time.Now()
|
|
q := NewQuery(ast.MustParseBody(query)).WithTime(clock).WithCompiler(ast.NewCompiler())
|
|
|
|
ctx := context.Background()
|
|
|
|
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(fmt.Sprintf(`
|
|
{
|
|
{
|
|
x: %v
|
|
}
|
|
}
|
|
`, clock.UnixNano()))
|
|
|
|
result := queryResultSetToTerm(qrs)
|
|
|
|
if !result.Equal(exp) {
|
|
t.Fatalf("expected %v but got %v", exp, result)
|
|
}
|
|
|
|
}
|