Function ASSERT_OK_AND_ASSIGN

cpp/src/arrow/engine/substrait/serde_test.cc:357–771

    ASSERT_OK_AND_ASSIGN(auto substrait_type, SerializeType(*source_type, &ext_set, {}));

    ASSERT_OK_AND_ASSIGN(auto actual_return,
                         DeserializeType(*substrait_type, ext_set, {}));

    EXPECT_EQ(*actual_return, *return_type);
  }
}

// Substrait does not store the time zone as part of the type.  That information is stored
// on the function instead.  As a result, that information is lost on a round-trip.
TEST(Substrait, TimestampTypes) {
  ExtensionSet ext_set;

  for (auto time_unit :
       {TimeUnit::NANO, TimeUnit::MICRO, TimeUnit::MILLI, TimeUnit::SECOND}) {
    for (auto time_zone : {"UTC", "America/New_York"}) {
      auto input_type = timestamp(time_unit, time_zone);
      ASSERT_OK_AND_ASSIGN(auto substrait_type, SerializeType(*input_type, &ext_set, {}));

      auto expected_return = timestamp(time_unit, TimestampTzTimezoneString());
      ASSERT_OK_AND_ASSIGN(auto actual_return,
                           DeserializeType(*substrait_type, ext_set, {}));

      EXPECT_EQ(*actual_return, *expected_return);
    }
    auto input_type = timestamp(time_unit);
    ASSERT_OK_AND_ASSIGN(auto substrait_type, SerializeType(*input_type, &ext_set, {}));

    ASSERT_OK_AND_ASSIGN(auto actual_return,
                         DeserializeType(*substrait_type, ext_set, {}));

    EXPECT_EQ(*actual_return, *input_type);
  }
}

TEST(Substrait, NamedStruct) {
  ExtensionSet ext_set;

  ASSERT_OK_AND_ASSIGN(auto buf,
                       internal::SubstraitFromJSON("NamedStruct", R"({
    "struct": {
      "types": [
        {"i64": {}},
        {"list": {"type": {"string":{}} }},
        {"struct": {
          "types": [
            {"fp32": {"nullability": "NULLABILITY_REQUIRED"}},
            {"string": {}}
          ]
        }},
        {"list": {"type": {"string":{}} }},
      ]
    },
    "names": ["a", "b", "c", "d", "e", "f"]
  })",
                                                   /*ignore_unknown_fields=*/false));
  ASSERT_OK_AND_ASSIGN(auto schema, DeserializeSchema(*buf, ext_set));
  Schema expected_schema({
      field("a", int64()),