diff --git a/pallets/capacity/src/tests/eras_tests.rs b/pallets/capacity/src/tests/eras_tests.rs new file mode 100644 index 0000000000..2d362f7c55 --- /dev/null +++ b/pallets/capacity/src/tests/eras_tests.rs @@ -0,0 +1,25 @@ +use super::mock::*; +use crate::{ + tests::testing_utils::{run_to_block, system_run_to_block}, + Config, CurrentEraInfo, Error, Event, RewardEraInfo, +}; + +use frame_support::traits::Get; + +#[test] +fn start_new_era_if_needed() { + new_test_ext().execute_with(|| { + CurrentEraInfo::::set(RewardEraInfo { current_era: 1, era_start: 0 }); + system_run_to_block(9); + run_to_block(10); + let mut current_era_info = CurrentEraInfo::::get(); + assert_eq!(current_era_info.current_era, 2u32); + assert_eq!(current_era_info.era_start, 10u32); + + system_run_to_block(19); + run_to_block(20); + current_era_info = CurrentEraInfo::::get(); + assert_eq!(current_era_info.current_era, 3u32); + assert_eq!(current_era_info.era_start, 20u32); + }) +}