Reduce remaining program crates to boilerplate crates
This commit is contained in:
@@ -1,104 +0,0 @@
|
||||
use bincode::deserialize;
|
||||
use log::*;
|
||||
use solana_runtime::bank::Bank;
|
||||
use solana_sdk::genesis_block::GenesisBlock;
|
||||
use solana_sdk::hash::{hash, Hash};
|
||||
use solana_sdk::pubkey::Pubkey;
|
||||
use solana_sdk::signature::{Keypair, KeypairUtil, Signature};
|
||||
use solana_sdk::system_transaction::SystemTransaction;
|
||||
use solana_storage_api::{StorageTransaction, ENTRIES_PER_SEGMENT};
|
||||
|
||||
fn get_storage_entry_height(bank: &Bank, account: &Pubkey) -> u64 {
|
||||
match bank.get_account(&account) {
|
||||
Some(storage_system_account) => {
|
||||
let state = deserialize(&storage_system_account.data);
|
||||
if let Ok(state) = state {
|
||||
let state: solana_storage_api::StorageProgramState = state;
|
||||
return state.entry_height;
|
||||
}
|
||||
}
|
||||
None => {
|
||||
info!("error in reading entry_height");
|
||||
}
|
||||
}
|
||||
0
|
||||
}
|
||||
|
||||
fn get_storage_blockhash(bank: &Bank, account: &Pubkey) -> Hash {
|
||||
if let Some(storage_system_account) = bank.get_account(&account) {
|
||||
let state = deserialize(&storage_system_account.data);
|
||||
if let Ok(state) = state {
|
||||
let state: solana_storage_api::StorageProgramState = state;
|
||||
return state.hash;
|
||||
}
|
||||
}
|
||||
Hash::default()
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_bank_storage() {
|
||||
let (mut genesis_block, alice) = GenesisBlock::new(1000);
|
||||
genesis_block.native_programs.push((
|
||||
"solana_storage_program".to_string(),
|
||||
solana_storage_api::id(),
|
||||
));
|
||||
let bank = Bank::new(&genesis_block);
|
||||
|
||||
let bob = Keypair::new();
|
||||
let jack = Keypair::new();
|
||||
let jill = Keypair::new();
|
||||
|
||||
let x = 42;
|
||||
let blockhash = genesis_block.hash();
|
||||
let x2 = x * 2;
|
||||
let storage_blockhash = hash(&[x2]);
|
||||
|
||||
bank.register_tick(&blockhash);
|
||||
|
||||
bank.transfer(10, &alice, &jill.pubkey(), blockhash)
|
||||
.unwrap();
|
||||
|
||||
bank.transfer(10, &alice, &bob.pubkey(), blockhash).unwrap();
|
||||
bank.transfer(10, &alice, &jack.pubkey(), blockhash)
|
||||
.unwrap();
|
||||
|
||||
let tx = SystemTransaction::new_program_account(
|
||||
&alice,
|
||||
&bob.pubkey(),
|
||||
blockhash,
|
||||
1,
|
||||
4 * 1024,
|
||||
&solana_storage_api::id(),
|
||||
0,
|
||||
);
|
||||
|
||||
bank.process_transaction(&tx).unwrap();
|
||||
|
||||
let tx = StorageTransaction::new_advertise_recent_blockhash(
|
||||
&bob,
|
||||
storage_blockhash,
|
||||
blockhash,
|
||||
ENTRIES_PER_SEGMENT,
|
||||
);
|
||||
|
||||
bank.process_transaction(&tx).unwrap();
|
||||
|
||||
let entry_height = 0;
|
||||
let tx = StorageTransaction::new_mining_proof(
|
||||
&bob,
|
||||
Hash::default(),
|
||||
blockhash,
|
||||
entry_height,
|
||||
Signature::default(),
|
||||
);
|
||||
let _result = bank.process_transaction(&tx).unwrap();
|
||||
|
||||
assert_eq!(
|
||||
get_storage_entry_height(&bank, &bob.pubkey()),
|
||||
ENTRIES_PER_SEGMENT
|
||||
);
|
||||
assert_eq!(
|
||||
get_storage_blockhash(&bank, &bob.pubkey()),
|
||||
storage_blockhash
|
||||
);
|
||||
}
|
Reference in New Issue
Block a user