git.delta.rocks / unique-network / refs/commits / 7bc3b32154c9

78  });78  });
7979
80  describe('stake extrinsic', () => {80  describe('stake extrinsic', () => {
81    itSub('should "lock" staking balance, add it to "staked" map, and increase "totalStaked" amount', async ({helper}) => {81    itSub('should "freeze" staking balance, add it to "staked" map, and increase "totalStaked" amount', async ({helper}) => {
82      const [staker, recepient] = await getAccounts(2);82      const [staker, recepient] = await getAccounts(2);
83      const totalStakedBefore = await helper.staking.getTotalStaked();83      const totalStakedBefore = await helper.staking.getTotalStaked();
8484
89      // Staker balance is: frozen: 100, reserved: 0n...89      // Staker balance is: frozen: 100, reserved: 0n...
90      // ...so he can not transfer 90090      // ...so he can not transfer 900
91      expect(await helper.balance.getSubstrateFull(staker.address)).to.contain({frozen: 100n * nominal, reserved: 0n});91      expect(await helper.balance.getSubstrateFull(staker.address)).to.contain({frozen: 100n * nominal, reserved: 0n});
92      expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: 100n * nominal, reasons: 'All'}]);92      expect(await helper.balance.getFrozen(staker.address)).to.deep.eq([{id: 'appstakeappstake', amount: 100n * nominal}]);
93      await expect(helper.balance.transferToSubstrate(staker, recepient.address, 900n * nominal)).to.be.rejectedWith(/^Token: Frozen$/);93      await expect(helper.balance.transferToSubstrate(staker, recepient.address, 900n * nominal)).to.be.rejectedWith(/^Token: Frozen$/);
9494
95      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(100n * nominal);95      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(100n * nominal);
144        }144        }
145      });145      });
146    });146    });
147147    // Now AppPromo makes freezes. Probably this test should be changed\removed.
148    itSub('should allow to stake() if balance is locked with different id', async ({helper}) => {148    itSub.skip('should allow to stake() if balance is locked with different id', async ({helper}) => {
149      const [staker] = await getAccounts(1);149      const [staker] = await getAccounts(1);
150150
151      // staker has tokens locked with vesting id:151      // staker has tokens locked with vesting id:
218218
219        // Right after unstake tokens are still locked219        // Right after unstake tokens are still locked
220        expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.eq(0);220        expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.eq(0);
221        expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: STAKE_AMOUNT, reasons: 'All'}]);221        expect(await helper.balance.getFrozen(staker.address)).to.deep.eq([{id: 'appstakeappstake', amount: STAKE_AMOUNT}]);
222        expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, frozen: STAKE_AMOUNT});222        expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, frozen: STAKE_AMOUNT});
223        // Staker can not transfer223        // Staker can not transfer
224        await expect(helper.balance.transferToSubstrate(staker, recepient.address, 100n * nominal)).to.be.rejectedWith(/^Token: Frozen$/);224        await expect(helper.balance.transferToSubstrate(staker, recepient.address, 100n * nominal)).to.be.rejectedWith(/^Token: Frozen$/);

--- a/tests/src/util/playgrounds/unique.ts
+++ b/tests/src/util/playgrounds/unique.ts
@@ -2404,6 +2404,11 @@
     const locks = (await this.helper.callRpc('api.query.balances.locks', [address])).toHuman();
     return locks.map((lock: any) => { return {id: lock.id, amount: BigInt(lock.amount.replace(/,/g, '')), reasons: lock.reasons}; });
   }
+
+  async getFrozen(address: TSubstrateAccount): Promise<[{ id: string, amount: bigint}]> {
+    const locks = (await this.helper.callRpc('api.query.balances.freezes', [address])).toHuman();
+    return locks.map((lock: any) => { return {id: lock.id, amount: BigInt(lock.amount.replace(/,/g, ''))}; });
+  }
 }
 
 class EthereumBalanceGroup<T extends ChainHelperBase> extends HelperGroup<T> {
@@ -2497,6 +2502,15 @@
   }
 
   /**
+   * Get frozen balances
+   * @param address substrate address
+   * @returns frozen balances with id via api.query.balances.freezes
+   */
+  getFrozen(address: TSubstrateAccount) {
+    return this.subBalanceGroup.getFrozen(address);
+  }
+
+  /**
    * Get ethereum address balance
    * @param address ethereum address
    * @example getEthereum("0x9F0583DbB855d...")