add rate limit test helper

Author: camfairchild

pallets/subtensor/tests/mock.rs

@@ -10,6 +10,7 @@ use frame_support::{
 use frame_system as system;
 use frame_system::{limits, EnsureNever, EnsureRoot, RawOrigin};
 use pallet_collective::MemberCount;
+use pallet_subtensor::utils::rate_limiting::TransactionType;
 use sp_core::{Get, H256, U256};
 use sp_runtime::Perbill;
 use sp_runtime::{
@@ -601,3 +602,13 @@ pub fn is_within_tolerance(actual: u64, expected: u64, tolerance: u64) -> bool {
     };
     difference <= tolerance
 }
+
+// Helper function to wait for the rate limit
+#[allow(dead_code)]
+pub fn step_rate_limit(transaction_type: &TransactionType, netuid: u16) {
+    // Check rate limit
+    let limit = SubtensorModule::get_rate_limit(transaction_type, netuid);
+
+    // Step that many blocks
+    step_block(limit as u16);
+}