mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-28 07:57:01 +00:00
This commit is contained in:
@@ -636,7 +636,7 @@ mod tests {
|
||||
let reader = BufReader::new(Cursor::new(data.to_vec()));
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
encrypted
|
||||
}
|
||||
|
||||
@@ -674,14 +674,14 @@ mod tests {
|
||||
// Encrypt
|
||||
let mut encrypt_reader = encrypt_reader;
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
// Decrypt using DecryptReader
|
||||
let reader = Cursor::new(encrypted.clone());
|
||||
let decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
let mut decrypt_reader = decrypt_reader;
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
assert_eq!(&decrypted, data);
|
||||
}
|
||||
@@ -700,7 +700,7 @@ mod tests {
|
||||
let encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypt_reader = encrypt_reader;
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
// Now test DecryptReader
|
||||
|
||||
@@ -708,7 +708,7 @@ mod tests {
|
||||
let decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
let mut decrypt_reader = decrypt_reader;
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
assert_eq!(&decrypted, data);
|
||||
}
|
||||
@@ -728,13 +728,13 @@ mod tests {
|
||||
let encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypt_reader = encrypt_reader;
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
let reader = std::io::Cursor::new(encrypted.clone());
|
||||
let decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
let mut decrypt_reader = decrypt_reader;
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
assert_eq!(&decrypted, &data);
|
||||
}
|
||||
@@ -752,12 +752,12 @@ mod tests {
|
||||
let reader = Cursor::new(data.clone());
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
let reader = ChunkedCursor::new(encrypted, 3);
|
||||
let mut decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
assert_eq!(decrypted, data);
|
||||
}
|
||||
@@ -775,12 +775,12 @@ mod tests {
|
||||
let reader = Cursor::new(data.clone());
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
let reader = PendingChunkedCursor::new(encrypted, 3);
|
||||
let mut decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
assert_eq!(decrypted, data);
|
||||
}
|
||||
@@ -798,7 +798,7 @@ mod tests {
|
||||
let reader = Cursor::new(data.clone());
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
let reader = ChunkedCursor::new(encrypted, 8192);
|
||||
let decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
@@ -806,7 +806,7 @@ mod tests {
|
||||
|
||||
let mut decrypted = Vec::new();
|
||||
while let Some(chunk) = stream.next().await {
|
||||
let bytes = chunk.unwrap();
|
||||
let bytes = chunk.expect("operation should succeed");
|
||||
decrypted.extend_from_slice(&bytes);
|
||||
}
|
||||
|
||||
@@ -826,7 +826,7 @@ mod tests {
|
||||
let reader = Cursor::new(data.clone());
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
let reader = ChunkedCursor::new(encrypted, 8192);
|
||||
let decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
@@ -835,7 +835,7 @@ mod tests {
|
||||
|
||||
let mut decrypted = Vec::new();
|
||||
while let Some(chunk) = stream.next().await {
|
||||
let bytes = chunk.unwrap();
|
||||
let bytes = chunk.expect("operation should succeed");
|
||||
decrypted.extend_from_slice(&bytes);
|
||||
}
|
||||
|
||||
@@ -857,7 +857,7 @@ mod tests {
|
||||
let reader = BufReader::new(Cursor::new(data.to_vec()));
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
encrypted
|
||||
}
|
||||
|
||||
@@ -871,7 +871,7 @@ mod tests {
|
||||
let reader = BufReader::new(Cursor::new(combined));
|
||||
let mut decrypt_reader = DecryptReader::new_multipart(reader, key, base_nonce, vec![1, 2]);
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
let mut expected = Vec::with_capacity(part_one.len() + part_two.len());
|
||||
expected.extend_from_slice(&part_one);
|
||||
@@ -891,7 +891,7 @@ mod tests {
|
||||
let reader = Cursor::new(data);
|
||||
let mut encrypt_reader = EncryptReader::new(reader, key, nonce);
|
||||
let mut encrypted = Vec::new();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.unwrap();
|
||||
encrypt_reader.read_to_end(&mut encrypted).await.expect("operation should succeed");
|
||||
|
||||
let payloads = extract_encrypted_payloads(&encrypted);
|
||||
assert!(payloads.len() >= 2);
|
||||
@@ -920,7 +920,7 @@ mod tests {
|
||||
let reader = Cursor::new(encrypted);
|
||||
let mut decrypt_reader = DecryptReader::new(reader, key, nonce);
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
assert_eq!(decrypted, data);
|
||||
}
|
||||
@@ -945,7 +945,7 @@ mod tests {
|
||||
let reader = BufReader::new(Cursor::new(combined));
|
||||
let mut decrypt_reader = DecryptReader::new_multipart(reader, key, base_nonce, vec![1, 2]);
|
||||
let mut decrypted = Vec::new();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.unwrap();
|
||||
decrypt_reader.read_to_end(&mut decrypted).await.expect("operation should succeed");
|
||||
|
||||
let mut expected = Vec::with_capacity(part_one.len() + part_two.len());
|
||||
expected.extend_from_slice(&part_one);
|
||||
|
||||
Reference in New Issue
Block a user