357 lines
14 KiB
Rust
357 lines
14 KiB
Rust
#![allow(unused_imports)]
|
|
|
|
use ethers_contract::ContractFactory;
|
|
use ethers_core::{abi::Abi, types::*, utils::parse_ether};
|
|
use ethers_middleware::signer::SignerMiddleware;
|
|
use ethers_providers::{Http, JsonRpcClient, Middleware, Provider};
|
|
use ethers_signers::{coins_bip39::English, LocalWallet, MnemonicBuilder, Signer};
|
|
use ethers_solc::Solc;
|
|
use once_cell::sync::Lazy;
|
|
use std::{
|
|
convert::TryFrom,
|
|
sync::{atomic::AtomicU8, Arc},
|
|
time::Duration,
|
|
};
|
|
|
|
#[allow(dead_code)]
|
|
static WALLETS: Lazy<TestWallets> = Lazy::new(|| {
|
|
TestWallets {
|
|
mnemonic: MnemonicBuilder::default()
|
|
// Please don't drain this :)
|
|
.phrase("impose air often almost medal sudden finish quote dwarf devote theme layer"),
|
|
next: Default::default(),
|
|
}
|
|
});
|
|
|
|
#[derive(Debug, Default)]
|
|
#[allow(dead_code)]
|
|
struct TestWallets {
|
|
mnemonic: MnemonicBuilder<English>,
|
|
next: AtomicU8,
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
impl TestWallets {
|
|
/// Helper for funding the wallets with an instantiated provider
|
|
#[allow(unused)]
|
|
pub async fn fund<T: JsonRpcClient, U: Into<u32>>(&self, provider: &Provider<T>, n: U) {
|
|
let addrs = (0..n.into()).map(|i| self.get(i).address()).collect::<Vec<_>>();
|
|
// hardcoded funder address private key, GOERLI
|
|
let signer = "9867bd0f8d9e16c57f5251b35a73f6f903eb8eee1bdc7f15256d0dc09d1945fb"
|
|
.parse::<LocalWallet>()
|
|
.unwrap()
|
|
.with_chain_id(provider.get_chainid().await.unwrap().as_u64());
|
|
let provider = SignerMiddleware::new(provider, signer);
|
|
let addr = provider.address();
|
|
|
|
let mut nonce = provider.get_transaction_count(addr, None).await.unwrap();
|
|
let mut pending_txs = Vec::new();
|
|
for addr in addrs {
|
|
println!("Funding wallet {addr:?}");
|
|
let tx = TransactionRequest::new()
|
|
.nonce(nonce)
|
|
.to(addr)
|
|
// 0.1 eth per wallet
|
|
.value(parse_ether("1").unwrap());
|
|
pending_txs.push(
|
|
provider.send_transaction(tx, Some(BlockNumber::Pending.into())).await.unwrap(),
|
|
);
|
|
nonce += 1.into();
|
|
}
|
|
|
|
futures_util::future::join_all(pending_txs).await;
|
|
}
|
|
|
|
pub fn next(&self) -> LocalWallet {
|
|
let idx = self.next.fetch_add(1, std::sync::atomic::Ordering::SeqCst);
|
|
|
|
// println!("Got wallet {:?}", wallet.address());
|
|
self.get(idx)
|
|
}
|
|
|
|
pub fn get<T: Into<u32>>(&self, idx: T) -> LocalWallet {
|
|
self.mnemonic
|
|
.clone()
|
|
.index(idx)
|
|
.expect("index not found")
|
|
.build()
|
|
.expect("cannot build wallet")
|
|
}
|
|
}
|
|
|
|
#[cfg(not(feature = "celo"))]
|
|
mod eth_tests {
|
|
use super::*;
|
|
use ethers_core::utils::Anvil;
|
|
use ethers_providers::GOERLI;
|
|
|
|
#[tokio::test]
|
|
async fn send_eth() {
|
|
let anvil = Anvil::new().spawn();
|
|
|
|
// this private key belongs to the above mnemonic
|
|
let wallet: LocalWallet = anvil.keys()[0].clone().into();
|
|
let wallet2: LocalWallet = anvil.keys()[1].clone().into();
|
|
|
|
// connect to the network
|
|
let provider = Provider::<Http>::try_from(anvil.endpoint())
|
|
.unwrap()
|
|
.interval(Duration::from_millis(10u64));
|
|
let chain_id = provider.get_chainid().await.unwrap().as_u64();
|
|
let provider = SignerMiddleware::new_with_provider_chain(provider, wallet).await.unwrap();
|
|
|
|
// craft the transaction
|
|
let tx = TransactionRequest::new().to(wallet2.address()).value(10000).chain_id(chain_id);
|
|
|
|
let balance_before = provider.get_balance(provider.address(), None).await.unwrap();
|
|
|
|
// send it!
|
|
provider.send_transaction(tx, None).await.unwrap();
|
|
|
|
let balance_after = provider.get_balance(provider.address(), None).await.unwrap();
|
|
|
|
assert!(balance_before > balance_after);
|
|
}
|
|
|
|
// hardhat compatibility test, to show hardhat rejects tx signed for other chains
|
|
#[tokio::test]
|
|
#[ignore]
|
|
async fn send_with_chain_id_hardhat() {
|
|
let wallet: LocalWallet =
|
|
"ac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80".parse().unwrap();
|
|
let provider = Provider::try_from("http://localhost:8545").unwrap();
|
|
let client = SignerMiddleware::new(provider, wallet);
|
|
|
|
let tx = TransactionRequest::new().to(Address::random()).value(100u64);
|
|
let res = client.send_transaction(tx, None).await;
|
|
|
|
let err = res.unwrap_err();
|
|
assert!(err.to_string().contains(
|
|
"Trying to send an incompatible EIP-155 transaction, signed for another chain."
|
|
));
|
|
}
|
|
|
|
#[tokio::test]
|
|
#[ignore]
|
|
async fn send_with_chain_id_anvil() {
|
|
let wallet: LocalWallet =
|
|
"ac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80".parse().unwrap();
|
|
let provider = Provider::try_from("http://localhost:8545").unwrap();
|
|
let client = SignerMiddleware::new(provider, wallet);
|
|
|
|
let tx = TransactionRequest::new().to(Address::random()).value(100u64);
|
|
let res = client.send_transaction(tx, None).await;
|
|
|
|
let _err = res.unwrap_err();
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn pending_txs_with_confirmations_testnet() {
|
|
let provider = GOERLI.provider().interval(Duration::from_millis(3000));
|
|
let chain_id = provider.get_chainid().await.unwrap();
|
|
let wallet = WALLETS.next().with_chain_id(chain_id.as_u64());
|
|
let address = wallet.address();
|
|
let provider = SignerMiddleware::new(provider, wallet);
|
|
generic_pending_txs_test(provider, address).await;
|
|
}
|
|
|
|
// different keys to avoid nonce errors
|
|
#[tokio::test]
|
|
async fn websocket_pending_txs_with_confirmations_testnet() {
|
|
let provider = GOERLI.ws().await.interval(Duration::from_millis(3000));
|
|
let chain_id = provider.get_chainid().await.unwrap();
|
|
let wallet = WALLETS.next().with_chain_id(chain_id.as_u64());
|
|
let address = wallet.address();
|
|
let provider = SignerMiddleware::new(provider, wallet);
|
|
generic_pending_txs_test(provider, address).await;
|
|
}
|
|
|
|
async fn generic_pending_txs_test<M: Middleware>(provider: M, who: Address) {
|
|
let tx = TransactionRequest::new().to(who).from(who);
|
|
let pending_tx = provider.send_transaction(tx, None).await.unwrap();
|
|
let tx_hash = *pending_tx;
|
|
let receipt = pending_tx.confirmations(1).await.unwrap().unwrap();
|
|
// got the correct receipt
|
|
assert_eq!(receipt.transaction_hash, tx_hash);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn typed_txs() {
|
|
let provider = GOERLI.provider();
|
|
|
|
let chain_id = provider.get_chainid().await.unwrap();
|
|
let wallet = WALLETS.next().with_chain_id(chain_id.as_u64());
|
|
let address = wallet.address();
|
|
// our wallet
|
|
let provider = SignerMiddleware::new(provider, wallet);
|
|
|
|
// Uncomment the below and run this test to re-fund the wallets if they get drained.
|
|
// Would be ideal if we'd have a way to do this automatically, but this should be
|
|
// happening rarely enough that it doesn't matter.
|
|
// WALLETS.fund(provider.provider(), 10u32).await;
|
|
|
|
async fn check_tx<P: JsonRpcClient + Clone>(
|
|
pending_tx: ethers_providers::PendingTransaction<'_, P>,
|
|
expected: u64,
|
|
) {
|
|
let provider = pending_tx.provider();
|
|
let receipt = pending_tx.await.unwrap().unwrap();
|
|
let tx = provider.get_transaction(receipt.transaction_hash).await.unwrap().unwrap();
|
|
assert_eq!(receipt.transaction_type, Some(expected.into()));
|
|
assert_eq!(tx.transaction_type, Some(expected.into()));
|
|
}
|
|
|
|
let nonce = provider.get_transaction_count(address, None).await.unwrap();
|
|
let bn = Some(BlockNumber::Pending.into());
|
|
let gas_price = provider.get_gas_price().await.unwrap() * 125 / 100;
|
|
|
|
let tx =
|
|
TransactionRequest::new().from(address).to(address).nonce(nonce).gas_price(gas_price);
|
|
let tx1 = provider.send_transaction(tx.clone(), bn).await.unwrap();
|
|
|
|
let tx = tx.clone().from(address).to(address).nonce(nonce + 1).with_access_list(vec![]);
|
|
let tx2 = provider.send_transaction(tx, bn).await.unwrap();
|
|
|
|
let tx = Eip1559TransactionRequest::new()
|
|
.from(address)
|
|
.to(address)
|
|
.nonce(nonce + 2)
|
|
.max_fee_per_gas(gas_price)
|
|
.max_priority_fee_per_gas(gas_price);
|
|
let tx3 = provider.send_transaction(tx, bn).await.unwrap();
|
|
|
|
futures_util::join!(check_tx(tx1, 0), check_tx(tx2, 1), check_tx(tx3, 2),);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn send_transaction_handles_tx_from_field() {
|
|
// launch anvil
|
|
let anvil = Anvil::new().spawn();
|
|
|
|
// grab 2 wallets
|
|
let signer: LocalWallet = anvil.keys()[0].clone().into();
|
|
let other: LocalWallet = anvil.keys()[1].clone().into();
|
|
|
|
// connect to the network
|
|
let provider = Provider::try_from(anvil.endpoint()).unwrap();
|
|
let provider =
|
|
SignerMiddleware::new_with_provider_chain(provider, signer.clone()).await.unwrap();
|
|
|
|
// sending a TransactionRequest with a from field of None should result
|
|
// in a transaction from the signer address
|
|
let request_from_none = TransactionRequest::new();
|
|
let receipt = provider
|
|
.send_transaction(request_from_none, None)
|
|
.await
|
|
.unwrap()
|
|
.await
|
|
.unwrap()
|
|
.unwrap();
|
|
let sent_tx = provider.get_transaction(receipt.transaction_hash).await.unwrap().unwrap();
|
|
|
|
assert_eq!(sent_tx.from, signer.address());
|
|
|
|
// sending a TransactionRequest with the signer as the from address should
|
|
// result in a transaction from the signer address
|
|
let request_from_signer = TransactionRequest::new().from(signer.address());
|
|
let receipt = provider
|
|
.send_transaction(request_from_signer, None)
|
|
.await
|
|
.unwrap()
|
|
.await
|
|
.unwrap()
|
|
.unwrap();
|
|
let sent_tx = provider.get_transaction(receipt.transaction_hash).await.unwrap().unwrap();
|
|
|
|
assert_eq!(sent_tx.from, signer.address());
|
|
|
|
// sending a TransactionRequest with a from address that is not the signer
|
|
// should result in a transaction from the specified address
|
|
let request_from_other = TransactionRequest::new().from(other.address());
|
|
let receipt = provider
|
|
.send_transaction(request_from_other, None)
|
|
.await
|
|
.unwrap()
|
|
.await
|
|
.unwrap()
|
|
.unwrap();
|
|
let sent_tx = provider.get_transaction(receipt.transaction_hash).await.unwrap().unwrap();
|
|
|
|
assert_eq!(sent_tx.from, other.address());
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "celo")]
|
|
mod celo_tests {
|
|
use super::*;
|
|
|
|
#[tokio::test]
|
|
async fn test_send_transaction() {
|
|
// Celo testnet
|
|
let provider = Provider::<Http>::try_from("https://alfajores-forno.celo-testnet.org")
|
|
.unwrap()
|
|
.interval(Duration::from_millis(3000u64));
|
|
let chain_id = provider.get_chainid().await.unwrap().as_u64();
|
|
|
|
// Funded with https://celo.org/developers/faucet
|
|
// Please do not drain this account :)
|
|
let wallet = "d652abb81e8c686edba621a895531b1f291289b63b5ef09a94f686a5ecdd5db1"
|
|
.parse::<LocalWallet>()
|
|
.unwrap()
|
|
.with_chain_id(chain_id);
|
|
let client = SignerMiddleware::new(provider, wallet);
|
|
|
|
let balance_before = client.get_balance(client.address(), None).await.unwrap();
|
|
let tx = TransactionRequest::pay(client.address(), 100);
|
|
let _receipt =
|
|
client.send_transaction(tx, None).await.unwrap().confirmations(3).await.unwrap();
|
|
let balance_after = client.get_balance(client.address(), None).await.unwrap();
|
|
assert!(balance_before > balance_after);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn deploy_and_call_contract() {
|
|
// compiles the given contract and returns the ABI and Bytecode
|
|
fn compile_contract(path: &str, name: &str) -> (Abi, Bytes) {
|
|
let path = format!("./tests/solidity-contracts/{path}");
|
|
let compiled = Solc::default().compile_source(&path).unwrap();
|
|
let contract = compiled.get(&path, name).expect("could not find contract");
|
|
let (abi, bin, _) = contract.into_parts_or_default();
|
|
(abi, bin)
|
|
}
|
|
|
|
let (abi, bytecode) = compile_contract("SimpleStorage.sol", "SimpleStorage");
|
|
|
|
// Celo testnet
|
|
let provider = Provider::<Http>::try_from("https://alfajores-forno.celo-testnet.org")
|
|
.unwrap()
|
|
.interval(Duration::from_millis(6000));
|
|
let chain_id = provider.get_chainid().await.unwrap().as_u64();
|
|
|
|
// Funded with https://celo.org/developers/faucet
|
|
let wallet = "58ea5643a78c36926ad5128a6b0d8dfcc7fc705788a993b1c724be3469bc9697"
|
|
.parse::<LocalWallet>()
|
|
.unwrap()
|
|
.with_chain_id(chain_id);
|
|
let client = SignerMiddleware::new_with_provider_chain(provider, wallet).await.unwrap();
|
|
let client = Arc::new(client);
|
|
|
|
let factory = ContractFactory::new(abi, bytecode, client);
|
|
let deployer = factory.deploy(()).unwrap().legacy();
|
|
let contract = deployer.block(BlockNumber::Pending).send().await.unwrap();
|
|
|
|
let value: U256 = contract.method("value", ()).unwrap().call().await.unwrap();
|
|
assert_eq!(value, 0.into());
|
|
|
|
// make a state mutating transaction
|
|
// gas estimation costs are sometimes under-reported on celo,
|
|
// so we manually set it to avoid failures
|
|
let call = contract.method::<_, H256>("setValue", U256::from(1)).unwrap().gas(100000);
|
|
let pending_tx = call.send().await.unwrap();
|
|
let _receipt = pending_tx.await.unwrap();
|
|
|
|
let value: U256 = contract.method("value", ()).unwrap().call().await.unwrap();
|
|
assert_eq!(value, 1.into());
|
|
}
|
|
}
|