add test that initialized accounts cannot be claimed

nssa/src/program.rs

@@ -207,6 +207,15 @@ mod tests {
                 elf: CHAIN_CALLER_ELF.to_vec(),
             }
         }
+
+        pub fn claimer() -> Self {
+            use test_program_methods::{CLAIMER_ELF, CLAIMER_ID};
+
+            Program {
+                id: CLAIMER_ID,
+                elf: CLAIMER_ELF.to_vec(),
+            }
+        }
     }
 
     #[test]

nssa/src/state.rs

@@ -477,6 +477,7 @@ pub mod tests {
             self.insert_program(Program::minter());
             self.insert_program(Program::burner());
             self.insert_program(Program::chain_caller());
+            self.insert_program(Program::claimer());
             self
         }
 
@@ -2214,4 +2215,30 @@ pub mod tests {
         assert_eq!(from_post.balance, initial_balance - amount);
         assert_eq!(to_post, expected_to_post);
     }
+
+    #[test]
+    fn test_claiming_mechanism_cannot_claim_initialied_accounts() {
+        let claimer = Program::claimer();
+        let mut state = V02State::new_with_genesis_accounts(&[], &[]).with_test_programs();
+        let account_id = AccountId::new([2; 32]);
+
+        // Insert an account with non-default program owner
+        state.force_insert_account(
+            account_id,
+            Account {
+                program_owner: [1, 2, 3, 4, 5, 6, 7, 8],
+                ..Account::default()
+            },
+        );
+
+        let message =
+            public_transaction::Message::try_new(claimer.id(), vec![account_id], vec![], ())
+                .unwrap();
+        let witness_set = public_transaction::WitnessSet::for_message(&message, &[]);
+        let tx = PublicTransaction::new(message, witness_set);
+
+        let result = state.transition_from_public_transaction(&tx);
+
+        assert!(matches!(result, Err(NssaError::InvalidProgramBehavior)))
+    }
 }

nssa/test_program_methods/guest/src/bin/claimer.rs

@@ -0,0 +1,19 @@
+use nssa_core::program::{AccountPostState, ProgramInput, read_nssa_inputs, write_nssa_outputs};
+
+type Instruction = ();
+
+fn main() {
+    let ProgramInput {
+        pre_states,
+        instruction: _,
+    } = read_nssa_inputs::<Instruction>();
+
+    let [pre] = match pre_states.try_into() {
+        Ok(array) => array,
+        Err(_) => return,
+    };
+
+    let account_post = AccountPostState::new_claimed(pre.account.clone());
+
+    write_nssa_outputs(vec![pre], vec![account_post]);
+}