2019-02-19 15:54:38 -08:00
|
|
|
use bincode::deserialize;
|
|
|
|
use log::info;
|
|
|
|
use solana_runtime::bank::Bank;
|
|
|
|
use solana_sdk::genesis_block::GenesisBlock;
|
|
|
|
use solana_sdk::hash::{hash, Hash};
|
|
|
|
use solana_sdk::signature::{Keypair, KeypairUtil, Signature};
|
|
|
|
use solana_sdk::storage_program;
|
|
|
|
use solana_sdk::storage_program::{StorageTransaction, ENTRIES_PER_SEGMENT};
|
|
|
|
|
|
|
|
fn get_storage_entry_height(bank: &Bank) -> u64 {
|
|
|
|
match bank.get_account(&storage_program::system_id()) {
|
|
|
|
Some(storage_system_account) => {
|
|
|
|
let state = deserialize(&storage_system_account.userdata);
|
|
|
|
if let Ok(state) = state {
|
|
|
|
let state: storage_program::StorageProgramState = state;
|
|
|
|
return state.entry_height;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
None => {
|
|
|
|
info!("error in reading entry_height");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
0
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_storage_last_id(bank: &Bank) -> Hash {
|
|
|
|
if let Some(storage_system_account) = bank.get_account(&storage_program::system_id()) {
|
|
|
|
let state = deserialize(&storage_system_account.userdata);
|
|
|
|
if let Ok(state) = state {
|
|
|
|
let state: storage_program::StorageProgramState = state;
|
|
|
|
return state.id;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Hash::default()
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2019-02-22 14:16:15 -08:00
|
|
|
#[ignore]
|
2019-02-19 15:54:38 -08:00
|
|
|
fn test_bank_storage() {
|
|
|
|
let (genesis_block, alice) = GenesisBlock::new(1000);
|
|
|
|
let bank = Bank::new(&genesis_block);
|
|
|
|
|
|
|
|
let bob = Keypair::new();
|
|
|
|
let jack = Keypair::new();
|
|
|
|
let jill = Keypair::new();
|
|
|
|
|
|
|
|
let x = 42;
|
|
|
|
let last_id = hash(&[x]);
|
|
|
|
let x2 = x * 2;
|
|
|
|
let storage_last_id = hash(&[x2]);
|
|
|
|
|
|
|
|
bank.register_tick(&last_id);
|
|
|
|
|
|
|
|
bank.transfer(10, &alice, jill.pubkey(), last_id).unwrap();
|
|
|
|
|
|
|
|
bank.transfer(10, &alice, bob.pubkey(), last_id).unwrap();
|
|
|
|
bank.transfer(10, &alice, jack.pubkey(), last_id).unwrap();
|
|
|
|
|
|
|
|
let tx = StorageTransaction::new_advertise_last_id(
|
|
|
|
&bob,
|
|
|
|
storage_last_id,
|
|
|
|
last_id,
|
|
|
|
ENTRIES_PER_SEGMENT,
|
|
|
|
);
|
|
|
|
|
|
|
|
bank.process_transaction(&tx).unwrap();
|
|
|
|
|
|
|
|
let entry_height = 0;
|
|
|
|
|
|
|
|
let tx = StorageTransaction::new_mining_proof(
|
|
|
|
&jack,
|
|
|
|
Hash::default(),
|
|
|
|
last_id,
|
|
|
|
entry_height,
|
|
|
|
Signature::default(),
|
|
|
|
);
|
|
|
|
|
|
|
|
bank.process_transaction(&tx).unwrap();
|
|
|
|
|
|
|
|
assert_eq!(get_storage_entry_height(&bank), ENTRIES_PER_SEGMENT);
|
|
|
|
assert_eq!(get_storage_last_id(&bank), storage_last_id);
|
|
|
|
}
|