check unclaimed amount before and after migration of vesting contracts

neutron-org · Aug 14, 2023 · 3e896c3 · 3e896c3
1 parent 5d03f39
commit 3e896c3
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/testcases/run_in_band/tge.auction_migration.test.ts b/src/testcases/run_in_band/tge.auction_migration.test.ts
@@ -1603,6 +1603,7 @@ describe('Neutron / TGE / Auction / Lockdrop migration', () => {
       let claimUsdcLP;
       const votingPowerBeforeLp: Record<string, number> = {};
       let totalUnclaimedAtHeightBeforeMigration: number;
+      let unclaimedAtHeightBeforeMigration: number;
       let unclaimedHeightBeforeMigration: number;
 
       it('should save voting power before migration: lp', async () => {
@@ -1645,6 +1646,12 @@ describe('Neutron / TGE / Auction / Lockdrop migration', () => {
             tge.contracts.vestingAtomLp,
             unclaimedHeightBeforeMigration,
           );
+        unclaimedAtHeightBeforeMigration = await queryUnclaimmedAmountAtHeight(
+          cmInstantiator.chain,
+          tge.contracts.vestingAtomLp,
+          unclaimedHeightBeforeMigration,
+          cmInstantiator.wallet.address.toString(),
+        );
 
         const [
           vestingInfoAtom,
@@ -1885,9 +1892,19 @@ describe('Neutron / TGE / Auction / Lockdrop migration', () => {
             tge.contracts.vestingAtomLp,
             unclaimedHeightBeforeMigration,
           );
+        const unclaimedAmountAfterMigration =
+          await queryUnclaimmedAmountAtHeight(
+            cmInstantiator.chain,
+            tge.contracts.vestingAtomLp,
+            unclaimedHeightBeforeMigration,
+            cmInstantiator.wallet.address.toString(),
+          );
         expect(totalUnclaimedAmountAfterMigration).toEqual(
           totalUnclaimedAtHeightBeforeMigration,
         );
+        expect(unclaimedAmountAfterMigration).toEqual(
+          unclaimedAtHeightBeforeMigration,
+        );
 
         const total = await queryTotalUnclaimedAmountAtHeight(
           cmInstantiator.chain,