81 lines
3.2 KiB
Rust
81 lines
3.2 KiB
Rust
use evm_client::provider::Transaction;
|
|
use evm_client::provider::{parse_signature_rs_v, get_balance};
|
|
use ethers_core::types::{Address, U256};
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
mod native_tests {
|
|
use super::*;
|
|
use vault::{SessionManager, KeyType};
|
|
use tempfile::TempDir;
|
|
use kvstore::native::NativeStore;
|
|
use alloy_primitives::keccak256;
|
|
|
|
#[tokio::test]
|
|
async fn test_vault_sessionmanager_balance_for_new_keypair() {
|
|
let tmp_dir = TempDir::new().expect("create temp dir");
|
|
let store = NativeStore::open(tmp_dir.path().to_str().unwrap()).expect("Failed to open native store");
|
|
let mut vault = vault::Vault::new(store.clone());
|
|
let keyspace = "testspace";
|
|
let password = b"testpass";
|
|
// 1. Create keyspace
|
|
vault.create_keyspace(keyspace, password, None).await.expect("create keyspace");
|
|
// 2. Add secp256k1 keypair
|
|
let key_id = vault.add_keypair(keyspace, password, Some(KeyType::Secp256k1), None).await.expect("add keypair");
|
|
// 3. Create SessionManager and unlock keyspace
|
|
let mut session = SessionManager::new(vault);
|
|
session.unlock_keyspace(keyspace, password).await.expect("unlock keyspace");
|
|
session.select_keyspace(keyspace).expect("select keyspace");
|
|
session.select_keypair(&key_id).expect("select keypair");
|
|
let kp = session.current_keypair().expect("current keypair");
|
|
// 4. Derive Ethereum address from public key (same as in evm_client)
|
|
let pubkey = &kp.public_key;
|
|
// Remove leading 0x04 if present (uncompressed SEC1)
|
|
let pubkey = if pubkey.len() == 65 && pubkey[0] == 0x04 { &pubkey[1..] } else { pubkey.as_slice() };
|
|
let hash = keccak256(pubkey);
|
|
let address = Address::from_slice(&hash[12..]);
|
|
// 5. Query balance
|
|
let url = "https://ethereum.blockpi.network/v1/rpc/public";
|
|
let balance = get_balance(url, address).await.expect("Failed to get balance");
|
|
assert_eq!(balance, ethers_core::types::U256::zero(), "New keypair should have zero balance");
|
|
}
|
|
}
|
|
|
|
|
|
use ethers_core::types::Bytes;
|
|
|
|
#[test]
|
|
fn test_rlp_encode_unsigned() {
|
|
let tx = Transaction {
|
|
nonce: U256::from(1),
|
|
to: Address::zero(),
|
|
value: U256::from(100),
|
|
gas: U256::from(21000),
|
|
gas_price: U256::from(1),
|
|
data: Bytes::new(),
|
|
chain_id: 1u64,
|
|
};
|
|
let rlp = tx.rlp_encode_unsigned();
|
|
assert!(!rlp.is_empty());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_signature_rs_v() {
|
|
let mut sig = [0u8; 65];
|
|
sig[31] = 1; sig[63] = 2; sig[64] = 27;
|
|
let (r, s, v) = parse_signature_rs_v(&sig, 1).unwrap();
|
|
assert_eq!(r, U256::from(1));
|
|
assert_eq!(s, U256::from(2));
|
|
assert_eq!(v, 27 + 1 * 2 + 8);
|
|
}
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
#[tokio::test]
|
|
async fn test_get_balance_real_address() {
|
|
// Vitalik's address
|
|
let address = "d8dA6BF26964aF9D7eEd9e03E53415D37aA96045";
|
|
let address = ethers_core::types::Address::from_slice(&hex::decode(address).unwrap());
|
|
let url = "https://ethereum.blockpi.network/v1/rpc/public";
|
|
let balance = get_balance(url, address).await.expect("Failed to get balance");
|
|
assert!(balance > ethers_core::types::U256::zero(), "Vitalik's balance should be greater than zero");
|
|
}
|