feat: Rebrand Polkadot/Substrate references to PezkuwiChain

This commit systematically rebrands various references from Parity Technologies'
Polkadot/Substrate ecosystem to PezkuwiChain within the kurdistan-sdk.

Key changes include:
- Updated external repository URLs (zombienet-sdk, parity-db, parity-scale-codec, wasm-instrument) to point to pezkuwichain forks.
- Modified internal documentation and code comments to reflect PezkuwiChain naming and structure.
- Replaced direct references to  with  or specific paths within the  for XCM, Pezkuwi, and other modules.
- Cleaned up deprecated  issue and PR references in various  and  files, particularly in  and  modules.
- Adjusted image and logo URLs in documentation to point to PezkuwiChain assets.
- Removed or rephrased comments related to external Polkadot/Substrate PRs and issues.

This is a significant step towards fully customizing the SDK for the PezkuwiChain ecosystem.

bizinikiwi/pezframe/election-provider-multi-phase/test-staking-e2e/src/mock.rs

@@ -0,0 +1,992 @@
+// This file is part of Bizinikiwi.
+
+// Copyright (C) Parity Technologies (UK) Ltd.
+// SPDX-License-Identifier: Apache-2.0
+
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// 	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#![allow(dead_code)]
+
+use pezframe_support::{
+	assert_ok, parameter_types, traits,
+	traits::{Hooks, UnfilteredDispatchable, VariantCountOf},
+	weights::constants,
+	PalletId,
+};
+use pezframe_system::EnsureRoot;
+use pezsp_core::{ConstBool, ConstU32, Get};
+use pezsp_npos_elections::{ElectionScore, VoteWeight};
+use pezsp_runtime::{
+	offchain::{
+		testing::{OffchainState, PoolState, TestOffchainExt, TestTransactionPoolExt},
+		OffchainDbExt, OffchainWorkerExt, TransactionPoolExt,
+	},
+	testing,
+	traits::Zero,
+	transaction_validity, BuildStorage, PerU16, Perbill, Percent,
+};
+use pezsp_staking::{
+	offence::{OffenceDetails, OnOffenceHandler},
+	Agent, DelegationInterface, EraIndex, SessionIndex, StakingInterface,
+};
+use std::collections::BTreeMap;
+
+use codec::Decode;
+use pezframe_election_provider_support::{
+	bounds::ElectionBoundsBuilder, onchain, ElectionDataProvider, ExtendedBalance,
+	SequentialPhragmen, Weight,
+};
+use pezpallet_election_provider_multi_phase::{
+	unsigned::MinerConfig, Call, CurrentPhase, ElectionCompute, GeometricDepositBase,
+	QueuedSolution, SolutionAccuracyOf,
+};
+use pezpallet_staking::{ActiveEra, CurrentEra, ErasStartSessionIndex, StakerStatus};
+use parking_lot::RwLock;
+use std::sync::Arc;
+
+use crate::{log, log_current_time};
+use pezframe_support::{derive_impl, traits::Nothing};
+
+pub const INIT_TIMESTAMP: BlockNumber = 30_000;
+pub const BLOCK_TIME: BlockNumber = 1000;
+
+type Block = pezframe_system::mocking::MockBlockU32<Runtime>;
+type Extrinsic = pezsp_runtime::testing::TestXt<RuntimeCall, ()>;
+
+pezframe_support::construct_runtime!(
+	pub enum Runtime {
+		System: pezframe_system,
+		ElectionProviderMultiPhase: pezpallet_election_provider_multi_phase,
+		Staking: pezpallet_staking,
+		DelegatedStaking: pezpallet_delegated_staking,
+		Pools: pezpallet_nomination_pools,
+		Balances: pezpallet_balances,
+		BagsList: pezpallet_bags_list,
+		Session: pezpallet_session,
+		Historical: pezpallet_session::historical,
+		Timestamp: pezpallet_timestamp,
+	}
+);
+
+pub(crate) type AccountId = u128;
+pub(crate) type AccountIndex = u32;
+pub(crate) type BlockNumber = u32;
+pub(crate) type Balance = u64;
+pub(crate) type VoterIndex = u16;
+pub(crate) type TargetIndex = u16;
+pub(crate) type Moment = u32;
+
+#[derive_impl(pezframe_system::config_preludes::TestDefaultConfig)]
+impl pezframe_system::Config for Runtime {
+	type AccountId = AccountId;
+	type Block = Block;
+	type AccountData = pezpallet_balances::AccountData<Balance>;
+	type Lookup = pezsp_runtime::traits::IdentityLookup<Self::AccountId>;
+}
+
+const NORMAL_DISPATCH_RATIO: Perbill = Perbill::from_percent(75);
+parameter_types! {
+	pub static ExistentialDeposit: Balance = 1;
+	pub BlockWeights: pezframe_system::limits::BlockWeights = pezframe_system::limits::BlockWeights
+		::with_sensible_defaults(
+			Weight::from_parts(2u64 * constants::WEIGHT_REF_TIME_PER_SECOND, u64::MAX),
+			NORMAL_DISPATCH_RATIO,
+		);
+}
+
+#[derive_impl(pezpallet_balances::config_preludes::TestDefaultConfig)]
+impl pezpallet_balances::Config for Runtime {
+	type ExistentialDeposit = ExistentialDeposit;
+	type AccountStore = System;
+	type MaxFreezes = VariantCountOf<RuntimeFreezeReason>;
+	type RuntimeHoldReason = RuntimeHoldReason;
+	type RuntimeFreezeReason = RuntimeFreezeReason;
+	type FreezeIdentifier = RuntimeFreezeReason;
+}
+
+impl pezpallet_timestamp::Config for Runtime {
+	type Moment = Moment;
+	type OnTimestampSet = ();
+	type MinimumPeriod = traits::ConstU32<5>;
+	type WeightInfo = ();
+}
+
+parameter_types! {
+	pub static Period: u32 = 30;
+	pub static Offset: u32 = 0;
+}
+
+pezsp_runtime::impl_opaque_keys! {
+	pub struct SessionKeys {
+		pub other: OtherSessionHandler,
+	}
+}
+
+impl pezpallet_session::Config for Runtime {
+	type SessionManager = pezpallet_session::historical::NoteHistoricalRoot<Runtime, Staking>;
+	type Keys = SessionKeys;
+	type ShouldEndSession = pezpallet_session::PeriodicSessions<Period, Offset>;
+	type NextSessionRotation = pezpallet_session::PeriodicSessions<Period, Offset>;
+	type SessionHandler = (OtherSessionHandler,);
+	type RuntimeEvent = RuntimeEvent;
+	type ValidatorId = AccountId;
+	type ValidatorIdOf = pezsp_runtime::traits::ConvertInto;
+	type DisablingStrategy = pezpallet_session::disabling::UpToLimitWithReEnablingDisablingStrategy<
+		SLASHING_DISABLING_FACTOR,
+	>;
+	type WeightInfo = ();
+	type Currency = Balances;
+	type KeyDeposit = ();
+}
+impl pezpallet_session::historical::Config for Runtime {
+	type RuntimeEvent = RuntimeEvent;
+	type FullIdentification = ();
+	type FullIdentificationOf = pezpallet_staking::UnitIdentificationOf<Self>;
+}
+
+pezframe_election_provider_support::generate_solution_type!(
+	#[compact]
+	pub struct MockNposSolution::<
+		VoterIndex = VoterIndex,
+		TargetIndex = TargetIndex,
+		Accuracy = PerU16,
+		MaxVoters = ConstU32::<2_000>
+	>(16)
+);
+
+parameter_types! {
+	pub static SignedPhase: BlockNumber = 10;
+	pub static UnsignedPhase: BlockNumber = 10;
+	// we expect a minimum of 3 blocks in signed phase and unsigned phases before trying
+	// entering in emergency phase after the election failed.
+	pub static MinBlocksBeforeEmergency: BlockNumber = 3;
+	pub static MaxActiveValidators: u32 = 1000;
+	pub static OffchainRepeat: u32 = 5;
+	pub static MinerMaxLength: u32 = 256;
+	pub static MinerMaxWeight: Weight = BlockWeights::get().max_block;
+	pub static TransactionPriority: transaction_validity::TransactionPriority = 1;
+	#[derive(Debug)]
+	pub static MaxWinners: u32 = 100;
+	#[derive(Debug)]
+	pub static MaxBackersPerWinner: u32 = 100;
+	pub static MaxVotesPerVoter: u32 = 16;
+	pub static SignedFixedDeposit: Balance = 1;
+	pub static SignedDepositIncreaseFactor: Percent = Percent::from_percent(10);
+	pub static ElectionBounds: pezframe_election_provider_support::bounds::ElectionBounds = ElectionBoundsBuilder::default()
+		.voters_count(1_000.into()).targets_count(1_000.into()).build();
+}
+
+impl pezpallet_election_provider_multi_phase::Config for Runtime {
+	type RuntimeEvent = RuntimeEvent;
+	type Currency = Balances;
+	type EstimateCallFee = pezframe_support::traits::ConstU64<8>;
+	type SignedPhase = SignedPhase;
+	type UnsignedPhase = UnsignedPhase;
+	type BetterSignedThreshold = ();
+	type OffchainRepeat = OffchainRepeat;
+	type MinerTxPriority = TransactionPriority;
+	type MinerConfig = Self;
+	type SignedMaxSubmissions = ConstU32<10>;
+	type SignedRewardBase = ();
+	type SignedDepositBase =
+		GeometricDepositBase<Balance, SignedFixedDeposit, SignedDepositIncreaseFactor>;
+	type SignedDepositByte = ();
+	type SignedMaxRefunds = ConstU32<3>;
+	type SignedDepositWeight = ();
+	type SignedMaxWeight = ();
+	type SlashHandler = ();
+	type RewardHandler = ();
+	type DataProvider = Staking;
+	type Fallback = pezframe_election_provider_support::NoElection<(
+		AccountId,
+		BlockNumber,
+		Staking,
+		MaxWinners,
+		MaxBackersPerWinner,
+	)>;
+	type GovernanceFallback = onchain::OnChainExecution<OnChainSeqPhragmen>;
+	type Solver = SequentialPhragmen<AccountId, SolutionAccuracyOf<Runtime>, ()>;
+	type ForceOrigin = EnsureRoot<AccountId>;
+	type MaxWinners = MaxWinners;
+	type MaxBackersPerWinner = MaxBackersPerWinner;
+	type ElectionBounds = ElectionBounds;
+	type BenchmarkingConfig = NoopElectionProviderBenchmarkConfig;
+	type WeightInfo = ();
+}
+
+impl MinerConfig for Runtime {
+	type AccountId = AccountId;
+	type Solution = MockNposSolution;
+	type MaxVotesPerVoter =
+	<<Self as pezpallet_election_provider_multi_phase::Config>::DataProvider as ElectionDataProvider>::MaxVotesPerVoter;
+	type MaxLength = MinerMaxLength;
+	type MaxWeight = MinerMaxWeight;
+	type MaxWinners = MaxWinners;
+	type MaxBackersPerWinner = MaxBackersPerWinner;
+
+	fn solution_weight(_v: u32, _t: u32, _a: u32, _d: u32) -> Weight {
+		Weight::zero()
+	}
+}
+
+const THRESHOLDS: [VoteWeight; 9] = [10, 20, 30, 40, 50, 60, 1_000, 2_000, 10_000];
+
+parameter_types! {
+	pub static BagThresholds: &'static [pezsp_npos_elections::VoteWeight] = &THRESHOLDS;
+	pub const SessionsPerEra: pezsp_staking::SessionIndex = 2;
+	pub static BondingDuration: pezsp_staking::EraIndex = 28;
+	pub const SlashDeferDuration: pezsp_staking::EraIndex = 7; // 1/4 the bonding duration.
+}
+
+impl pezpallet_bags_list::Config for Runtime {
+	type RuntimeEvent = RuntimeEvent;
+	type WeightInfo = ();
+	type ScoreProvider = Staking;
+	type BagThresholds = BagThresholds;
+	type Score = VoteWeight;
+	type MaxAutoRebagPerBlock = ();
+}
+
+pub struct BalanceToU256;
+impl pezsp_runtime::traits::Convert<Balance, pezsp_core::U256> for BalanceToU256 {
+	fn convert(n: Balance) -> pezsp_core::U256 {
+		n.into()
+	}
+}
+
+pub struct U256ToBalance;
+impl pezsp_runtime::traits::Convert<pezsp_core::U256, Balance> for U256ToBalance {
+	fn convert(n: pezsp_core::U256) -> Balance {
+		n.try_into().unwrap()
+	}
+}
+
+parameter_types! {
+	pub const PoolsPalletId: pezframe_support::PalletId = pezframe_support::PalletId(*b"py/nopls");
+	pub static MaxUnbonding: u32 = 8;
+}
+
+impl pezpallet_nomination_pools::Config for Runtime {
+	type RuntimeEvent = RuntimeEvent;
+	type WeightInfo = ();
+	type Currency = Balances;
+	type RuntimeFreezeReason = RuntimeFreezeReason;
+	type RewardCounter = pezsp_runtime::FixedU128;
+	type BalanceToU256 = BalanceToU256;
+	type U256ToBalance = U256ToBalance;
+	type StakeAdapter =
+		pezpallet_nomination_pools::adapter::DelegateStake<Self, Staking, DelegatedStaking>;
+	type PostUnbondingPoolsWindow = ConstU32<2>;
+	type PalletId = PoolsPalletId;
+	type MaxMetadataLen = ConstU32<256>;
+	type MaxUnbonding = MaxUnbonding;
+	type MaxPointsToBalance = pezframe_support::traits::ConstU8<10>;
+	type AdminOrigin = pezframe_system::EnsureRoot<Self::AccountId>;
+	type BlockNumberProvider = System;
+	type Filter = Nothing;
+}
+
+parameter_types! {
+	pub const DelegatedStakingPalletId: PalletId = PalletId(*b"py/dlstk");
+	pub const SlashRewardFraction: Perbill = Perbill::from_percent(1);
+}
+
+impl pezpallet_delegated_staking::Config for Runtime {
+	type RuntimeEvent = RuntimeEvent;
+	type PalletId = DelegatedStakingPalletId;
+	type Currency = Balances;
+	type OnSlash = ();
+	type SlashRewardFraction = SlashRewardFraction;
+	type RuntimeHoldReason = RuntimeHoldReason;
+	type CoreStaking = Staking;
+}
+
+parameter_types! {
+	pub static MaxUnlockingChunks: u32 = 32;
+}
+
+/// Upper limit on the number of NPOS nominations.
+const MAX_QUOTA_NOMINATIONS: u32 = 16;
+/// Disabling factor set explicitly to byzantine threshold
+pub(crate) const SLASHING_DISABLING_FACTOR: usize = 3;
+
+#[derive_impl(pezpallet_staking::config_preludes::TestDefaultConfig)]
+impl pezpallet_staking::Config for Runtime {
+	type OldCurrency = Balances;
+	type Currency = Balances;
+	type CurrencyBalance = Balance;
+	type UnixTime = Timestamp;
+	type SessionsPerEra = SessionsPerEra;
+	type BondingDuration = BondingDuration;
+	type SlashDeferDuration = SlashDeferDuration;
+	type AdminOrigin = EnsureRoot<AccountId>; // root can cancel slashes
+	type SessionInterface = Self;
+	type EraPayout = ();
+	type NextNewSession = Session;
+	type MaxExposurePageSize = ConstU32<256>;
+	type ElectionProvider = ElectionProviderMultiPhase;
+	type GenesisElectionProvider = onchain::OnChainExecution<OnChainSeqPhragmen>;
+	type VoterList = BagsList;
+	type NominationsQuota = pezpallet_staking::FixedNominationsQuota<MAX_QUOTA_NOMINATIONS>;
+	type TargetList = pezpallet_staking::UseValidatorsMap<Self>;
+	type MaxUnlockingChunks = MaxUnlockingChunks;
+	type EventListeners = (Pools, DelegatedStaking);
+	type WeightInfo = pezpallet_staking::weights::BizinikiwiWeight<Runtime>;
+	type BenchmarkingConfig = pezpallet_staking::TestBenchmarkingConfig;
+}
+
+impl<LocalCall> pezframe_system::offchain::CreateTransactionBase<LocalCall> for Runtime
+where
+	RuntimeCall: From<LocalCall>,
+{
+	type RuntimeCall = RuntimeCall;
+	type Extrinsic = Extrinsic;
+}
+
+impl<LocalCall> pezframe_system::offchain::CreateBare<LocalCall> for Runtime
+where
+	RuntimeCall: From<LocalCall>,
+{
+	fn create_bare(call: Self::RuntimeCall) -> Self::Extrinsic {
+		Extrinsic::new_bare(call)
+	}
+}
+
+pub struct OnChainSeqPhragmen;
+
+parameter_types! {
+	pub static VotersBound: u32 = 600;
+	pub static TargetsBound: u32 = 400;
+}
+
+impl onchain::Config for OnChainSeqPhragmen {
+	type MaxWinnersPerPage = MaxWinners;
+	type MaxBackersPerWinner = MaxBackersPerWinner;
+	type Sort = ConstBool<true>;
+	type System = Runtime;
+	type Solver = SequentialPhragmen<
+		AccountId,
+		pezpallet_election_provider_multi_phase::SolutionAccuracyOf<Runtime>,
+	>;
+	type DataProvider = Staking;
+	type WeightInfo = ();
+	type Bounds = ElectionBounds;
+}
+
+pub struct NoopElectionProviderBenchmarkConfig;
+
+impl pezpallet_election_provider_multi_phase::BenchmarkingConfig
+	for NoopElectionProviderBenchmarkConfig
+{
+	const VOTERS: [u32; 2] = [0, 0];
+	const TARGETS: [u32; 2] = [0, 0];
+	const ACTIVE_VOTERS: [u32; 2] = [0, 0];
+	const DESIRED_TARGETS: [u32; 2] = [0, 0];
+	const SNAPSHOT_MAXIMUM_VOTERS: u32 = 0;
+	const MINER_MAXIMUM_VOTERS: u32 = 0;
+	const MAXIMUM_TARGETS: u32 = 0;
+}
+
+pub struct OtherSessionHandler;
+impl traits::OneSessionHandler<AccountId> for OtherSessionHandler {
+	type Key = testing::UintAuthorityId;
+
+	fn on_genesis_session<'a, I: 'a>(_: I)
+	where
+		I: Iterator<Item = (&'a AccountId, Self::Key)>,
+		AccountId: 'a,
+	{
+	}
+
+	fn on_new_session<'a, I: 'a>(_: bool, _: I, _: I)
+	where
+		I: Iterator<Item = (&'a AccountId, Self::Key)>,
+		AccountId: 'a,
+	{
+	}
+
+	fn on_disabled(_validator_index: u32) {}
+}
+
+impl pezsp_runtime::BoundToRuntimeAppPublic for OtherSessionHandler {
+	type Public = testing::UintAuthorityId;
+}
+
+pub struct StakingExtBuilder {
+	validator_count: u32,
+	minimum_validator_count: u32,
+	min_nominator_bond: Balance,
+	min_validator_bond: Balance,
+	status: BTreeMap<AccountId, StakerStatus<AccountId>>,
+	stakes: BTreeMap<AccountId, Balance>,
+	stakers: Vec<(AccountId, AccountId, Balance, StakerStatus<AccountId>)>,
+}
+
+impl Default for StakingExtBuilder {
+	fn default() -> Self {
+		let stakers = vec![
+			// (stash, ctrl, stake, status)
+			// these two will be elected in the default test where we elect 2.
+			(11, 11, 1000, StakerStatus::<AccountId>::Validator),
+			(21, 21, 1000, StakerStatus::<AccountId>::Validator),
+			// loser validators if validator_count() is default.
+			(31, 31, 500, StakerStatus::<AccountId>::Validator),
+			(41, 41, 1500, StakerStatus::<AccountId>::Validator),
+			(51, 51, 1500, StakerStatus::<AccountId>::Validator),
+			(61, 61, 1500, StakerStatus::<AccountId>::Validator),
+			(71, 71, 1500, StakerStatus::<AccountId>::Validator),
+			(81, 81, 1500, StakerStatus::<AccountId>::Validator),
+			(91, 91, 1500, StakerStatus::<AccountId>::Validator),
+			(101, 101, 500, StakerStatus::<AccountId>::Validator),
+			// an idle validator
+			(201, 201, 1000, StakerStatus::<AccountId>::Idle),
+		];
+
+		Self {
+			validator_count: 2,
+			minimum_validator_count: 0,
+			min_nominator_bond: ExistentialDeposit::get(),
+			min_validator_bond: ExistentialDeposit::get(),
+			status: Default::default(),
+			stakes: Default::default(),
+			stakers,
+		}
+	}
+}
+
+impl StakingExtBuilder {
+	pub fn validator_count(mut self, n: u32) -> Self {
+		self.validator_count = n;
+		self
+	}
+	pub fn max_unlocking(self, max: u32) -> Self {
+		<MaxUnlockingChunks>::set(max);
+		self
+	}
+	pub fn bonding_duration(self, eras: EraIndex) -> Self {
+		<BondingDuration>::set(eras);
+		self
+	}
+}
+
+pub struct EpmExtBuilder {}
+
+impl Default for EpmExtBuilder {
+	fn default() -> Self {
+		EpmExtBuilder {}
+	}
+}
+
+impl EpmExtBuilder {
+	pub fn disable_emergency_throttling(self) -> Self {
+		<MinBlocksBeforeEmergency>::set(0);
+		self
+	}
+
+	pub fn phases(self, signed: BlockNumber, unsigned: BlockNumber) -> Self {
+		<SignedPhase>::set(signed);
+		<UnsignedPhase>::set(unsigned);
+		self
+	}
+}
+
+pub struct PoolsExtBuilder {}
+
+impl Default for PoolsExtBuilder {
+	fn default() -> Self {
+		PoolsExtBuilder {}
+	}
+}
+
+impl PoolsExtBuilder {
+	pub fn max_unbonding(self, max: u32) -> Self {
+		<MaxUnbonding>::set(max);
+		self
+	}
+}
+
+pub struct BalancesExtBuilder {
+	balances: Vec<(AccountId, Balance)>,
+}
+
+impl Default for BalancesExtBuilder {
+	fn default() -> Self {
+		let balances = vec![
+			// (account_id, balance)
+			(1, 10),
+			(2, 20),
+			(3, 300),
+			(4, 400),
+			// controllers (still used in some tests. Soon to be deprecated).
+			(10, 100),
+			(20, 100),
+			(30, 100),
+			(40, 100),
+			(50, 100),
+			(60, 100),
+			(70, 100),
+			(80, 100),
+			(90, 100),
+			(100, 100),
+			(200, 100),
+			// stashes
+			(11, 1100),
+			(21, 2000),
+			(31, 3000),
+			(41, 4000),
+			(51, 5000),
+			(61, 6000),
+			(71, 7000),
+			(81, 8000),
+			(91, 9000),
+			(101, 10000),
+			(201, 20000),
+			// This allows us to have a total_payout different from 0.
+			(999, 1_000_000_000_000),
+		];
+		Self { balances }
+	}
+}
+
+pub struct ExtBuilder {
+	staking_builder: StakingExtBuilder,
+	epm_builder: EpmExtBuilder,
+	balances_builder: BalancesExtBuilder,
+	pools_builder: PoolsExtBuilder,
+}
+
+impl Default for ExtBuilder {
+	fn default() -> Self {
+		Self {
+			staking_builder: StakingExtBuilder::default(),
+			epm_builder: EpmExtBuilder::default(),
+			balances_builder: BalancesExtBuilder::default(),
+			pools_builder: PoolsExtBuilder::default(),
+		}
+	}
+}
+
+impl ExtBuilder {
+	pub fn build(&self) -> pezsp_io::TestExternalities {
+		pezsp_tracing::try_init_simple();
+		let mut storage =
+			pezframe_system::GenesisConfig::<Runtime>::default().build_storage().unwrap();
+
+		let _ = pezpallet_balances::GenesisConfig::<Runtime> {
+			balances: self.balances_builder.balances.clone(),
+			..Default::default()
+		}
+		.assimilate_storage(&mut storage);
+
+		let mut stakers = self.staking_builder.stakers.clone();
+		self.staking_builder.status.clone().into_iter().for_each(|(stash, status)| {
+			let (_, _, _, ref mut prev_status) = stakers
+				.iter_mut()
+				.find(|s| s.0 == stash)
+				.expect("set_status staker should exist; qed");
+			*prev_status = status;
+		});
+		// replaced any of the stakes if needed.
+		self.staking_builder.stakes.clone().into_iter().for_each(|(stash, stake)| {
+			let (_, _, ref mut prev_stake, _) = stakers
+				.iter_mut()
+				.find(|s| s.0 == stash)
+				.expect("set_stake staker should exits; qed.");
+			*prev_stake = stake;
+		});
+
+		let _ = pezpallet_staking::GenesisConfig::<Runtime> {
+			stakers: stakers.clone(),
+			validator_count: self.staking_builder.validator_count,
+			minimum_validator_count: self.staking_builder.minimum_validator_count,
+			slash_reward_fraction: Perbill::from_percent(10),
+			min_nominator_bond: self.staking_builder.min_nominator_bond,
+			min_validator_bond: self.staking_builder.min_validator_bond,
+			..Default::default()
+		}
+		.assimilate_storage(&mut storage);
+
+		let _ = pezpallet_session::GenesisConfig::<Runtime> {
+			// set the keys for the first session.
+			keys: stakers
+				.into_iter()
+				.map(|(id, ..)| (id, id, SessionKeys { other: (id as AccountId as u64).into() }))
+				.collect(),
+			..Default::default()
+		}
+		.assimilate_storage(&mut storage);
+
+		let mut ext = pezsp_io::TestExternalities::from(storage);
+
+		// We consider all test to start after timestamp is initialized This must be ensured by
+		// having `timestamp::on_initialize` called before `staking::on_initialize`.
+		ext.execute_with(|| {
+			System::set_block_number(1);
+			Session::on_initialize(1);
+			<Staking as Hooks<u32>>::on_initialize(1);
+			Timestamp::set_timestamp(INIT_TIMESTAMP);
+		});
+
+		ext
+	}
+
+	pub fn staking(mut self, builder: StakingExtBuilder) -> Self {
+		self.staking_builder = builder;
+		self
+	}
+
+	pub fn epm(mut self, builder: EpmExtBuilder) -> Self {
+		self.epm_builder = builder;
+		self
+	}
+
+	pub fn pools(mut self, builder: PoolsExtBuilder) -> Self {
+		self.pools_builder = builder;
+		self
+	}
+
+	pub fn balances(mut self, builder: BalancesExtBuilder) -> Self {
+		self.balances_builder = builder;
+		self
+	}
+
+	pub fn build_offchainify(
+		self,
+	) -> (pezsp_io::TestExternalities, Arc<RwLock<PoolState>>, Arc<RwLock<OffchainState>>) {
+		// add offchain and pool externality extensions.
+		let mut ext = self.build();
+		let (offchain, offchain_state) = TestOffchainExt::new();
+		let (pool, pool_state) = TestTransactionPoolExt::new();
+
+		ext.register_extension(OffchainDbExt::new(offchain.clone()));
+		ext.register_extension(OffchainWorkerExt::new(offchain));
+		ext.register_extension(TransactionPoolExt::new(pool));
+
+		(ext, pool_state, offchain_state)
+	}
+}
+
+pub(crate) fn execute_with(mut ext: pezsp_io::TestExternalities, test: impl FnOnce() -> ()) {
+	ext.execute_with(test);
+
+	#[cfg(feature = "try-runtime")]
+	ext.execute_with(|| {
+		let bn = System::block_number();
+
+		assert_ok!(<ElectionProviderMultiPhase as Hooks<BlockNumber>>::try_state(bn));
+		assert_ok!(<Staking as Hooks<BlockNumber>>::try_state(bn));
+		assert_ok!(<Pools as Hooks<BlockNumber>>::try_state(bn));
+		assert_ok!(<Session as Hooks<BlockNumber>>::try_state(bn));
+	});
+}
+
+// Progress to given block, triggering session and era changes as we progress and ensuring that
+// there is a solution queued when expected.
+pub fn roll_to(n: BlockNumber, delay_solution: bool) {
+	for b in (System::block_number()) + 1..=n {
+		System::set_block_number(b);
+		Session::on_initialize(b);
+		Timestamp::set_timestamp(System::block_number() * BLOCK_TIME + INIT_TIMESTAMP);
+
+		// TODO(gpestana): implement a realistic OCW worker instead of simulating it
+		// https://github.com/pezkuwichain/kurdistan-sdk/issues/10
+		// if there's no solution queued and the solution should not be delayed, try mining and
+		// queue a solution.
+		if CurrentPhase::<Runtime>::get().is_signed() && !delay_solution {
+			let _ = try_queue_solution(ElectionCompute::Signed).map_err(|e| {
+				log!(info, "failed to mine/queue solution: {:?}", e);
+			});
+		}
+		ElectionProviderMultiPhase::on_initialize(b);
+
+		Staking::on_initialize(b);
+		if b != n {
+			Staking::on_finalize(System::block_number());
+		}
+		Pools::on_initialize(b);
+
+		log_current_time();
+	}
+}
+
+// Progress to given block, triggering session and era changes as we progress and ensuring that
+// there is a solution queued when expected.
+pub fn roll_to_with_ocw(n: BlockNumber, pool: Arc<RwLock<PoolState>>, delay_solution: bool) {
+	for b in (System::block_number()) + 1..=n {
+		System::set_block_number(b);
+		Session::on_initialize(b);
+		Timestamp::set_timestamp(System::block_number() * BLOCK_TIME + INIT_TIMESTAMP);
+
+		ElectionProviderMultiPhase::on_initialize(b);
+		ElectionProviderMultiPhase::offchain_worker(b);
+
+		if !delay_solution && pool.read().transactions.len() > 0 {
+			// decode submit_unsigned callable that may be queued in the pool by ocw. skip all
+			// other extrinsics in the pool.
+			for encoded in &pool.read().transactions {
+				let extrinsic = Extrinsic::decode(&mut &encoded[..]).unwrap();
+
+				let _ = match extrinsic.function {
+					RuntimeCall::ElectionProviderMultiPhase(
+						call @ Call::submit_unsigned { .. },
+					) => {
+						// call submit_unsigned callable in OCW pool.
+						crate::assert_ok!(call.dispatch_bypass_filter(RuntimeOrigin::none()));
+					},
+					_ => (),
+				};
+			}
+
+			pool.try_write().unwrap().transactions.clear();
+		}
+
+		Staking::on_initialize(b);
+		if b != n {
+			Staking::on_finalize(System::block_number());
+		}
+
+		log_current_time();
+	}
+}
+// helper to progress one block ahead.
+pub fn roll_one(pool: Arc<RwLock<PoolState>>, delay_solution: bool) {
+	let bn = System::block_number().saturating_add(1);
+	roll_to_with_ocw(bn, pool, delay_solution);
+}
+
+/// Progresses from the current block number (whatever that may be) to the block where the session
+/// `session_index` starts.
+pub(crate) fn start_session(
+	session_index: SessionIndex,
+	pool: Arc<RwLock<PoolState>>,
+	delay_solution: bool,
+) {
+	let end = if Offset::get().is_zero() {
+		Period::get() * session_index
+	} else {
+		Offset::get() * session_index + Period::get() * session_index
+	};
+
+	assert!(end >= System::block_number());
+
+	roll_to_with_ocw(end, pool, delay_solution);
+
+	// session must have progressed properly.
+	assert_eq!(
+		Session::current_index(),
+		session_index,
+		"current session index = {}, expected = {}",
+		Session::current_index(),
+		session_index,
+	);
+}
+
+/// Go one session forward.
+pub(crate) fn advance_session(pool: Arc<RwLock<PoolState>>) {
+	let current_index = Session::current_index();
+	start_session(current_index + 1, pool, false);
+}
+
+pub(crate) fn advance_session_delayed_solution(pool: Arc<RwLock<PoolState>>) {
+	let current_index = Session::current_index();
+	start_session(current_index + 1, pool, true);
+}
+
+pub(crate) fn start_next_active_era(pool: Arc<RwLock<PoolState>>) -> Result<(), ()> {
+	start_active_era(active_era() + 1, pool, false)
+}
+
+pub(crate) fn start_next_active_era_delayed_solution(
+	pool: Arc<RwLock<PoolState>>,
+) -> Result<(), ()> {
+	start_active_era(active_era() + 1, pool, true)
+}
+
+pub(crate) fn advance_eras(n: usize, pool: Arc<RwLock<PoolState>>) {
+	for _ in 0..n {
+		assert_ok!(start_next_active_era(pool.clone()));
+	}
+}
+
+/// Progress until the given era.
+pub(crate) fn start_active_era(
+	era_index: EraIndex,
+	pool: Arc<RwLock<PoolState>>,
+	delay_solution: bool,
+) -> Result<(), ()> {
+	let era_before = current_era();
+
+	start_session((era_index * <SessionsPerEra as Get<u32>>::get()).into(), pool, delay_solution);
+
+	log!(
+		info,
+		"start_active_era - era_before: {}, current era: {} -> progress to: {} -> after era: {}",
+		era_before,
+		active_era(),
+		era_index,
+		current_era(),
+	);
+
+	// if the solution was not delayed, era should have progressed.
+	if !delay_solution && (active_era() != era_index || current_era() != active_era()) {
+		Err(())
+	} else {
+		Ok(())
+	}
+}
+
+pub(crate) fn active_era() -> EraIndex {
+	ActiveEra::<Runtime>::get().unwrap().index
+}
+
+pub(crate) fn current_era() -> EraIndex {
+	CurrentEra::<Runtime>::get().unwrap()
+}
+
+// Fast forward until EPM signed phase.
+pub fn roll_to_epm_signed() {
+	while !matches!(
+		CurrentPhase::<Runtime>::get(),
+		pezpallet_election_provider_multi_phase::Phase::Signed
+	) {
+		roll_to(System::block_number() + 1, false);
+	}
+}
+
+// Fast forward until EPM unsigned phase.
+pub fn roll_to_epm_unsigned() {
+	while !matches!(
+		CurrentPhase::<Runtime>::get(),
+		pezpallet_election_provider_multi_phase::Phase::Unsigned(_)
+	) {
+		roll_to(System::block_number() + 1, false);
+	}
+}
+
+// Fast forward until EPM off.
+pub fn roll_to_epm_off() {
+	while !matches!(
+		CurrentPhase::<Runtime>::get(),
+		pezpallet_election_provider_multi_phase::Phase::Off
+	) {
+		roll_to(System::block_number() + 1, false);
+	}
+}
+
+// Queue a solution based on the current snapshot.
+pub(crate) fn try_queue_solution(when: ElectionCompute) -> Result<(), String> {
+	let raw_solution = ElectionProviderMultiPhase::mine_solution()
+		.map_err(|e| format!("error mining solution: {:?}", e))?;
+
+	ElectionProviderMultiPhase::feasibility_check(raw_solution.0, when)
+		.map(|ready| {
+			QueuedSolution::<Runtime>::put(ready);
+		})
+		.map_err(|e| format!("error in solution feasibility: {:?}", e))
+}
+
+pub(crate) fn on_offence_now(
+	offenders: &[OffenceDetails<
+		AccountId,
+		pezpallet_session::historical::IdentificationTuple<Runtime>,
+	>],
+	slash_fraction: &[Perbill],
+) {
+	let now = ActiveEra::<Runtime>::get().unwrap().index;
+	let _ = <Staking as OnOffenceHandler<_, _, _>>::on_offence(
+		offenders,
+		slash_fraction,
+		ErasStartSessionIndex::<Runtime>::get(now).unwrap(),
+	);
+}
+
+// Add offence to validator, slash it.
+pub(crate) fn add_slash(who: &AccountId) {
+	on_offence_now(
+		&[OffenceDetails { offender: (*who, ()), reporters: vec![] }],
+		&[Perbill::from_percent(10)],
+	);
+}
+
+// Slashes 1/2 of the active set. Returns the `AccountId`s of the slashed validators.
+pub(crate) fn slash_half_the_active_set() -> Vec<AccountId> {
+	let mut slashed = Session::validators();
+	slashed.truncate(slashed.len() / 2);
+
+	for v in slashed.iter() {
+		add_slash(v);
+	}
+
+	slashed
+}
+
+// Slashes a percentage of the active nominators that haven't been slashed yet, with
+// a minimum of 1 validator slash.
+pub(crate) fn slash_percentage(percentage: Perbill) -> Vec<AccountId> {
+	let validators = Session::validators();
+	let mut remaining_slashes = (percentage * validators.len() as u32).max(1);
+	let mut slashed = vec![];
+
+	for v in validators.into_iter() {
+		if remaining_slashes != 0 {
+			add_slash(&v);
+			slashed.push(v);
+			remaining_slashes -= 1;
+		}
+	}
+	slashed
+}
+
+pub(crate) fn set_minimum_election_score(
+	minimal_stake: ExtendedBalance,
+	sum_stake: ExtendedBalance,
+	sum_stake_squared: ExtendedBalance,
+) -> Result<(), ()> {
+	let election_score = ElectionScore { minimal_stake, sum_stake, sum_stake_squared };
+	ElectionProviderMultiPhase::set_minimum_untrusted_score(
+		RuntimeOrigin::root(),
+		Some(election_score),
+	)
+	.map(|_| ())
+	.map_err(|_| ())
+}
+
+pub(crate) fn staked_amount_for(account_id: AccountId) -> Balance {
+	Staking::total_stake(&account_id).expect("account must be staker")
+}
+
+pub(crate) fn delegated_balance_for(account_id: AccountId) -> Balance {
+	DelegatedStaking::agent_balance(Agent::from(account_id)).unwrap_or_default()
+}
+
+pub(crate) fn staking_events() -> Vec<pezpallet_staking::Event<Runtime>> {
+	System::events()
+		.into_iter()
+		.map(|r| r.event)
+		.filter_map(|e| if let RuntimeEvent::Staking(inner) = e { Some(inner) } else { None })
+		.collect::<Vec<_>>()
+}
+
+pub(crate) fn epm_events() -> Vec<pezpallet_election_provider_multi_phase::Event<Runtime>> {
+	System::events()
+		.into_iter()
+		.map(|r| r.event)
+		.filter_map(|e| {
+			if let RuntimeEvent::ElectionProviderMultiPhase(inner) = e {
+				Some(inner)
+			} else {
+				None
+			}
+		})
+		.collect::<Vec<_>>()
+}