@@ -14,7 +14,7 @@ use crate::model::{MithrilNetworkConfiguration, MithrilNetworkConfigurationForEp
1414/// Trait to retrieve protocol configuration
1515#[ cfg_attr( test, mockall:: automock) ]
1616#[ async_trait]
17- pub trait ProtocolConfigurationRetriever : Sync + Send {
17+ pub trait ProtocolConfigurationRetrieverFromAggregator : Sync + Send {
1818 /// Retrieves protocol configuration for a given epoch from the aggregator
1919 async fn retrieve_protocol_configuration (
2020 & self ,
@@ -24,12 +24,14 @@ pub trait ProtocolConfigurationRetriever: Sync + Send {
2424
2525/// Structure implementing MithrilNetworkConfigurationProvider using HTTP.
2626pub struct HttpMithrilNetworkConfigurationProvider {
27- protocol_configuration_retriever : Arc < dyn ProtocolConfigurationRetriever > ,
27+ protocol_configuration_retriever : Arc < dyn ProtocolConfigurationRetrieverFromAggregator > ,
2828}
2929
3030impl HttpMithrilNetworkConfigurationProvider {
3131 /// HttpMithrilNetworkConfigurationProvider factory
32- pub fn new ( protocol_configuration_retriever : Arc < dyn ProtocolConfigurationRetriever > ) -> Self {
32+ pub fn new (
33+ protocol_configuration_retriever : Arc < dyn ProtocolConfigurationRetrieverFromAggregator > ,
34+ ) -> Self {
3335 Self {
3436 protocol_configuration_retriever,
3537 }
@@ -94,15 +96,17 @@ mod tests {
9496
9597 use crate :: {
9698 http_client:: http_impl:: {
97- HttpMithrilNetworkConfigurationProvider , MockProtocolConfigurationRetriever ,
99+ HttpMithrilNetworkConfigurationProvider ,
100+ MockProtocolConfigurationRetrieverFromAggregator ,
98101 } ,
99102 interface:: MithrilNetworkConfigurationProvider ,
100103 } ;
101104
102105 #[ tokio:: test]
103106 async fn test_get_network_configuration_retrieve_configurations_for_aggregation_next_aggregation_and_registration ( )
104107 {
105- let mut protocol_configuration_retriever = MockProtocolConfigurationRetriever :: new ( ) ;
108+ let mut protocol_configuration_retriever =
109+ MockProtocolConfigurationRetrieverFromAggregator :: new ( ) ;
106110
107111 protocol_configuration_retriever
108112 . expect_retrieve_protocol_configuration ( )
0 commit comments