reth_rpc/eth/helpers/
state.rs

Help
1//! Contains RPC handler implementations specific to state.
2
3use reth_chainspec::{ChainSpecProvider, EthereumHardforks};
4use reth_storage_api::{BlockReader, StateProviderFactory};
5use reth_transaction_pool::TransactionPool;
6
7use reth_rpc_eth_api::{
8    helpers::{EthState, LoadState, SpawnBlocking},
9    RpcNodeCoreExt,
10};
11
12use crate::EthApi;
13
14impl<Provider, Pool, Network, EvmConfig> EthState for EthApi<Provider, Pool, Network, EvmConfig>
15where
16    Self: LoadState + SpawnBlocking,
17    Provider: BlockReader,
18{
19    fn max_proof_window(&self) -> u64 {
20        self.inner.eth_proof_window()
21    }
22}
23
24impl<Provider, Pool, Network, EvmConfig> LoadState for EthApi<Provider, Pool, Network, EvmConfig>
25where
26    Self: RpcNodeCoreExt<
27        Provider: BlockReader
28                      + StateProviderFactory
29                      + ChainSpecProvider<ChainSpec: EthereumHardforks>,
30        Pool: TransactionPool,
31    >,
32    Provider: BlockReader,
33{
34}
35
36#[cfg(test)]
37mod tests {
38    use super::*;
39    use alloy_eips::eip1559::ETHEREUM_BLOCK_GAS_LIMIT_30M;
40    use alloy_primitives::{Address, StorageKey, StorageValue, U256};
41    use reth_evm_ethereum::EthEvmConfig;
42    use reth_network_api::noop::NoopNetwork;
43    use reth_provider::test_utils::{ExtendedAccount, MockEthProvider, NoopProvider};
44    use reth_rpc_eth_api::helpers::EthState;
45    use reth_rpc_eth_types::{
46        EthStateCache, FeeHistoryCache, FeeHistoryCacheConfig, GasPriceOracle,
47    };
48    use reth_rpc_server_types::constants::{
49        DEFAULT_ETH_PROOF_WINDOW, DEFAULT_MAX_SIMULATE_BLOCKS, DEFAULT_PROOF_PERMITS,
50    };
51    use reth_tasks::pool::BlockingTaskPool;
52    use reth_transaction_pool::test_utils::{testing_pool, TestPool};
53    use std::collections::HashMap;
54
55    fn noop_eth_api() -> EthApi<NoopProvider, TestPool, NoopNetwork, EthEvmConfig> {
56        let pool = testing_pool();
57        let evm_config = EthEvmConfig::mainnet();
58
59        let cache = EthStateCache::spawn(NoopProvider::default(), Default::default());
60        EthApi::new(
61            NoopProvider::default(),
62            pool,
63            NoopNetwork::default(),
64            cache.clone(),
65            GasPriceOracle::new(NoopProvider::default(), Default::default(), cache),
66            ETHEREUM_BLOCK_GAS_LIMIT_30M,
67            DEFAULT_MAX_SIMULATE_BLOCKS,
68            DEFAULT_ETH_PROOF_WINDOW,
69            BlockingTaskPool::build().expect("failed to build tracing pool"),
70            FeeHistoryCache::new(FeeHistoryCacheConfig::default()),
71            evm_config,
72            DEFAULT_PROOF_PERMITS,
73        )
74    }
75
76    fn mock_eth_api(
77        accounts: HashMap<Address, ExtendedAccount>,
78    ) -> EthApi<MockEthProvider, TestPool, (), EthEvmConfig> {
79        let pool = testing_pool();
80        let mock_provider = MockEthProvider::default();
81
82        let evm_config = EthEvmConfig::new(mock_provider.chain_spec());
83        mock_provider.extend_accounts(accounts);
84
85        let cache = EthStateCache::spawn(mock_provider.clone(), Default::default());
86        EthApi::new(
87            mock_provider.clone(),
88            pool,
89            (),
90            cache.clone(),
91            GasPriceOracle::new(mock_provider, Default::default(), cache),
92            ETHEREUM_BLOCK_GAS_LIMIT_30M,
93            DEFAULT_MAX_SIMULATE_BLOCKS,
94            DEFAULT_ETH_PROOF_WINDOW + 1,
95            BlockingTaskPool::build().expect("failed to build tracing pool"),
96            FeeHistoryCache::new(FeeHistoryCacheConfig::default()),
97            evm_config,
98            DEFAULT_PROOF_PERMITS,
99        )
100    }
101
102    #[tokio::test]
103    async fn test_storage() {
104        // === Noop ===
105        let eth_api = noop_eth_api();
106        let address = Address::random();
107        let storage = eth_api.storage_at(address, U256::ZERO.into(), None).await.unwrap();
108        assert_eq!(storage, U256::ZERO.to_be_bytes());
109
110        // === Mock ===
111        let storage_value = StorageValue::from(1337);
112        let storage_key = StorageKey::random();
113        let storage = HashMap::from([(storage_key, storage_value)]);
114
115        let accounts =
116            HashMap::from([(address, ExtendedAccount::new(0, U256::ZERO).extend_storage(storage))]);
117        let eth_api = mock_eth_api(accounts);
118
119        let storage_key: U256 = storage_key.into();
120        let storage = eth_api.storage_at(address, storage_key.into(), None).await.unwrap();
121        assert_eq!(storage, storage_value.to_be_bytes());
122    }
123
124    #[tokio::test]
125    async fn test_get_account_missing() {
126        let eth_api = noop_eth_api();
127        let address = Address::random();
128        let account = eth_api.get_account(address, Default::default()).await.unwrap();
129        assert!(account.is_none());
130    }
131}
reth_rpc/eth/helpers/state.rs

reth_rpc/eth/helpers/
state.rs
