diff --git a/integration-testing/src/flight_client_scenarios/integration_test.rs b/integration-testing/src/flight_client_scenarios/integration_test.rs index 19a4b1a4ef1..a731a06eab1 100644 --- a/integration-testing/src/flight_client_scenarios/integration_test.rs +++ b/integration-testing/src/flight_client_scenarios/integration_test.rs @@ -276,7 +276,7 @@ async fn receive_batch_flight_data( { let length = data.data_body.len(); let mut reader = std::io::Cursor::new(&data.data_body); - read::read_dictionary(batch, fields, ipc_schema, dictionaries, &mut reader, 0, length as u64) + read::read_dictionary(batch, fields, ipc_schema, dictionaries, &mut reader, 0, length as u64, &mut Default::default()) .expect("Error reading dictionary"); data = resp.next().await?.ok()?; diff --git a/integration-testing/src/flight_server_scenarios/integration_test.rs b/integration-testing/src/flight_server_scenarios/integration_test.rs index 07453f5c724..10424759e87 100644 --- a/integration-testing/src/flight_server_scenarios/integration_test.rs +++ b/integration-testing/src/flight_server_scenarios/integration_test.rs @@ -296,6 +296,7 @@ async fn record_batch_from_message( &mut reader, 0, length as u64, + &mut Default::default() ); arrow_batch_result.map_err(|e| Status::internal(format!("Could not convert to Chunk: {:?}", e))) @@ -312,7 +313,7 @@ async fn dictionary_from_message( let mut reader = std::io::Cursor::new(data_body); let dictionary_batch_result = - ipc::read::read_dictionary(dict_batch, fields, ipc_schema, dictionaries, &mut reader, 0, length as u64); + ipc::read::read_dictionary(dict_batch, fields, ipc_schema, dictionaries, &mut reader, 0, length as u64, &mut Default::default()); dictionary_batch_result .map_err(|e| Status::internal(format!("Could not convert to Dictionary: {:?}", e))) }