feat: light program test skip startup logs

ananas-block · ananas-block · commit dd35a6066f63 · 2025-06-13T01:42:28.000+01:00
diff --git a/sdk-libs/program-test/src/program_test/config.rs b/sdk-libs/program-test/src/program_test/config.rs
@@ -27,6 +27,8 @@ pub struct ProgramTestConfig {
     pub v2_address_tree_config: Option<InitAddressTreeAccountsInstructionData>,
     pub skip_protocol_init: bool,
     pub log_failed_tx: bool,
+    pub no_logs: bool,
+    pub skip_startup_logs: bool,
 }
 
 impl ProgramTestConfig {
@@ -117,6 +119,8 @@ impl Default for ProgramTestConfig {
             skip_protocol_init: false,
             skip_v1_trees: false,
             log_failed_tx: true,
+            no_logs: false,
+            skip_startup_logs: true,
         }
     }
 }
diff --git a/sdk-libs/program-test/src/program_test/light_program_test.rs b/sdk-libs/program-test/src/program_test/light_program_test.rs
@@ -1,7 +1,6 @@
 use std::fmt::{self, Debug, Formatter};
 
 use account_compression::{AddressMerkleTreeAccount, QueueAccount};
-use forester_utils::utils::airdrop_lamports;
 use light_client::{
     indexer::{AddressMerkleTreeAccounts, StateMerkleTreeAccounts},
     rpc::{merkle_tree::MerkleTreeExt, RpcError},
@@ -61,16 +60,24 @@ impl LightProgramTest {
             config: config.clone(),
         };
         let keypairs = TestKeypairs::program_test_default();
-        airdrop_lamports(
-            &mut context,
-            &keypairs.governance_authority.pubkey(),
-            100_000_000_000,
-        )
-        .await?;
-        airdrop_lamports(&mut context, &keypairs.forester.pubkey(), 10_000_000_000).await?;
+
+        context
+            .context
+            .airdrop(&keypairs.governance_authority.pubkey(), 100_000_000_000_000)
+            .expect("governance_authority airdrop failed.");
+        context
+            .context
+            .airdrop(&keypairs.forester.pubkey(), 10_000_000_000)
+            .expect("forester airdrop failed.");
 
         if !config.skip_protocol_init {
+            if context.config.skip_startup_logs {
+                context.config.no_logs = true;
+            }
             initialize_accounts(&mut context, &config, &keypairs).await?;
+            if context.config.skip_startup_logs {
+                context.config.no_logs = false;
+            }
             let batch_size = config
                 .v2_state_tree_config
                 .as_ref()
diff --git a/sdk-libs/program-test/src/program_test/rpc.rs b/sdk-libs/program-test/src/program_test/rpc.rs
@@ -146,10 +146,12 @@ impl Rpc for LightProgramTest {
 
                 RpcError::TransactionError(x.err)
             })?;
-            #[cfg(debug_assertions)]
-            {
-                if std::env::var("RUST_BACKTRACE").is_ok() {
-                    println!("{}", _res.pretty_logs());
+            if !self.config.no_logs {
+                #[cfg(debug_assertions)]
+                {
+                    if std::env::var("RUST_BACKTRACE").is_ok() {
+                        println!("{}", _res.pretty_logs());
+                    }
                 }
             }
         }
@@ -408,14 +410,17 @@ impl LightProgramTest {
 
             RpcError::TransactionError(x.err)
         })?;
-        #[cfg(debug_assertions)]
-        {
-            if std::env::var("RUST_BACKTRACE").is_ok() {
-                // Print all tx logs and events.
-                println!("{}", _res.pretty_logs());
-                println!("event:\n {:?}", event);
+        if !self.config.no_logs {
+            #[cfg(debug_assertions)]
+            {
+                if std::env::var("RUST_BACKTRACE").is_ok() {
+                    // Print all tx logs and events.
+                    println!("{}", _res.pretty_logs());
+                    println!("event:\n {:?}", event);
+                }
             }
         }
+
         let slot = self.context.get_sysvar::<Clock>().slot;
         let event = event.map(|e| (e, signature, slot));
 

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,8 @@ pub struct ProgramTestConfig {`
`27`	`27`	`pub v2_address_tree_config: Option<InitAddressTreeAccountsInstructionData>,`
`28`	`28`	`pub skip_protocol_init: bool,`
`29`	`29`	`pub log_failed_tx: bool,`
	`30`	`+ pub no_logs: bool,`
	`31`	`+ pub skip_startup_logs: bool,`
`30`	`32`	`}`
`31`	`33`
`32`	`34`	`impl ProgramTestConfig {`
`@@ -117,6 +119,8 @@ impl Default for ProgramTestConfig {`
`117`	`119`	`skip_protocol_init: false,`
`118`	`120`	`skip_v1_trees: false,`
`119`	`121`	`log_failed_tx: true,`
	`122`	`+ no_logs: false,`
	`123`	`+ skip_startup_logs: true,`
`120`	`124`	`}`
`121`	`125`	`}`
`122`	`126`	`}`