init and upgrade staking test to 2.0

yanganto · Feb 27, 2020 · 8e78068 · 8e78068
1 parent 68ed099
commit 8e78068
Show file tree

Hide file tree

Showing 2 changed files with 4,408 additions and 4,430 deletions.
diff --git a/frame/staking/src/mock.rs b/frame/staking/src/mock.rs
@@ -416,6 +416,14 @@ pub fn check_exposure(stash: AccountId) {
 	// );
 }
 
+pub fn assert_ledger_consistent(stash: u64) {
+	assert_is_stash(stash);
+	let ledger = Staking::ledger(stash - 1).unwrap();
+
+	assert_eq!(ledger.active_ring, ledger.ring_staking_lock.staking_amount);
+	assert_eq!(ledger.active_kton, ledger.kton_staking_lock.staking_amount);
+}
+
 /// Check that for each nominator: slashable_balance > sum(used_balance)
 /// Note: we might not consume all of a nominator's balance, but we MUST NOT over spend it.
 pub fn check_nominator_exposure(stash: AccountId) {