add authorities_len for aura (paritytech#2040)

duniter · Oct 26, 2023 · 335f881 · 335f881
1 parent 77ebcae
commit 335f881
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/substrate/frame/aura/src/lib.rs b/substrate/frame/aura/src/lib.rs
@@ -230,6 +230,11 @@ impl<T: Config> Pallet<T> {
 		}
 	}
 
+	/// Return current authorities length.
+	pub fn authorities_len() -> usize {
+		Authorities::<T>::decode_len().unwrap_or(0)
+	}
+
 	/// Get the current slot from the pre-runtime digests.
 	fn current_slot_from_digests() -> Option<Slot> {
 		let digest = frame_system::Pallet::<T>::digest();
@@ -363,7 +368,7 @@ impl<T: Config> FindAuthor<u32> for Pallet<T> {
 		for (id, mut data) in digests.into_iter() {
 			if id == AURA_ENGINE_ID {
 				let slot = Slot::decode(&mut data).ok()?;
-				let author_index = *slot % Self::authorities().len() as u64;
+				let author_index = *slot % Self::authorities_len() as u64;
 				return Some(author_index as u32)
 			}
 		}

diff --git a/substrate/frame/aura/src/tests.rs b/substrate/frame/aura/src/tests.rs
@@ -29,7 +29,8 @@ use sp_runtime::{Digest, DigestItem};
 fn initial_values() {
 	build_ext_and_execute_test(vec![0, 1, 2, 3], || {
 		assert_eq!(Aura::current_slot(), 0u64);
-		assert_eq!(Aura::authorities().len(), 4);
+		assert_eq!(Aura::authorities().len(), Aura::authorities_len());
+		assert_eq!(Aura::authorities_len(), 4);
 	});
 }