Skip to content

Commit e28d896

Browse files
committed
refactor(protocol-config, signer): rename EpochConfiguration to MithrilNetworkConfigurationForEpoch
1 parent 4099356 commit e28d896

File tree

8 files changed

+48
-46
lines changed

8 files changed

+48
-46
lines changed

internal/mithril-protocol-config/src/http_client/http_impl.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ use mithril_common::entities::Epoch;
99
use mithril_common::messages::ProtocolConfigurationMessage;
1010

1111
use crate::interface::MithrilNetworkConfigurationProvider;
12-
use crate::model::{EpochConfiguration, MithrilNetworkConfiguration};
12+
use crate::model::{MithrilNetworkConfiguration, MithrilNetworkConfigurationForEpoch};
1313

1414
/// Trait to retrieve protocol configuration
1515
#[cfg_attr(test, mockall::automock)]
@@ -49,7 +49,7 @@ impl MithrilNetworkConfigurationProvider for HttpMithrilNetworkConfigurationProv
4949
let next_aggregation_epoch = epoch.offset_to_next_signer_retrieval_epoch();
5050
let registration_epoch = epoch.offset_to_next_signer_retrieval_epoch().next();
5151

52-
let configuration_for_aggregation: EpochConfiguration = self
52+
let configuration_for_aggregation: MithrilNetworkConfigurationForEpoch = self
5353
.protocol_configuration_retriever
5454
.retrieve_protocol_configuration(aggregation_epoch)
5555
.await?

internal/mithril-protocol-config/src/model.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,20 +24,20 @@ pub struct MithrilNetworkConfiguration {
2424
pub epoch: Epoch,
2525

2626
/// Configuration for aggregation
27-
pub configuration_for_aggregation: EpochConfiguration,
27+
pub configuration_for_aggregation: MithrilNetworkConfigurationForEpoch,
2828

2929
/// Configuration for next aggregation
30-
pub configuration_for_next_aggregation: EpochConfiguration,
30+
pub configuration_for_next_aggregation: MithrilNetworkConfigurationForEpoch,
3131

3232
/// Configuration for registration
33-
pub configuration_for_registration: EpochConfiguration,
33+
pub configuration_for_registration: MithrilNetworkConfigurationForEpoch,
3434
}
3535

3636
//A epoch configuration
3737
#[derive(PartialEq, Clone, Debug)]
3838

