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

@@ -101,6 +101,7 @@ impl Payload {
     }
 
     /// 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 @@ impl Payload {
         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 @@ impl Payload {
         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 @@ mod plaintext_recovery {
     }
 
     #[test]
+    #[cfg(not(feature = "mock"))]
     fn it_fails_to_recover_plaintext_from_invalid_payload() {
         let message = vec![42u8; 160];