Method TestAsyncGetFlightInfo

cpp/src/arrow/flight/test_definitions.cc:1616–1706

}

void ErrorHandlingTest::TestAsyncGetFlightInfo() {
  if (!supports_async()) {
    GTEST_SKIP() << "Transport does not support async";
  }
  // Server-side still does all the junk around trying to translate Arrow
  // status codes, so this test is a little indirect

  for (const auto code : kStatusCodes) {
    ARROW_SCOPED_TRACE("C++ status code: ", static_cast<int>(code), ": ",
                       Status::CodeAsString(code));

    // Just the status code
    {
      auto descr = FlightDescriptor::Path(
          {std::to_string(static_cast<int>(code)), "Expected message"});
      auto listener = std::make_shared<TransportStatusListener>();

      client_->GetFlightInfoAsync(descr, listener);
      EXPECT_FINISHES(listener->future);
      auto detail = TransportStatusDetail::Unwrap(listener->future.status());
      ASSERT_TRUE(detail.has_value());

      EXPECT_EQ(detail->get().code(), kTransportStatusCodes.at(code));
      // Exact equality - should have no extra junk in the message
      EXPECT_EQ(detail->get().message(), "Expected message");
    }

    // Custom status detail
    {
      auto descr = FlightDescriptor::Path(
          {std::to_string(static_cast<int>(code)), "Expected message", ""});
      auto listener = std::make_shared<TransportStatusListener>();

      client_->GetFlightInfoAsync(descr, listener);
      EXPECT_FINISHES(listener->future);
      auto detail = TransportStatusDetail::Unwrap(listener->future.status());
      ASSERT_TRUE(detail.has_value());

      EXPECT_EQ(detail->get().code(), kTransportStatusCodes.at(code));
      // The server-side arrow::Status-to-TransportStatus conversion puts the
      // detail into the main error message.
      EXPECT_EQ(detail->get().message(),
                "Expected message. Detail: Custom status detail");

      std::string_view arrow_code, arrow_message;
      for (const auto& [key, value] : detail->get().details()) {
        if (key == "x-arrow-status") {
          arrow_code = value;
        } else if (key == "x-arrow-status-message-bin") {
          arrow_message = value;
        }
      }
      EXPECT_EQ(arrow_code, std::to_string(static_cast<int>(code)));
      EXPECT_EQ(arrow_message, "Expected message");
    }

    // Flight status detail
    for (const auto flight_code : kFlightStatusCodes) {