3939
/// A network configuration available for an epoch
40-
pub struct EpochConfiguration {
40+
pub struct MithrilNetworkConfigurationForEpoch {
4141
/// Cryptographic protocol parameters (`k`, `m` and `phi_f`)
4242
pub protocol_parameters: ProtocolParameters,
4343

@@ -48,9 +48,9 @@ pub struct EpochConfiguration {
4848
pub signed_entity_types_config: SignedEntityTypeConfiguration,
4949
}
5050

51-
impl From<ProtocolConfigurationMessage> for EpochConfiguration {
51+
impl From<ProtocolConfigurationMessage> for MithrilNetworkConfigurationForEpoch {
5252
fn from(message: ProtocolConfigurationMessage) -> Self {
53-
EpochConfiguration {
53+
MithrilNetworkConfigurationForEpoch {
5454
protocol_parameters: message.protocol_parameters,
5555
enabled_signed_entity_types: message.available_signed_entity_types,
5656
signed_entity_types_config: SignedEntityTypeConfiguration {

internal/mithril-protocol-config/src/test/double/configuration_provider.rs

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -4,29 +4,29 @@ use tokio::sync::RwLock;
44

55
use crate::{
66
interface::MithrilNetworkConfigurationProvider,
7-
model::{EpochConfiguration, MithrilNetworkConfiguration},
7+
model::{MithrilNetworkConfigurationForEpoch, MithrilNetworkConfiguration},
88
};
99
use async_trait::async_trait;
1010
use mithril_common::{StdResult, entities::Epoch};
1111

1212
/// A fake [MithrilNetworkConfigurationProvider] that return [MithrilNetworkConfiguration]
1313
pub struct FakeMithrilNetworkConfigurationProvider {
1414
/// Configuration for aggregation
15-
pub configuration_for_aggregation: RwLock<EpochConfiguration>,
15+
pub configuration_for_aggregation: RwLock<MithrilNetworkConfigurationForEpoch>,
1616

1717
/// Configuration for next aggregation
18-
pub configuration_for_next_aggregation: RwLock<EpochConfiguration>,
18+
pub configuration_for_next_aggregation: RwLock<MithrilNetworkConfigurationForEpoch>,
1919

2020
/// Configuration for registration
21-
pub configuration_for_registration: RwLock<EpochConfiguration>,
21+
pub configuration_for_registration: RwLock<MithrilNetworkConfigurationForEpoch>,
2222
}
2323

2424
impl FakeMithrilNetworkConfigurationProvider {
2525
/// FakeMithrilNetworkConfigurationProvider factory
2626
pub fn new(
27-
configuration_for_aggregation: EpochConfiguration,
28-
configuration_for_next_aggregation: EpochConfiguration,
29-
configuration_for_registration: EpochConfiguration,
27+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch,
28+
configuration_for_next_aggregation: MithrilNetworkConfigurationForEpoch,
29+
configuration_for_registration: MithrilNetworkConfigurationForEpoch,
3030
) -> Self {
3131
Self {
3232
configuration_for_aggregation: RwLock::new(configuration_for_aggregation),
@@ -36,7 +36,7 @@ impl FakeMithrilNetworkConfigurationProvider {
3636
}
3737

3838
///Change the configuration of the aggregation
39-
pub async fn change_aggregation_configuration(&self, conf: EpochConfiguration) {
39+
pub async fn change_aggregation_configuration(&self, conf: MithrilNetworkConfigurationForEpoch) {
4040
let mut configuration_for_aggregation = self.configuration_for_aggregation.write().await;
4141
*configuration_for_aggregation = conf;
4242
}
@@ -75,13 +75,13 @@ mod tests {
7575

7676
use crate::{
7777
interface::MithrilNetworkConfigurationProvider,
78-
model::{EpochConfiguration, SignedEntityTypeConfiguration},
78+
model::{MithrilNetworkConfigurationForEpoch, SignedEntityTypeConfiguration},
7979
test::double::configuration_provider::FakeMithrilNetworkConfigurationProvider,
8080
};
8181

8282
#[tokio::test]
8383
async fn test_get_network_configuration() {
84-
let configuration_for_aggregation = EpochConfiguration {
84+
let configuration_for_aggregation = MithrilNetworkConfigurationForEpoch {
8585
protocol_parameters: ProtocolParameters {
8686
k: 1,
8787
m: 11,
@@ -96,7 +96,7 @@ mod tests {
9696
},
9797
};
9898

99-
let configuration_for_next_aggregation = EpochConfiguration {
99+
let configuration_for_next_aggregation = MithrilNetworkConfigurationForEpoch {
100100
protocol_parameters: ProtocolParameters {
101101
k: 2,
102102
m: 22,
@@ -111,7 +111,7 @@ mod tests {
111111
},
112112
};
113113

114-
let configuration_for_registration = EpochConfiguration {
114+
let configuration_for_registration = MithrilNetworkConfigurationForEpoch {
115115
protocol_parameters: ProtocolParameters {
116116
k: 3,
117117
m: 33,

internal/mithril-protocol-config/src/test/double/dummies.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ use mithril_common::{
66
};
77

88
use crate::model::{
9-
EpochConfiguration, MithrilNetworkConfiguration, SignedEntityTypeConfiguration,
9+
MithrilNetworkConfigurationForEpoch, MithrilNetworkConfiguration, SignedEntityTypeConfiguration,
1010
};
1111

1212
impl Dummy for MithrilNetworkConfiguration {
@@ -16,14 +16,14 @@ impl Dummy for MithrilNetworkConfiguration {
1616

1717
Self {
1818
epoch: beacon.epoch,
19-
configuration_for_aggregation: EpochConfiguration::dummy(),
20-
configuration_for_next_aggregation: EpochConfiguration::dummy(),
21-
configuration_for_registration: EpochConfiguration::dummy(),
19+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch::dummy(),
20+
configuration_for_next_aggregation: MithrilNetworkConfigurationForEpoch::dummy(),
21+
configuration_for_registration: MithrilNetworkConfigurationForEpoch::dummy(),
2222
}
2323
}
2424
}
2525

26-
impl Dummy for EpochConfiguration {
26+
impl Dummy for MithrilNetworkConfigurationForEpoch {
2727
/// Return a dummy for [EpochConfiguration] (test-only).
2828
fn dummy() -> Self {
2929
Self {

mithril-signer/src/runtime/runner.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -372,7 +372,7 @@ impl Runner for SignerRunner {
372372

373373
#[cfg(test)]
374374
mod tests {
375-
use mithril_protocol_config::model::EpochConfiguration;
375+
use mithril_protocol_config::model::MithrilNetworkConfigurationForEpoch;
376376
use mockall::mock;
377377
use mockall::predicate::eq;
378378
use std::collections::BTreeSet;
@@ -557,9 +557,9 @@ mod tests {
557557
));
558558
let kes_signer = None;
559559

560-
let configuration_for_aggregation = EpochConfiguration::dummy();
561-
let configuration_for_next_aggregation = EpochConfiguration::dummy();
562-
let configuration_for_registration = EpochConfiguration::dummy();
560+
let configuration_for_aggregation = MithrilNetworkConfigurationForEpoch::dummy();
561+
let configuration_for_next_aggregation = MithrilNetworkConfigurationForEpoch::dummy();
562+
let configuration_for_registration = MithrilNetworkConfigurationForEpoch::dummy();
563563

564564
let network_configuration_service = Arc::new(FakeMithrilNetworkConfigurationProvider::new(
565565
configuration_for_aggregation,
@@ -776,7 +776,7 @@ mod tests {
776776

777777
let mithril_network_configuration = MithrilNetworkConfiguration {
778778
epoch,
779-
configuration_for_aggregation: EpochConfiguration {
779+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
780780
enabled_signed_entity_types: BTreeSet::from([
781781
SignedEntityTypeDiscriminants::MithrilStakeDistribution,
782782
SignedEntityTypeDiscriminants::CardanoTransactions,

mithril-signer/src/services/cardano_transactions/preloader_checker.rs

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,9 @@ mod tests {
5555
entities::{Epoch, SignedEntityTypeDiscriminants, TimePoint},
5656
test::double::Dummy,
5757
};
58-
use mithril_protocol_config::model::{EpochConfiguration, MithrilNetworkConfiguration};
58+
use mithril_protocol_config::model::{
59+
MithrilNetworkConfiguration, MithrilNetworkConfigurationForEpoch,
60+
};
5961
use mockall::mock;
6062
use std::collections::BTreeSet;
6163

@@ -88,7 +90,7 @@ mod tests {
8890
.times(1)
8991
.returning(|_| {
9092
Ok(MithrilNetworkConfiguration {
91-
configuration_for_aggregation: EpochConfiguration {
93+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
9294
enabled_signed_entity_types: BTreeSet::from([
9395
SignedEntityTypeDiscriminants::MithrilStakeDistribution,
9496
]),
@@ -122,7 +124,7 @@ mod tests {
122124
.times(1)
123125
.returning(|_| {
124126
Ok(MithrilNetworkConfiguration {
125-
configuration_for_aggregation: EpochConfiguration {
127+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
126128
enabled_signed_entity_types: BTreeSet::from([
127129
SignedEntityTypeDiscriminants::CardanoTransactions,
128130
]),

mithril-signer/src/services/epoch_service.rs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -437,7 +437,7 @@ mod tests {
437437
double::{Dummy, fake_data},
438438
};
439439

440-
use mithril_protocol_config::model::{EpochConfiguration, SignedEntityTypeConfiguration};
440+
use mithril_protocol_config::model::{MithrilNetworkConfigurationForEpoch, SignedEntityTypeConfiguration};
441441

442442
use crate::database::repository::{ProtocolInitializerRepository, StakePoolStore};
443443
use crate::database::test_helper::main_db_connection;
@@ -486,7 +486,7 @@ mod tests {
486486
let epoch = Epoch(12);
487487
let mithril_network_configuration = MithrilNetworkConfiguration {
488488
epoch,
489-
configuration_for_aggregation: EpochConfiguration {
489+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
490490
enabled_signed_entity_types: BTreeSet::new(),
491491
..Dummy::dummy()
492492
},
@@ -673,7 +673,7 @@ mod tests {
673673
let epoch = Epoch(12);
674674
let mithril_network_configuration = MithrilNetworkConfiguration {
675675
epoch,
676-
configuration_for_aggregation: EpochConfiguration {
676+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
677677
enabled_signed_entity_types: BTreeSet::from([
678678
SignedEntityTypeDiscriminants::CardanoImmutableFilesFull,
679679
]),
@@ -790,7 +790,7 @@ mod tests {
790790
// MithrilNetworkConfiguration
791791
let mithril_network_configuration = MithrilNetworkConfiguration {
792792
epoch,
793-
configuration_for_aggregation: EpochConfiguration {
793+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
794794
enabled_signed_entity_types: BTreeSet::new(),
795795
..Dummy::dummy()
796796
},
@@ -863,7 +863,7 @@ mod tests {
863863

864864
let mithril_network_configuration = MithrilNetworkConfiguration {
865865
epoch,
866-
configuration_for_aggregation: EpochConfiguration {
866+
configuration_for_aggregation: MithrilNetworkConfigurationForEpoch {
867867
enabled_signed_entity_types: BTreeSet::new(),
868868
..Dummy::dummy()
869869
},
@@ -909,14 +909,14 @@ mod tests {
909909
let current_signers = signers[1..3].to_vec();
910910
let next_signers = signers[2..5].to_vec();
911911

912-
let configuration_for_aggregation = EpochConfiguration {
912+
let configuration_for_aggregation = MithrilNetworkConfigurationForEpoch {
913913
signed_entity_types_config: SignedEntityTypeConfiguration {
914914
cardano_transactions: None,
915915
},
916916
enabled_signed_entity_types: BTreeSet::new(),
917917
..Dummy::dummy()
918918
};
919-
let configuration_for_next_aggregation = EpochConfiguration::dummy();
919+
let configuration_for_next_aggregation = MithrilNetworkConfigurationForEpoch::dummy();
920920

921921
epoch_service
922922
.write()
@@ -944,7 +944,7 @@ mod tests {
944944
let allowed_discriminants =
945945
BTreeSet::from([SignedEntityTypeDiscriminants::CardanoImmutableFilesFull]);
946946

947-
let configuration_for_aggregation = EpochConfiguration {
947+
let configuration_for_aggregation = MithrilNetworkConfigurationForEpoch {
948948
enabled_signed_entity_types: allowed_discriminants.clone(),
949949
..Dummy::dummy()
950950
};

mithril-signer/tests/test_extensions/state_machine_tester.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
use anyhow::anyhow;
33
use mithril_metric::{MetricCollector, MetricsServiceExporter};
44
use mithril_protocol_config::{
5-
model::{EpochConfiguration, SignedEntityTypeConfiguration},
5+
model::{MithrilNetworkConfigurationForEpoch, SignedEntityTypeConfiguration},
66
test::double::configuration_provider::FakeMithrilNetworkConfigurationProvider,
77
};
88
use prometheus_parse::Value;
@@ -175,17 +175,17 @@ impl StateMachineTester {
175175
ticker_service.clone(),
176176
));
177177

178-
let configuration_for_aggregation = EpochConfiguration {
178+
let configuration_for_aggregation = MithrilNetworkConfigurationForEpoch {
179179
signed_entity_types_config: SignedEntityTypeConfiguration {
180180
cardano_transactions: Some(cardano_transactions_signing_config.clone()),
181181
},
182182
enabled_signed_entity_types: SignedEntityTypeDiscriminants::all(),
183183
..Dummy::dummy()
184184
};
185185

186-
let configuration_for_next_aggregation = EpochConfiguration::dummy();
186+
let configuration_for_next_aggregation = MithrilNetworkConfigurationForEpoch::dummy();
187187

188-
let configuration_for_registration = EpochConfiguration {
188+
let configuration_for_registration = MithrilNetworkConfigurationForEpoch {
189189
protocol_parameters: fake_data::protocol_parameters(),
190190
..Dummy::dummy()
191191
};
@@ -486,7 +486,7 @@ impl StateMachineTester {
486486
&mut self,
487487
discriminants: &[SignedEntityTypeDiscriminants],
488488
) -> &mut Self {
489-
let config = EpochConfiguration {
489+
let config = MithrilNetworkConfigurationForEpoch {
490490
enabled_signed_entity_types: BTreeSet::from_iter(discriminants.iter().cloned()),
491491
..self
492492
.network_configuration_service

0 commit comments

Comments
 (0)