gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add extra authorized withdrawer test

Browse Source
This commit is contained in:
Michael Vines 2021-06-16 15:36:53 -07:00
parent fa04531c7a
commit db37680f6f
1 changed files with 30 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
programs/stake/src/stake_state.rs
View File

@ -2766,11 +2766,7 @@ mod tests {
// Lockup expired: custodian cannot change it
assert_eq!(
stake_keyed_account.set_lockup(
&LockupArgs {
unix_timestamp: Some(3),
epoch: None,
custodian: None,
},
&LockupArgs::default(),
&vec![custodian].into_iter().collect(),
Some(&Clock {
unix_timestamp: UnixTimestamp::MAX,
@ -2784,11 +2780,7 @@ mod tests {
// Lockup expired: authorized withdrawer can change it
assert_eq!(
stake_keyed_account.set_lockup(
&LockupArgs {
unix_timestamp: Some(3),
epoch: None,
custodian: None,
},
&LockupArgs::default(),
&vec![stake_pubkey].into_iter().collect(),
Some(&Clock {
unix_timestamp: UnixTimestamp::MAX,
@ -2798,6 +2790,34 @@ mod tests {
),
Ok(())
);
// Change authorized withdrawer
let new_withdraw_authority = solana_sdk::pubkey::new_rand();
assert_eq!(
stake_keyed_account.authorize(
&vec![stake_pubkey].into_iter().collect(),
&new_withdraw_authority,
StakeAuthorize::Withdrawer,
false,
&Clock::default(),
None
),
Ok(())
);
// Previous authorized withdrawer cannot change the lockup anymore
assert_eq!(
stake_keyed_account.set_lockup(
&LockupArgs::default(),
&vec![stake_pubkey].into_iter().collect(),
Some(&Clock {
unix_timestamp: UnixTimestamp::MAX,
epoch: Epoch::MAX,
..Clock::default()
})
),
Err(InstructionError::MissingRequiredSignature)
);
}
#[test]