1329 lines
47 KiB
Rust
1329 lines
47 KiB
Rust
use crate::{
|
|
client_error::{ClientError, ClientErrorKind, Result as ClientResult},
|
|
generic_rpc_client_request::GenericRpcClientRequest,
|
|
mock_rpc_client_request::{MockRpcClientRequest, Mocks},
|
|
rpc_client_request::RpcClientRequest,
|
|
rpc_request::{RpcError, RpcRequest},
|
|
rpc_response::*,
|
|
};
|
|
use bincode::serialize;
|
|
use indicatif::{ProgressBar, ProgressStyle};
|
|
use log::*;
|
|
use serde_json::{json, Value};
|
|
use solana_sdk::{
|
|
account::Account,
|
|
clock::{Slot, UnixTimestamp, DEFAULT_TICKS_PER_SECOND, DEFAULT_TICKS_PER_SLOT},
|
|
commitment_config::CommitmentConfig,
|
|
epoch_schedule::EpochSchedule,
|
|
fee_calculator::{FeeCalculator, FeeRateGovernor},
|
|
hash::Hash,
|
|
inflation::Inflation,
|
|
pubkey::Pubkey,
|
|
signature::Signature,
|
|
signers::Signers,
|
|
transaction::{self, Transaction, TransactionError},
|
|
};
|
|
use solana_transaction_status::{ConfirmedBlock, TransactionEncoding, TransactionStatus};
|
|
use solana_vote_program::vote_state::MAX_LOCKOUT_HISTORY;
|
|
use std::{
|
|
error,
|
|
net::SocketAddr,
|
|
thread::sleep,
|
|
time::{Duration, Instant},
|
|
};
|
|
|
|
pub struct RpcClient {
|
|
client: Box<dyn GenericRpcClientRequest + Send + Sync>,
|
|
}
|
|
|
|
impl RpcClient {
|
|
pub fn new(url: String) -> Self {
|
|
Self {
|
|
client: Box::new(RpcClientRequest::new(url)),
|
|
}
|
|
}
|
|
|
|
pub fn new_mock(url: String) -> Self {
|
|
Self {
|
|
client: Box::new(MockRpcClientRequest::new(url)),
|
|
}
|
|
}
|
|
|
|
pub fn new_mock_with_mocks(url: String, mocks: Mocks) -> Self {
|
|
Self {
|
|
client: Box::new(MockRpcClientRequest::new_with_mocks(url, mocks)),
|
|
}
|
|
}
|
|
|
|
pub fn new_socket(addr: SocketAddr) -> Self {
|
|
Self::new(get_rpc_request_str(addr, false))
|
|
}
|
|
|
|
pub fn new_socket_with_timeout(addr: SocketAddr, timeout: Duration) -> Self {
|
|
let url = get_rpc_request_str(addr, false);
|
|
Self {
|
|
client: Box::new(RpcClientRequest::new_with_timeout(url, timeout)),
|
|
}
|
|
}
|
|
|
|
pub fn confirm_transaction(&self, signature: &Signature) -> ClientResult<bool> {
|
|
Ok(self
|
|
.confirm_transaction_with_commitment(signature, CommitmentConfig::default())?
|
|
.value)
|
|
}
|
|
|
|
pub fn confirm_transaction_with_commitment(
|
|
&self,
|
|
signature: &Signature,
|
|
commitment_config: CommitmentConfig,
|
|
) -> RpcResult<bool> {
|
|
let Response { context, value } = self.get_signature_statuses(&[*signature])?;
|
|
|
|
Ok(Response {
|
|
context,
|
|
value: value[0]
|
|
.as_ref()
|
|
.filter(|result| result.satisfies_commitment(commitment_config))
|
|
.map(|result| result.status.is_ok())
|
|
.unwrap_or_default(),
|
|
})
|
|
}
|
|
|
|
pub fn send_transaction(&self, transaction: &Transaction) -> ClientResult<Signature> {
|
|
let serialized_encoded = bs58::encode(serialize(transaction).unwrap()).into_string();
|
|
let response =
|
|
self.client
|
|
.send(&RpcRequest::SendTransaction, json!([serialized_encoded]), 5)?;
|
|
|
|
match response.as_str() {
|
|
None => {
|
|
Err(RpcError::ForUser("Received result of an unexpected type".to_string()).into())
|
|
}
|
|
Some(signature_base58_str) => {
|
|
let signature = signature_base58_str.parse::<Signature>().map_err(|err| {
|
|
Into::<ClientError>::into(RpcError::ParseError(err.to_string()))
|
|
})?;
|
|
// A mismatching RPC response signature indicates an issue with the RPC node, and
|
|
// should not be passed along to confirmation methods. The transaction may or may
|
|
// not have been submitted to the cluster, so callers should verify the success of
|
|
// the correct transaction signature independently.
|
|
if signature != transaction.signatures[0] {
|
|
Err(RpcError::RpcRequestError(format!(
|
|
"RPC node returned mismatched signature {:?}, expected {:?}",
|
|
signature, transaction.signatures[0]
|
|
))
|
|
.into())
|
|
} else {
|
|
Ok(transaction.signatures[0])
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn get_signature_status(
|
|
&self,
|
|
signature: &Signature,
|
|
) -> ClientResult<Option<transaction::Result<()>>> {
|
|
self.get_signature_status_with_commitment(signature, CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn get_signature_statuses(
|
|
&self,
|
|
signatures: &[Signature],
|
|
) -> RpcResult<Vec<Option<TransactionStatus>>> {
|
|
let signatures: Vec<_> = signatures.iter().map(|s| s.to_string()).collect();
|
|
let signature_status =
|
|
self.client
|
|
.send(&RpcRequest::GetSignatureStatuses, json!([signatures]), 5)?;
|
|
Ok(serde_json::from_value(signature_status)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetSignatureStatuses"))?)
|
|
}
|
|
|
|
pub fn get_signature_status_with_commitment(
|
|
&self,
|
|
signature: &Signature,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<Option<transaction::Result<()>>> {
|
|
let signature_status = self.client.send(
|
|
&RpcRequest::GetSignatureStatuses,
|
|
json!([[signature.to_string()]]),
|
|
5,
|
|
)?;
|
|
let result: Response<Vec<Option<TransactionStatus>>> =
|
|
serde_json::from_value(signature_status)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetSignatureStatuses"))?;
|
|
Ok(result.value[0]
|
|
.clone()
|
|
.filter(|result| result.satisfies_commitment(commitment_config))
|
|
.map(|status_meta| status_meta.status))
|
|
}
|
|
|
|
pub fn get_slot(&self) -> ClientResult<Slot> {
|
|
self.get_slot_with_commitment(CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn get_slot_with_commitment(
|
|
&self,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<Slot> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetSlot, json!([commitment_config]), 0)
|
|
.map_err(|err| err.into_with_command("GetSlot"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetSlot"))
|
|
}
|
|
|
|
pub fn total_supply(&self) -> ClientResult<u64> {
|
|
self.total_supply_with_commitment(CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn total_supply_with_commitment(
|
|
&self,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<u64> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetTotalSupply, json!([commitment_config]), 0)
|
|
.map_err(|err| err.into_with_command("GetTotalSupply"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetTotalSupply"))
|
|
}
|
|
|
|
pub fn get_vote_accounts(&self) -> ClientResult<RpcVoteAccountStatus> {
|
|
self.get_vote_accounts_with_commitment(CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn get_vote_accounts_with_commitment(
|
|
&self,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<RpcVoteAccountStatus> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetVoteAccounts, json!([commitment_config]), 0)
|
|
.map_err(|err| err.into_with_command("GetVoteAccounts"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetVoteAccounts"))
|
|
}
|
|
|
|
pub fn get_cluster_nodes(&self) -> ClientResult<Vec<RpcContactInfo>> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetClusterNodes, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetClusterNodes"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetClusterNodes"))
|
|
}
|
|
|
|
pub fn get_confirmed_block(&self, slot: Slot) -> ClientResult<ConfirmedBlock> {
|
|
self.get_confirmed_block_with_encoding(slot, TransactionEncoding::Json)
|
|
}
|
|
|
|
pub fn get_confirmed_block_with_encoding(
|
|
&self,
|
|
slot: Slot,
|
|
encoding: TransactionEncoding,
|
|
) -> ClientResult<ConfirmedBlock> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetConfirmedBlock, json!([slot, encoding]), 0)
|
|
.map_err(|err| err.into_with_command("GetConfirmedBlock"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetConfirmedBlock"))
|
|
}
|
|
|
|
pub fn get_confirmed_blocks(
|
|
&self,
|
|
start_slot: Slot,
|
|
end_slot: Option<Slot>,
|
|
) -> ClientResult<Vec<Slot>> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetConfirmedBlocks,
|
|
json!([start_slot, end_slot]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetConfirmedBlocks"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetConfirmedBlocks"))
|
|
}
|
|
|
|
pub fn get_block_time(&self, slot: Slot) -> ClientResult<UnixTimestamp> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetBlockTime, json!([slot]), 0);
|
|
|
|
response
|
|
.map(|result_json| {
|
|
if result_json.is_null() {
|
|
return Err(RpcError::ForUser(format!("Block Not Found: slot={}", slot)).into());
|
|
}
|
|
let result = serde_json::from_value(result_json)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetBlockTime"))?;
|
|
trace!("Response block timestamp {:?} {:?}", slot, result);
|
|
Ok(result)
|
|
})
|
|
.map_err(|err| err.into_with_command("GetBlockTime"))?
|
|
}
|
|
|
|
pub fn get_epoch_info(&self) -> ClientResult<RpcEpochInfo> {
|
|
self.get_epoch_info_with_commitment(CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn get_epoch_info_with_commitment(
|
|
&self,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<RpcEpochInfo> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetEpochInfo, json!([commitment_config]), 0)
|
|
.map_err(|err| err.into_with_command("GetEpochInfo"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetEpochInfo"))
|
|
}
|
|
|
|
pub fn get_leader_schedule(
|
|
&self,
|
|
slot: Option<Slot>,
|
|
) -> ClientResult<Option<RpcLeaderSchedule>> {
|
|
self.get_leader_schedule_with_commitment(slot, CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn get_leader_schedule_with_commitment(
|
|
&self,
|
|
slot: Option<Slot>,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<Option<RpcLeaderSchedule>> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetLeaderSchedule,
|
|
json!([slot, commitment_config]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetLeaderSchedule"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetLeaderSchedule"))
|
|
}
|
|
|
|
pub fn get_epoch_schedule(&self) -> ClientResult<EpochSchedule> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetEpochSchedule, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetEpochSchedule"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetEpochSchedule"))
|
|
}
|
|
|
|
pub fn get_identity(&self) -> ClientResult<Pubkey> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetIdentity, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetIdentity"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetIdentity"))
|
|
.and_then(|rpc_identity: RpcIdentity| {
|
|
rpc_identity.identity.parse::<Pubkey>().map_err(|_| {
|
|
ClientError::new_with_command(
|
|
RpcError::ParseError("Pubkey".to_string()).into(),
|
|
"GetIdentity",
|
|
)
|
|
})
|
|
})
|
|
}
|
|
|
|
pub fn get_inflation(&self) -> ClientResult<Inflation> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetInflation, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetInflation"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetInflation"))
|
|
}
|
|
|
|
pub fn get_version(&self) -> ClientResult<RpcVersionInfo> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetVersion, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetVersion"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetVersion"))
|
|
}
|
|
|
|
pub fn minimum_ledger_slot(&self) -> ClientResult<Slot> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::MinimumLedgerSlot, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("MinimumLedgerSlot"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "MinimumLedgerSlot"))
|
|
}
|
|
|
|
pub fn send_and_confirm_transaction<T: Signers>(
|
|
&self,
|
|
transaction: &mut Transaction,
|
|
signer_keys: &T,
|
|
) -> ClientResult<Signature> {
|
|
let mut send_retries = 20;
|
|
loop {
|
|
let mut status_retries = 15;
|
|
let signature = self.send_transaction(transaction)?;
|
|
let status = loop {
|
|
let status = self.get_signature_status(&signature)?;
|
|
if status.is_none() {
|
|
status_retries -= 1;
|
|
if status_retries == 0 {
|
|
break status;
|
|
}
|
|
} else {
|
|
break status;
|
|
}
|
|
if cfg!(not(test)) {
|
|
// Retry twice a second
|
|
sleep(Duration::from_millis(500));
|
|
}
|
|
};
|
|
send_retries = if let Some(result) = status.clone() {
|
|
match result {
|
|
Ok(_) => return Ok(signature),
|
|
Err(TransactionError::AccountInUse) => {
|
|
// Fetch a new blockhash and re-sign the transaction before sending it again
|
|
self.resign_transaction(transaction, signer_keys)?;
|
|
send_retries - 1
|
|
}
|
|
Err(_) => 0,
|
|
}
|
|
} else {
|
|
send_retries - 1
|
|
};
|
|
if send_retries == 0 {
|
|
if let Some(err) = status {
|
|
return Err(err.unwrap_err().into());
|
|
} else {
|
|
return Err(
|
|
RpcError::ForUser("unable to confirm transaction. This can happen in situations such as transaction expiration and insufficient fee-payer funds".to_string()).into(),
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn send_and_confirm_transactions<T: Signers>(
|
|
&self,
|
|
mut transactions: Vec<Transaction>,
|
|
signer_keys: &T,
|
|
) -> Result<(), Box<dyn error::Error>> {
|
|
let mut send_retries = 5;
|
|
loop {
|
|
let mut status_retries = 15;
|
|
|
|
// Send all transactions
|
|
let mut transactions_signatures = vec![];
|
|
for transaction in transactions {
|
|
if cfg!(not(test)) {
|
|
// Delay ~1 tick between write transactions in an attempt to reduce AccountInUse errors
|
|
// when all the write transactions modify the same program account (eg, deploying a
|
|
// new program)
|
|
sleep(Duration::from_millis(1000 / DEFAULT_TICKS_PER_SECOND));
|
|
}
|
|
|
|
let signature = self.send_transaction(&transaction).ok();
|
|
transactions_signatures.push((transaction, signature))
|
|
}
|
|
|
|
// Collect statuses for all the transactions, drop those that are confirmed
|
|
while status_retries > 0 {
|
|
status_retries -= 1;
|
|
|
|
if cfg!(not(test)) {
|
|
// Retry twice a second
|
|
sleep(Duration::from_millis(500));
|
|
}
|
|
|
|
transactions_signatures = transactions_signatures
|
|
.into_iter()
|
|
.filter(|(_transaction, signature)| {
|
|
if let Some(signature) = signature {
|
|
if let Ok(status) = self.get_signature_status(&signature) {
|
|
if status.is_none() {
|
|
return false;
|
|
}
|
|
return status.unwrap().is_err();
|
|
}
|
|
}
|
|
true
|
|
})
|
|
.collect();
|
|
|
|
if transactions_signatures.is_empty() {
|
|
return Ok(());
|
|
}
|
|
}
|
|
|
|
if send_retries == 0 {
|
|
return Err(RpcError::ForUser("Transactions failed".to_string()).into());
|
|
}
|
|
send_retries -= 1;
|
|
|
|
// Re-sign any failed transactions with a new blockhash and retry
|
|
let (blockhash, _fee_calculator) =
|
|
self.get_new_blockhash(&transactions_signatures[0].0.message().recent_blockhash)?;
|
|
transactions = vec![];
|
|
for (mut transaction, _) in transactions_signatures.into_iter() {
|
|
transaction.try_sign(signer_keys, blockhash)?;
|
|
transactions.push(transaction);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn resign_transaction<T: Signers>(
|
|
&self,
|
|
tx: &mut Transaction,
|
|
signer_keys: &T,
|
|
) -> ClientResult<()> {
|
|
let (blockhash, _fee_calculator) =
|
|
self.get_new_blockhash(&tx.message().recent_blockhash)?;
|
|
tx.try_sign(signer_keys, blockhash)?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn retry_get_balance(
|
|
&self,
|
|
pubkey: &Pubkey,
|
|
retries: usize,
|
|
) -> Result<Option<u64>, Box<dyn error::Error>> {
|
|
let balance_json = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetBalance,
|
|
json!([pubkey.to_string()]),
|
|
retries,
|
|
)
|
|
.map_err(|err| err.into_with_command("RetryGetBalance"))?;
|
|
|
|
Ok(Some(
|
|
serde_json::from_value::<Response<u64>>(balance_json)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "RetryGetBalance"))?
|
|
.value,
|
|
))
|
|
}
|
|
|
|
pub fn get_account(&self, pubkey: &Pubkey) -> ClientResult<Account> {
|
|
self.get_account_with_commitment(pubkey, CommitmentConfig::default())?
|
|
.value
|
|
.ok_or_else(|| RpcError::ForUser(format!("AccountNotFound: pubkey={}", pubkey)).into())
|
|
}
|
|
|
|
pub fn get_account_with_commitment(
|
|
&self,
|
|
pubkey: &Pubkey,
|
|
commitment_config: CommitmentConfig,
|
|
) -> RpcResult<Option<Account>> {
|
|
let response = self.client.send(
|
|
&RpcRequest::GetAccountInfo,
|
|
json!([pubkey.to_string(), commitment_config]),
|
|
0,
|
|
);
|
|
|
|
response
|
|
.map(|result_json| {
|
|
if result_json.is_null() {
|
|
return Err(
|
|
RpcError::ForUser(format!("AccountNotFound: pubkey={}", pubkey)).into(),
|
|
);
|
|
}
|
|
let Response {
|
|
context,
|
|
value: rpc_account,
|
|
} = serde_json::from_value::<Response<Option<RpcAccount>>>(result_json)?;
|
|
trace!("Response account {:?} {:?}", pubkey, rpc_account);
|
|
let account = rpc_account.and_then(|rpc_account| rpc_account.decode().ok());
|
|
Ok(Response {
|
|
context,
|
|
value: account,
|
|
})
|
|
})
|
|
.map_err(|err| {
|
|
Into::<ClientError>::into(RpcError::ForUser(format!(
|
|
"AccountNotFound: pubkey={}: {}",
|
|
pubkey, err
|
|
)))
|
|
})?
|
|
}
|
|
|
|
pub fn get_account_data(&self, pubkey: &Pubkey) -> ClientResult<Vec<u8>> {
|
|
Ok(self.get_account(pubkey)?.data)
|
|
}
|
|
|
|
pub fn get_minimum_balance_for_rent_exemption(&self, data_len: usize) -> ClientResult<u64> {
|
|
let minimum_balance_json = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetMinimumBalanceForRentExemption,
|
|
json!([data_len]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetMinimumBalanceForRentExemption"))?;
|
|
|
|
let minimum_balance: u64 = serde_json::from_value(minimum_balance_json).map_err(|err| {
|
|
ClientError::new_with_command(err.into(), "GetMinimumBalanceForRentExemption")
|
|
})?;
|
|
trace!(
|
|
"Response minimum balance {:?} {:?}",
|
|
data_len,
|
|
minimum_balance
|
|
);
|
|
Ok(minimum_balance)
|
|
}
|
|
|
|
/// Request the balance of the account `pubkey`.
|
|
pub fn get_balance(&self, pubkey: &Pubkey) -> ClientResult<u64> {
|
|
Ok(self
|
|
.get_balance_with_commitment(pubkey, CommitmentConfig::default())?
|
|
.value)
|
|
}
|
|
|
|
pub fn get_balance_with_commitment(
|
|
&self,
|
|
pubkey: &Pubkey,
|
|
commitment_config: CommitmentConfig,
|
|
) -> RpcResult<u64> {
|
|
let balance_json = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetBalance,
|
|
json!([pubkey.to_string(), commitment_config]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetBalance"))?;
|
|
|
|
serde_json::from_value::<Response<u64>>(balance_json)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetBalance"))
|
|
}
|
|
|
|
pub fn get_program_accounts(&self, pubkey: &Pubkey) -> ClientResult<Vec<(Pubkey, Account)>> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetProgramAccounts,
|
|
json!([pubkey.to_string()]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetProgramAccounts"))?;
|
|
|
|
let accounts: Vec<RpcKeyedAccount> =
|
|
serde_json::from_value::<Vec<RpcKeyedAccount>>(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetProgramAccounts"))?;
|
|
|
|
let mut pubkey_accounts: Vec<(Pubkey, Account)> = Vec::new();
|
|
for RpcKeyedAccount { pubkey, account } in accounts.into_iter() {
|
|
let pubkey = pubkey.parse().map_err(|_| {
|
|
ClientError::new_with_command(
|
|
RpcError::ParseError("Pubkey".to_string()).into(),
|
|
"GetProgramAccounts",
|
|
)
|
|
})?;
|
|
pubkey_accounts.push((pubkey, account.decode().unwrap()));
|
|
}
|
|
Ok(pubkey_accounts)
|
|
}
|
|
|
|
/// Request the transaction count.
|
|
pub fn get_transaction_count(&self) -> ClientResult<u64> {
|
|
self.get_transaction_count_with_commitment(CommitmentConfig::default())
|
|
}
|
|
|
|
pub fn get_transaction_count_with_commitment(
|
|
&self,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<u64> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetTransactionCount,
|
|
json!([commitment_config]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetTransactionCount"))?;
|
|
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetTransactionCount"))
|
|
}
|
|
|
|
pub fn get_recent_blockhash(&self) -> ClientResult<(Hash, FeeCalculator)> {
|
|
Ok(self
|
|
.get_recent_blockhash_with_commitment(CommitmentConfig::default())?
|
|
.value)
|
|
}
|
|
|
|
pub fn get_recent_blockhash_with_commitment(
|
|
&self,
|
|
commitment_config: CommitmentConfig,
|
|
) -> RpcResult<(Hash, FeeCalculator)> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetRecentBlockhash,
|
|
json!([commitment_config]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetRecentBlockhash"))?;
|
|
|
|
let Response {
|
|
context,
|
|
value:
|
|
RpcBlockhashFeeCalculator {
|
|
blockhash,
|
|
fee_calculator,
|
|
},
|
|
} = serde_json::from_value::<Response<RpcBlockhashFeeCalculator>>(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetRecentBlockhash"))?;
|
|
let blockhash = blockhash.parse().map_err(|_| {
|
|
ClientError::new_with_command(
|
|
RpcError::ParseError("Hash".to_string()).into(),
|
|
"GetRecentBlockhash",
|
|
)
|
|
})?;
|
|
Ok(Response {
|
|
context,
|
|
value: (blockhash, fee_calculator),
|
|
})
|
|
}
|
|
|
|
pub fn get_fee_calculator_for_blockhash(
|
|
&self,
|
|
blockhash: &Hash,
|
|
) -> ClientResult<Option<FeeCalculator>> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetFeeCalculatorForBlockhash,
|
|
json!([blockhash.to_string()]),
|
|
0,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetFeeCalculatorForBlockhash"))?;
|
|
let Response { value, .. } = serde_json::from_value::<Response<Option<RpcFeeCalculator>>>(
|
|
response,
|
|
)
|
|
.map_err(|e| ClientError::new_with_command(e.into(), "GetFeeCalculatorForBlockhash"))?;
|
|
Ok(value.map(|rf| rf.fee_calculator))
|
|
}
|
|
|
|
pub fn get_fee_rate_governor(&self) -> RpcResult<FeeRateGovernor> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetFeeRateGovernor, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetFeeRateGovernor"))?;
|
|
let Response {
|
|
context,
|
|
value: RpcFeeRateGovernor { fee_rate_governor },
|
|
} = serde_json::from_value::<Response<RpcFeeRateGovernor>>(response)
|
|
.map_err(|e| ClientError::new_with_command(e.into(), "GetFeeRateGovernor"))?;
|
|
Ok(Response {
|
|
context,
|
|
value: fee_rate_governor,
|
|
})
|
|
}
|
|
|
|
pub fn get_new_blockhash(&self, blockhash: &Hash) -> ClientResult<(Hash, FeeCalculator)> {
|
|
let mut num_retries = 0;
|
|
let start = Instant::now();
|
|
while start.elapsed().as_secs() < 5 {
|
|
if let Ok((new_blockhash, fee_calculator)) = self.get_recent_blockhash() {
|
|
if new_blockhash != *blockhash {
|
|
return Ok((new_blockhash, fee_calculator));
|
|
}
|
|
}
|
|
debug!("Got same blockhash ({:?}), will retry...", blockhash);
|
|
|
|
// Retry ~twice during a slot
|
|
sleep(Duration::from_millis(
|
|
500 * DEFAULT_TICKS_PER_SLOT / DEFAULT_TICKS_PER_SECOND,
|
|
));
|
|
num_retries += 1;
|
|
}
|
|
Err(RpcError::ForUser(format!(
|
|
"Unable to get new blockhash after {}ms (retried {} times), stuck at {}",
|
|
start.elapsed().as_millis(),
|
|
num_retries,
|
|
blockhash
|
|
))
|
|
.into())
|
|
}
|
|
|
|
pub fn get_genesis_hash(&self) -> ClientResult<Hash> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::GetGenesisHash, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("GetGenesisHash"))?;
|
|
|
|
let hash = serde_json::from_value::<String>(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetGenesisHash"))?;
|
|
|
|
let hash = hash.parse().map_err(|_| {
|
|
ClientError::new_with_command(
|
|
RpcError::ParseError("Hash".to_string()).into(),
|
|
"GetGenesisHash",
|
|
)
|
|
})?;
|
|
Ok(hash)
|
|
}
|
|
|
|
pub fn poll_balance_with_timeout_and_commitment(
|
|
&self,
|
|
pubkey: &Pubkey,
|
|
polling_frequency: &Duration,
|
|
timeout: &Duration,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<u64> {
|
|
let now = Instant::now();
|
|
loop {
|
|
match self.get_balance_with_commitment(&pubkey, commitment_config.clone()) {
|
|
Ok(bal) => {
|
|
return Ok(bal.value);
|
|
}
|
|
Err(e) => {
|
|
sleep(*polling_frequency);
|
|
if now.elapsed() > *timeout {
|
|
return Err(e);
|
|
}
|
|
}
|
|
};
|
|
}
|
|
}
|
|
|
|
pub fn poll_get_balance_with_commitment(
|
|
&self,
|
|
pubkey: &Pubkey,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<u64> {
|
|
self.poll_balance_with_timeout_and_commitment(
|
|
pubkey,
|
|
&Duration::from_millis(100),
|
|
&Duration::from_secs(1),
|
|
commitment_config,
|
|
)
|
|
}
|
|
|
|
pub fn wait_for_balance_with_commitment(
|
|
&self,
|
|
pubkey: &Pubkey,
|
|
expected_balance: Option<u64>,
|
|
commitment_config: CommitmentConfig,
|
|
) -> Option<u64> {
|
|
const LAST: usize = 30;
|
|
for run in 0..LAST {
|
|
let balance_result =
|
|
self.poll_get_balance_with_commitment(pubkey, commitment_config.clone());
|
|
if expected_balance.is_none() {
|
|
return balance_result.ok();
|
|
}
|
|
trace!(
|
|
"retry_get_balance[{}] {:?} {:?}",
|
|
run,
|
|
balance_result,
|
|
expected_balance
|
|
);
|
|
if let (Some(expected_balance), Ok(balance_result)) = (expected_balance, balance_result)
|
|
{
|
|
if expected_balance == balance_result {
|
|
return Some(balance_result);
|
|
}
|
|
}
|
|
}
|
|
None
|
|
}
|
|
|
|
/// Poll the server to confirm a transaction.
|
|
pub fn poll_for_signature(&self, signature: &Signature) -> ClientResult<()> {
|
|
self.poll_for_signature_with_commitment(signature, CommitmentConfig::default())
|
|
}
|
|
|
|
/// Poll the server to confirm a transaction.
|
|
pub fn poll_for_signature_with_commitment(
|
|
&self,
|
|
signature: &Signature,
|
|
commitment_config: CommitmentConfig,
|
|
) -> ClientResult<()> {
|
|
let now = Instant::now();
|
|
loop {
|
|
if let Ok(Some(_)) =
|
|
self.get_signature_status_with_commitment(&signature, commitment_config.clone())
|
|
{
|
|
break;
|
|
}
|
|
if now.elapsed().as_secs() > 15 {
|
|
return Err(RpcError::ForUser(format!(
|
|
"signature not found after {} seconds",
|
|
now.elapsed().as_secs()
|
|
))
|
|
.into());
|
|
}
|
|
sleep(Duration::from_millis(250));
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
/// Check a signature in the bank.
|
|
pub fn check_signature(&self, signature: &Signature) -> bool {
|
|
trace!("check_signature: {:?}", signature);
|
|
|
|
for _ in 0..30 {
|
|
let response =
|
|
self.confirm_transaction_with_commitment(signature, CommitmentConfig::recent());
|
|
match response {
|
|
Ok(Response {
|
|
value: signature_status,
|
|
..
|
|
}) => {
|
|
if signature_status {
|
|
trace!("Response found signature");
|
|
} else {
|
|
trace!("Response signature not found");
|
|
}
|
|
|
|
return signature_status;
|
|
}
|
|
Err(err) => {
|
|
debug!("check_signature request failed: {:?}", err);
|
|
}
|
|
};
|
|
sleep(Duration::from_millis(250));
|
|
}
|
|
|
|
panic!("Couldn't check signature of {}", signature);
|
|
}
|
|
|
|
/// Poll the server to confirm a transaction.
|
|
pub fn poll_for_signature_confirmation(
|
|
&self,
|
|
signature: &Signature,
|
|
min_confirmed_blocks: usize,
|
|
) -> ClientResult<usize> {
|
|
let mut now = Instant::now();
|
|
let mut confirmed_blocks = 0;
|
|
loop {
|
|
let response = self.get_num_blocks_since_signature_confirmation(signature);
|
|
match response {
|
|
Ok(count) => {
|
|
if confirmed_blocks != count {
|
|
info!(
|
|
"signature {} confirmed {} out of {} after {} ms",
|
|
signature,
|
|
count,
|
|
min_confirmed_blocks,
|
|
now.elapsed().as_millis()
|
|
);
|
|
now = Instant::now();
|
|
confirmed_blocks = count;
|
|
}
|
|
if count >= min_confirmed_blocks {
|
|
break;
|
|
}
|
|
}
|
|
Err(err) => {
|
|
debug!("check_confirmations request failed: {:?}", err);
|
|
}
|
|
};
|
|
if now.elapsed().as_secs() > 20 {
|
|
info!(
|
|
"signature {} confirmed {} out of {} failed after {} ms",
|
|
signature,
|
|
confirmed_blocks,
|
|
min_confirmed_blocks,
|
|
now.elapsed().as_millis()
|
|
);
|
|
if confirmed_blocks > 0 {
|
|
return Ok(confirmed_blocks);
|
|
} else {
|
|
return Err(RpcError::ForUser(format!(
|
|
"signature not found after {} seconds",
|
|
now.elapsed().as_secs()
|
|
))
|
|
.into());
|
|
}
|
|
}
|
|
sleep(Duration::from_millis(250));
|
|
}
|
|
Ok(confirmed_blocks)
|
|
}
|
|
|
|
pub fn get_num_blocks_since_signature_confirmation(
|
|
&self,
|
|
signature: &Signature,
|
|
) -> ClientResult<usize> {
|
|
let response = self
|
|
.client
|
|
.send(
|
|
&RpcRequest::GetSignatureStatuses,
|
|
json!([[signature.to_string()]]),
|
|
5,
|
|
)
|
|
.map_err(|err| err.into_with_command("GetSignatureStatuses"))?;
|
|
let result: Response<Vec<Option<TransactionStatus>>> = serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "GetSignatureStatuses"))?;
|
|
|
|
let confirmations = result.value[0]
|
|
.clone()
|
|
.ok_or_else(|| {
|
|
ClientError::new_with_command(
|
|
ClientErrorKind::Custom("signature not found".to_string()),
|
|
"GetSignatureStatuses",
|
|
)
|
|
})?
|
|
.confirmations
|
|
.unwrap_or(MAX_LOCKOUT_HISTORY + 1);
|
|
Ok(confirmations)
|
|
}
|
|
|
|
pub fn send_and_confirm_transaction_with_spinner<T: Signers>(
|
|
&self,
|
|
transaction: &mut Transaction,
|
|
signer_keys: &T,
|
|
) -> ClientResult<Signature> {
|
|
let mut confirmations = 0;
|
|
|
|
let progress_bar = new_spinner_progress_bar();
|
|
progress_bar.set_message(&format!(
|
|
"[{}/{}] Waiting for confirmations",
|
|
confirmations,
|
|
MAX_LOCKOUT_HISTORY + 1,
|
|
));
|
|
|
|
let mut send_retries = 20;
|
|
let signature = loop {
|
|
let mut status_retries = 15;
|
|
let (signature, status) = loop {
|
|
let signature = self.send_transaction(transaction)?;
|
|
|
|
// Get recent commitment in order to count confirmations for successful transactions
|
|
let status = self
|
|
.get_signature_status_with_commitment(&signature, CommitmentConfig::recent())?;
|
|
if status.is_none() {
|
|
status_retries -= 1;
|
|
if status_retries == 0 {
|
|
break (signature, status);
|
|
}
|
|
} else {
|
|
break (signature, status);
|
|
}
|
|
|
|
if cfg!(not(test)) {
|
|
sleep(Duration::from_millis(500));
|
|
}
|
|
};
|
|
send_retries = if let Some(result) = status.clone() {
|
|
match result {
|
|
Ok(_) => 0,
|
|
Err(TransactionError::AccountInUse) => {
|
|
// Fetch a new blockhash and re-sign the transaction before sending it again
|
|
self.resign_transaction(transaction, signer_keys)?;
|
|
send_retries - 1
|
|
}
|
|
// If transaction errors, return right away; no point in counting confirmations
|
|
Err(_) => 0,
|
|
}
|
|
} else {
|
|
send_retries - 1
|
|
};
|
|
if send_retries == 0 {
|
|
if let Some(result) = status {
|
|
match result {
|
|
Ok(_) => {
|
|
break signature;
|
|
}
|
|
Err(err) => {
|
|
return Err(err.into());
|
|
}
|
|
}
|
|
} else {
|
|
return Err(
|
|
RpcError::ForUser("unable to confirm transaction. This can happen in situations such as transaction expiration and insufficient fee-payer funds".to_string()).into(),
|
|
);
|
|
}
|
|
}
|
|
};
|
|
loop {
|
|
// Return when default (max) commitment is reached
|
|
// Failed transactions have already been eliminated, `is_some` check is sufficient
|
|
if self.get_signature_status(&signature)?.is_some() {
|
|
progress_bar.set_message("Transaction confirmed");
|
|
progress_bar.finish_and_clear();
|
|
return Ok(signature);
|
|
}
|
|
progress_bar.set_message(&format!(
|
|
"[{}/{}] Waiting for confirmations",
|
|
confirmations + 1,
|
|
MAX_LOCKOUT_HISTORY + 1,
|
|
));
|
|
sleep(Duration::from_millis(500));
|
|
confirmations = self.get_num_blocks_since_signature_confirmation(&signature)?;
|
|
}
|
|
}
|
|
|
|
pub fn validator_exit(&self) -> ClientResult<bool> {
|
|
let response = self
|
|
.client
|
|
.send(&RpcRequest::ValidatorExit, Value::Null, 0)
|
|
.map_err(|err| err.into_with_command("ValidatorExit"))?;
|
|
serde_json::from_value(response)
|
|
.map_err(|err| ClientError::new_with_command(err.into(), "ValidatorExit"))
|
|
}
|
|
|
|
pub fn send(&self, request: &RpcRequest, params: Value, retries: usize) -> ClientResult<Value> {
|
|
assert!(params.is_array() || params.is_null());
|
|
self.client.send(request, params, retries)
|
|
}
|
|
}
|
|
|
|
fn new_spinner_progress_bar() -> ProgressBar {
|
|
let progress_bar = ProgressBar::new(42);
|
|
progress_bar
|
|
.set_style(ProgressStyle::default_spinner().template("{spinner:.green} {wide_msg}"));
|
|
progress_bar.enable_steady_tick(100);
|
|
progress_bar
|
|
}
|
|
|
|
pub fn get_rpc_request_str(rpc_addr: SocketAddr, tls: bool) -> String {
|
|
if tls {
|
|
format!("https://{}", rpc_addr)
|
|
} else {
|
|
format!("http://{}", rpc_addr)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::{client_error::ClientErrorKind, mock_rpc_client_request::PUBKEY};
|
|
use assert_matches::assert_matches;
|
|
use jsonrpc_core::{Error, IoHandler, Params};
|
|
use jsonrpc_http_server::{AccessControlAllowOrigin, DomainsValidation, ServerBuilder};
|
|
use serde_json::Number;
|
|
use solana_logger;
|
|
use solana_sdk::{
|
|
instruction::InstructionError, signature::Keypair, system_transaction,
|
|
transaction::TransactionError,
|
|
};
|
|
use std::{io, sync::mpsc::channel, thread};
|
|
|
|
#[test]
|
|
fn test_send() {
|
|
let (sender, receiver) = channel();
|
|
thread::spawn(move || {
|
|
let rpc_addr = "0.0.0.0:0".parse().unwrap();
|
|
let mut io = IoHandler::default();
|
|
// Successful request
|
|
io.add_method("getBalance", |_params: Params| {
|
|
Ok(Value::Number(Number::from(50)))
|
|
});
|
|
// Failed request
|
|
io.add_method("getRecentBlockhash", |params: Params| {
|
|
if params != Params::None {
|
|
Err(Error::invalid_request())
|
|
} else {
|
|
Ok(Value::String(
|
|
"deadbeefXjn8o3yroDHxUtKsZZgoy4GPkPPXfouKNHhx".to_string(),
|
|
))
|
|
}
|
|
});
|
|
|
|
let server = ServerBuilder::new(io)
|
|
.threads(1)
|
|
.cors(DomainsValidation::AllowOnly(vec![
|
|
AccessControlAllowOrigin::Any,
|
|
]))
|
|
.start_http(&rpc_addr)
|
|
.expect("Unable to start RPC server");
|
|
sender.send(*server.address()).unwrap();
|
|
server.wait();
|
|
});
|
|
|
|
let rpc_addr = receiver.recv().unwrap();
|
|
let rpc_client = RpcClient::new_socket(rpc_addr);
|
|
|
|
let balance = rpc_client.send(
|
|
&RpcRequest::GetBalance,
|
|
json!(["deadbeefXjn8o3yroDHxUtKsZZgoy4GPkPPXfouKNHhx"]),
|
|
0,
|
|
);
|
|
assert_eq!(balance.unwrap().as_u64().unwrap(), 50);
|
|
|
|
let blockhash = rpc_client.send(&RpcRequest::GetRecentBlockhash, Value::Null, 0);
|
|
assert_eq!(
|
|
blockhash.unwrap().as_str().unwrap(),
|
|
"deadbeefXjn8o3yroDHxUtKsZZgoy4GPkPPXfouKNHhx"
|
|
);
|
|
|
|
// Send erroneous parameter
|
|
let blockhash = rpc_client.send(&RpcRequest::GetRecentBlockhash, json!(["parameter"]), 0);
|
|
assert_eq!(blockhash.is_err(), true);
|
|
}
|
|
|
|
#[test]
|
|
fn test_retry_send() {
|
|
solana_logger::setup();
|
|
let (sender, receiver) = channel();
|
|
thread::spawn(move || {
|
|
// 1. Pick a random port
|
|
// 2. Tell the client to start using it
|
|
// 3. Delay for 1.5 seconds before starting the server to ensure the client will fail
|
|
// and need to retry
|
|
let rpc_addr: SocketAddr = "0.0.0.0:4242".parse().unwrap();
|
|
sender.send(rpc_addr.clone()).unwrap();
|
|
sleep(Duration::from_millis(1500));
|
|
|
|
let mut io = IoHandler::default();
|
|
io.add_method("getBalance", move |_params: Params| {
|
|
Ok(Value::Number(Number::from(5)))
|
|
});
|
|
let server = ServerBuilder::new(io)
|
|
.threads(1)
|
|
.cors(DomainsValidation::AllowOnly(vec![
|
|
AccessControlAllowOrigin::Any,
|
|
]))
|
|
.start_http(&rpc_addr)
|
|
.expect("Unable to start RPC server");
|
|
server.wait();
|
|
});
|
|
|
|
let rpc_addr = receiver.recv().unwrap();
|
|
let rpc_client = RpcClient::new_socket(rpc_addr);
|
|
|
|
let balance = rpc_client.send(
|
|
&RpcRequest::GetBalance,
|
|
json!(["deadbeefXjn8o3yroDHxUtKsZZgoy4GPkPPXfouKNHhw"]),
|
|
10,
|
|
);
|
|
assert_eq!(balance.unwrap().as_u64().unwrap(), 5);
|
|
}
|
|
|
|
#[test]
|
|
fn test_send_transaction() {
|
|
let rpc_client = RpcClient::new_mock("succeeds".to_string());
|
|
|
|
let key = Keypair::new();
|
|
let to = Pubkey::new_rand();
|
|
let blockhash = Hash::default();
|
|
let tx = system_transaction::transfer(&key, &to, 50, blockhash);
|
|
|
|
let signature = rpc_client.send_transaction(&tx);
|
|
assert_eq!(signature.unwrap(), tx.signatures[0]);
|
|
|
|
let rpc_client = RpcClient::new_mock("fails".to_string());
|
|
|
|
let signature = rpc_client.send_transaction(&tx);
|
|
assert!(signature.is_err());
|
|
|
|
// Test bad signature returned from rpc node
|
|
let rpc_client = RpcClient::new_mock("malicious".to_string());
|
|
let signature = rpc_client.send_transaction(&tx);
|
|
assert!(signature.is_err());
|
|
}
|
|
#[test]
|
|
fn test_get_recent_blockhash() {
|
|
let rpc_client = RpcClient::new_mock("succeeds".to_string());
|
|
|
|
let expected_blockhash: Hash = PUBKEY.parse().unwrap();
|
|
|
|
let (blockhash, _fee_calculator) = rpc_client.get_recent_blockhash().expect("blockhash ok");
|
|
assert_eq!(blockhash, expected_blockhash);
|
|
|
|
let rpc_client = RpcClient::new_mock("fails".to_string());
|
|
|
|
assert!(rpc_client.get_recent_blockhash().is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_get_signature_status() {
|
|
let signature = Signature::default();
|
|
|
|
let rpc_client = RpcClient::new_mock("succeeds".to_string());
|
|
let status = rpc_client.get_signature_status(&signature).unwrap();
|
|
assert_eq!(status, Some(Ok(())));
|
|
|
|
let rpc_client = RpcClient::new_mock("sig_not_found".to_string());
|
|
let status = rpc_client.get_signature_status(&signature).unwrap();
|
|
assert_eq!(status, None);
|
|
|
|
let rpc_client = RpcClient::new_mock("account_in_use".to_string());
|
|
let status = rpc_client.get_signature_status(&signature).unwrap();
|
|
assert_eq!(status, Some(Err(TransactionError::AccountInUse)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_send_and_confirm_transaction() {
|
|
let rpc_client = RpcClient::new_mock("succeeds".to_string());
|
|
|
|
let key = Keypair::new();
|
|
let to = Pubkey::new_rand();
|
|
let blockhash = Hash::default();
|
|
let mut tx = system_transaction::transfer(&key, &to, 50, blockhash);
|
|
|
|
let result = rpc_client.send_and_confirm_transaction(&mut tx, &[&key]);
|
|
result.unwrap();
|
|
|
|
let rpc_client = RpcClient::new_mock("account_in_use".to_string());
|
|
let result = rpc_client.send_and_confirm_transaction(&mut tx, &[&key]);
|
|
assert!(result.is_err());
|
|
|
|
let rpc_client = RpcClient::new_mock("instruction_error".to_string());
|
|
let result = rpc_client.send_and_confirm_transaction(&mut tx, &[&key]);
|
|
assert_matches!(
|
|
result.unwrap_err().kind(),
|
|
ClientErrorKind::TransactionError(TransactionError::InstructionError(
|
|
0,
|
|
InstructionError::UninitializedAccount
|
|
))
|
|
);
|
|
|
|
let rpc_client = RpcClient::new_mock("sig_not_found".to_string());
|
|
let result = rpc_client.send_and_confirm_transaction(&mut tx, &[&key]);
|
|
if let ClientErrorKind::Io(err) = result.unwrap_err().kind() {
|
|
assert_eq!(err.kind(), io::ErrorKind::Other);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_resign_transaction() {
|
|
let rpc_client = RpcClient::new_mock("succeeds".to_string());
|
|
|
|
let key = Keypair::new();
|
|
let to = Pubkey::new_rand();
|
|
let blockhash: Hash = "HUu3LwEzGRsUkuJS121jzkPJW39Kq62pXCTmTa1F9jDL"
|
|
.parse()
|
|
.unwrap();
|
|
let prev_tx = system_transaction::transfer(&key, &to, 50, blockhash);
|
|
let mut tx = system_transaction::transfer(&key, &to, 50, blockhash);
|
|
|
|
rpc_client.resign_transaction(&mut tx, &[&key]).unwrap();
|
|
|
|
assert_ne!(prev_tx, tx);
|
|
assert_ne!(prev_tx.signatures, tx.signatures);
|
|
assert_ne!(
|
|
prev_tx.message().recent_blockhash,
|
|
tx.message().recent_blockhash
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rpc_client_thread() {
|
|
let rpc_client = RpcClient::new_mock("succeeds".to_string());
|
|
thread::spawn(move || rpc_client);
|
|
}
|
|
}
|