ewilliams28/releases
1
0
Fork 0
mirror of https://github.com/open-policy-agent/opa.git synced 2026-08-12 19:32:48 -06:00
Code Issues Packages Projects Releases Wiki Activity

ast: Make Module.String() include if/contains for v1 modules (#7000)

Browse Source 
Fixes: #6973
Signed-off-by: Johan Fylling <johan.dev@fylling.se>
This commit is contained in:
Johan Fylling
2024-09-06 15:09:02 +02:00
committed by GitHub
parent 3587ccf4b1
commit 7cd3fecae7
4 changed files with 151 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compile/compile_test.go
+4 -2
View File
@@ -1246,9 +1246,11 @@ func TestCompilerOptimizationWithConfiguredNamespace(t *testing.T) {
t.Fatalf("expected two modules but got: %v", len(compiler.bundle.Modules))
}
// The compiler will strip the rego.v1 import from the module, so we need to compare it
// to a pure v1 module that doesn't require the import.
optimizedExp := ast.MustParseModuleWithOpts(`package custom
__not1_0_2__ = true if { data.test.q = _; _ }`,
ast.ParserOptions{AllFutureKeywords: true})
ast.ParserOptions{RegoVersion: ast.RegoV1})
if optimizedExp.String() != compiler.bundle.Modules[0].Parsed.String() {
t.Fatalf("expected optimized module to be:\n\n%v\n\ngot:\n\n%v", optimizedExp, compiler.bundle.Modules[0])
@@ -1258,7 +1260,7 @@ func TestCompilerOptimizationWithConfiguredNamespace(t *testing.T) {
k = {1, 2, 3} if { true }
p = true if { not data.custom.__not1_0_2__ }
q = true if { __local0__3 = input.a; data.test.k[__local0__3] = _; _; __local1__3 = input.b; data.test.k[__local1__3] = _; _ }`,
ast.ParserOptions{AllFutureKeywords: true})
ast.ParserOptions{RegoVersion: ast.RegoV1})
if expected.String() != compiler.bundle.Modules[1].Parsed.String() {
t.Fatalf("expected module to be:\n\n%v\n\ngot:\n\n%v", expected, compiler.bundle.Modules[1])