feat(aaa): add benchmark

ajuna-network · Mar 22, 2023 · 111309b · 111309b
1 parent 20d6d15
commit 111309b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/pallets/ajuna-awesome-avatars/src/benchmarking.rs b/pallets/ajuna-awesome-avatars/src/benchmarking.rs
@@ -348,6 +348,20 @@ benchmarks! {
 		assert_last_event::<T>(Event::FreeMintsSet { target, how_many }.into())
 	}
 
+	fix_variation {
+		let name = "player";
+		create_avatars::<T>(name, 1)?;
+
+		let player = account::<T>(name);
+		let avatar_id = AAvatars::<T>::owners(&player)[0];
+		let (_owner, original_avatar) = AAvatars::<T>::avatars(avatar_id).unwrap();
+	}: _(RawOrigin::Signed(player), avatar_id)
+	verify {
+		let (_owner, updated_avatar) = AAvatars::<T>::avatars(avatar_id).unwrap();
+		assert!(original_avatar.dna[1] & 0b0000_1111 != original_avatar.dna[2] & 0b0000_1111);
+		assert!(updated_avatar.dna[1] & 0b0000_1111 == updated_avatar.dna[2] & 0b0000_1111);
+	}
+
 	impl_benchmark_test_suite!(
 		AAvatars, crate::mock::ExtBuilder::default().build(), crate::mock::Test
 	);

diff --git a/pallets/ajuna-awesome-avatars/src/tests.rs b/pallets/ajuna-awesome-avatars/src/tests.rs
@@ -3035,7 +3035,7 @@ mod fix_variation {
 			.execute_with(|| {
 				run_to_block(season.start);
 
-				let avatar_id = create_avatars(ALICE, 1)[0];
+				let avatar_id = create_avatars(1, ALICE, 1)[0];
 
 				let (_, mut avatar_before) = Avatars::<Test>::get(avatar_id).unwrap();