diff --git a/runtime/common/src/handlers.rs b/runtime/common/src/handlers.rs index 1942c41959b502d1b0a56143051cf479ea8eec88..f72a59a3c07ab8d00afa533227061c66ad4972ca 100644 --- a/runtime/common/src/handlers.rs +++ b/runtime/common/src/handlers.rs @@ -24,11 +24,13 @@ use pallet_smith_members::SmithRemovalReason; pub struct OnNewSessionHandler<Runtime>(core::marker::PhantomData<Runtime>); impl<Runtime> pallet_authority_members::traits::OnNewSession for OnNewSessionHandler<Runtime> where - Runtime: pallet_provide_randomness::Config + pallet_distance::Config, + Runtime: + pallet_provide_randomness::Config + pallet_distance::Config + pallet_smith_members::Config, { fn on_new_session(index: sp_staking::SessionIndex) { pallet_provide_randomness::Pallet::<Runtime>::on_new_epoch(); pallet_distance::Pallet::<Runtime>::on_new_session(index); + pallet_smith_members::Pallet::<Runtime>::on_new_session(index); } } diff --git a/runtime/gdev/tests/integration_tests.rs b/runtime/gdev/tests/integration_tests.rs index 4f0b442f02922e3c08b0685c1a90d201367cea7e..3864d3a012d39f363b9c8c0b42c2b5279672804b 100644 --- a/runtime/gdev/tests/integration_tests.rs +++ b/runtime/gdev/tests/integration_tests.rs @@ -246,6 +246,7 @@ fn test_session_change() { ExtBuilder::new(1, 3, 4).build().execute_with(|| { assert_eq!(<Runtime as pallet_babe::Config>::EpochDuration::get(), 25); assert_eq!(Session::current_index(), 0); + assert_eq!(SmithMembers::current_session(), 0); assert_eq!(Babe::epoch_index(), 0); assert_eq!(Babe::current_epoch_start(), 0u64); run_to_block(2); @@ -253,33 +254,42 @@ fn test_session_change() { assert_eq!(Babe::epoch_index(), 0); run_to_block(24); assert_eq!(Session::current_index(), 0); + assert_eq!(SmithMembers::current_session(), 0); assert_eq!(Babe::epoch_index(), 0); run_to_block(25); assert_eq!(Session::current_index(), 1); + assert_eq!(SmithMembers::current_session(), 1); assert_eq!(Babe::epoch_index(), 1); assert_eq!(Babe::current_epoch_start(), 25u64); run_to_block(26); assert_eq!(Session::current_index(), 1); + assert_eq!(SmithMembers::current_session(), 1); assert_eq!(Babe::epoch_index(), 1); run_to_block(50); assert_eq!(Session::current_index(), 2); + assert_eq!(SmithMembers::current_session(), 2); assert_eq!(Babe::epoch_index(), 2); assert_eq!(Babe::current_epoch_start(), 50u64); run_to_block(51); assert_eq!(Session::current_index(), 2); + assert_eq!(SmithMembers::current_session(), 2); assert_eq!(Babe::epoch_index(), 2); run_to_block(52); assert_eq!(Session::current_index(), 2); + assert_eq!(SmithMembers::current_session(), 2); assert_eq!(Babe::epoch_index(), 2); run_to_block(60); assert_eq!(Session::current_index(), 2); + assert_eq!(SmithMembers::current_session(), 2); assert_eq!(Babe::epoch_index(), 2); assert_eq!(Babe::current_epoch_start(), 50u64); run_to_block(75); assert_eq!(Session::current_index(), 3); + assert_eq!(SmithMembers::current_session(), 3); assert_eq!(Babe::epoch_index(), 3); run_to_block(100); assert_eq!(Session::current_index(), 4); + assert_eq!(SmithMembers::current_session(), 4); assert_eq!(Babe::epoch_index(), 4); }) }