chore: adding dummy proof size for spammer

Signed-off-by: Simon Paitrault <[email protected]>
topos-protocol · Aug 2, 2023 · c8060e6 · c8060e6
1 parent 226fc90
commit c8060e6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/topos-certificate-spammer/Cargo.toml b/crates/topos-certificate-spammer/Cargo.toml
@@ -20,6 +20,7 @@ tracing-opentelemetry.workspace = true
 opentelemetry.workspace = true
 tiny-keccak.workspace = true
 uuid.workspace = true
+lazy_static.workspace = true
 
 toml = "0.5.9"
 

diff --git a/crates/topos-certificate-spammer/src/utils.rs b/crates/topos-certificate-spammer/src/utils.rs
@@ -2,6 +2,10 @@ use topos_core::uci::{Certificate, SubnetId};
 
 use crate::{error::Error, SourceSubnet};
 
+lazy_static::lazy_static! {
+    static ref PROOF_SIZE: usize = std::env::var("PROOF_SIZE_KB").map(|v| v.parse::<usize>().unwrap_or(0)).unwrap_or(0) * 1024;
+}
+
 pub fn generate_random_32b_array() -> [u8; 32] {
     (0..32)
         .map(|_| rand::random::<u8>())
@@ -22,7 +26,7 @@ pub fn generate_test_certificate(
         generate_random_32b_array(),
         target_subnet_ids,
         0,
-        Vec::new(),
+        vec![254u8; *PROOF_SIZE],
     )?;
     new_cert
         .update_signature(&source_subnet.signing_key)