Don't make Slot.0 public

input-output-hk · May 24, 2019 · ed4835f · ed4835f
1 parent 5157b4a
commit ed4835f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/chain-impl-mockchain/src/leadership/genesis/mod.rs b/chain-impl-mockchain/src/leadership/genesis/mod.rs
@@ -145,10 +145,6 @@ mod test {
     use crate::stake::PoolStakeDistribution;
     use crate::stake::StakePoolInfo;
     use chain_addr::Discrimination;
-    use chain_time::{
-        era::{Epoch, TimeEra},
-        timeframe::Slot,
-    };
     use std::collections::HashMap;
 
     fn make_pool(ledger: &mut Ledger) -> (StakePoolId, SecretKey<Curve25519_2HashDH>) {
@@ -213,9 +209,7 @@ mod test {
             );
         }
 
-        let era = TimeEra::new(Slot(0), Epoch(0), slots_per_epoch);
-
-        let mut date = BlockDate::first();
+        let mut date = ledger.date;
 
         let mut empty_slots = 0;
 
@@ -247,7 +241,7 @@ mod test {
             if any_small {
                 times_selected_small += 1;
             }
-            date = date.next(&era);
+            date = date.next(&ledger.settings.era);
         }
 
         for (pool_id, (_pool_vrf_private_key, times_selected, stake)) in pools.iter_mut() {

diff --git a/chain-time/src/timeframe.rs b/chain-time/src/timeframe.rs
@@ -6,7 +6,7 @@ use std::time::{Duration, SystemTime};
 /// The slots are not comparable to others slots made on a
 /// different time frame
 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
-pub struct Slot(pub u64);
+pub struct Slot(pub(crate) u64);
 
 impl From<Slot> for u64 {
     fn from(s: Slot) -> u64 {