add Account::new_data (#4701)

* add account_new_data

* fixup

* fixup
This commit is contained in:
Rob Walker
2019-06-17 15:58:05 -07:00
committed by GitHub
parent fe04fb4cd3
commit d86358eedc
4 changed files with 40 additions and 17 deletions

View File

@ -122,7 +122,7 @@ fn test_proof_bounds() {
Hash::default(),
);
// the proof is for segment 0, need to move the slot into segment 2
let mut current_account = current::create_account(1);
let mut current_account = current::create_account(1, 0, 0, 0);
Current::to(
&Current {
slot: SLOTS_PER_SEGMENT * 2,
@ -152,7 +152,7 @@ fn test_serialize_overflow() {
let tick_pubkey = Pubkey::new_rand();
let mut keyed_accounts = Vec::new();
let mut user_account = Account::default();
let mut current_account = current::create_account(1);
let mut current_account = current::create_account(1, 0, 0, 0);
keyed_accounts.push(KeyedAccount::new(&pubkey, true, &mut user_account));
keyed_accounts.push(KeyedAccount::new(&tick_pubkey, false, &mut current_account));
@ -181,7 +181,7 @@ fn test_invalid_accounts_len() {
Hash::default(),
);
// move tick height into segment 1
let mut current_account = current::create_account(1);
let mut current_account = current::create_account(1, 0, 0, 0);
Current::to(
&Current {
slot: 16,
@ -240,7 +240,7 @@ fn test_submit_mining_ok() {
Hash::default(),
);
// move slot into segment 1
let mut current_account = current::create_account(1);
let mut current_account = current::create_account(1, 0, 0, 0);
Current::to(
&Current {
slot: SLOTS_PER_SEGMENT,

View File

@ -32,7 +32,7 @@ use solana_sdk::inflation::Inflation;
use solana_sdk::native_loader;
use solana_sdk::pubkey::Pubkey;
use solana_sdk::signature::{Keypair, Signature};
use solana_sdk::syscall::current::{self, Current};
use solana_sdk::syscall::current;
use solana_sdk::syscall::fees::{self, Fees};
use solana_sdk::syscall::slot_hashes::{self, SlotHashes};
use solana_sdk::syscall::tick_height::{self, TickHeight};
@ -402,15 +402,15 @@ impl Bank {
}
fn update_current(&self) {
let mut account = current::create_account(1);
let current = Current {
slot: self.slot,
epoch: self.epoch_schedule.get_epoch(self.slot),
stakers_epoch: self.epoch_schedule.get_stakers_epoch(self.slot),
};
current.to(&mut account).unwrap();
self.store_account(&current::id(), &account);
self.store_account(
&current::id(),
&current::create_account(
1,
self.slot,
self.epoch_schedule.get_epoch(self.slot),
self.epoch_schedule.get_stakers_epoch(self.slot),
),
);
}
fn update_slot_hashes(&self) {

View File

@ -46,6 +46,20 @@ impl Account {
}
}
pub fn new_data<T: serde::Serialize>(
lamports: u64,
state: &T,
owner: &Pubkey,
) -> Result<Account, bincode::Error> {
let data = bincode::serialize(state)?;
Ok(Account {
lamports,
data,
owner: *owner,
executable: false,
})
}
pub fn deserialize_data<T: serde::de::DeserializeOwned>(&self) -> Result<T, bincode::Error> {
bincode::deserialize(&self.data)
}

View File

@ -32,8 +32,17 @@ impl Current {
}
}
pub fn create_account(lamports: u64) -> Account {
Account::new(lamports, Current::size_of(), &syscall::id())
pub fn create_account(lamports: u64, slot: u64, epoch: u64, stakers_epoch: u64) -> Account {
Account::new_data(
lamports,
&Current {
slot,
epoch,
stakers_epoch,
},
&syscall::id(),
)
.unwrap()
}
#[cfg(test)]
@ -42,7 +51,7 @@ mod tests {
#[test]
fn test_create_account() {
let account = create_account(1);
let account = create_account(1, 0, 0, 0);
let current = Current::from(&account).unwrap();
assert_eq!(current, Current::default());
}