Membership pallet benchmark

pallets/membership/src/benchmarking.rs

+// Copyright 2021-2022 Axiom-Team
+//
+// This file is part of Duniter-v2S.
+//
+// Duniter-v2S is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, version 3 of the License.
+//
+// Duniter-v2S is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with Duniter-v2S. If not, see <https://www.gnu.org/licenses/>.
+
+#![cfg(feature = "runtime-benchmarks")]
+
+use super::*;
+
+use frame_benchmarking::{account, benchmarks_instance_pallet, whitelisted_caller};
+use frame_system::RawOrigin;
+use sp_runtime::traits::Convert;
+
+#[cfg(test)]
+use maplit::btreemap;
+
+use crate::Pallet;
+
+fn assert_has_event<T: Config<I>, I: 'static>(generic_event: <T as Config<I>>::RuntimeEvent) {
+    frame_system::Pallet::<T>::assert_has_event(generic_event.into());
+}
+
+benchmarks_instance_pallet! {
+    where_clause {
+        where
+            T::IdtyId: From<u32>,
+    }
+    force_request_membership {
+        let idty: T::IdtyId = 5.into();
+    }: _<T::RuntimeOrigin>(RawOrigin::Root.into(), idty, T::MetaData ::default() )
+    verify {
+        assert_has_event::<T, I>(Event::<T, I>::MembershipRequested(idty).into());
+    }
+    request_membership {
+        // Only possible in subwot
+        let idty: T::IdtyId = 4.into();
+        Membership::<T, I>::take(idty);
+        let caller: T::AccountId = T::AccountIdOf::convert(idty.clone()).unwrap();
+        let caller_origin: <T as frame_system::Config>::RuntimeOrigin = RawOrigin::Signed(caller.clone()).into();
+    }: _<T::RuntimeOrigin>(caller_origin, T::MetaData ::default())
+    verify {
+        assert_has_event::<T, I>(Event::<T, I>::MembershipRequested(idty).into());
+    }
+    claim_membership {
+        let idty: T::IdtyId = 4.into();
+        Membership::<T, I>::take(idty);
+        PendingMembership::<T, I>::insert(idty.clone(), T::MetaData::default());
+        let caller: T::AccountId = T::AccountIdOf::convert(idty.clone()).unwrap();
+        let caller_origin: <T as frame_system::Config>::RuntimeOrigin = RawOrigin::Signed(caller.clone()).into();
+    }: _<T::RuntimeOrigin>(caller_origin, Some(idty))
+    verify {
+        assert_has_event::<T, I>(Event::<T, I>::MembershipAcquired(idty).into());
+    }
+    renew_membership {
+        let idty: T::IdtyId = 3.into();
+        let caller: T::AccountId = T::AccountIdOf::convert(idty.clone()).unwrap();
+        let caller_origin: <T as frame_system::Config>::RuntimeOrigin = RawOrigin::Signed(caller.clone()).into();
+    }: _<T::RuntimeOrigin>(caller_origin, Some(idty))
+    verify {
+        assert_has_event::<T, I>(Event::<T, I>::MembershipRenewed(idty).into());
+    }
+    revoke_membership {
+        let idty: T::IdtyId = 3.into();
+        let caller: T::AccountId = T::AccountIdOf::convert(idty.clone()).unwrap();
+        let caller_origin: <T as frame_system::Config>::RuntimeOrigin = RawOrigin::Signed(caller.clone()).into();
+    }: _<T::RuntimeOrigin>(caller_origin, Some(idty))
+    verify {
+        assert_has_event::<T, I>(Event::<T, I>::MembershipRevoked(idty).into());
+    }
+
+    /*impl_benchmark_test_suite!(
+        Pallet,
+        crate::mock::new_test_ext(crate::mock::DefaultMembershipConfig {
+        memberships: btreemap![
+            0 => MembershipData {
+                expire_on: 3,
+            }
+        ],}),
+        crate::mock::Test
+    ); */
+}