diff --git a/mithril-aggregator/src/command_args.rs b/mithril-aggregator/src/command_args.rs index a839a54e748..9a9d896c360 100644 --- a/mithril-aggregator/src/command_args.rs +++ b/mithril-aggregator/src/command_args.rs @@ -18,6 +18,7 @@ use tokio::{ }; use mithril_common::{ + api::APIVersionProvider, certificate_chain::MithrilCertificateVerifier, chain_observer::{CardanoCliRunner, ChainObserver}, crypto_helper::{ @@ -446,6 +447,8 @@ impl ServeCommand { let event_transmitter = Arc::new(TransmitterService::new(tx)); let mut event_store = event_store::EventStore::new(rx); + let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone())); + // Init dependency manager let dependency_manager = DependencyManager { config: config.clone(), @@ -470,6 +473,7 @@ impl ServeCommand { era_checker: era_checker.clone(), era_reader: era_reader.clone(), event_transmitter, + api_version_provider, }; let dependency_manager = Arc::new(dependency_manager); diff --git a/mithril-aggregator/src/dependency.rs b/mithril-aggregator/src/dependency.rs index 42b99ff3327..0aa23815739 100644 --- a/mithril-aggregator/src/dependency.rs +++ b/mithril-aggregator/src/dependency.rs @@ -1,4 +1,5 @@ use mithril_common::{ + api::APIVersionProvider, certificate_chain::CertificateVerifier, chain_observer::ChainObserver, crypto_helper::ProtocolGenesisVerifier, @@ -93,6 +94,9 @@ pub struct DependencyManager { /// Event Transmitter Service pub event_transmitter: Arc>, + + /// API Version provider + pub api_version_provider: Arc, } #[doc(hidden)] @@ -271,6 +275,7 @@ pub mod tests { SingleSignatureStore, SnapshotUploaderType, VerificationKeyStore, }; use mithril_common::{ + api::APIVersionProvider, certificate_chain::MithrilCertificateVerifier, chain_observer::FakeObserver, crypto_helper::{key_encode_hex, ProtocolGenesisSigner}, @@ -374,6 +379,8 @@ pub mod tests { Arc::new(TransmitterService::new(tx)) }; + let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone())); + let dependency_manager = DependencyManager { config, snapshot_store, @@ -397,6 +404,7 @@ pub mod tests { era_checker, era_reader, event_transmitter, + api_version_provider, }; let config = AggregatorConfig::new( diff --git a/mithril-aggregator/tests/test_extensions/dependency.rs b/mithril-aggregator/tests/test_extensions/dependency.rs index 63584735c10..084974961d1 100644 --- a/mithril-aggregator/tests/test_extensions/dependency.rs +++ b/mithril-aggregator/tests/test_extensions/dependency.rs @@ -5,6 +5,7 @@ use mithril_aggregator::{ MultiSignerImpl, ProtocolParametersStore, SingleSignatureStore, SnapshotUploaderType, VerificationKeyStore, }; +use mithril_common::api::APIVersionProvider; use mithril_common::certificate_chain::MithrilCertificateVerifier; use mithril_common::chain_observer::FakeObserver; use mithril_common::crypto_helper::{key_encode_hex, ProtocolGenesisSigner}; @@ -118,6 +119,8 @@ pub async fn initialize_dependencies( let (tx, rx) = tokio::sync::mpsc::unbounded_channel(); let event_transmitter = Arc::new(TransmitterService::new(tx)); + let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone())); + let dependency_manager = DependencyManager { config, snapshot_store, @@ -141,6 +144,7 @@ pub async fn initialize_dependencies( era_checker, era_reader, event_transmitter, + api_version_provider, }; let config = AggregatorConfig::new(