token 2022: add InitializeMember instruction from SPL Token Group i…

…nterface
solana-labs · Nov 16, 2023 · 9ab3658 · 9ab3658
1 parent 5f75990
commit 9ab3658
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 29 deletions.
diff --git a/token/client/src/token.rs b/token/client/src/token.rs
@@ -1,3 +1,5 @@
+use spl_token_group_interface::state::TokenGroupMember;
+
 use {
     crate::{
         client::{ProgramClient, ProgramClientError, SendTransaction, SimulateTransaction},
@@ -3834,26 +3836,6 @@ where
         .await
     }
 
-    async fn get_additional_rent_for_new_member(&self) -> TokenResult<u64> {
-        let account = self.get_account(self.pubkey).await?;
-        let account_lamports = account.lamports;
-        let mint_state = self.unpack_mint_info(account)?;
-        let new_account_len = mint_state
-            .try_get_account_len()?
-            .checked_add(ExtensionType::try_calculate_account_len::<Mint>(&[
-                ExtensionType::TokenGroupMember,
-            ])?)
-            .ok_or(TokenError::Program(
-                spl_token_2022::error::TokenError::Overflow.into(),
-            ))?;
-        let new_rent_exempt_minimum = self
-            .client
-            .get_minimum_balance_for_rent_exemption(new_account_len)
-            .await
-            .map_err(TokenError::Client)?;
-        Ok(new_rent_exempt_minimum.saturating_sub(account_lamports))
-    }
-
     /// Initialize a token-group member on a mint
     #[allow(clippy::too_many_arguments)]
     pub async fn token_group_initialize_member_with_rent_transfer<S: Signers>(
@@ -3864,7 +3846,9 @@ where
         group_update_authority: &Pubkey,
         signing_keypairs: &S,
     ) -> TokenResult<T::Output> {
-        let additional_lamports = self.get_additional_rent_for_new_member().await?;
+        let additional_lamports = self
+            .get_additional_rent_for_fixed_len_extension::<TokenGroupMember>()
+            .await?;
         let mut instructions = vec![];
         if additional_lamports > 0 {
             instructions.push(system_instruction::transfer(

diff --git a/token/program-2022-test/tests/token_group_initialize_member.rs b/token/program-2022-test/tests/token_group_initialize_member.rs
@@ -246,24 +246,19 @@ async fn success_initialize_member() {
     // fail double-init
     let error = member1_token_context
         .token
-        .token_group_initialize_member_with_rent_transfer(
-            &payer.pubkey(),
+        .token_group_initialize_member(
             &member1_token_context.mint_authority.pubkey(),
             &group_mint_keypair.pubkey(),
             &group_authority.pubkey(),
-            &[
-                &payer,
-                &member1_token_context.mint_authority,
-                &group_authority,
-            ],
+            &[&member1_token_context.mint_authority, &group_authority],
         )
         .await
         .unwrap_err();
     assert_eq!(
         error,
         TokenClientError::Client(Box::new(TransportError::TransactionError(
             TransactionError::InstructionError(
-                1,
+                0,
                 InstructionError::Custom(TokenError::ExtensionAlreadyInitialized as u32)
             )
         )))