Skip to content

Commit 63b9771

Browse files
committed
lint
1 parent af5a576 commit 63b9771

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

parquet/src/util/test_common/encryption_util.rs

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,16 +18,13 @@
1818
use crate::arrow::arrow_reader::{
1919
ArrowReaderMetadata, ArrowReaderOptions, ParquetRecordBatchReaderBuilder,
2020
};
21-
use crate::arrow::ParquetRecordBatchStreamBuilder;
22-
use crate::encryption::decrypt::FileDecryptionProperties;
23-
use crate::errors::ParquetError;
24-
use crate::file::metadata::FileMetaData;
2521
use arrow_array::cast::AsArray;
2622
use arrow_array::{types, RecordBatch};
27-
use futures::TryStreamExt;
23+
2824
use std::fs::File;
2925
use crate::arrow::ArrowWriter;
3026
use crate::encryption::encrypt::FileEncryptionProperties;
27+
use crate::encryption::decrypt::FileDecryptionProperties;
3128
use crate::file::properties::WriterProperties;
3229

3330
/// Tests reading an encrypted file from the parquet-testing repository
@@ -139,18 +136,24 @@ fn verify_encryption_test_data(
139136
}
140137

141138
#[cfg(feature = "encryption")]
142-
pub fn read_and_roundtrip_to_encrypted_file(path: &str, decryption_properties: FileDecryptionProperties, encryption_properties: FileEncryptionProperties) {
139+
pub fn read_and_roundtrip_to_encrypted_file(
140+
path: &str,
141+
decryption_properties: FileDecryptionProperties,
142+
encryption_properties: FileEncryptionProperties,
143+
) {
143144
let temp_file = tempfile::tempfile().unwrap();
144145

145146
// read example data
146147
let file = File::open(path).unwrap();
147-
let options =
148-
ArrowReaderOptions::default().with_file_decryption_properties(decryption_properties.clone());
148+
let options = ArrowReaderOptions::default()
149+
.with_file_decryption_properties(decryption_properties.clone());
149150
let metadata = ArrowReaderMetadata::load(&file, options.clone()).unwrap();
150151

151152
let builder = ParquetRecordBatchReaderBuilder::try_new_with_options(file, options).unwrap();
152153
let batch_reader = builder.build().unwrap();
153-
let batches = batch_reader.collect::<crate::errors::Result<Vec<RecordBatch>, _>>().unwrap();
154+
let batches = batch_reader
155+
.collect::<crate::errors::Result<Vec<RecordBatch>, _>>()
156+
.unwrap();
154157

155158
// write example data
156159
let props = WriterProperties::builder()

0 commit comments

Comments
 (0)