From 555293a54f50aac07cf798e3703ec66f383ec8ef Mon Sep 17 00:00:00 2001
From: Ruslan Piasetskyi <ruslan.piasetskyi@gmail.com>
Date: Wed, 14 Sep 2022 15:36:15 +0200
Subject: [PATCH] digest: add hash_rt_outsize_serialization_test macro

---
 digest/src/dev.rs | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/digest/src/dev.rs b/digest/src/dev.rs
index cbb163d38..8f2ba8be6 100644
--- a/digest/src/dev.rs
+++ b/digest/src/dev.rs
@@ -73,6 +73,43 @@ macro_rules! hash_serialization_test {
     };
 }
 
+/// Define hash function serialization test
+#[macro_export]
+#[cfg_attr(docsrs, doc(cfg(feature = "dev")))]
+macro_rules! hash_rt_outsize_serialization_test {
+    ($name:ident, $hasher:ty, $expected_serialized_state:expr) => {
+        #[test]
+        fn $name() {
+            use digest::{
+                crypto_common::{BlockSizeUser, SerializableState},
+                typenum::Unsigned,
+                Digest, Update, VariableOutput,
+            };
+            const HASH_OUTPUT_SIZE: usize = <$hasher>::MAX_OUTPUT_SIZE - 1;
+
+            let mut h = <$hasher>::new(HASH_OUTPUT_SIZE).unwrap();
+
+            h.update(&[0x13; <$hasher as BlockSizeUser>::BlockSize::USIZE + 1]);
+
+            let serialized_state = h.serialize();
+            assert_eq!(serialized_state.as_slice(), $expected_serialized_state);
+
+            let mut h = <$hasher>::deserialize(&serialized_state).unwrap();
+
+            h.update(&[0x13; <$hasher as BlockSizeUser>::BlockSize::USIZE + 1]);
+            let mut output1 = [0; HASH_OUTPUT_SIZE];
+            h.finalize_variable(&mut output1).unwrap();
+
+            let mut h = <$hasher>::new(HASH_OUTPUT_SIZE).unwrap();
+            h.update(&[0x13; 2 * (<$hasher as BlockSizeUser>::BlockSize::USIZE + 1)]);
+            let mut output2 = [0; HASH_OUTPUT_SIZE];
+            h.finalize_variable(&mut output2).unwrap();
+
+            assert_eq!(output1, output2);
+        }
+    };
+}
+
 /// Define [`Update`][crate::Update] impl benchmark
 #[macro_export]
 #[cfg_attr(docsrs, doc(cfg(feature = "dev")))]