blockworks-foundation
/
solana-with-rpc-optimizations
mirror of https://github.com/blockworks-foundation/solana-with-rpc-optimizations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor test truncate data (#15851) 

* refactor test truncate data

* retain existing behavior
This commit is contained in:
Jeff Washington (jwash) 2021-03-17 15:09:26 -05:00 committed by GitHub
parent 59c19d9fbf
commit d30c9ef351
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
programs/bpf_loader/src/lib.rs
View File

@ -1521,6 +1521,11 @@ mod tests {
);
}
fn truncate_data(account: &mut AccountSharedData, len: usize) {
// when account data becomes copy on write, this operation will be more complicated
account.data.truncate(len);
}
#[test]
fn test_bpf_loader_upgradeable_deploy_with_max_len() {
let (genesis_config, mint_keypair) = create_genesis_config(1_000_000_000);
@ -2006,9 +2011,10 @@ mod tests {
// Test Bad ELF data
bank.clear_signatures();
let mut modified_buffer_account = buffer_account;
modified_buffer_account
.data
.truncate(UpgradeableLoaderState::buffer_len(1).unwrap());
truncate_data(
&mut modified_buffer_account,
UpgradeableLoaderState::buffer_len(1).unwrap(),
);
bank.store_account(&buffer_address, &modified_buffer_account);
bank.store_account(&program_keypair.pubkey(), &AccountSharedData::default());
bank.store_account(&programdata_address, &AccountSharedData::default());
@ -2050,7 +2056,7 @@ mod tests {
modified_buffer_account.data_as_mut_slice()
[UpgradeableLoaderState::buffer_data_offset().unwrap()..]
.copy_from_slice(&elf);
modified_buffer_account.data.truncate(5);
truncate_data(&mut modified_buffer_account, 5);
bank.store_account(&buffer_address, &modified_buffer_account);
bank.store_account(&program_keypair.pubkey(), &AccountSharedData::default());
bank.store_account(&programdata_address, &AccountSharedData::default());
@ -2687,7 +2693,7 @@ mod tests {
authority_address: Some(upgrade_authority_address),
})
.unwrap();
buffer_account.borrow_mut().data.truncate(5);
truncate_data(&mut buffer_account.borrow_mut(), 5);
assert_eq!(
Err(InstructionError::InvalidAccountData),
process_instruction(