Initial implementation of policy coverage (#146)

Signed-off-by: Anand Krishnamoorthi <anakrish@microsoft.com>

tests/coverage/mod.rs

@@ -0,0 +1,81 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+use std::collections::BTreeSet;
+
+use regorus::*;
+
+use anyhow::Result;
+use test_generator::test_resources;
+
+#[derive(serde::Deserialize)]
+struct TestCase {
+    data: Option<Value>,
+    input: Option<Value>,
+    modules: Vec<String>,
+    note: String,
+    query: String,
+    uncovered: Vec<BTreeSet<u32>>,
+    skip: Option<bool>,
+}
+
+#[derive(serde::Deserialize)]
+struct YamlTest {
+    cases: Vec<TestCase>,
+}
+
+fn yaml_test_impl(file: &str) -> Result<()> {
+    let yaml_str = std::fs::read_to_string(file)?;
+    let test: YamlTest = serde_yaml::from_str(&yaml_str)?;
+
+    println!("running {file}");
+
+    for case in test.cases.into_iter() {
+        print!("case {} ", case.note);
+        if case.skip == Some(true) {
+            println!("skipped");
+            continue;
+        }
+
+        let mut engine = Engine::new();
+
+        for (idx, rego) in case.modules.iter().enumerate() {
+            engine.add_policy(format!("rego_{idx}"), rego.clone())?;
+        }
+
+        if let Some(data) = case.data {
+            engine.add_data(data)?;
+        }
+
+        if let Some(input) = case.input {
+            engine.set_input(input);
+        }
+
+        let _ = engine.eval_query(case.query.clone(), false)?;
+        let report = engine.get_coverage_report()?;
+
+        for (idx, uncovered) in case.uncovered.into_iter().enumerate() {
+            assert_eq!(uncovered, report.files[idx].uncovered);
+        }
+
+        println!("passed");
+    }
+
+    Ok(())
+}
+
+fn yaml_test(file: &str) -> Result<()> {
+    match yaml_test_impl(file) {
+        Ok(_) => Ok(()),
+        Err(e) => {
+            // If Err is returned, it doesn't always get printed by cargo test.
+            // Therefore, panic with the error.
+            panic!("{}", e);
+        }
+    }
+}
+
+#[test_resources("tests/coverage/*.yaml")]
+fn run(path: &str) {
+    yaml_test(path).unwrap()
+}

tests/coverage/tests.yaml

@@ -0,0 +1,20 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+cases:
+  - note: basic
+    modules:
+      - |
+        package test
+
+        x = 1
+        
+        y = k {
+          input.x == 5
+          k = input.k
+        }
+    query: data.test
+    uncovered: [
+      [ 5, 7 ]
+    ]
+