Function ExpectResult

extensions/math_ext_test.cc:181–210

}

void ExpectResult(const TestCase& test_case) {
  Expr expr;
  Activation activation;
  activation.InsertValue("a", test_case.arg1);
  if (test_case.arg2.has_value()) {
    activation.InsertValue("b", *test_case.arg2);
    expr = CallExprTwoArgs(test_case.operation);
  } else {
    expr = CallExprOneArg(test_case.operation);
  }

  SourceInfo source_info;
  InterpreterOptions options;
  std::unique_ptr<CelExpressionBuilder> builder =
      CreateCelExpressionBuilder(options);
  ASSERT_OK(RegisterMathExtensionFunctions(builder->GetRegistry(), options));
  ASSERT_OK_AND_ASSIGN(auto cel_expression,
                       builder->CreateExpression(&expr, &source_info));

  google::protobuf::Arena arena;
  ASSERT_OK_AND_ASSIGN(auto value,
                       cel_expression->Evaluate(activation, &arena));
  if (!test_case.result.IsError()) {
    EXPECT_THAT(value, EqualsCelValue(test_case.result));
  } else {
    auto expected = test_case.result.ErrorOrDie();
    EXPECT_THAT(*value.ErrorOrDie(),
                StatusIs(expected->code(), HasSubstr(expected->message())));
  }
}

using MathExtParamsTest = testing::TestWithParam<TestCase>;
TEST_P(MathExtParamsTest, MinMaxTests) { ExpectResult(GetParam()); }