Skip to content

Commit e1713bd

Browse files
authored
improve oracle benchmarks (#1036)
1 parent 23b4f57 commit e1713bd

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

oracle/src/benchmarking.rs

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
1-
use super::*;
2-
use crate::Pallet as Oracle;
1+
pub use crate::*;
32

4-
use frame_benchmarking::v2::*;
3+
use frame_benchmarking::v2::{impl_test_function, instance_benchmarks, whitelisted_caller};
54

65
use frame_support::assert_ok;
76
use frame_system::{Pallet as System, RawOrigin};
@@ -36,18 +35,18 @@ mod benchmarks {
3635
// Feed some values before running `on_finalize` hook
3736
System::<T>::set_block_number(1u32.into());
3837
let values = T::BenchmarkHelper::get_currency_id_value_pairs();
39-
assert_ok!(Oracle::<T, I>::feed_values(RawOrigin::Signed(caller).into(), values));
38+
assert_ok!(Pallet::<T, I>::feed_values(RawOrigin::Signed(caller).into(), values));
4039

4140
#[block]
4241
{
43-
Oracle::<T, I>::on_finalize(System::<T>::block_number());
42+
Pallet::<T, I>::on_finalize(System::<T>::block_number());
4443
}
4544

4645
assert!(!HasDispatched::<T, I>::exists());
4746
}
4847

4948
impl_benchmark_test_suite! {
50-
Oracle,
49+
Pallet,
5150
crate::mock::new_test_ext(),
5251
crate::mock::Test,
5352
}

oracle/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ use sp_std::{prelude::*, vec};
4343
pub use crate::default_combine_data::DefaultCombineData;
4444

4545
#[cfg(feature = "runtime-benchmarks")]
46-
mod benchmarking;
46+
pub mod benchmarking;
4747

4848
mod default_combine_data;
4949
mod mock;

0 commit comments

Comments
 (0)