Function ASSERT_OK_AND_ASSIGN

cpp/src/arrow/flight/flight_test.cc:271–390

TEST(TestFlight, ServerCallContextIncomingHeaders) {
  auto server = TestFlightServer::Make();
  ASSERT_OK_AND_ASSIGN(auto location, Location::ForGrpcTcp("localhost", 0));
  FlightServerOptions options(location);
  ASSERT_OK(server->Init(options));

  ASSERT_OK_AND_ASSIGN(auto client, FlightClient::Connect(server->location()));
  Action action;
  action.type = "list-incoming-headers";
  action.body = Buffer::FromString("test-header");
  FlightCallOptions call_options;
  call_options.headers.emplace_back("test-header1", "value1");
  call_options.headers.emplace_back("test-header2", "value2");
  ASSERT_OK_AND_ASSIGN(auto stream, client->DoAction(call_options, action));
  ASSERT_OK_AND_ASSIGN(auto result, stream->Next());
  ASSERT_NE(result.get(), nullptr);
  ASSERT_EQ(result->body->ToString(), "test-header1: value1");
  ASSERT_OK_AND_ASSIGN(result, stream->Next());
  ASSERT_NE(result.get(), nullptr);
  ASSERT_EQ(result->body->ToString(), "test-header2: value2");
  ASSERT_OK_AND_ASSIGN(result, stream->Next());
  ASSERT_EQ(result.get(), nullptr);
}

// ----------------------------------------------------------------------
// Client tests

class TestFlightClient : public ::testing::Test {
 public:
  void SetUp() {
    server_ = TestFlightServer::Make();

    ASSERT_OK_AND_ASSIGN(auto location, Location::ForGrpcTcp("localhost", 0));
    FlightServerOptions options(location);
    ASSERT_OK(server_->Init(options));

    ASSERT_OK(ConnectClient());
  }

  void TearDown() {
    ASSERT_OK(client_->Close());
    ASSERT_OK(server_->Shutdown());
  }

  Status ConnectClient() {
    ARROW_ASSIGN_OR_RAISE(auto location,
                          Location::ForGrpcTcp("localhost", server_->port()));
    return FlightClient::Connect(location).Value(&client_);
  }

  template <typename EndpointCheckFunc>
  void CheckDoGet(const FlightDescriptor& descr,
                  const RecordBatchVector& expected_batches,
                  EndpointCheckFunc&& check_endpoints) {
    auto expected_schema = expected_batches[0]->schema();

    ASSERT_OK_AND_ASSIGN(auto info, client_->GetFlightInfo(descr));
    check_endpoints(info->endpoints());

    ipc::DictionaryMemo dict_memo;