Mock feature

Cargo.toml

@@ -38,3 +38,5 @@ criterion = "0.5.1"
 name = "benchmarks"
 harness = false
 
+[features]
+mock = []

src/packet/builder.rs

@@ -47,7 +47,7 @@ impl<'a> SphinxPacketBuilder<'a> {
     }
 }
 
-impl<'a> Default for SphinxPacketBuilder<'a> {
+impl Default for SphinxPacketBuilder<'_> {
     fn default() -> Self {
         SphinxPacketBuilder {
             payload_size: DEFAULT_PAYLOAD_SIZE,

src/payload/mod.rs

@@ -15,10 +15,10 @@
 use crate::constants::SECURITY_PARAMETER;
 use crate::header::keys::PayloadKey;
 use crate::{Error, ErrorKind, Result};
 use arrayref::array_ref;
 use blake2::VarBlake2b;
 use chacha::ChaCha; // we might want to swap this one with a different implementation
 use lioness::Lioness;
 
 // payload consists of security parameter long zero-padding, plaintext and '1' byte to indicate start of padding
 // (it can optionally be followed by zero-padding
@@ -89,10 +89,10 @@
     /// Note: this function should only ever be called in [`encapsulate_message`] after
     /// [`validate_parameters`] was performed.
     fn set_final_payload(plaintext_message: &[u8], payload_size: usize) -> Self {
         let final_payload: Vec<u8> = std::iter::repeat(0u8)
             .take(SECURITY_PARAMETER) // start with zero-padding
             .chain(plaintext_message.iter().cloned()) // put the plaintext
             .chain(std::iter::repeat(1u8).take(1)) // add single 1 byte to indicate start of padding
             .chain(std::iter::repeat(0u8)) // and fill everything else with zeroes
             .take(payload_size) // take however much we need (remember, iterators are lazy)
             .collect();
@@ -101,6 +101,7 @@
     }
 
     /// Tries to add an additional layer of encryption onto self.
+    #[cfg(not(feature = "mock"))]
     fn add_encryption_layer(mut self, payload_enc_key: &PayloadKey) -> Result<Self> {
         let lioness_cipher = Lioness::<VarBlake2b, ChaCha>::new_raw(array_ref!(
             payload_enc_key,
@@ -117,7 +118,15 @@
         Ok(self)
     }
 
+    #[cfg(feature = "mock")]
+    fn add_encryption_layer(mut self, payload_enc_key: &PayloadKey) -> Result<Self> {
+        // Mock implementation for testing purposes
+        // noop
+        Ok(self)
+    }
+
     /// Tries to remove single layer of encryption from self.
+    #[cfg(not(feature = "mock"))]
     pub fn unwrap(mut self, payload_key: &PayloadKey) -> Result<Self> {
         let lioness_cipher = Lioness::<VarBlake2b, ChaCha>::new_raw(array_ref!(
             payload_key,
@@ -133,6 +142,13 @@
         Ok(self)
     }
 
+    #[cfg(feature = "mock")]
+    pub fn unwrap(mut self, payload_key: &PayloadKey) -> Result<Self> {
+        // Mock implementation for testing purposes
+        // noop
+        Ok(self)
+    }
+
     /// After calling [`unwrap`] required number of times with correct `payload_keys`, tries to parse
     /// the resultant payload content into original encapsulated plaintext message.
     pub fn recover_plaintext(self) -> Result<Vec<u8>> {
@@ -451,6 +467,7 @@
     }
 
     #[test]
+    #[cfg(not(feature = "mock"))]
     fn it_fails_to_recover_plaintext_from_invalid_payload() {
         let message = vec![42u8; 160];