gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use retry_transfer to test multiple times for replicator tokens

Browse Source 
May fix failures in CI where replicator is trying to do an airdrop.
This commit is contained in:
Stephen Akridge 2018-12-10 11:21:28 -08:00 committed by sakridge
parent 4481efd51e
commit 172e511e56
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/replicator.rs
View File

@ -12,6 +12,7 @@ use crate::rpc_request::{RpcClient, RpcRequest};
use crate::service::Service;
use crate::store_ledger_stage::StoreLedgerStage;
use crate::streamer::BlobReceiver;
use crate::thin_client::retry_get_balance;
use crate::window;
use crate::window_service::window_service;
use rand::thread_rng;
@ -194,7 +195,7 @@ impl Replicator {
let mut client = mk_client(&leader);
if client.get_balance(&keypair.pubkey()).is_err() {
if retry_get_balance(&mut client, &keypair.pubkey(), None).is_none() {
let mut drone_addr = leader_info.tpu;
drone_addr.set_port(DRONE_PORT);

11
tests/replicator.rs
View File

@ -10,6 +10,8 @@ use solana::ledger::{create_tmp_genesis, get_tmp_ledger_path, read_ledger};
use solana::logger;
use solana::replicator::Replicator;
use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::system_transaction::SystemTransaction;
use solana_sdk::transaction::Transaction;
use std::fs::remove_dir_all;
use std::sync::Arc;
use std::thread::sleep;
@ -53,8 +55,15 @@ fn test_replicator_startup() {
let replicator_keypair = Keypair::new();
let amount = 1;
let mut tx = Transaction::system_new(
&mint.keypair(),
replicator_keypair.pubkey(),
amount,
last_id,
);
leader_client
.transfer(1, &mint.keypair(), replicator_keypair.pubkey(), &last_id)
.retry_transfer(&mint.keypair(), &mut tx, 5)
.unwrap();
info!("starting replicator node");