From e913579d97607564011aa11f81555f538589cdd7 Mon Sep 17 00:00:00 2001
From: Branislav Kontur <bkontur@gmail.com>
Date: Wed, 19 Jul 2023 10:55:19 +0200
Subject: [PATCH] Added `construct_benchmark_partials` for BridgeHubs (#2901)

---
 cumulus/polkadot-parachain/src/command.rs | 44 +++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/cumulus/polkadot-parachain/src/command.rs b/cumulus/polkadot-parachain/src/command.rs
index 4bc19f86e4..42ae887db6 100644
--- a/cumulus/polkadot-parachain/src/command.rs
+++ b/cumulus/polkadot-parachain/src/command.rs
@@ -405,6 +405,50 @@ macro_rules! construct_benchmark_partials {
 				)?;
 				$code
 			},
+			Runtime::BridgeHub(bridge_hub_runtime_type) => match bridge_hub_runtime_type {
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::Polkadot |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::PolkadotLocal |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::PolkadotDevelopment => {
+					let $partials = new_partial::<chain_spec::bridge_hubs::polkadot::RuntimeApi, _>(
+						&$config,
+						crate::service::aura_build_import_queue::<_, AuraId>,
+					)?;
+					$code
+				},
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::Kusama |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::KusamaLocal |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::KusamaDevelopment => {
+					let $partials = new_partial::<chain_spec::bridge_hubs::kusama::RuntimeApi, _>(
+						&$config,
+						crate::service::aura_build_import_queue::<_, AuraId>,
+					)?;
+					$code
+				},
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::Westend => {
+					let $partials = new_partial::<chain_spec::bridge_hubs::westend::RuntimeApi, _>(
+						&$config,
+						crate::service::aura_build_import_queue::<_, AuraId>,
+					)?;
+					$code
+				},
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::Rococo |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::RococoLocal |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::RococoDevelopment => {
+					let $partials = new_partial::<chain_spec::bridge_hubs::rococo::RuntimeApi, _>(
+						&$config,
+						crate::service::aura_build_import_queue::<_, AuraId>,
+					)?;
+					$code
+				},
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::Wococo |
+				chain_spec::bridge_hubs::BridgeHubRuntimeType::WococoLocal => {
+					let $partials = new_partial::<chain_spec::bridge_hubs::wococo::RuntimeApi, _>(
+						&$config,
+						crate::service::aura_build_import_queue::<_, AuraId>,
+					)?;
+					$code
+				},
+			},
 			Runtime::CollectivesPolkadot | Runtime::CollectivesWestend => {
 				let $partials = new_partial::<collectives_polkadot_runtime::RuntimeApi, _>(
 					&$config,