From 0f1f47ac80f1e69fad4608322cab1c7352007c2c Mon Sep 17 00:00:00 2001
From: Roman Krasiuk <rokrassyuk@gmail.com>
Date: Wed, 22 Nov 2023 04:56:44 -0800
Subject: [PATCH] test(provider): utility functions for initializing provider
 factory (#5530)

---
 crates/storage/provider/src/test_utils/mod.rs | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/crates/storage/provider/src/test_utils/mod.rs b/crates/storage/provider/src/test_utils/mod.rs
index bbbe973908..0da47c4794 100644
--- a/crates/storage/provider/src/test_utils/mod.rs
+++ b/crates/storage/provider/src/test_utils/mod.rs
@@ -1,3 +1,11 @@
+use crate::ProviderFactory;
+use reth_db::{
+    test_utils::{create_test_rw_db, TempDatabase},
+    DatabaseEnv,
+};
+use reth_primitives::{ChainSpec, MAINNET};
+use std::sync::Arc;
+
 pub mod blocks;
 mod events;
 mod executor;
@@ -8,3 +16,16 @@ pub use events::TestCanonStateSubscriptions;
 pub use executor::{TestExecutor, TestExecutorFactory};
 pub use mock::{ExtendedAccount, MockEthProvider};
 pub use noop::NoopProvider;
+
+/// Creates test provider factory with mainnet chain spec.
+pub fn create_test_provider_factory() -> ProviderFactory<Arc<TempDatabase<DatabaseEnv>>> {
+    create_test_provider_factory_with_chain_spec(MAINNET.clone())
+}
+
+/// Creates test provider factory with provided chain spec.
+pub fn create_test_provider_factory_with_chain_spec(
+    chain_spec: Arc<ChainSpec>,
+) -> ProviderFactory<Arc<TempDatabase<DatabaseEnv>>> {
+    let db = create_test_rw_db();
+    ProviderFactory::new(db, chain_spec)
+}