Function TEST_F

eval/eval/comprehension_step_test.cc:103–139

}

TEST_F(ListKeysStepTest, MapPartiallyUnknown) {
  ExecutionPath path;
  auto result = CreateIdentStep("var", 0);
  ASSERT_OK(result);
  path.push_back(*std::move(result));
  ComprehensionInitStep* init_step = new ComprehensionInitStep(1);
  init_step->set_error_jump_offset(1);
  path.push_back(absl::WrapUnique(init_step));
  path.push_back(std::make_unique<GetListKeysResultStep>());

  auto expression =
      MakeExpression(std::move(path), /*unknown_attributes=*/true);

  Activation activation;
  Arena arena;
  Struct value;
  (*value.mutable_fields())["key1"].set_number_value(1.0);
  (*value.mutable_fields())["key2"].set_number_value(2.0);
  (*value.mutable_fields())["key3"].set_number_value(3.0);

  activation.InsertValue("var", CelProtoWrapper::CreateMessage(&value, &arena));
  activation.set_unknown_attribute_patterns({CelAttributePattern(
      "var",
      {CreateCelAttributeQualifierPattern(CelValue::CreateStringView("key2")),
       CreateCelAttributeQualifierPattern(CelValue::CreateStringView("foo")),
       CelAttributeQualifierPattern::CreateWildcard()})});

  auto eval_result = expression->Evaluate(activation, &arena);

  ASSERT_OK(eval_result);
  ASSERT_TRUE(eval_result->IsUnknownSet());
  const auto& attrs = eval_result->UnknownSetOrDie()->unknown_attributes();

  EXPECT_THAT(attrs, SizeIs(1));
  EXPECT_THAT(attrs.begin()->variable_name(), Eq("var"));
  EXPECT_THAT(attrs.begin()->qualifier_path(), SizeIs(0));
}

TEST_F(ListKeysStepTest, ErrorPassedThrough) {
  ExecutionPath path;