test: add mint check tests for migration

Author: Oighty

tests/test-utils.ts

@@ -143,6 +143,7 @@ export async function createMintInstruction(
   defaultAccountState = AccountState.Initialized,
   vault?: PublicKey,
   mintTokens = false,
+  decimals = 6,
 ) {
   // mint size with extensions
   const mintLen = getMintLen([
@@ -164,7 +165,7 @@ export async function createMintInstruction(
     createInitializeScaledUiAmountConfigInstruction(mint, extensionAuth, 1.0, TOKEN_2022_PROGRAM_ID),
     createInitializeDefaultAccountStateInstruction(mint, AccountState.Initialized, TOKEN_2022_PROGRAM_ID),
     createInitializePermanentDelegateInstruction(mint, extensionAuth, TOKEN_2022_PROGRAM_ID),
-    createInitializeMintInstruction(mint, 6, payer.publicKey, payer.publicKey, TOKEN_2022_PROGRAM_ID),
+    createInitializeMintInstruction(mint, decimals, payer.publicKey, payer.publicKey, TOKEN_2022_PROGRAM_ID),
   ];
 
   const tokenAccount = getAssociatedTokenAddressSync(mint, payer.publicKey, false, TOKEN_2022_PROGRAM_ID);

tests/unit/earn.test.ts

@@ -463,6 +463,9 @@ class EarnTest<V extends Variant = Variant.New> {
       this.getEarnGlobalAccount(),
       mint.publicKey,
       AccountState.Frozen,
+      undefined,
+      false,
+      decimals,
     );
 
     let tx = new Transaction();
@@ -1309,6 +1312,40 @@ for (const variant of VARIANTS) {
               .rpc(),
           );
         });
+
+        test('New m mint has different decimals - reverts', async () => {
+          const wrongMint = new Keypair();
+          await $.createMMint(wrongMint, $.nonAdmin, 9);
+
+          await $.expectAnchorError(
+            $.earn.methods
+              .initialize()
+              .accountsPartial({
+                admin: $.admin.publicKey,
+                mMint: wrongMint.publicKey,
+                oldMMint: $.oldMMint!.publicKey,
+              })
+              .signers([$.admin])
+              .rpc(),
+            'ConstraintMintDecimals',
+          );
+        });
+
+        test('New m mint supply is too large - reverts', async () => {
+          await $.mintM($.admin.publicKey, new BN(100_000_000)); // mint 100 m tokens to admin
+
+          await $.expectSystemError(
+            $.earn.methods
+              .initialize()
+              .accountsPartial({
+                admin: $.admin.publicKey,
+                mMint: $.mMint.publicKey,
+                oldMMint: $.oldMMint!.publicKey,
+              })
+              .signers([$.admin])
+              .rpc(),
+          );
+        });
       }
 
       // given the admin signs the transaction