Function TEST_P

extensions/lists_functions_test.cc:71–118

class ListsFunctionsTest : public testing::TestWithParam<TestInfo> {};

TEST_P(ListsFunctionsTest, EndToEnd) {
  const TestInfo& test_info = GetParam();
  RecordProperty("cel_expression", test_info.expr);
  if (!test_info.err.empty()) {
    RecordProperty("cel_expected_error", test_info.err);
  }

  ASSERT_OK_AND_ASSIGN(auto source, cel::NewSource(test_info.expr, "<input>"));

  MacroRegistry macro_registry;
  ParserOptions parser_options{.add_macro_calls = true};
  ASSERT_THAT(RegisterStandardMacros(macro_registry, parser_options), IsOk());
  ASSERT_THAT(RegisterListsMacros(macro_registry, parser_options), IsOk());
  ASSERT_OK_AND_ASSIGN(ParsedExpr parsed_expr,
                       google::api::expr::parser::Parse(*source, macro_registry,
                                                        parser_options));
  Expr expr = parsed_expr.expr();
  SourceInfo source_info = parsed_expr.source_info();

  google::protobuf::Arena arena;
  const auto options = RuntimeOptions{};
  ASSERT_OK_AND_ASSIGN(auto builder,
                       CreateStandardRuntimeBuilder(
                           internal::GetTestingDescriptorPool(), options));

  // Needed to resolve namespaced functions when evaluating a ParsedExpr.
  ASSERT_THAT(cel::EnableReferenceResolver(
                  builder, cel::ReferenceResolverEnabled::kAlways),
              IsOk());
  EXPECT_THAT(RegisterListsFunctions(builder.function_registry(), options),
              IsOk());
  ASSERT_OK_AND_ASSIGN(auto runtime, std::move(builder).Build());

  ASSERT_OK_AND_ASSIGN(std::unique_ptr<Program> program,
                       ProtobufRuntimeAdapter::CreateProgram(*runtime, expr));

  Activation activation;
  ASSERT_OK_AND_ASSIGN(Value result, program->Evaluate(&arena, activation));
  if (!test_info.err.empty()) {
    EXPECT_THAT(result,
                ErrorValueIs(StatusIs(testing::_, HasSubstr(test_info.err))));
    return;
  }
  ASSERT_TRUE(result.IsBool())
      << test_info.expr << " -> " << result.DebugString();
  EXPECT_TRUE(result.GetBool().NativeValue())
      << test_info.expr << " -> " << result.DebugString();
}

INSTANTIATE_TEST_SUITE_P(
    ListsFunctionsTest, ListsFunctionsTest,