add pre-ecsec tests

This commit is contained in:
akildemir
2024-09-01 12:13:58 +03:00
parent a2df54dd6a
commit 9aa7a3798b
6 changed files with 469 additions and 16 deletions

View File

@@ -390,6 +390,7 @@ pub mod pallet {
let allocation_per_key_share = Self::allocation_per_key_share(network).unwrap().0;
let mut participants = vec![];
let mut total_allocated_stake = 0;
{
let mut iter = SortedAllocationsIter::<T>::new(network);
let mut key_shares = 0;
@@ -400,6 +401,7 @@ pub mod pallet {
(amount.0 / allocation_per_key_share).min(u64::from(MAX_KEY_SHARES_PER_SET));
participants.push((key, these_key_shares));
total_allocated_stake += amount.0;
key_shares += these_key_shares;
}
amortize_excess_key_shares(&mut participants);
@@ -413,6 +415,10 @@ pub mod pallet {
Pallet::<T>::deposit_event(Event::NewSet { set });
Participants::<T>::set(network, Some(participants.try_into().unwrap()));
if network == NetworkId::Serai {
TotalAllocatedStake::<T>::set(network, Some(Amount(total_allocated_stake)));
}
SessionBeginBlock::<T>::set(
network,
session,
@@ -705,7 +711,7 @@ pub mod pallet {
(!Keys::<T>::contains_key(ValidatorSet { network, session: Session(session.0 - 1) }))
}
fn new_session() {
pub fn new_session() {
for network in serai_primitives::NETWORKS {
// If this network hasn't started sessions yet, don't start one now
let Some(current_session) = Self::session(network) else { continue };