Fix tests

Author: kianenigma

frame/election-provider-multi-phase/src/benchmarking.rs

@@ -494,6 +494,6 @@ frame_benchmarking::benchmarks! {
 
 impl_benchmark_test_suite!(
 	MultiPhase,
-	crate::mock::ExtBuilder::default().build(),
+	crate::mock::ExtBuilder::default().build_offchainify(10).0,
 	crate::mock::Runtime,
 );

frame/election-provider-multi-phase/src/lib.rs

@@ -913,7 +913,7 @@ pub mod pallet {
 
 			// Note: we don't `rotate_round` at this point; the next call to
 			// `ElectionProvider::elect` will succeed and take care of that.
-			
+
 			let solution = ReadySolution {
 				supports,
 				score: [0, 0, 0],

frame/election-provider-multi-phase/src/mock.rs

@@ -438,6 +438,32 @@ impl ElectionDataProvider<AccountId, u64> for StakingMock {
 		Targets::set(targets);
 		Voters::set(voters);
 	}
+
+	#[cfg(any(feature = "runtime-benchmarks", test))]
+	fn clear() {
+		Targets::set(vec![]);
+		Voters::set(vec![]);
+	}
+
+	#[cfg(any(feature = "runtime-benchmarks", test))]
+	fn add_voter(voter: AccountId, weight: VoteWeight, targets: Vec<AccountId>) {
+		let mut current = Voters::get();
+		current.push((voter, weight, targets));
+		Voters::set(current);
+	}
+
+	#[cfg(any(feature = "runtime-benchmarks", test))]
+	fn add_target(target: AccountId) {
+		let mut current = Targets::get();
+		current.push(target);
+		Targets::set(current);
+
+		// to be on-par with staking, we add a self vote as well. the stake is really not that
+		// important.
+		let mut current = Voters::get();
+		current.push((target, ExistentialDeposit::get() as u64, vec![target]));
+		Voters::set(current);
+	}
 }
 
 impl ExtBuilder {