Function roundtrip_dictionary

arrow-flight/tests/encode_decode.rs:431–438

Encodes input as a FlightData stream, and then decodes it using FlightRecordBatchStream and validates the decoded record batches match the expected input. When is resolved, it should be possible to use `roundtrip`

(input: Vec<RecordBatch>)

/// When <https://github.com/apache/arrow-rs/issues/3389> is resolved,
/// it should be possible to use `roundtrip`
async fn roundtrip_dictionary(input: Vec<RecordBatch>) {
    let schema = Arc::new(prepare_schema_for_flight(input[0].schema_ref()));
    let expected_output: Vec<_> = input
        .iter()
        .map(|batch| prepare_batch_for_flight(batch, schema.clone()).unwrap())
        .collect();
    roundtrip_with_encoder(FlightDataEncoderBuilder::default(), input, expected_output).await
}

async fn roundtrip_with_encoder(
    encoder: FlightDataEncoderBuilder,