Support settable drone lamport cap (#3675)

This commit is contained in:
Jack May
2019-04-08 12:37:01 -07:00
committed by GitHub
parent 44b391096d
commit e551f6b552
6 changed files with 17 additions and 9 deletions

View File

@@ -243,10 +243,18 @@ pub fn request_airdrop_transaction(
}
// For integration tests. Listens on random open port and reports port to Sender.
pub fn run_local_drone(mint_keypair: Keypair, sender: Sender<SocketAddr>) {
pub fn run_local_drone(
mint_keypair: Keypair,
sender: Sender<SocketAddr>,
request_cap_input: Option<u64>,
) {
thread::spawn(move || {
let drone_addr = socketaddr!(0, 0);
let drone = Arc::new(Mutex::new(Drone::new(mint_keypair, None, None)));
let drone = Arc::new(Mutex::new(Drone::new(
mint_keypair,
None,
request_cap_input,
)));
let socket = TcpListener::bind(&drone_addr).unwrap();
sender.send(socket.local_addr().unwrap()).unwrap();
info!("Drone started. Listening on: {}", drone_addr);

View File

@@ -19,7 +19,7 @@ fn test_local_drone() {
let expected_tx = Transaction::new(&[&keypair], message, blockhash);
let (sender, receiver) = channel();
run_local_drone(keypair, sender);
run_local_drone(keypair, sender, None);
let drone_addr = receiver.recv().unwrap();
let result = request_airdrop_transaction(&drone_addr, &to, lamports, blockhash);