Add tests for native mint

program/tests/helpers/common.rs

@@ -247,3 +247,43 @@ pub fn setup_validation_state_account(
         },
     }
 }
+
+pub fn setup_native_token(balance: u64, owner: &TransferAuthority) -> (KeyedAccount, Account) {
+    let native_mint = KeyedAccount {
+        key: spl_token_2022::native_mint::id(),
+        account: setup_mint(
+            TokenProgram::SplToken2022,
+            &Rent::default(),
+            Pubkey::new_unique(),
+        ),
+    };
+
+    let account_size =
+        ExtensionType::try_calculate_account_len::<spl_token_2022::state::Account>(&[]).unwrap();
+    let mut account_data = vec![0; account_size];
+    let mut state = StateWithExtensionsMut::<spl_token_2022::state::Account>::unpack_uninitialized(
+        &mut account_data,
+    )
+    .unwrap();
+
+    state.base = spl_token_2022::state::Account {
+        mint: native_mint.key,
+        amount: balance,
+        owner: owner.keyed_account.key,
+        state: AccountState::Initialized,
+        is_native: COption::Some(20),
+        ..Default::default()
+    };
+    state.pack_base();
+
+    let native_token_account = Account {
+        lamports: Rent::default()
+            .minimum_balance(spl_token_2022::state::Account::LEN)
+            .checked_add(balance)
+            .unwrap(),
+        data: account_data,
+        owner: spl_token_2022::id(),
+        ..Default::default()
+    };
+    (native_mint, native_token_account)
+}

program/tests/test_unwrap.rs

@@ -1,7 +1,7 @@
 use {
     crate::helpers::{
         common::{
-            setup_counter, setup_multisig, setup_transfer_hook_account,
+            setup_counter, setup_multisig, setup_native_token, setup_transfer_hook_account,
             setup_validation_state_account, unwrapped_mint_with_transfer_hook, MINT_SUPPLY,
         },
         create_mint_builder::{CreateMintBuilder, KeyedAccount, TokenProgram},
@@ -319,3 +319,40 @@ fn test_unwrap_with_transfer_hook() {
     let count = unwrap_result.extra_accounts[0].clone().account.data[0];
     assert_eq!(count, 1)
 }
+
+#[test]
+fn test_successfully_unwraps_to_native_mint() {
+    let source_starting_amount = 50_000;
+    let recipient_starting_amount = 25_000;
+    let escrow_starting_amount = 42_000;
+    let unwrap_amount = 40_000;
+
+    let transfer_authority = TransferAuthority {
+        keyed_account: Default::default(),
+        signers: vec![],
+    };
+
+    let (native_mint, native_token_account) =
+        setup_native_token(recipient_starting_amount, &transfer_authority);
+
+    let wrap_result = UnwrapBuilder::default()
+        .unwrapped_token_program(TokenProgram::SplToken2022)
+        .unwrapped_mint(native_mint)
+        .transfer_authority(transfer_authority)
+        .recipient_token_account(native_token_account)
+        .wrapped_token_program(TokenProgram::SplToken)
+        .escrow_starting_amount(escrow_starting_amount)
+        .wrapped_token_starting_amount(source_starting_amount)
+        .recipient_starting_amount(recipient_starting_amount)
+        .unwrap_amount(unwrap_amount)
+        .check(Check::success())
+        .execute();
+
+    assert_unwrap_result(
+        source_starting_amount,
+        recipient_starting_amount,
+        escrow_starting_amount,
+        unwrap_amount,
+        &wrap_result,
+    );
+}

program/tests/test_wrap.rs

@@ -1,7 +1,7 @@
 use {
     crate::helpers::{
         common::{
-            setup_counter, setup_multisig, setup_transfer_hook_account,
+            setup_counter, setup_multisig, setup_native_token, setup_transfer_hook_account,
             setup_validation_state_account, unwrapped_mint_with_transfer_hook, MINT_SUPPLY,
         },
         create_mint_builder::{CreateMintBuilder, KeyedAccount, TokenProgram},
@@ -258,3 +258,28 @@ fn test_wrap_with_transfer_hook() {
     let count = wrap_result.extra_accounts[0].clone().account.data[0];
     assert_eq!(count, 1)
 }
+
+#[test]
+fn test_successfully_wraps_native_mint() {
+    let starting_amount = 50_000;
+    let wrap_amount = 12_555;
+
+    let transfer_authority = TransferAuthority {
+        keyed_account: Default::default(),
+        signers: vec![],
+    };
+
+    let (native_mint, native_token_account) = setup_native_token(wrap_amount, &transfer_authority);
+
+    let wrap_result = WrapBuilder::default()
+        .unwrapped_token_program(TokenProgram::SplToken2022)
+        .unwrapped_token_account(native_token_account)
+        .unwrapped_mint(native_mint)
+        .transfer_authority(transfer_authority)
+        .wrapped_token_program(TokenProgram::SplToken)
+        .recipient_starting_amount(starting_amount)
+        .wrap_amount(wrap_amount)
+        .execute();
+
+    assert_wrap_result(starting_amount, wrap_amount, &wrap_result);
+}