fix: compatibility with datafusion

apache · May 29, 2024 · 2aa2e5f · 2aa2e5f
1 parent cc1ef1f
commit 2aa2e5f
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/crates/iceberg/src/writer/base_writer/equality_delete_writer.rs b/crates/iceberg/src/writer/base_writer/equality_delete_writer.rs
@@ -232,7 +232,6 @@ impl FieldProjector {
 mod test {
     use arrow_select::concat::concat_batches;
     use bytes::Bytes;
-    use futures::AsyncReadExt;
     use itertools::Itertools;
     use std::{collections::HashMap, sync::Arc};
 
@@ -266,15 +265,11 @@ mod test {
         assert_eq!(data_file.file_format, DataFileFormat::Parquet);
 
         // read the written file
-        let mut input_file = file_io
-            .new_input(data_file.file_path.clone())
-            .unwrap()
-            .reader()
-            .await
-            .unwrap();
-        let mut res = vec![];
-        let file_size = input_file.read_to_end(&mut res).await.unwrap();
-        let reader_builder = ParquetRecordBatchReaderBuilder::try_new(Bytes::from(res)).unwrap();
+        let input_file = file_io.new_input(data_file.file_path.clone()).unwrap();
+        // read the written file
+        let input_content = input_file.read().await.unwrap();
+        let reader_builder =
+            ParquetRecordBatchReaderBuilder::try_new(input_content.clone()).unwrap();
         let metadata = reader_builder.metadata().clone();
 
         // check data