1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use futures::{stream, StreamExt};
use std::sync::{Arc, Mutex};
use solana_sdk::{
  signature::{Keypair, Signer},
  native_token::sol_to_lamports,
};
use crate::{
  program_test::ProgramTest,
};

#[derive(Default)]
pub struct TestAccount {
  pub participants: Vec<Keypair>,
}

impl TestAccount {
  pub async fn new(pt: &mut ProgramTest, count: i32) -> Self {
    let participants: Vec<Keypair> = (0..count)
      .map(|_| Keypair::new())
      .collect();
      
    // fund the newly created accounts
    let pt = Arc::new(Mutex::new(pt));
    stream::iter(participants.iter())
      .for_each(|account| {
        let pt = Arc::clone(&pt);

        async move {
          let mut lock = pt.lock().unwrap();
          lock.transfer_sol(&account.pubkey(), sol_to_lamports(1_f64)).await
        }
      })
      .await;

    Self {
      participants,
    }
  }
}