Create new Pythagoras era

input-output-hk · May 26, 2023 · 9673acc · 9673acc
1 parent c271d62
commit 9673acc
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/mithril-aggregator/tests/certificate_chain.rs b/mithril-aggregator/tests/certificate_chain.rs
@@ -2,7 +2,10 @@ mod test_extensions;
 
 use mithril_aggregator::{Configuration, VerificationKeyStorer};
 use mithril_common::{
-    chain_observer::ChainObserver, entities::ProtocolParameters, test_utils::MithrilFixtureBuilder,
+    chain_observer::ChainObserver,
+    entities::{Epoch, ProtocolParameters},
+    era::{EraMarker, SupportedEra},
+    test_utils::MithrilFixtureBuilder,
 };
 use test_extensions::{utilities::get_test_dir, RuntimeTester, SignedEntityTypeDiscriminants};
 
@@ -21,6 +24,14 @@ async fn certificate_chain() {
     let mut tester = RuntimeTester::build(configuration).await;
     let observer = tester.observer.clone();
 
+    comment!("use Pythagoras era");
+    tester
+        .set_era_markers(vec![EraMarker::new(
+            &SupportedEra::Pythagoras.to_string(),
+            Some(Epoch(0)),
+        )])
+        .await;
+
     comment!("Create signers & declare stake distribution");
     let fixture = MithrilFixtureBuilder::default()
         .with_signers(5)

diff --git a/mithril-aggregator/tests/create_certificate.rs b/mithril-aggregator/tests/create_certificate.rs
@@ -4,7 +4,8 @@ use std::collections::BTreeSet;
 
 use mithril_aggregator::Configuration;
 use mithril_common::{
-    entities::{ProtocolMessagePartKey, ProtocolParameters},
+    entities::{Epoch, ProtocolMessagePartKey, ProtocolParameters},
+    era::{EraMarker, SupportedEra},
     test_utils::MithrilFixtureBuilder,
 };
 use test_extensions::{utilities::get_test_dir, RuntimeTester, SignedEntityTypeDiscriminants};
@@ -24,6 +25,14 @@ async fn create_certificate() {
     let mut tester = RuntimeTester::build(configuration).await;
     let observer = tester.observer.clone();
 
+    comment!("use Pythagoras era");
+    tester
+        .set_era_markers(vec![EraMarker::new(
+            &SupportedEra::Pythagoras.to_string(),
+            Some(Epoch(0)),
+        )])
+        .await;
+
     comment!("create signers & declare stake distribution");
     let fixture = MithrilFixtureBuilder::default()
         .with_signers(10)

diff --git a/mithril-common/src/era/supported_era.rs b/mithril-common/src/era/supported_era.rs
@@ -28,6 +28,9 @@ impl UnsupportedEraError {
 pub enum SupportedEra {
     /// Thales era
     Thales,
+
+    /// Pythagoras era
+    Pythagoras,
 }
 
 impl SupportedEra {