Function verifySourceInfoTransfer

policy/composer_test.go:99–114

verifySourceInfoTransfer checks that each offset range in the compiledRule has a corresponding node in composed

(t *testing.T, compiledRule *CompiledRule, composed *cel.Ast)

// verifySourceInfoTransfer checks that each offset range in the compiledRule has a corresponding node in composed
func verifySourceInfoTransfer(t *testing.T, compiledRule *CompiledRule, composed *cel.Ast) {
	t.Helper()
	dstRanges := make(map[ast.OffsetRange]ast.Expr)
	check := func(a *cel.Ast) {
		ast.PostOrderVisit(a.NativeRep().Expr(), &transferChecker{
			t:       t,
			srcInfo: a.NativeRep().SourceInfo(),
			dstInfo: composed.NativeRep().SourceInfo(),
			ranges:  &dstRanges})
	}
	ast.PostOrderVisit(composed.NativeRep().Expr(), &collectRanges{sourceInfo: composed.NativeRep().SourceInfo(), ranges: &dstRanges})
	for _, match := range compiledRule.matches {
		check(match.cond)
		check(match.output.expr)
	}
}

type collectRanges struct {
	sourceInfo *ast.SourceInfo