sudo & sudo_as

frame/sudo/src/benchmarking.rs

@@ -15,7 +15,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-// Benchmarks for Sudo Pallet
+//! Benchmarks for Sudo Pallet
 
 use super::*;
 use crate::Pallet;
@@ -46,5 +46,42 @@ mod benchmarks {
 		assert_last_event::<T>(Event::KeyChanged { old_sudoer: Some(caller) }.into());
 	}
 
+	#[benchmark]
+	fn sudo() {
+		let caller: T::AccountId = whitelisted_caller();
+		Key::<T>::put(caller.clone());
+
+		let call: <T as Config>::RuntimeCall = frame_system::Call::remark { remark: vec![] }.into();
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(caller.clone()), Box::new(call.clone()));
+
+		let res = call.dispatch_bypass_filter(RawOrigin::Signed(caller.clone()).into());
+
+		assert_last_event::<T>(
+			Event::Sudid { sudo_result: res.map(|_| ()).map_err(|e| e.error) }.into(),
+		)
+	}
+
+	#[benchmark]
+	fn sudo_as() {
+		let caller: T::AccountId = whitelisted_caller();
+		Key::<T>::put(caller.clone());
+
+		let call: <T as Config>::RuntimeCall = frame_system::Call::remark { remark: vec![] }.into();
+
+		let who: T::AccountId = account("as", 0, SEED);
+		let who_lookup = T::Lookup::unlookup(who.clone());
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(caller), who_lookup, Box::new(call.clone()));
+
+		let res = call.dispatch_bypass_filter(RawOrigin::Signed(who).into());
+
+		assert_last_event::<T>(
+			Event::SudoAsDone { sudo_result: res.map(|_| ()).map_err(|e| e.error) }.into(),
+		)
+	}
+
 	impl_benchmark_test_suite!(Pallet, crate::mock::new_bench_ext(), crate::mock::Test);
 }

frame/sudo/src/lib.rs

@@ -134,7 +134,8 @@ pub mod pallet {
 		/// A sudo-able call.
 		type RuntimeCall: Parameter
 			+ UnfilteredDispatchable<RuntimeOrigin = Self::RuntimeOrigin>
-			+ GetDispatchInfo;
+			+ GetDispatchInfo
+			+ From<frame_system::Call<Self>>;
 
 		/// Type representing the weight of this pallet
 		type WeightInfo: WeightInfo;
@@ -203,7 +204,7 @@ pub mod pallet {
 		/// ## Complexity
 		/// - O(1).
 		#[pallet::call_index(2)]
-		#[pallet::weight(T::WeightInfo::set_key())] // FIXME
+		#[pallet::weight(T::WeightInfo::set_key())]
 		pub fn set_key(
 			origin: OriginFor<T>,
 			new: AccountIdLookupOf<T>,