1234567891011121314151617import {IKeyringPair} from '@polkadot/types/types';18import {itSub, usingPlaygrounds, expect} from './util';1920describe('Vesting', () => {21 let donor: IKeyringPair;22 let nominal: bigint;2324 before(async () => {25 await usingPlaygrounds(async (helper, privateKey) => {26 donor = await privateKey({filename: __filename});27 nominal = helper.balance.getOneTokenNominal();28 });29 });3031 itSub.only('can perform vestedTransfer and claim tokens', async ({helper}) => {32 33 const [sender, recepient] = await helper.arrange.createAccounts([1000n, 1n], donor);34 const currentRelayBlock = await helper.chain.getRelayBlockNumber();35 const schedule1 = {startRelayBlock: currentRelayBlock + 4n, periodBlocks: 4n, periodCount: 2n, perPeriod: 50n * nominal};36 const schedule2 = {startRelayBlock: currentRelayBlock + 8n, periodBlocks: 8n, periodCount: 2n, perPeriod: 100n * nominal};3738 39 await helper.balance.vestedTransfer(sender, recepient.address, schedule1);40 await helper.balance.vestedTransfer(sender, recepient.address, schedule2);41 let schedule = await helper.balance.getVestingSchedules(recepient.address);4243 44 let balanceSender = await helper.balance.getSubstrateFull(sender.address);45 expect(balanceSender.free / nominal).to.eq(699n);46 expect(balanceSender.feeFrozen).to.eq(0n);47 expect(balanceSender.miscFrozen).to.eq(0n);48 expect(balanceSender.reserved).to.eq(0n);4950 51 let balanceRecepient = await helper.balance.getSubstrateFull(recepient.address);52 expect(balanceRecepient.free).to.eq(301n * nominal);53 expect(balanceRecepient.feeFrozen).to.eq(300n * nominal);54 expect(balanceRecepient.miscFrozen).to.eq(300n * nominal);55 expect(balanceRecepient.reserved).to.eq(0n);5657 58 expect(schedule).to.has.length(2);59 expect(schedule[0]).to.deep.eq(schedule1);60 expect(schedule[1]).to.deep.eq(schedule2);6162 await helper.wait.forRelayBlockNumber(currentRelayBlock + 8n);63 await helper.balance.claim(recepient);6465 66 balanceRecepient = await helper.balance.getSubstrateFull(recepient.address);67 expect(balanceRecepient.free / nominal).to.eq(300n);68 expect(balanceRecepient.feeFrozen).to.eq(250n * nominal);69 expect(balanceRecepient.miscFrozen).to.eq(250n * nominal);70 expect(balanceRecepient.reserved).to.eq(0n);71 72 await helper.wait.forRelayBlockNumber(currentRelayBlock + 16n);73 await helper.balance.claim(recepient);7475 76 balanceRecepient = await helper.balance.getSubstrateFull(recepient.address);77 expect(balanceRecepient.free / nominal).to.eq(300n);78 expect(balanceRecepient.feeFrozen).to.eq(100n * nominal);79 expect(balanceRecepient.miscFrozen).to.eq(100n * nominal);80 expect(balanceRecepient.reserved).to.eq(0n);81 82 83 schedule = await helper.balance.getVestingSchedules(recepient.address);84 expect(schedule).to.has.length(1);85 expect(schedule[0]).to.deep.eq(schedule2);8687 await helper.wait.forRelayBlockNumber(currentRelayBlock + 24n);88 await helper.balance.claim(recepient);8990 91 balanceRecepient = await helper.balance.getSubstrateFull(recepient.address);92 expect(balanceRecepient.free / nominal).to.eq(300n);93 expect(balanceRecepient.feeFrozen).to.eq(0n);94 expect(balanceRecepient.miscFrozen).to.eq(0n);95 expect(balanceRecepient.reserved).to.eq(0n);96 97 98 balanceSender = await helper.balance.getSubstrateFull(sender.address);99 expect(balanceSender.free / nominal).to.eq(699n);100 expect(balanceSender.feeFrozen).to.eq(0n);101 expect(balanceSender.miscFrozen).to.eq(0n);102 expect(balanceSender.reserved).to.eq(0n);103 });104});