chore: adding dummy proof size for spammer

Signed-off-by: Simon Paitrault <[email protected]>
topos-protocol · Jul 28, 2023 · afce88a · afce88a
1 parent 2963509
commit afce88a
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/lib.rs b/crates/topos-certificate-spammer/src/lib.rs
@@ -13,6 +13,10 @@ use topos_tce_proxy::client::{TceClient, TceClientBuilder};
 use tracing::{debug, error, info, info_span, Instrument, Span};
 use tracing_opentelemetry::OpenTelemetrySpanExt;
 
+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;
+}
+
 #[derive(Debug, thiserror::Error)]
 pub enum Error {
     #[error("target nodes are not specified")]
@@ -98,7 +102,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)