Merge pull request #214 from neutron-org/feat/disable-vesting-token-c…

…hange-once-set [NTRN-105] feat: add vesting token immutability once set tests
neutron-org · Oct 11, 2023 · a65b977 · a65b977
2 parents 0f677a7 + fb5a3ea
commit a65b977
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/src/testcases/parallel/tge.investors_vesting_vault.test.ts b/src/testcases/parallel/tge.investors_vesting_vault.test.ts
@@ -783,6 +783,23 @@ describe('Neutron / TGE / Investors vesting vault', () => {
         ).rejects.toThrow(/Unauthorized/);
       });
 
+      test('set vesting token not allowed more than once', async () => {
+        await expect(
+          cmManager.executeContract(
+            contractAddresses[INVESTORS_VESTING_CONTRACT_KEY],
+            JSON.stringify({
+              set_vesting_token: {
+                vesting_token: {
+                  native_token: {
+                    denom: IBC_ATOM_DENOM,
+                  },
+                },
+              },
+            }),
+          ),
+        ).rejects.toThrow(/Vesting token is already set!/);
+      });
+
       describe('remove vesting accounts is permissioned', () => {
         test('removal not allowed to a stranger', async () => {
           await expect(

diff --git a/src/testcases/parallel/tge.vesting_lp_vault.test.ts b/src/testcases/parallel/tge.vesting_lp_vault.test.ts
@@ -978,6 +978,38 @@ describe('Neutron / TGE / Vesting LP vault', () => {
         ).rejects.toThrow(/Unauthorized/);
       });
 
+      test('set vesting token not allowed more than once', async () => {
+        await expect(
+          cmManager.executeContract(
+            contractAddresses[VESTING_LP_USDC_CONTRACT_KEY],
+            JSON.stringify({
+              set_vesting_token: {
+                vesting_token: {
+                  native_token: {
+                    denom: NEUTRON_DENOM,
+                  },
+                },
+              },
+            }),
+          ),
+        ).rejects.toThrow(/Vesting token is already set!/);
+
+        await expect(
+          cmManager.executeContract(
+            contractAddresses[VESTING_LP_ATOM_CONTRACT_KEY],
+            JSON.stringify({
+              set_vesting_token: {
+                vesting_token: {
+                  native_token: {
+                    denom: NEUTRON_DENOM,
+                  },
+                },
+              },
+            }),
+          ),
+        ).rejects.toThrow(/Vesting token is already set!/);
+      });
+
       describe('register vesting accounts not allowed to a stranger', () => {
         test('via send cw20', async () => {
           // create a random cw20 token with allocation to user1