Pass state db

chainwayxyz · Feb 4, 2025 · f8c4da4 · f8c4da4
1 parent 303171a
commit f8c4da4
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/bin/citrea/tests/e2e/mod.rs b/bin/citrea/tests/e2e/mod.rs
@@ -7,7 +7,7 @@ use citrea_common::{BatchProverConfig, SequencerConfig};
 use citrea_evm::smart_contracts::SimpleStorageContract;
 use citrea_primitives::forks::fork_from_block_number;
 use citrea_stf::genesis_config::GenesisPaths;
-use citrea_storage_ops::PruningConfig;
+use citrea_storage_ops::pruning::PruningConfig;
 use reth_primitives::BlockNumberOrTag;
 use sov_mock_da::{MockAddress, MockDaService};
 use sov_rollup_interface::rpc::{LastVerifiedBatchProofResponse, SoftConfirmationStatus};

diff --git a/bin/citrea/tests/e2e/pruning.rs b/bin/citrea/tests/e2e/pruning.rs
@@ -4,7 +4,7 @@ use std::str::FromStr;
 
 use alloy_primitives::Address;
 /// Testing if the sequencer and full node can handle system transactions correctly (the full node should have the same system transactions as the sequencer)
-use citrea_storage_ops::PruningConfig;
+use citrea_storage_ops::pruning::PruningConfig;
 use futures::FutureExt;
 use reth_primitives::BlockNumberOrTag;
 use sov_mock_da::{MockAddress, MockDaService};

diff --git a/bin/citrea/tests/test_helpers/mod.rs b/bin/citrea/tests/test_helpers/mod.rs
@@ -15,7 +15,7 @@ use citrea_common::{
 use citrea_light_client_prover::da_block_handler::StartVariant;
 use citrea_primitives::TEST_PRIVATE_KEY;
 use citrea_stf::genesis_config::GenesisPaths;
-use citrea_storage_ops::PruningConfig;
+use citrea_storage_ops::pruning::PruningConfig;
 use sov_db::ledger_db::SharedLedgerOps;
 use sov_db::rocks_db_config::RocksdbConfig;
 use sov_db::schema::tables::{

diff --git a/crates/storage-ops/src/tests.rs b/crates/storage-ops/src/tests.rs
@@ -5,6 +5,7 @@ use std::time::Duration;
 use sov_db::ledger_db::LedgerDB;
 use sov_db::native_db::NativeDB;
 use sov_db::rocks_db_config::RocksdbConfig;
+use sov_db::state_db::StateDB;
 use sov_prover_storage_manager::SnapshotManager;
 use tokio::sync::broadcast;
 use tokio_util::sync::CancellationToken;
@@ -21,9 +22,11 @@ async fn test_pruner_simple_run() {
     let rocksdb_config = RocksdbConfig::new(tmpdir.path(), None, None);
     let ledger_db = LedgerDB::with_config(&rocksdb_config).unwrap();
     let native_db = NativeDB::<SnapshotManager>::setup_schema_db(&rocksdb_config).unwrap();
+    let state_db = StateDB::<SnapshotManager>::setup_schema_db(&rocksdb_config).unwrap();
     let pruner = Pruner::new(
         PruningConfig { distance: 5 },
         ledger_db,
+        Arc::new(state_db),
         Arc::new(native_db),
     );
     let pruner_service = PrunerService::new(pruner, 0, receiver);