benchmarking.rs

// Copyright 2021-2023 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::v2::*;
use frame_system::RawOrigin;

use crate::Pallet;

#[benchmarks(
        where
            T::IdtyIndex: From<u32>
)]
mod benchmarks {
    use super::*;

    fn assert_has_event<T: Config>(generic_event: <T as Config>::RuntimeEvent) {
        frame_system::Pallet::<T>::assert_has_event(generic_event.into());
    }

    #[benchmark]
    fn invite_smith() {
        let issuer: T::IdtyIndex = 1.into();
        let caller: T::AccountId = T::IdtyAttr::owner_key(issuer).unwrap();
        Pallet::<T>::on_smith_goes_online(1.into());
        let receiver: T::IdtyIndex = 4.into();

        #[extrinsic_call]
        _(RawOrigin::Signed(caller), receiver);

        assert_has_event::<T>(Event::<T>::InvitationSent { receiver, issuer }.into());
    }

    #[benchmark]
    fn accept_invitation() -> Result<(), BenchmarkError> {
        let issuer: T::IdtyIndex = 1.into();
        let caller: T::AccountId = T::IdtyAttr::owner_key(issuer).unwrap();
        Pallet::<T>::on_smith_goes_online(1.into());
        let caller_origin: <T as frame_system::Config>::RuntimeOrigin =
            RawOrigin::Signed(caller.clone()).into();
        let receiver: T::IdtyIndex = 4.into();
        Pallet::<T>::invite_smith(caller_origin, receiver)?;
        let issuer: T::IdtyIndex = 4.into();
        let caller: T::AccountId = T::IdtyAttr::owner_key(issuer).unwrap();

        #[extrinsic_call]
        _(RawOrigin::Signed(caller));

        assert_has_event::<T>(
            Event::<T>::InvitationAccepted {
                idty_index: receiver,
            }
            .into(),
        );
        Ok(())
    }

    #[benchmark]
    fn certify_smith() -> Result<(), BenchmarkError> {
        let issuer: T::IdtyIndex = 1.into();
        let caller: T::AccountId = T::IdtyAttr::owner_key(issuer).unwrap();
        Pallet::<T>::on_smith_goes_online(1.into());
        let caller_origin: <T as frame_system::Config>::RuntimeOrigin =
            RawOrigin::Signed(caller.clone()).into();
        let receiver: T::IdtyIndex = 4.into();
        Pallet::<T>::invite_smith(caller_origin, receiver)?;
        let issuer: T::IdtyIndex = receiver;
        let caller: T::AccountId = T::IdtyAttr::owner_key(issuer).unwrap();
        let caller_origin: <T as frame_system::Config>::RuntimeOrigin =
            RawOrigin::Signed(caller.clone()).into();
        Pallet::<T>::accept_invitation(caller_origin)?;
        let issuer: T::IdtyIndex = 1.into();
        let caller: T::AccountId = T::IdtyAttr::owner_key(issuer).unwrap();

        #[extrinsic_call]
        _(RawOrigin::Signed(caller), receiver);

        assert_has_event::<T>(Event::<T>::SmithCertAdded { receiver, issuer }.into());
        Ok(())
    }

    #[benchmark]
    fn on_removed_wot_member() {
        let idty: T::IdtyIndex = 1.into();
        assert!(Smiths::<T>::get(idty).is_some());

        #[block]
        {
            Pallet::<T>::on_removed_wot_member(idty);
        }
    }

    #[benchmark]
    fn on_removed_wot_member_empty() {
        let idty: T::IdtyIndex = 100.into();
        assert!(Smiths::<T>::get(idty).is_none());

        #[block]
        {
            Pallet::<T>::on_removed_wot_member(idty);
        }
    }

    impl_benchmark_test_suite!(
        Pallet,
        crate::mock::new_test_ext(crate::GenesisConfig {
            initial_smiths: maplit::btreemap![
                1 => (false, vec![2, 3]),
                2 => (false, vec![1, 3]),
                3 => (false, vec![1, 2]),
            ],
        }),
        crate::mock::Runtime
    );
}