Function TEST

eval/tests/unknowns_end_to_end_test.cc:697–751

  })pb";

TEST(UnknownsIterAttrTest, IterAttributeTrail) {
  InterpreterOptions options;
  Expr expr;
  Activation activation;
  Arena arena;

  protobuf::Value element;
  protobuf::Value& value =
      element.mutable_struct_value()->mutable_fields()->operator[]("elem1");
  value.set_number_value(1);
  protobuf::ListValue list;
  *list.add_values() = element;
  *list.add_values() = element;
  *list.add_values() = element;

  options.unknown_processing = UnknownProcessingOptions::kAttributeAndFunction;
  auto builder = CreateCelExpressionBuilder(options);
  ASSERT_THAT(RegisterBuiltinFunctions(builder->GetRegistry()), IsOk());
  ASSERT_THAT(builder->GetRegistry()->RegisterLazyFunction(
                  CreateDescriptor("Fn", CelValue::Type::kMap)),
              IsOk());
  ASSERT_TRUE(
      google::protobuf::TextFormat::ParseFromString(kListCompExistsWithAttrExpr, &expr))
      << "error parsing expr";

  // var.exists(x, Fn(x))
  auto plan = builder->CreateExpression(&expr, nullptr).value();

  activation.InsertValue("var", CelProtoWrapper::CreateMessage(&list, &arena));

  // var[1]['elem1'] is unknown
  activation.set_unknown_attribute_patterns({CelAttributePattern(
      "var", {
                 CreateCelAttributeQualifierPattern(CelValue::CreateInt64(1)),
                 CreateCelAttributeQualifierPattern(
                     CelValue::CreateStringView("elem1")),
             })});

  ASSERT_THAT(activation.InsertFunction(std::make_unique<FunctionImpl>(
                  "Fn", FunctionResponse::kFalse, CelValue::Type::kMap)),
              IsOk());

  CelValue response = plan->Evaluate(activation, &arena).value();

  ASSERT_TRUE(response.IsUnknownSet()) << CelValue::TypeName(response.type());
  ASSERT_EQ(response.UnknownSetOrDie()->unknown_attributes().size(), 1);
  // 'var[1]' is partially unknown when we make the function call so we treat it
  // as unknown.
  ASSERT_EQ(response.UnknownSetOrDie()
                ->unknown_attributes()
                .begin()
                ->qualifier_path()
                .size(),
            1);
}

TEST(UnknownsIterAttrTest, IterAttributeTrailMapKeyTypes) {
  InterpreterOptions options;