jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(test-utils): Make MockEthProvider generic over Transaction (#13853)

Browse Source 
Co-authored-by: Matthias Seitz <matthias.seitz@outlook.de>
This commit is contained in:
Dhanraj Avhad
2025-01-23 04:48:43 +05:30
committed by GitHub
parent 69375780f5
commit cc8558fa11
1 changed files with 48 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
crates/storage/provider/src/test_utils/mock.rs
View File

@ -42,9 +42,9 @@ use std::{
/// A mock implementation for Provider interfaces. /// A mock implementation for Provider interfaces.
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct MockEthProvider { pub struct MockEthProvider<T = TransactionSigned> {
/// Local block store /// Local block store
pub blocks: Arc<Mutex<HashMap<B256, Block>>>, pub blocks: Arc<Mutex<HashMap<B256, Block<T>>>>,
/// Local header store /// Local header store
pub headers: Arc<Mutex<HashMap<B256, Header>>>, pub headers: Arc<Mutex<HashMap<B256, Header>>>,
/// Local account store /// Local account store
@ -55,7 +55,52 @@ pub struct MockEthProvider {
pub state_roots: Arc<Mutex<Vec<B256>>>, pub state_roots: Arc<Mutex<Vec<B256>>>,
} }
impl Default for MockEthProvider { impl<T> MockEthProvider<T> {
/// Add block to local block store
pub fn add_block(&self, hash: B256, block: Block<T>) {
self.add_header(hash, block.header.clone());
self.blocks.lock().insert(hash, block);
}
/// Add multiple blocks to local block store
pub fn extend_blocks(&self, iter: impl IntoIterator<Item = (B256, Block<T>)>) {
for (hash, block) in iter {
self.add_header(hash, block.header.clone());
self.add_block(hash, block)
}
}
/// Add header to local header store
pub fn add_header(&self, hash: B256, header: Header) {
self.headers.lock().insert(hash, header);
}
/// Add multiple headers to local header store
pub fn extend_headers(&self, iter: impl IntoIterator<Item = (B256, Header)>) {
for (hash, header) in iter {
self.add_header(hash, header)
}
}
/// Add account to local account store
pub fn add_account(&self, address: Address, account: ExtendedAccount) {
self.accounts.lock().insert(address, account);
}
/// Add account to local account store
pub fn extend_accounts(&self, iter: impl IntoIterator<Item = (Address, ExtendedAccount)>) {
for (address, account) in iter {
self.add_account(address, account)
}
}
/// Add state root to local state root store
pub fn add_state_root(&self, state_root: B256) {
self.state_roots.lock().push(state_root);
}
}
impl<T> Default for MockEthProvider<T> {
fn default() -> Self { fn default() -> Self {
Self { Self {
blocks: Default::default(), blocks: Default::default(),
@ -104,51 +149,6 @@ impl ExtendedAccount {
} }
} }
impl MockEthProvider {
/// Add block to local block store
pub fn add_block(&self, hash: B256, block: Block) {
self.add_header(hash, block.header.clone());
self.blocks.lock().insert(hash, block);
}
/// Add multiple blocks to local block store
pub fn extend_blocks(&self, iter: impl IntoIterator<Item = (B256, Block)>) {
for (hash, block) in iter {
self.add_header(hash, block.header.clone());
self.add_block(hash, block)
}
}
/// Add header to local header store
pub fn add_header(&self, hash: B256, header: Header) {
self.headers.lock().insert(hash, header);
}
/// Add multiple headers to local header store
pub fn extend_headers(&self, iter: impl IntoIterator<Item = (B256, Header)>) {
for (hash, header) in iter {
self.add_header(hash, header)
}
}
/// Add account to local account store
pub fn add_account(&self, address: Address, account: ExtendedAccount) {
self.accounts.lock().insert(address, account);
}
/// Add account to local account store
pub fn extend_accounts(&self, iter: impl IntoIterator<Item = (Address, ExtendedAccount)>) {
for (address, account) in iter {
self.add_account(address, account)
}
}
/// Add state root to local state root store
pub fn add_state_root(&self, state_root: B256) {
self.state_roots.lock().push(state_root);
}
}
/// Mock node. /// Mock node.
#[derive(Debug)] #[derive(Debug)]
pub struct MockNode; pub struct MockNode;