From 02dfbfb8709ea39731da74c9e88bf1384e868061 Mon Sep 17 00:00:00 2001 From: Lee Smet Date: Mon, 25 Sep 2023 13:34:26 +0200 Subject: [PATCH] Fix PacketBuffer tests Signed-off-by: Lee Smet --- src/crypto.rs | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/crypto.rs b/src/crypto.rs index 422a6b8..9eb83a8 100644 --- a/src/crypto.rs +++ b/src/crypto.rs @@ -286,7 +286,7 @@ impl fmt::Debug for PacketBuffer { #[cfg(test)] mod tests { - use super::{PacketBuffer, SecretKey, AES_NONCE_SIZE, AES_TAG_SIZE}; + use super::{PacketBuffer, SecretKey, AES_NONCE_SIZE, AES_TAG_SIZE, DATA_HEADER_SIZE}; #[test] /// Test if encryption works in general. We just create some random value and encrypt it. @@ -307,7 +307,10 @@ mod tests { // We only care that this does not panic. let res = ss.encrypt(pb); // At the same time, check expected size. - assert_eq!(res.len(), data.len() + AES_TAG_SIZE + AES_NONCE_SIZE); + assert_eq!( + res.len(), + data.len() + DATA_HEADER_SIZE + AES_TAG_SIZE + AES_NONCE_SIZE + ); } #[test] @@ -335,6 +338,6 @@ mod tests { let original = ss2.decrypt(res).expect("Decryption works"); - assert_eq!(&original, &data[..]); + assert_eq!(&*original, &data[..]); } }