Expand CPI returning error test (#13672)

2020-11-18 14:18:49 -08:00
parent f25c969ad8
commit 15057b087c
2 changed files with 19 additions and 1 deletions
--- a/programs/bpf/rust/invoked/src/processor.rs
+++ b/programs/bpf/rust/invoked/src/processor.rs
@ -107,6 +107,14 @@ fn process_instruction(
        }
        TEST_RETURN_ERROR => {
            info!("return error");
+            const ARGUMENT_INDEX: usize = 0;
+
+            // modify lamports that should be dropped
+            assert_eq!(10, **accounts[ARGUMENT_INDEX].try_borrow_lamports()?);
+            **accounts[ARGUMENT_INDEX].try_borrow_mut_lamports()? += 1;
+            // modify data that should be dropped
+            assert_eq!(0, accounts[ARGUMENT_INDEX].try_borrow_mut_data()?[0]);
+            accounts[ARGUMENT_INDEX].try_borrow_mut_data()?[0] = 1;
            return Err(ProgramError::Custom(42));
        }
        TEST_DERIVED_SIGNERS => {