rego-test-assertions

Tiny Rego library providing helper functions for unit testing. The library primarily contains various assertion functions, which will print the expected result vs. the outcome to the console on failure. This allows you to quickly grasp what went wrong in your unit tests, resulting in a faster test iteration process!

This assert library is what I’ve been wanting since I started using OPA. Fixed something in ten minutes, while eating a sandwich, I’d previously been poking at for a couple of hours. You’ve made my day!

— Duncan Thomas, Groupon

Usage

Simply download the test assertions library policy file and place it somewhere opa test checks for policy and tests:

Alternatively, use ODM to add the library to your project:

odm depend rego-test-assertions --no-namespace \
    git+https://github.com/anderseknert/rego-test-assertions

In order to use the test assertion functions, import the test.assert package:

import data.test.assert

Functions

Once imported, all functions may now be referenced like assert.<function>. Using the assert package prefix avoids having these functions clash with other built-ins and custom functions, and makes it clear in your test code what the purpose of these functions is. As an added bonus, you won't need repeated import statements to import each function separetely.

Function	Arguments	Example console output
assert.equals	expected, result	expected equals: "foo" got: "bar"
assert.not_equals	expected, result	expected not equals: 1 got: 1
assert.all_equals	coll, value	expected all items to have value 2, failed for [1]
assert.none_equals	coll, value	expected no items to have value 2, failed for [2, 2]
assert.has	item, coll	expected string "foo" in array got: ["bar", "baz"]
assert.not_has	item, coll	expected string "foo" not in set got {"foo", "x"}
assert.empty	coll	expected empty set got: {"admin", "dba"}
assert.not_empty	coll	expected empty array
assert.starts_with	str, search	expected "test" to start with "a"
assert.ends_with	str, search	expected "test" to end with "a"
assert.all_starts_with	coll, search	expected all strings to start with "a", failed for ["b"]
assert.all_ends_with	coll, search	expected all strings to end with "a", failed for ["b"]
assert.none_starts_swith	coll, search	expected no strings to start with "a", failed for ["a"]
assert.none_ends_swith	coll, search	expected no strings to end with "a", failed for ["a"]
assert.fail	msg	fail with provided message!

Example

Policy

package rego.example

deny[msg] {
    msg := "I'll always deny that"
}

violation[msg] {
    input.user.name == "banned"
    msg := "You're banned!"
}

Test without assertions

package rego.example_test

import future.keywords.in

import data.example.deny
import data.example.violation

test_empty_without_assertion {
    count(deny) == 0
}

test_in_without_assertion {
    "You're banned!" in violation with input.user.name as "bob"
}

❯ opa test example_test.rego
example_test.rego:
data.rego.example_test.test_empty_without_assertion: FAIL (178.375µs)
data.rego.example_test.test_in_without_assertion: FAIL (99.416µs)

--------------------------------------------------------------------------------
FAIL: 2/2

Test with assertions

package rego.example_test

import data.example.deny
import data.example.violation

import data.test.assert

test_empty_with_assertion {
    assert.empty(deny)
}

test_in_with_assertion {
    assert.has("You're banned!", violation) with input.user.name as "bob"
}

❯ opa test example_test.rego
example_test.rego:
data.rego.example_test.test_empty_with_assertion: FAIL (206.5µs)

  expected empty set got {"I'll always deny that"}

data.rego.example_test.test_in_with_assertion: FAIL (138.792µs)

  expected string "You're banned!" in set got set()

--------------------------------------------------------------------------------
FAIL: 2/2

Prerequisites

OPA v0.39.0 and above — a bug in older versions caused the print function to error in else blocks, which this library makes heavy use of.