test_with.rego

package p

single_line_with {
  fn(1)   with input.a as "a"
}

multi_line_with {
    fn(1) with input.a as "a"
                with input.b as "b"
            with input.c as {
                "foo": "bar",
            }
                with input.d as [
                    1,
                    2,
                    3]
}

mock_f(_) = 123

func_replacements {
    count(array.concat(input.x, [])) with input.x as "foo"
    with array.concat as true
    with count as mock_f
}

original(x) = x+1

more_func_replacements {
    original(1) with original as mock_f
    original(1) with original as 1234
}