mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-13 04:41:02 +00:00
Gav
110 lines
2.9 KiB
Rust
110 lines
2.9 KiB
Rust
use keyedvec::KeyedVec;
|
|
use storage::{Storage, storage_into};
|
|
use primitives::{BlockNumber, Balance, AccountID};
|
|
use runtime::consensus;
|
|
|
|
/// The length of a staking era in blocks.
|
|
pub fn era_length() -> BlockNumber {
|
|
sessions_per_era() * consensus::session_length()
|
|
}
|
|
|
|
/// The length of a staking era in sessions.
|
|
pub fn sessions_per_era() -> BlockNumber {
|
|
storage_into(b"sta\0spe")
|
|
}
|
|
|
|
/// The era has changed - enact new staking set.
|
|
///
|
|
/// NOTE: This always happens on a session change.
|
|
pub fn next_era() {
|
|
unimplemented!()
|
|
}
|
|
|
|
/// The balance of a given account.
|
|
pub fn balance(who: &AccountID) -> Balance {
|
|
storage_into(&who.to_keyed_vec(b"sta\0bal\0"))
|
|
}
|
|
|
|
/// Transfer some unlocked staking balance to another staker.
|
|
pub fn transfer_stake(transactor: &AccountID, dest: &AccountID, value: Balance) {
|
|
let from_key = transactor.to_keyed_vec(b"sta\0bal\0");
|
|
let from_balance: Balance = storage_into(&from_key);
|
|
assert!(from_balance >= value);
|
|
let to_key = dest.to_keyed_vec(b"sta\0bal\0");
|
|
let to_balance: Balance = storage_into(&to_key);
|
|
assert!(to_balance + value > to_balance); // no overflow
|
|
(from_balance - value).store(&from_key);
|
|
(to_balance + value).store(&to_key);
|
|
}
|
|
|
|
/// Declare the desire to stake for the transactor.
|
|
///
|
|
/// Effects will be felt at the beginning of the next era.
|
|
pub fn stake(_transactor: &AccountID) {
|
|
unimplemented!()
|
|
}
|
|
|
|
/// Retract the desire to stake for the transactor.
|
|
///
|
|
/// Effects will be felt at the beginning of the next era.
|
|
pub fn unstake(_transactor: &AccountID) {
|
|
unimplemented!()
|
|
}
|
|
|
|
/// Hook to be called prior to transaction processing.
|
|
pub fn pre_transactions() {
|
|
consensus::pre_transactions();
|
|
}
|
|
|
|
/// Hook to be called after to transaction processing.
|
|
pub fn post_transactions() {
|
|
// TODO: check block number and call next_era if necessary.
|
|
consensus::post_transactions();
|
|
}
|
|
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use runtime_support::with_externalities;
|
|
use testing::TestExternalities;
|
|
use primitives::{AccountID};
|
|
use runtime::staking;
|
|
|
|
macro_rules! map {
|
|
($( $name:expr => $value:expr ),*) => (
|
|
vec![ $( ( $name, $value ) ),* ].into_iter().collect()
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn staking_balance_works() {
|
|
let one: AccountID = [1u8; 32];
|
|
let two: AccountID = [2u8; 32];
|
|
|
|
let mut t = TestExternalities { storage: map![
|
|
{ let mut r = b"sta\0bal\0".to_vec(); r.extend_from_slice(&one); r } => vec![42u8, 0, 0, 0, 0, 0, 0, 0]
|
|
], };
|
|
|
|
with_externalities(&mut t, || {
|
|
assert_eq!(staking::balance(&one), 42);
|
|
assert_eq!(staking::balance(&two), 0);
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn staking_balance_transfer_works() {
|
|
let one: AccountID = [1u8; 32];
|
|
let two: AccountID = [2u8; 32];
|
|
|
|
let mut t = TestExternalities { storage: map![
|
|
{ let mut r = b"sta\0bal\0".to_vec(); r.extend_from_slice(&one); r } => vec![111u8, 0, 0, 0, 0, 0, 0, 0]
|
|
], };
|
|
|
|
with_externalities(&mut t, || {
|
|
staking::transfer_stake(&one, &two, 69);
|
|
assert_eq!(staking::balance(&one), 42);
|
|
assert_eq!(staking::balance(&two), 69);
|
|
});
|
|
}
|
|
}
|