feat: add testnet init xtask command (#1632)

* feat: add testnet init xtask command

* fix tests

* fix lint

* fix: registry program test

Author: ananas-block

Cargo.lock

@@ -70,6 +70,23 @@ impl InitAddressTreeAccountsInstructionData {
             close_threshold: None,
         }
     }
+
+    pub fn testnet_default() -> Self {
+        Self {
+            index: 0,
+            program_owner: None,
+            forester: None,
+            bloom_filter_num_iters: 3,
+            input_queue_batch_size: 2000,
+            input_queue_zkp_batch_size: DEFAULT_ZKP_BATCH_SIZE,
+            height: 40,
+            root_history_capacity: 20,
+            bloom_filter_capacity: 20_000 * 8,
+            network_fee: Some(5000),
+            rollover_threshold: Some(95),
+            close_threshold: None,
+        }
+    }
 }
 
 impl Default for InitAddressTreeAccountsInstructionData {

program-libs/batched-merkle-tree/src/initialize_address_tree.rs

@@ -86,6 +86,26 @@ impl InitStateTreeAccountsInstructionData {
             close_threshold: None,
         }
     }
+
+    pub fn testnet_default() -> Self {
+        Self {
+            index: 0,
+            program_owner: None,
+            forester: None,
+            additional_bytes: DEFAULT_CPI_CONTEXT_ACCOUNT_SIZE,
+            bloom_filter_num_iters: 3,
+            input_queue_batch_size: 2000,
+            output_queue_batch_size: 2000,
+            input_queue_zkp_batch_size: DEFAULT_ZKP_BATCH_SIZE,
+            output_queue_zkp_batch_size: DEFAULT_ZKP_BATCH_SIZE,
+            height: DEFAULT_BATCH_STATE_TREE_HEIGHT,
+            root_history_capacity: 20,
+            bloom_filter_capacity: 20_000 * 8,
+            network_fee: Some(5000),
+            rollover_threshold: Some(95),
+            close_threshold: None,
+        }
+    }
 }
 
 impl Default for InitStateTreeAccountsInstructionData {

program-libs/batched-merkle-tree/src/initialize_state_tree.rs

@@ -521,8 +521,7 @@ async fn test_initialize_protocol_config() {
             0,
         )
         .await;
-        let expected_error_code = RegistryError::InvalidNetworkFee as u32 + 6000;
-        assert_rpc_error(result, 2, expected_error_code).unwrap();
+        assert_rpc_error(result, 3, RegistryError::InvalidNetworkFee.into()).unwrap();
     }
 }
 

program-tests/registry-test/tests/tests.rs

@@ -1,6 +1,9 @@
 #![cfg(feature = "test-sbf")]
 
-use account_compression::errors::AccountCompressionErrorCode;
+use account_compression::{
+    errors::AccountCompressionErrorCode, AddressMerkleTreeConfig, AddressQueueConfig,
+    NullifierQueueConfig, StateMerkleTreeConfig,
+};
 use anchor_lang::{AnchorSerialize, InstructionData, ToAccountMetas};
 use light_batched_merkle_tree::{
     errors::BatchedMerkleTreeError,
@@ -151,7 +154,7 @@ async fn invoke_failing_test() {
         }
     }
     for mut num_addresses in 0..=2 {
-        for j in 0..6 {
+        for (j, option) in options.iter().enumerate() {
             // there is no combined circuit instantiation for 8 inputs and addresses
             if j == 5 {
                 num_addresses = 0;
@@ -162,7 +165,7 @@ async fn invoke_failing_test() {
                     &mut test_indexer,
                     &payer,
                     &env,
-                    options[j],
+                    *option,
                     0,
                     num_addresses,
                     num_outputs,
@@ -1678,8 +1681,13 @@ async fn regenerate_accounts() {
         protocol_config,
         true,
         skip_register_programs,
+        false,
+        StateMerkleTreeConfig::default(),
+        NullifierQueueConfig::default(),
+        AddressMerkleTreeConfig::default(),
+        AddressQueueConfig::default(),
         InitStateTreeAccountsInstructionData::test_default(),
-        InitAddressTreeAccountsInstructionData::test_default(),
+        Some(InitAddressTreeAccountsInstructionData::test_default()),
     )
     .await;
     let keypairs = EnvAccountKeypairs::for_regenerate_accounts();
@@ -2121,16 +2129,12 @@ async fn batch_invoke_test() {
         let input_compressed_accounts = vec![
             compressed_account_with_context_1.compressed_account,
             compressed_account_with_context_2.compressed_account,
-        ]
-        .iter()
-        .cloned()
-        .collect::<Vec<_>>();
+        ];
 
-        let merkle_context = [
+        let merkle_context = vec![
             compressed_account_with_context_1.merkle_context,
             compressed_account_with_context_2.merkle_context,
-        ]
-        .to_vec();
+        ];
         let output_compressed_accounts = vec![
             CompressedAccount {
                 lamports: 0,

program-tests/system-test/tests/test.rs

@@ -15,19 +15,16 @@ no-entrypoint = []
 no-idl = []
 no-log-ix-name = []
 cpi = ["no-entrypoint"]
-custom-heap = ["light-heap"]
-mem-profiling = []
-default = ["custom-heap", "mem-profiling"]
+default = []
 test-sbf = []
 bench-sbf = []
 sdk = []
 
 [dependencies]
 aligned-sized = { workspace = true }
-anchor-lang = { workspace = true , features = ["init-if-needed"]}
-light-heap = { workspace = true, optional = true }
-account-compression = { workspace =  true  }
-light-system-program = { workspace = true , features = ["cpi"]  }
+anchor-lang = { workspace = true, features = ["init-if-needed"] }
+account-compression = { workspace = true }
+light-system-program = { workspace = true, features = ["cpi"] }
 solana-security-txt = "1.1.0"
 light-merkle-tree-metadata = { workspace = true }
 light-batched-merkle-tree = { workspace = true }

programs/registry/Cargo.toml

@@ -70,6 +70,29 @@ impl Default for ProtocolConfig {
     }
 }
 
+impl ProtocolConfig {
+    pub fn testnet_default() -> Self {
+        Self {
+            genesis_slot: 0,
+            min_weight: 1,
+            slot_length: 60,
+            registration_phase_length: 100,
+            active_phase_length: 1000,
+            report_work_phase_length: 100,
+            network_fee: 5000,
+            cpi_context_size: 20 * 1024 + 8,
+            finalize_counter_limit: 100,
+            place_holder: Pubkey::default(),
+            place_holder_a: 0,
+            place_holder_b: 0,
+            place_holder_c: 0,
+            place_holder_d: 0,
+            place_holder_e: 0,
+            place_holder_f: 0,
+        }
+    }
+}
+
 #[derive(Debug, Default, Clone, PartialEq, Eq)]
 pub enum EpochState {
     Registration,