Fix the policy benchmark runner (#994)

google · Aug 13, 2024 · 5780f08 · 5780f08
1 parent 0e3ead6
commit 5780f08
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/policy/compiler_test.go b/policy/compiler_test.go
@@ -240,8 +240,16 @@ func (r *runner) bench(b *testing.B) {
 		for _, tst := range s.Tests {
 			tc := tst
 			b.Run(fmt.Sprintf("%s/%s/%s", r.name, section, tc.Name), func(b *testing.B) {
+				input := map[string]any{}
+				for k, v := range tc.Input {
+					if v.Expr == "" {
+						input[k] = v.Value
+						continue
+					}
+					input[k] = r.eval(b, v.Expr)
+				}
 				for i := 0; i < b.N; i++ {
-					_, _, err := r.prg.Eval(tc.Input)
+					_, _, err := r.prg.Eval(input)
 					if err != nil {
 						b.Fatalf("policy eval failed: %v", err)
 					}