diff --git a/oracle/src/benchmarking.rs b/oracle/src/benchmarking.rs index 7c8cc8f25..591016638 100644 --- a/oracle/src/benchmarking.rs +++ b/oracle/src/benchmarking.rs @@ -1,7 +1,6 @@ -use super::*; -use crate::Pallet as Oracle; +pub use crate::*; -use frame_benchmarking::v2::*; +use frame_benchmarking::v2::{impl_test_function, instance_benchmarks, whitelisted_caller}; use frame_support::assert_ok; use frame_system::{Pallet as System, RawOrigin}; @@ -36,18 +35,18 @@ mod benchmarks { // Feed some values before running `on_finalize` hook System::::set_block_number(1u32.into()); let values = T::BenchmarkHelper::get_currency_id_value_pairs(); - assert_ok!(Oracle::::feed_values(RawOrigin::Signed(caller).into(), values)); + assert_ok!(Pallet::::feed_values(RawOrigin::Signed(caller).into(), values)); #[block] { - Oracle::::on_finalize(System::::block_number()); + Pallet::::on_finalize(System::::block_number()); } assert!(!HasDispatched::::exists()); } impl_benchmark_test_suite! { - Oracle, + Pallet, crate::mock::new_test_ext(), crate::mock::Test, } diff --git a/oracle/src/lib.rs b/oracle/src/lib.rs index dc5041f8a..869829434 100644 --- a/oracle/src/lib.rs +++ b/oracle/src/lib.rs @@ -43,7 +43,7 @@ use sp_std::{prelude::*, vec}; pub use crate::default_combine_data::DefaultCombineData; #[cfg(feature = "runtime-benchmarks")] -mod benchmarking; +pub mod benchmarking; mod default_combine_data; mod mock;