github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-02-19 03:04:27 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(stages): retain RocksDB TempDir in TestStageDB to prevent premature deletion (#21444)

Browse Source
This commit is contained in:
ethfanWilliam
2026-01-26 23:43:11 +04:00
committed by GitHub
parent 94235d64a8
commit f1459fcf91
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
crates/stages/stages/src/test_utils/test_db.rs
View File

@@ -38,15 +38,17 @@ use tempfile::TempDir;
pub struct TestStageDB {
pub factory: ProviderFactory<MockNodeTypesWithDB>,
pub temp_static_files_dir: TempDir,
pub temp_rocksdb_dir: TempDir,
}
impl Default for TestStageDB {
/// Create a new instance of [`TestStageDB`]
fn default() -> Self {
let (static_dir, static_dir_path) = create_test_static_files_dir();
let (_, rocksdb_dir_path) = create_test_rocksdb_dir();
let (rocksdb_dir, rocksdb_dir_path) = create_test_rocksdb_dir();
Self {
temp_static_files_dir: static_dir,
temp_rocksdb_dir: rocksdb_dir,
factory: ProviderFactory::new(
create_test_rw_db(),
MAINNET.clone(),
@@ -61,10 +63,11 @@ impl Default for TestStageDB {
impl TestStageDB {
pub fn new(path: &Path) -> Self {
let (static_dir, static_dir_path) = create_test_static_files_dir();
let (_, rocksdb_dir_path) = create_test_rocksdb_dir();
let (rocksdb_dir, rocksdb_dir_path) = create_test_rocksdb_dir();
Self {
temp_static_files_dir: static_dir,
temp_rocksdb_dir: rocksdb_dir,
factory: ProviderFactory::new(
create_test_rw_db_with_path(path),
MAINNET.clone(),