chore: use create_test_provider_factory in more places (#9463)

This commit is contained in:
Dan Cline
2024-07-12 05:43:27 -04:00
committed by GitHub
parent 91aa8bb9d9
commit df9cb81ea6
2 changed files with 7 additions and 21 deletions

View File

@ -361,22 +361,13 @@ impl<DB: Database> Pruner<DB, ProviderFactory<DB>> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use crate::Pruner; use crate::Pruner;
use reth_chainspec::MAINNET;
use reth_db::test_utils::{create_test_rw_db, create_test_static_files_dir};
use reth_exex_types::FinishedExExHeight; use reth_exex_types::FinishedExExHeight;
use reth_provider::{providers::StaticFileProvider, ProviderFactory}; use reth_provider::{test_utils::create_test_provider_factory, ProviderFactory};
#[test] #[test]
fn is_pruning_needed() { fn is_pruning_needed() {
let db = create_test_rw_db(); let provider_factory = create_test_provider_factory();
let (_static_dir, static_dir_path) = create_test_static_files_dir();
let provider_factory = ProviderFactory::new(
db,
MAINNET.clone(),
StaticFileProvider::read_write(static_dir_path).unwrap(),
);
let (finished_exex_height_tx, finished_exex_height_rx) = let (finished_exex_height_tx, finished_exex_height_rx) =
tokio::sync::watch::channel(FinishedExExHeight::NoExExs); tokio::sync::watch::channel(FinishedExExHeight::NoExExs);

View File

@ -6,11 +6,10 @@ use crate::{
}; };
use alloy_rlp::Decodable; use alloy_rlp::Decodable;
use rayon::iter::{ParallelBridge, ParallelIterator}; use rayon::iter::{ParallelBridge, ParallelIterator};
use reth_db::test_utils::{create_test_rw_db, create_test_static_files_dir};
use reth_primitives::{BlockBody, SealedBlock, StaticFileSegment}; use reth_primitives::{BlockBody, SealedBlock, StaticFileSegment};
use reth_provider::{ use reth_provider::{
providers::{StaticFileProvider, StaticFileWriter}, providers::StaticFileWriter, test_utils::create_test_provider_factory_with_chain_spec,
HashingWriter, ProviderFactory, HashingWriter,
}; };
use reth_stages::{stages::ExecutionStage, ExecInput, Stage}; use reth_stages::{stages::ExecutionStage, ExecInput, Stage};
use std::{collections::BTreeMap, fs, path::Path, sync::Arc}; use std::{collections::BTreeMap, fs, path::Path, sync::Arc};
@ -84,13 +83,9 @@ impl Case for BlockchainTestCase {
.par_bridge() .par_bridge()
.try_for_each(|case| { .try_for_each(|case| {
// Create a new test database and initialize a provider for the test case. // Create a new test database and initialize a provider for the test case.
let db = create_test_rw_db(); let provider = create_test_provider_factory_with_chain_spec(Arc::new(
let (_static_files_dir, static_files_dir_path) = create_test_static_files_dir(); case.network.clone().into(),
let provider = ProviderFactory::new( ))
db.as_ref(),
Arc::new(case.network.clone().into()),
StaticFileProvider::read_write(static_files_dir_path).unwrap(),
)
.provider_rw() .provider_rw()
.unwrap(); .unwrap();