Function TEST_F

eval/compiler/instrumentation_test.cc:81–116

}

TEST_F(InstrumentationTest, Basic) {
  FlatExprBuilder builder(env_, options_);

  std::vector<int64_t> expr_ids;
  Instrumentation expr_id_recorder =
      [&expr_ids](int64_t expr_id, const cel::Value&) -> absl::Status {
    expr_ids.push_back(expr_id);
    return absl::OkStatus();
  };

  builder.AddProgramOptimizer(CreateInstrumentationExtension(
      [=](const cel::Ast&) -> Instrumentation { return expr_id_recorder; }));

  ASSERT_OK_AND_ASSIGN(ParsedExpr expr, Parse("1 + 2 + 3"));
  ASSERT_OK_AND_ASSIGN(auto ast,
                       cel::extensions::CreateAstFromParsedExpr(expr));
  ASSERT_OK_AND_ASSIGN(auto plan,
                       builder.CreateExpressionImpl(std::move(ast),
                                                    /*issues=*/nullptr));

  auto state = plan.MakeEvaluatorState(env_->descriptor_pool.get(),
                                       env_->MutableMessageFactory(), &arena_);
  cel::Activation activation;

  ASSERT_OK_AND_ASSIGN(auto value, plan.EvaluateWithCallback(
                                       activation, /*embedder_context=*/nullptr,
                                       EvaluationListener(), state));

  // AST for the test expression:
  //              + <4>
  //             /     \
  //          +<2>     3<5>
  //         /    \
  //      1<1>   2<3>
  EXPECT_THAT(expr_ids, ElementsAre(1, 3, 2, 5, 4));
}

TEST_F(InstrumentationTest, BasicWithConstFolding) {
  FlatExprBuilder builder(env_, options_);