Merge pull request #378 from liberland/BLOCKCHAIN-449/fix-weight-calc…

…ulation BLOCKCHAIN-449 - Fix weight for llm.remark call
liberland · May 16, 2024 · 9c6dafc · 9c6dafc
2 parents b3b68ae + 6a1cec9
commit 9c6dafc
Show file tree

Hide file tree

Showing 3 changed files with 176 additions and 147 deletions.
diff --git a/substrate/frame/llm/src/benchmarking.rs b/substrate/frame/llm/src/benchmarking.rs
@@ -100,6 +100,15 @@ benchmarks! {
 	verify {
 		assert_eq!(<<T as Config>::Currency as Currency<T::AccountId>>::total_balance(&user), amount);
 	}
+
+	remark {
+		let l in 0 .. 64;
+		let data: RemarkData = [1u8].repeat(l as usize).try_into().unwrap();
+	}: _(RawOrigin::Root, data.clone())
+	verify {
+		let e: <T as Config>::RuntimeEvent = Event::Remarked(data).into();
+		frame_system::Pallet::<T>::assert_last_event(e.into());
+	}
 }
 
 impl_benchmark_test_suite!(LLM, crate::mock::new_test_ext(), crate::mock::Test,);
diff --git a/substrate/frame/llm/src/lib.rs b/substrate/frame/llm/src/lib.rs
@@ -502,8 +502,8 @@ pub mod pallet {
 
 		/// Emit Remarked event. Used by Liberland tooling to annotate transfers.
 		#[pallet::call_index(7)]
-		#[pallet::weight(<T as Config>::WeightInfo::treasury_lld_transfer())]
-		pub fn remark(origin: OriginFor<T>, data: RemarkData) -> DispatchResult {
+		#[pallet::weight(<T as Config>::WeightInfo::remark(data.len() as u32))]
+		pub fn remark(_origin: OriginFor<T>, data: RemarkData) -> DispatchResult {
 			Self::deposit_event(Event::<T>::Remarked(data));
 			Ok(())
 		}