Function TEST_F

eval/compiler/flat_expr_builder_extensions_test.cc:116–141

}

TEST_F(PlannerContextTest, GetPlan) {
  Expr a;
  Expr b;
  Expr c;
  ProgramBuilder program_builder;

  ASSERT_OK_AND_ASSIGN(auto step_ptrs,
                       InitSimpleTree(a, b, c, program_builder));

  std::shared_ptr<google::protobuf::Arena> arena;
  PlannerContext context(env_, resolver_, options_,
                         type_registry_.GetComposedTypeProvider(),
                         issue_collector_, program_builder, arena);

  EXPECT_THAT(context.GetSubplan(b), ElementsAre(UniquePtrHolds(step_ptrs.b)));

  EXPECT_THAT(context.GetSubplan(c), ElementsAre(UniquePtrHolds(step_ptrs.c)));

  EXPECT_THAT(context.GetSubplan(a), ElementsAre(UniquePtrHolds(step_ptrs.b),
                                                 UniquePtrHolds(step_ptrs.c),
                                                 UniquePtrHolds(step_ptrs.a)));

  Expr d;
  EXPECT_FALSE(context.IsSubplanInspectable(d));
  EXPECT_THAT(context.GetSubplan(d), IsEmpty());
}

TEST_F(PlannerContextTest, ReplacePlan) {
  Expr a;