From 1345d5fb1a402e9bea66338f7c77b485d436338d Mon Sep 17 00:00:00 2001
From: Ayush Kumar Mishra <ayushmishra2005@gmail.com>
Date: Mon, 25 Jan 2021 17:41:30 +0530
Subject: [PATCH] #80 Added benchmarking method for add_registrar

---
 .../src/benchmarking.rs                       | 30 +++++++++++++++++++
 pallets/polkadex-identity-pallet/src/lib.rs   |  4 ++-
 runtime/testnet/src/lib.rs                    |  1 +
 3 files changed, 34 insertions(+), 1 deletion(-)
 create mode 100644 pallets/polkadex-identity-pallet/src/benchmarking.rs

diff --git a/pallets/polkadex-identity-pallet/src/benchmarking.rs b/pallets/polkadex-identity-pallet/src/benchmarking.rs
new file mode 100644
index 000000000..82f7773dd
--- /dev/null
+++ b/pallets/polkadex-identity-pallet/src/benchmarking.rs
@@ -0,0 +1,30 @@
+use super::*;
+
+use sp_std::prelude::*;
+use frame_system::RawOrigin;
+use frame_benchmarking::{benchmarks, whitelisted_caller, account};
+use orml_nft::Module as NftModule;
+use crate::Module as KittiesModule;
+
+benchmarks! {
+	_ {}
+
+	add_registrar {
+		let account_id = account("registrar", 0, 0);
+	}: _(RawOrigin::Root, account_id)
+
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::tests::{new_test_ext, Test};
+    use frame_support::assert_ok;
+
+    #[test]
+    fn test_benchmarks() {
+        new_test_ext().execute_with(|| {
+            assert_ok!(test_benchmark_add_registrar::<Test>());
+        });
+    }
+}
diff --git a/pallets/polkadex-identity-pallet/src/lib.rs b/pallets/polkadex-identity-pallet/src/lib.rs
index 3cb82aede..ef152da1b 100644
--- a/pallets/polkadex-identity-pallet/src/lib.rs
+++ b/pallets/polkadex-identity-pallet/src/lib.rs
@@ -14,6 +14,8 @@ use sp_std::prelude::*;
 mod mock;
 #[cfg(test)]
 mod test;
+#[cfg(feature = "runtime-benchmarks")]
+mod benchmarking;
 
 pub type RegistrarIndex = u32;
 
@@ -200,7 +202,7 @@ decl_module! {
     ///
     /// This function returns a status that, judgement is successfully passed or not.
     #[weight = 10_000]
-    fn provide_judgement_trader(origin,target: T::AccountId,judgement: Judgement) -> DispatchResult {
+    fn provide_judgement_trader(origin, target: T::AccountId, judgement: Judgement) -> DispatchResult {
 		    let registrar = ensure_signed(origin)?;
 		    ensure!(<Registrars<T>>::contains_key(&registrar), Error::<T>::SenderIsNotRegistrar); // Check for the existance
 		    ensure!(!SuperOf::<T>::contains_key(&target), Error::<T>::GivenAccountIsSubAccount);
diff --git a/runtime/testnet/src/lib.rs b/runtime/testnet/src/lib.rs
index 39404f3b4..7bc0b18f6 100644
--- a/runtime/testnet/src/lib.rs
+++ b/runtime/testnet/src/lib.rs
@@ -633,6 +633,7 @@ impl_runtime_apis! {
 			add_benchmark!(params, batches, pallet_collective, Council);
 			add_benchmark!(params, batches, pallet_timestamp, Timestamp);
 			add_benchmark!(params, batches, pallet_treasury, Treasury);
+			add_benchmark!(params, batches, pallet_idenity, PolkadexIdentity);
 
 			if batches.is_empty() { return Err("Benchmark not found for this pallet.".into()) }
 			Ok(batches)