use super::*;
use crate::{pallet as pallet_asset_rate, Pallet as AssetRate};
use codec::Encode;
use frame_benchmarking::v2::*;
use frame_support::assert_ok;
use frame_system::RawOrigin;
use sp_core::crypto::FromEntropy;
pub trait AssetKindFactory<AssetKind> {
fn create_asset_kind(seed: u32) -> AssetKind;
}
impl<AssetKind> AssetKindFactory<AssetKind> for ()
where
AssetKind: FromEntropy,
{
fn create_asset_kind(seed: u32) -> AssetKind {
AssetKind::from_entropy(&mut seed.encode().as_slice()).unwrap()
}
}
const SEED: u32 = 1;
fn default_conversion_rate() -> FixedU128 {
FixedU128::from_u32(1u32)
}
#[benchmarks]
mod benchmarks {
use super::*;
#[benchmark]
fn create() -> Result<(), BenchmarkError> {
let asset_kind: T::AssetKind = T::BenchmarkHelper::create_asset_kind(SEED);
#[extrinsic_call]
_(RawOrigin::Root, Box::new(asset_kind.clone()), default_conversion_rate());
assert_eq!(
pallet_asset_rate::ConversionRateToNative::<T>::get(asset_kind),
Some(default_conversion_rate())
);
Ok(())
}
#[benchmark]
fn update() -> Result<(), BenchmarkError> {
let asset_kind: T::AssetKind = T::BenchmarkHelper::create_asset_kind(SEED);
assert_ok!(AssetRate::<T>::create(
RawOrigin::Root.into(),
Box::new(asset_kind.clone()),
default_conversion_rate()
));
#[extrinsic_call]
_(RawOrigin::Root, Box::new(asset_kind.clone()), FixedU128::from_u32(2));
assert_eq!(
pallet_asset_rate::ConversionRateToNative::<T>::get(asset_kind),
Some(FixedU128::from_u32(2))
);
Ok(())
}
#[benchmark]
fn remove() -> Result<(), BenchmarkError> {
let asset_kind: T::AssetKind = T::BenchmarkHelper::create_asset_kind(SEED);
assert_ok!(AssetRate::<T>::create(
RawOrigin::Root.into(),
Box::new(asset_kind.clone()),
default_conversion_rate()
));
#[extrinsic_call]
_(RawOrigin::Root, Box::new(asset_kind.clone()));
assert!(pallet_asset_rate::ConversionRateToNative::<T>::get(asset_kind).is_none());
Ok(())
}
impl_benchmark_test_suite! { AssetRate, crate::mock::new_test_ext(), crate::mock::Test }
}