Use #[tokio::test(flavor = "multi_thread", worker_threads = 2)]

Author: snawaz

magicblock-committor-service/src/tasks/mod.rs

@@ -19,15 +19,10 @@ use magicblock_program::magic_scheduled_base_intent::{
 };
 use solana_account::{Account, ReadableAccount};
 use solana_pubkey::Pubkey;
-use solana_rpc_client::rpc_client::RpcClient;
-use solana_sdk::{
-    commitment_config::CommitmentConfig, instruction::Instruction,
-};
+use solana_sdk::instruction::Instruction;
 use thiserror::Error;
 
-use crate::{
-    config::ChainConfig, tasks::visitor::Visitor, ComputeBudgetConfig,
-};
+use crate::tasks::visitor::Visitor;
 
 pub mod args_task;
 pub mod buffer_task;
@@ -125,24 +120,33 @@ impl CommitTask {
         allow_undelegation: bool,
         committed_account: CommittedAccount,
     ) -> Self {
-        let chain_config =
-            ChainConfig::local(ComputeBudgetConfig::new(1_000_000));
-
-        let rpc_client = RpcClient::new_with_commitment(
-            chain_config.rpc_uri.to_string(),
-            CommitmentConfig {
-                commitment: chain_config.commitment,
-            },
-        );
-
         let fetched_account = if committed_account.account.data.len()
             > CommitTask::COMMIT_STATE_SIZE_THRESHOLD
         {
-            rpc_client.get_account(&committed_account.pubkey).ok()
+            //use solana_rpc_client::rpc_client::RpcClient;
+            //use solana_sdk::commitment_config::CommitmentConfig;
+
+            //use crate::{config::ChainConfig, ComputeBudgetConfig};
+
+            //let chain_config =
+            //    ChainConfig::local(ComputeBudgetConfig::new(1_000_000));
+
+            // let rpc_client = RpcClient::new_with_commitment(
+            //     chain_config.rpc_uri.to_string(),
+            //     CommitmentConfig {
+            //         commitment: chain_config.commitment,
+            //     },
+            // );
+
+            //  rpc_client.get_account(&committed_account.pubkey).ok()
+
+            None
         } else {
             None
         };
 
+        println!("fetched_account: {:#?}", fetched_account);
+
         Self {
             commit_id,
             allow_undelegation,

test-integration/test-committor-service/tests/test_delivery_preparator.rs

@@ -15,9 +15,12 @@ use crate::common::{create_commit_task, generate_random_bytes, TestFixture};
 
 mod common;
 
-#[tokio::test]
+//#[tokio::test]
+#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
 async fn test_prepare_10kb_buffer() {
+    println!("TestFixture::new()");
     let fixture = TestFixture::new().await;
+    println!("TestFixture::new() done");
     let preparator = fixture.create_delivery_preparator();
 
     let data = generate_random_bytes(10 * 1024);
@@ -76,6 +79,7 @@ async fn test_prepare_10kb_buffer() {
     );
 }
 
+//#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
 #[tokio::test]
 async fn test_prepare_multiple_buffers() {
     let fixture = TestFixture::new().await;
@@ -206,7 +210,7 @@ async fn test_lookup_tables() {
     }
 }
 
-#[tokio::test]
+#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
 async fn test_already_initialized_error_handled() {
     let fixture = TestFixture::new().await;
     let preparator = fixture.create_delivery_preparator();
@@ -288,7 +292,7 @@ async fn test_already_initialized_error_handled() {
     assert_eq!(account.data.as_slice(), data, "Unexpected account data");
 }
 
-#[tokio::test]
+#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
 async fn test_prepare_cleanup_and_reprepare_mixed_tasks() {
     use borsh::BorshDeserialize;