Function ASSERT_OK_AND_ASSIGN

eval/tests/benchmark_test.cc:284–488

void BM_PolicySymbolic(benchmark::State& state) {
  google::protobuf::Arena arena;
  ASSERT_OK_AND_ASSIGN(ParsedExpr parsed_expr, parser::Parse(R"cel(
   !(ip in ["10.0.1.4", "10.0.1.5", "10.0.1.6"]) &&
   ((path.startsWith("v1") && token in ["v1", "v2", "admin"]) ||
    (path.startsWith("v2") && token in ["v2", "admin"]) ||
    (path.startsWith("/admin") && token == "admin" && ip in [
       "10.0.1.1",  "10.0.1.2", "10.0.1.3"
    ])
   ))cel"));

  InterpreterOptions options = GetOptions(arena);
  options.constant_folding = true;
  options.constant_arena = &arena;

  auto builder = CreateCelExpressionBuilder(options);
  ASSERT_OK(RegisterBuiltinFunctions(builder->GetRegistry(), options));

  SourceInfo source_info;
  ASSERT_OK_AND_ASSIGN(auto cel_expr, builder->CreateExpression(
                                          &parsed_expr.expr(), &source_info));

  Activation activation;
  activation.InsertValue("ip", CelValue::CreateStringView(kIP));
  activation.InsertValue("path", CelValue::CreateStringView(kPath));
  activation.InsertValue("token", CelValue::CreateStringView(kToken));

  for (auto _ : state) {
    ASSERT_OK_AND_ASSIGN(CelValue result,
                         cel_expr->Evaluate(activation, &arena));
    ASSERT_TRUE(result.BoolOrDie());
  }
}

BENCHMARK(BM_PolicySymbolic);

class RequestMap : public CelMap {
 public:
  absl::optional<CelValue> operator[](CelValue key) const override {
    if (!key.IsString()) {
      return {};
    }
    auto value = key.StringOrDie().value();
    if (value == "ip") {
      return CelValue::CreateStringView(kIP);
    } else if (value == "path") {
      return CelValue::CreateStringView(kPath);
    } else if (value == "token") {
      return CelValue::CreateStringView(kToken);
    }
    return {};
  }
  int size() const override { return 3; }
  absl::StatusOr<const CelList*> ListKeys() const override {
    return absl::UnimplementedError("CelMap::ListKeys is not implemented");
  }
};

// Uses a lazily constructed map container for "ip", "path", and "token".
void BM_PolicySymbolicMap(benchmark::State& state) {