feat: add explain and verbose attributes to opa_test

ticketmaster · Apr 2, 2024 · 8fb76ad · 8fb76ad
1 parent da540fe
commit 8fb76ad
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 16 deletions.
diff --git a/MODULE.bazel.lock b/MODULE.bazel.lock
diff --git a/examples/simple/BUILD.bazel b/examples/simple/BUILD.bazel
@@ -28,6 +28,7 @@ opa_test(
     name = "simple_test",
     size = "small",
     srcs = ["main_test.rego"],
+    explain = "full",
     deps = [":simple"],
 )
 

diff --git a/opa/private/opa_test.bzl b/opa/private/opa_test.bzl
@@ -11,9 +11,14 @@ def _opa_test_impl(ctx):
 
     runfiles = ctx.runfiles(files = [toolchain.opa, bundle] + ctx.files.srcs)
 
+    args = ["--explain", ctx.attr.explain]
+
+    if ctx.attr.verbose:
+        args.append("--verbose")
+
     ctx.actions.write(
         output = test_file,
-        content = "%s test %s %s" % (toolchain.opa.short_path, bundle.short_path, " ".join([f.short_path for f in ctx.files.srcs])),
+        content = "%s test %s %s" % (toolchain.opa.short_path, bundle.short_path, " ".join(args + [f.short_path for f in ctx.files.srcs])),
         is_executable = True,
     )
 
@@ -37,6 +42,15 @@ opa_test = rule(
             providers = [OpaInfo],
             doc = "The bundle to test",
         ),
+        "explain": attr.string(
+            values = ["fails", "full", "notes", "debug"],
+            doc = "enable query explanations (default fails)",
+            default = "fails",
+        ),
+        "verbose": attr.bool(
+            doc = "set verbose reporting mode",
+            default = False,
+        ),
     },
     toolchains = ["//tools:toolchain_type"],
 )