Function TEST

env/env_runtime_test.cc:162–197

                         ValuesIn(GetEnvRuntimeTestCases()));

TEST(EnvRuntimeTest, RegisterExtensionFunctions) {
  auto descriptor_pool = cel::internal::GetSharedTestingDescriptorPool();
  Config config;
  ASSERT_THAT(config.AddExtensionConfig("math", 2), IsOk());

  Env env;
  env.SetDescriptorPool(descriptor_pool);
  RegisterStandardExtensions(env);
  env.SetConfig(config);
  ASSERT_OK_AND_ASSIGN(std::unique_ptr<Compiler> compiler, env.NewCompiler());
  ASSERT_OK_AND_ASSIGN(ValidationResult result,
                       compiler->Compile("math.sqrt(4) == 2.0"));
  EXPECT_THAT(result.GetIssues(), IsEmpty()) << result.FormatError();
  ASSERT_OK_AND_ASSIGN(std::unique_ptr<Ast> ast, result.ReleaseAst());

  EnvRuntime env_runtime;
  env_runtime.SetDescriptorPool(descriptor_pool);
  env_runtime.RegisterExtensionFunctions(
      "cel.lib.math", "math", 2,
      [](cel::RuntimeBuilder& runtime_builder,
         const cel::RuntimeOptions& opts) -> absl::Status {
        return cel::extensions::RegisterMathExtensionFunctions(
            runtime_builder.function_registry(), opts, 2);
      });
  env_runtime.SetConfig(config);
  ASSERT_OK_AND_ASSIGN(std::unique_ptr<Runtime> runtime,
                       env_runtime.NewRuntime());
  ASSERT_OK_AND_ASSIGN(std::unique_ptr<Program> program,
                       runtime->CreateProgram(std::move(ast)));
  ASSERT_NE(program, nullptr);

  google::protobuf::Arena arena;
  Activation activation;
  ASSERT_OK_AND_ASSIGN(Value value, program->Evaluate(&arena, activation));
  EXPECT_TRUE(value.GetBool());
}
}  // namespace
}  // namespace cel