processor: assign

Author: buffalojoec

program/src/processor.rs

@@ -255,6 +255,18 @@ fn process_create_account_with_seed(
     )
 }
 
+fn process_assign(accounts: &[AccountInfo], owner: Pubkey) -> ProgramResult {
+    accounts!(
+        accounts,
+        signers,
+        0 => account_info,
+    );
+
+    let address = Address::create(account_info.key, None)?;
+
+    assign(account_info, &address, &owner, &signers)
+}
+
 pub fn process(_program_id: &Pubkey, accounts: &[AccountInfo], input: &[u8]) -> ProgramResult {
     match solana_bincode::limited_deserialize::<SystemInstruction>(input, MAX_INPUT_LEN)
         .map_err(|_| ProgramError::InvalidInstructionData)?
@@ -277,6 +289,10 @@ pub fn process(_program_id: &Pubkey, accounts: &[AccountInfo], input: &[u8]) ->
             msg!("Instruction: CreateAccountWithSeed");
             process_create_account_with_seed(accounts, base, seed, lamports, space, owner)
         }
+        SystemInstruction::Assign { owner } => {
+            msg!("Instruction: Assign");
+            process_assign(accounts, owner)
+        }
         /* TODO: Remaining instruction implementations... */
         _ => Err(ProgramError::InvalidInstructionData),
     }

program/tests/assign.rs

@@ -0,0 +1,61 @@
+mod setup;
+
+use {
+    mollusk_svm::result::Check, solana_account::Account, solana_program_error::ProgramError,
+    solana_pubkey::Pubkey, solana_system_interface::instruction::assign,
+};
+
+const OWNER: Pubkey = Pubkey::new_from_array([8; 32]);
+
+#[test]
+fn fail_account_not_signer() {
+    let mollusk = setup::setup();
+
+    let pubkey = Pubkey::new_unique();
+
+    let mut instruction = assign(&pubkey, &OWNER);
+    instruction.accounts[0].is_signer = false;
+
+    mollusk.process_and_validate_instruction(
+        &instruction,
+        &[(pubkey, Account::default())],
+        &[Check::err(ProgramError::MissingRequiredSignature)],
+    );
+}
+
+#[test]
+fn success() {
+    let mollusk = setup::setup();
+
+    let pubkey = Pubkey::new_unique();
+
+    mollusk.process_and_validate_instruction(
+        &assign(&pubkey, &OWNER),
+        &[(pubkey, Account::default())],
+        &[
+            Check::success(),
+            Check::account(&pubkey).owner(&OWNER).build(),
+        ],
+    );
+}
+
+#[test]
+fn success_already_assigned() {
+    let mollusk = setup::setup();
+
+    let pubkey = Pubkey::new_unique();
+
+    let account = Account {
+        owner: OWNER, // Already assigned
+        ..Account::default()
+    };
+
+    mollusk.process_and_validate_instruction(
+        &assign(&pubkey, &OWNER),
+        &[(pubkey, account)],
+        &[
+            Check::success(),
+            Check::account(&pubkey).owner(&OWNER).build(),
+        ],
+    );
+}