new_with_options

apache · Feb 5, 2025 · ac4ac21 · ac4ac21
1 parent 925d86f
commit ac4ac21
Showing 1 changed file with 32 additions and 12 deletions.
diff --git a/parquet/src/arrow/async_reader/mod.rs b/parquet/src/arrow/async_reader/mod.rs
@@ -1278,9 +1278,14 @@ mod tests {
         };
 
         let options = ArrowReaderOptions::new().with_page_index(true);
-        let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
-            .await
-            .unwrap();
+        let builder = ParquetRecordBatchStreamBuilder::new_with_options(
+            async_reader,
+            options,
+            #[cfg(feature = "encryption")]
+            None,
+        )
+        .await
+        .unwrap();
 
         // The builder should have page and offset indexes loaded now
         let metadata_with_index = builder.metadata();
@@ -1392,9 +1397,14 @@ mod tests {
         };
 
         let options = ArrowReaderOptions::new().with_page_index(true);
-        let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
-            .await
-            .unwrap();
+        let builder = ParquetRecordBatchStreamBuilder::new_with_options(
+            async_reader,
+            options,
+            #[cfg(feature = "encryption")]
+            None,
+        )
+        .await
+        .unwrap();
 
         let selection = RowSelection::from(vec![
             RowSelector::skip(21),   // Skip first page
@@ -1475,9 +1485,14 @@ mod tests {
             };
 
             let options = ArrowReaderOptions::new().with_page_index(true);
-            let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
-                .await
-                .unwrap();
+            let builder = ParquetRecordBatchStreamBuilder::new_with_options(
+                async_reader,
+                options,
+                #[cfg(feature = "encryption")]
+                None,
+            )
+            .await
+            .unwrap();
 
             let col_idx: usize = rand.gen_range(0..13);
             let mask = ProjectionMask::leaves(builder.parquet_schema(), vec![col_idx]);
@@ -1546,9 +1561,14 @@ mod tests {
         };
 
         let options = ArrowReaderOptions::new().with_page_index(true);
-        let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
-            .await
-            .unwrap();
+        let builder = ParquetRecordBatchStreamBuilder::new_with_options(
+            async_reader,
+            options,
+            #[cfg(feature = "encryption")]
+            None,
+        )
+        .await
+        .unwrap();
 
         let col_idx: usize = rand.gen_range(0..13);
         let mask = ProjectionMask::leaves(builder.parquet_schema(), vec![col_idx]);