From 8f80c30e616805705c2568db9786cd88d9ee1833 Mon Sep 17 00:00:00 2001
From: bgallois <benjamin@gallois.cc>
Date: Thu, 21 Dec 2023 15:48:52 +0100
Subject: [PATCH] fix #166

---
 pallets/membership/src/benchmarking.rs | 4 ++--
 pallets/membership/src/lib.rs          | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pallets/membership/src/benchmarking.rs b/pallets/membership/src/benchmarking.rs
index 531bd3e13..e3a028dbb 100644
--- a/pallets/membership/src/benchmarking.rs
+++ b/pallets/membership/src/benchmarking.rs
@@ -76,9 +76,9 @@ benchmarks_instance_pallet! {
     }: {Pallet::<T, I>::on_initialize(BlockNumberFor::<T>::zero());}
 
     expire_memberships {
-        let i in 0..1024;
+        let i in 0..3; // Limited by the number of validators
         let mut idties: Vec<T::IdtyId> = Vec::new();
-        for j in 0..i {
+        for j in 1..i+1 {
             let j: T::IdtyId = j.into();
             Membership::<T, I>::insert(j, MembershipData::<T::BlockNumber>::default());
             idties.push(j);
diff --git a/pallets/membership/src/lib.rs b/pallets/membership/src/lib.rs
index 284a03244..08bbb81ad 100644
--- a/pallets/membership/src/lib.rs
+++ b/pallets/membership/src/lib.rs
@@ -308,7 +308,7 @@ pub mod pallet {
             for idty_id in MembershipsExpireOn::<T, I>::take(block_number) {
                 // remove membership (take)
                 Self::do_remove_membership(idty_id, MembershipRemovalReason::Expired);
-                expired_idty_count = 0;
+                expired_idty_count += 1;
             }
             T::WeightInfo::expire_memberships(expired_idty_count)
         }
-- 
GitLab