Avoid changing signature of ProgramTest::add_account

This commit is contained in:
Michael Vines 2021-03-15 23:29:40 -07:00
parent 82269f1351
commit 03180b502d
1 changed files with 9 additions and 8 deletions

View File

@ -481,8 +481,9 @@ impl ProgramTest {
} }
/// Add an account to the test environment /// Add an account to the test environment
pub fn add_account(&mut self, address: Pubkey, account: AccountSharedData) { pub fn add_account(&mut self, address: Pubkey, account: Account) {
self.accounts.push((address, account)); self.accounts
.push((address, AccountSharedData::from(account)));
} }
/// Add an account to the test environment with the account data in the provided `filename` /// Add an account to the test environment with the account data in the provided `filename`
@ -495,7 +496,7 @@ impl ProgramTest {
) { ) {
self.add_account( self.add_account(
address, address,
AccountSharedData::from(Account { Account {
lamports, lamports,
data: read_file(find_file(filename).unwrap_or_else(|| { data: read_file(find_file(filename).unwrap_or_else(|| {
panic!("Unable to locate {}", filename); panic!("Unable to locate {}", filename);
@ -503,7 +504,7 @@ impl ProgramTest {
owner, owner,
executable: false, executable: false,
rent_epoch: 0, rent_epoch: 0,
}), },
); );
} }
@ -518,14 +519,14 @@ impl ProgramTest {
) { ) {
self.add_account( self.add_account(
address, address,
AccountSharedData::from(Account { Account {
lamports, lamports,
data: base64::decode(data_base64) data: base64::decode(data_base64)
.unwrap_or_else(|err| panic!("Failed to base64 decode: {}", err)), .unwrap_or_else(|err| panic!("Failed to base64 decode: {}", err)),
owner, owner,
executable: false, executable: false,
rent_epoch: 0, rent_epoch: 0,
}), },
); );
} }
@ -581,13 +582,13 @@ impl ProgramTest {
self.add_account( self.add_account(
program_id, program_id,
AccountSharedData::from(Account { Account {
lamports: Rent::default().minimum_balance(data.len()).min(1), lamports: Rent::default().minimum_balance(data.len()).min(1),
data, data,
owner: loader, owner: loader,
executable: true, executable: true,
rent_epoch: 0, rent_epoch: 0,
}), },
); );
} else { } else {
info!("\"{}\" program loaded as native code", program_name); info!("\"{}\" program loaded as native code", program_name);