Add feature_set_override parameter to mock_process_instruction() (#24386)

This commit is contained in:
Brooks Prumo
2022-04-15 13:43:04 -05:00
committed by GitHub
parent 34418cb848
commit f33ad34531
6 changed files with 23 additions and 16 deletions

View File

@ -1199,6 +1199,7 @@ pub fn mock_process_instruction(
transaction_accounts: Vec<TransactionAccount>,
instruction_accounts: Vec<AccountMeta>,
sysvar_cache_override: Option<&SysvarCache>,
feature_set_override: Option<Arc<FeatureSet>>,
expected_result: Result<(), InstructionError>,
process_instruction: ProcessInstructionWithContext,
) -> Vec<AccountSharedData> {
@ -1218,6 +1219,9 @@ pub fn mock_process_instruction(
if let Some(sysvar_cache) = sysvar_cache_override {
invoke_context.sysvar_cache = Cow::Borrowed(sysvar_cache);
}
if let Some(feature_set) = feature_set_override {
invoke_context.feature_set = feature_set;
}
let result = invoke_context
.push(
&preparation.instruction_accounts,