From 2ae7ee51e0beb1920d4ab29e3736756321ccdee0 Mon Sep 17 00:00:00 2001
From: joshieDo <93316087+joshieDo@users.noreply.github.com>
Date: Fri, 25 Oct 2024 19:13:46 +0900
Subject: [PATCH] fix: increase `arbitrary::Unstructured` buffer size if
 `NotEnoughData` is thrown (#12069)

---
 crates/cli/commands/src/test_vectors/compact.rs | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/crates/cli/commands/src/test_vectors/compact.rs b/crates/cli/commands/src/test_vectors/compact.rs
index cda7d5bd57..162ee1ceaa 100644
--- a/crates/cli/commands/src/test_vectors/compact.rs
+++ b/crates/cli/commands/src/test_vectors/compact.rs
@@ -191,7 +191,21 @@ where
         runner.rng().fill_bytes(&mut bytes);
         compact_buffer.clear();
 
-        let obj = T::arbitrary(&mut arbitrary::Unstructured::new(&bytes))?;
+        // Sometimes type T, might require extra arbitrary data, so we retry it a few times.
+        let mut tries = 0;
+        let obj = loop {
+            match T::arbitrary(&mut arbitrary::Unstructured::new(&bytes)) {
+                Ok(obj) => break obj,
+                Err(err) => {
+                    if tries < 5 && matches!(err, arbitrary::Error::NotEnoughData) {
+                        tries += 1;
+                        bytes.extend(std::iter::repeat(0u8).take(256));
+                    } else {
+                        return Err(err)?
+                    }
+                }
+            }
+        };
         let res = obj.to_compact(&mut compact_buffer);
 
         if IDENTIFIER_TYPE.contains(&type_name) {