fix benchmark

KILTprotocol · Jun 29, 2022 · 60e5edc · 60e5edc
1 parent 3c70655
commit 60e5edc
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/pallets/pallet-did-lookup/src/benchmarking.rs b/pallets/pallet-did-lookup/src/benchmarking.rs
@@ -25,12 +25,12 @@ use crate::{
 
 use codec::Encode;
 use frame_benchmarking::{account, benchmarks, impl_benchmark_test_suite};
-use frame_support::{traits::{Currency, Get}, crypto::{ecdsa::ECDSAExt}};
+use frame_support::{traits::{Currency, Get}, crypto::ecdsa::ECDSAExt};
 use frame_system::RawOrigin;
 use kilt_support::traits::GenerateBenchmarkOrigin;
 use sha3::{Keccak256, Digest};
-use sp_io::crypto::sr25519_generate;
-use sp_runtime::{app_crypto::{sr25519, ecdsa, Pair}, KeyTypeId, AccountId32};
+use sp_io::crypto::{sr25519_generate, ecdsa_generate};
+use sp_runtime::{app_crypto::{sr25519}, KeyTypeId, AccountId32};
 
 const SEED: u32 = 0;
 
@@ -90,15 +90,19 @@ benchmarks! {
 		let previous_did: T::DidIdentifier = account("prev", 0, SEED + 1);
 		let expire_at: <T as frame_system::Config>::BlockNumber = 500_u32.into();
 
-		let eth_pair = ecdsa::Pair::generate().0;
-		let eth_account = AccountId20(eth_pair.public().to_eth_address().unwrap());
+		let eth_public_key = ecdsa_generate(KeyTypeId(*b"aura"), None);
+		let eth_account = AccountId20(eth_public_key.to_eth_address().unwrap());
 
 		let wrapped_payload = get_wrapped_payload(
 			&Encode::encode(&(&did, expire_at))[..], 
 			&eth_account,
 		);
 
-		let sig = eth_pair.sign_prehashed(&Keccak256::digest(&wrapped_payload).try_into().unwrap());
+		let sig = sp_io::crypto::ecdsa_sign_prehashed(
+			KeyTypeId(*b"aura"),
+			&eth_public_key,
+			&Keccak256::digest(&wrapped_payload).try_into().unwrap(),
+		).ok_or("Error while building signature.")?;
 
 		make_free_for_did::<T>(&caller);