Function CreateProgram

extensions/comprehensions_v2_test.cc:56–96

using ::testing::TestWithParam;

absl::StatusOr<std::unique_ptr<Program>> CreateProgram(
    const std::string& expression, bool enable_mutable_accumulator,
    int max_recursion_depth) {
  // Configure the compiler
  CEL_ASSIGN_OR_RETURN(
      auto compiler_builder,
      NewCompilerBuilder(internal::GetTestingDescriptorPool()));
  CEL_RETURN_IF_ERROR(compiler_builder->AddLibrary(StandardCheckerLibrary()));
  CEL_RETURN_IF_ERROR(compiler_builder->AddLibrary(OptionalCompilerLibrary()));
  CEL_RETURN_IF_ERROR(compiler_builder->AddLibrary(BindingsCompilerLibrary()));
  CEL_RETURN_IF_ERROR(compiler_builder->AddLibrary(StringsCompilerLibrary()));
  CEL_RETURN_IF_ERROR(compiler_builder->AddLibrary(
      extensions::ComprehensionsV2CompilerLibrary()));

  CEL_ASSIGN_OR_RETURN(auto compiler, std::move(*compiler_builder).Build());

  // Configure the runtime
  cel::RuntimeOptions options;
  options.enable_qualified_type_identifiers = true;
  options.enable_comprehension_list_append = enable_mutable_accumulator;
  options.enable_comprehension_mutable_map = enable_mutable_accumulator;
  options.max_recursion_depth = max_recursion_depth;

  CEL_ASSIGN_OR_RETURN(auto runtime_builder,
                       CreateStandardRuntimeBuilder(
                           internal::GetTestingDescriptorPool(), options));
  CEL_RETURN_IF_ERROR(EnableOptionalTypes(runtime_builder));
  CEL_RETURN_IF_ERROR(
      RegisterStringsFunctions(runtime_builder.function_registry(), options));
  CEL_RETURN_IF_ERROR(RegisterComprehensionsV2Functions(
      runtime_builder.function_registry(), options));
  CEL_ASSIGN_OR_RETURN(std::unique_ptr<const Runtime> runtime,
                       std::move(runtime_builder).Build());

  CEL_ASSIGN_OR_RETURN(ValidationResult result, compiler->Compile(expression));
  if (!result.IsValid()) {
    return absl::Status(absl::StatusCode::kInvalidArgument,
                        result.FormatError());
  }
  return runtime->CreateProgram(*result.ReleaseAst());
}

struct TestOptions {
  bool enable_mutable_accumulator;