Skip to content

Commit 44559b0

Browse files
committed
lint
1 parent aa8a5d8 commit 44559b0

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

parquet/src/util/test_common/encryption_util.rs

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,13 @@
1818
use crate::arrow::arrow_reader::{
1919
ArrowReaderMetadata, ArrowReaderOptions, ParquetRecordBatchReaderBuilder,
2020
};
21+
use crate::arrow::ArrowWriter;
2122
use crate::encryption::decryption::FileDecryptionProperties;
23+
use crate::encryption::encrypt::FileEncryptionProperties;
24+
use crate::file::properties::WriterProperties;
2225
use arrow_array::cast::AsArray;
2326
use arrow_array::{types, RecordBatch};
2427
use std::fs::File;
25-
use crate::arrow::ArrowWriter;
26-
use crate::encryption::encrypt::FileEncryptionProperties;
27-
use crate::file::properties::WriterProperties;
2828

2929
/// Tests reading an encrypted file from the parquet-testing repository
3030
pub fn verify_encryption_test_file_read(
@@ -102,18 +102,24 @@ pub fn verify_encryption_test_file_read(
102102
}
103103

104104
#[cfg(feature = "encryption")]
105-
pub fn read_and_roundtrip_to_encrypted_file(path: &str, decryption_properties: FileDecryptionProperties, encryption_properties: FileEncryptionProperties) {
105+
pub fn read_and_roundtrip_to_encrypted_file(
106+
path: &str,
107+
decryption_properties: FileDecryptionProperties,
108+
encryption_properties: FileEncryptionProperties,
109+
) {
106110
let temp_file = tempfile::tempfile().unwrap();
107111

108112
// read example data
109113
let file = File::open(path).unwrap();
110-
let options =
111-
ArrowReaderOptions::default().with_file_decryption_properties(decryption_properties.clone());
114+
let options = ArrowReaderOptions::default()
115+
.with_file_decryption_properties(decryption_properties.clone());
112116
let metadata = ArrowReaderMetadata::load(&file, options.clone()).unwrap();
113117

114118
let builder = ParquetRecordBatchReaderBuilder::try_new_with_options(file, options).unwrap();
115119
let batch_reader = builder.build().unwrap();
116-
let batches = batch_reader.collect::<crate::errors::Result<Vec<RecordBatch>, _>>().unwrap();
120+
let batches = batch_reader
121+
.collect::<crate::errors::Result<Vec<RecordBatch>, _>>()
122+
.unwrap();
117123

118124
// write example data
119125
let props = WriterProperties::builder()

0 commit comments

Comments
 (0)