chore: wire 'BlockRangeRootRetriever' in signer

input-output-hk · Apr 29, 2024 · 7fabf15 · 7fabf15
1 parent 6c3e222
commit 7fabf15
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 5 deletions.
diff --git a/mithril-signer/src/runtime/runner.rs b/mithril-signer/src/runtime/runner.rs
@@ -456,16 +456,17 @@ mod tests {
         api_version::APIVersionProvider,
         cardano_block_scanner::DumbBlockScanner,
         chain_observer::{ChainObserver, FakeObserver},
-        crypto_helper::ProtocolInitializer,
+        crypto_helper::{MKMap, MKMapNode, MKTreeNode, ProtocolInitializer},
         digesters::{DumbImmutableDigester, DumbImmutableFileObserver},
-        entities::{CardanoDbBeacon, Epoch, StakeDistribution},
+        entities::{BlockRange, CardanoDbBeacon, Epoch, ImmutableFileNumber, StakeDistribution},
         era::{
             adapters::{EraReaderAdapterType, EraReaderBootstrapAdapter},
             EraChecker, EraReader,
         },
         signable_builder::{
-            CardanoImmutableFilesFullSignableBuilder, CardanoTransactionsSignableBuilder,
-            MithrilSignableBuilderService, MithrilStakeDistributionSignableBuilder,
+            BlockRangeRootRetriever, CardanoImmutableFilesFullSignableBuilder,
+            CardanoTransactionsSignableBuilder, MithrilSignableBuilderService,
+            MithrilStakeDistributionSignableBuilder,
         },
         test_utils::{fake_data, MithrilFixtureBuilder},
         TimePointProvider, TimePointProviderImpl,
@@ -497,6 +498,23 @@ mod tests {
         }
     }
 
+    mock! {
+        pub BlockRangeRootRetrieverImpl { }
+
+        #[async_trait]
+        impl BlockRangeRootRetriever for BlockRangeRootRetrieverImpl {
+            async fn retrieve_block_range_roots(
+                &self,
+                up_to_beacon: ImmutableFileNumber,
+            ) -> StdResult<Box<dyn Iterator<Item = (BlockRange, MKTreeNode)>>>;
+
+            async fn compute_merkle_map_from_block_range_roots(
+                &self,
+                up_to_beacon: ImmutableFileNumber,
+            ) -> StdResult<MKMap<BlockRange, MKMapNode<BlockRange>>>;
+        }
+    }
+
     async fn init_services() -> SignerServices {
         let adapter: MemoryAdapter<Epoch, ProtocolInitializer> = MemoryAdapter::new(None).unwrap();
         let stake_distribution_signers = fake_data::signers_with_stakes(2);
@@ -543,8 +561,10 @@ mod tests {
             None,
             slog_scope::logger(),
         ));
+        let block_range_root_retriever = Arc::new(MockBlockRangeRootRetrieverImpl::new());
         let cardano_transactions_builder = Arc::new(CardanoTransactionsSignableBuilder::new(
             transaction_importer,
+            block_range_root_retriever,
             slog_scope::logger(),
         ));
         let signable_builder_service = Arc::new(MithrilSignableBuilderService::new(

diff --git a/mithril-signer/src/runtime/signer_services.rs b/mithril-signer/src/runtime/signer_services.rs
@@ -268,14 +268,16 @@ impl<'a> ServiceBuilder for ProductionServiceBuilder<'a> {
         ));
         let transactions_importer = CardanoTransactionsImporter::new(
             block_scanner,
-            transaction_store,
+            transaction_store.clone(),
             &self.config.db_directory,
             // Rescan the last immutable when importing transactions, it may have been partially imported
             Some(1),
             slog_scope::logger(),
         );
+        let block_range_root_retriever = transaction_store.clone();
         let cardano_transactions_builder = Arc::new(CardanoTransactionsSignableBuilder::new(
             Arc::new(transactions_importer),
+            block_range_root_retriever,
             slog_scope::logger(),
         ));
         let signable_builder_service = Arc::new(MithrilSignableBuilderService::new(

diff --git a/mithril-signer/tests/test_extensions/state_machine_tester.rs b/mithril-signer/tests/test_extensions/state_machine_tester.rs
@@ -161,8 +161,10 @@ impl StateMachineTester {
             None,
             slog_scope::logger(),
         ));
+        let block_range_root_retriever = transaction_store.clone();
         let cardano_transactions_builder = Arc::new(CardanoTransactionsSignableBuilder::new(
             transaction_importer,
+            block_range_root_retriever,
             slog_scope::logger(),
         ));
         let signable_builder_service = Arc::new(MithrilSignableBuilderService::new(