Function verifySourceInfoCoverage

policy/compiler_test.go:535–573

(t testing.TB, policy *Policy, ast *cel.Ast)

}

func verifySourceInfoCoverage(t testing.TB, policy *Policy, ast *cel.Ast) {
	t.Helper()
	info := ast.SourceInfo()

	exprLines := exprLinesFromPolicy(policy)
	coveredLines := make(map[int]bool)
	ids := ast.NativeRep().IDs()
	for id, offset := range info.GetPositions() {
		if offset <= 0 {
			t.Errorf("id %d has invalid offset %v", id, offset)
		}
		// Check that each position in the SourceInfo corresponds to a valid AST node.
		if !ids[id] {
			t.Errorf("id %d not found in AST", id)
		}
		loc, found := ast.Source().OffsetLocation(offset)
		if found {
			coveredLines[loc.Line()] = true
		} else {
			t.Errorf("invalid source location for offset %d", offset)
		}
	}
	// Verify that each source line inside an expression is covered by the at least one node in the
	// AST.
	for line := range exprLines {
		if !coveredLines[line] {
			t.Errorf("Line %d expected to be covered by SourceInfo, but was not", line)
		}
	}

	if t.Failed() {
		checked, err := cel.AstToCheckedExpr(ast)
		if err != nil {
			t.Logf("cel.AstToCheckedExpr() failed: %v", err)
		} else {
			t.Logf("AST:\n%s", prototext.Format(checked.GetExpr()))
		}
	}
}

// exprLinesFromPolicy returns a set of line numbers within a policy where expressions (variables,
// conditions, etc.) are defined.