@@ -186,7 +186,7 @@ TEST(BranchCoverage, IncrementsCounters) {
186186 EXPECT_TRUE (result.IsBool () && result.BoolOrDie () == true );
187187
188188 using Stats = BranchCoverage::NodeCoverageStats;
189- const NavigableAst & ast = coverage->ast ();
189+ const NavigableProtoAst & ast = coverage->ast ();
190190 auto root_node_stats = coverage->StatsForNode (ast.Root ().expr ()->id ());
191191
192192 EXPECT_THAT (root_node_stats, MatchesNodeStats (Stats{/* is_boolean=*/ true ,
@@ -195,7 +195,7 @@ TEST(BranchCoverage, IncrementsCounters) {
195195 /* boolean_false_count=*/ 0 ,
196196 /* error_count=*/ 0 }));
197197
198- const AstNode * ternary;
198+ const NavigableProtoAstNode * ternary;
199199 for (const auto & node : ast.Root ().DescendantsPreorder ()) {
200200 if (node.node_kind () == NodeKind::kCall &&
201201 node.expr ()->call_expr ().function () == cel::builtin::kTernary ) {
@@ -219,7 +219,7 @@ TEST(BranchCoverage, IncrementsCounters) {
219219 /* boolean_false_count=*/ 0 ,
220220 /* error_count=*/ 0 }));
221221
222- const AstNode * not_arg_expr;
222+ const NavigableProtoAstNode * not_arg_expr;
223223 for (const auto & node : ast.Root ().DescendantsPreorder ()) {
224224 if (node.node_kind () == NodeKind::kCall &&
225225 node.expr ()->call_expr ().function () == cel::builtin::kNot ) {
@@ -237,7 +237,7 @@ TEST(BranchCoverage, IncrementsCounters) {
237237 /* boolean_false_count=*/ 1 ,
238238 /* error_count=*/ 0 }));
239239
240- const AstNode * div_expr;
240+ const NavigableProtoAstNode * div_expr;
241241 for (const auto & node : ast.Root ().DescendantsPreorder ()) {
242242 if (node.node_kind () == NodeKind::kCall &&
243243 node.expr ()->call_expr ().function () == cel::builtin::kDivide ) {
@@ -306,7 +306,7 @@ TEST(BranchCoverage, AccumulatesAcrossRuns) {
306306 << result.DebugString ();
307307
308308 using Stats = BranchCoverage::NodeCoverageStats;
309- const NavigableAst & ast = coverage->ast ();
309+ const NavigableProtoAst & ast = coverage->ast ();
310310 auto root_node_stats = coverage->StatsForNode (ast.Root ().expr ()->id ());
311311
312312 EXPECT_THAT (root_node_stats, MatchesNodeStats (Stats{/* is_boolean=*/ true ,
@@ -315,7 +315,7 @@ TEST(BranchCoverage, AccumulatesAcrossRuns) {
315315 /* boolean_false_count=*/ 1 ,
316316 /* error_count=*/ 0 }));
317317
318- const AstNode * ternary;
318+ const NavigableProtoAstNode * ternary;
319319 for (const auto & node : ast.Root ().DescendantsPreorder ()) {
320320 if (node.node_kind () == NodeKind::kCall &&
321321 node.expr ()->call_expr ().function () == cel::builtin::kTernary ) {
@@ -378,7 +378,7 @@ TEST(BranchCoverage, CountsErrors) {
378378 EXPECT_TRUE (result.IsBool () && result.BoolOrDie () == false );
379379
380380 using Stats = BranchCoverage::NodeCoverageStats;
381- const NavigableAst & ast = coverage->ast ();
381+ const NavigableProtoAst & ast = coverage->ast ();
382382 auto root_node_stats = coverage->StatsForNode (ast.Root ().expr ()->id ());
383383
384384 EXPECT_THAT (root_node_stats, MatchesNodeStats (Stats{/* is_boolean=*/ true ,
@@ -387,7 +387,7 @@ TEST(BranchCoverage, CountsErrors) {
387387 /* boolean_false_count=*/ 1 ,
388388 /* error_count=*/ 0 }));
389389
390- const AstNode * ternary;
390+ const NavigableProtoAstNode * ternary;
391391 for (const auto & node : ast.Root ().DescendantsPreorder ()) {
392392 if (node.node_kind () == NodeKind::kCall &&
393393 node.expr ()->call_expr ().function () == cel::builtin::kTernary ) {
@@ -396,7 +396,7 @@ TEST(BranchCoverage, CountsErrors) {
396396 }
397397 }
398398
399- const AstNode * div_expr;
399+ const NavigableProtoAstNode * div_expr;
400400 for (const auto & node : ast.Root ().DescendantsPreorder ()) {
401401 if (node.node_kind () == NodeKind::kCall &&
402402 node.expr ()->call_expr ().function () == cel::builtin::kDivide ) {
0 commit comments