git.delta.rocks / unique-network / refs/commits / e5f3450abd04

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617import {IKeyringPair} from '@polkadot/types/types';18import {19  itSub, usingPlaygrounds, Pallets, requirePalletsOrSkip, LOCKING_PERIOD, UNLOCKING_PERIOD,20} from '../../util';21import {DevUniqueHelper} from '../../util/playgrounds/unique.dev';22import {itEth, expect, SponsoringMode} from '../../eth/util';2324let donor: IKeyringPair;25let palletAdmin: IKeyringPair;26let nominal: bigint;27let palletAddress: string;28let accounts: IKeyringPair[];29// App promotion periods:30// LOCKING_PERIOD = 12 blocks of relay31// UNLOCKING_PERIOD = 6 blocks of parachain3233describe('App promotion', () => {34  before(async function () {35    await usingPlaygrounds(async (helper, privateKey) => {36      requirePalletsOrSkip(this, helper, [Pallets.AppPromotion]);37      donor = await privateKey({filename: __filename});38      palletAddress = helper.arrange.calculatePalletAddress('appstake');39      palletAdmin = await privateKey('//PromotionAdmin');40      nominal = helper.balance.getOneTokenNominal();4142      const accountBalances = new Array(100);43      accountBalances.fill(1000n);44      accounts = await helper.arrange.createAccounts(accountBalances, donor); // create accounts-pool to speed up tests45    });46  });4748  describe('stake extrinsic', () => {49    itSub('should "lock" staking balance, add it to "staked" map, and increase "totalStaked" amount', async ({helper}) => {50      const [staker, recepient] = [accounts.pop()!, accounts.pop()!];51      const totalStakedBefore = await helper.staking.getTotalStaked();5253      // Minimum stake amount is 100:54      await expect(helper.staking.stake(staker, 100n * nominal - 1n)).to.be.rejected;55      await helper.staking.stake(staker, 100n * nominal);5657      // Staker balance is: miscFrozen: 100, feeFrozen: 100, reserved: 0n...58      // ...so he can not transfer 90059      expect(await helper.balance.getSubstrateFull(staker.address)).to.contain({miscFrozen: 100n * nominal, feeFrozen: 100n * nominal, reserved: 0n});60      expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: 100n * nominal, reasons: 'All'}]);61      await expect(helper.balance.transferToSubstrate(staker, recepient.address, 900n * nominal)).to.be.rejectedWith('balances.LiquidityRestrictions');6263      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(100n * nominal);64      expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);65      // it is potentially flaky test. Promotion can credited some tokens. Maybe we need to use closeTo?66      expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore + 100n * nominal); // total tokens amount staked in app-promotion increased676869      await helper.staking.stake(staker, 200n * nominal);70      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(300n * nominal);71      const totalStakedPerBlock = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});72      expect(totalStakedPerBlock[0].amount).to.equal(100n * nominal);73      expect(totalStakedPerBlock[1].amount).to.equal(200n * nominal);74    });7576    itSub('should allow to create maximum 10 stakes for account', async ({helper}) => {77      const [staker] = await helper.arrange.createAccounts([2000n], donor);78      for (let i = 0; i < 10; i++) {79        await helper.staking.stake(staker, 100n * nominal);80      }8182      // can have 10 stakes83      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(1000n * nominal);84      expect(await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).to.have.length(10);8586      await expect(helper.staking.stake(staker, 100n * nominal)).to.be.rejectedWith('appPromotion.NoPermission');8788      // After unstake can stake again89      await helper.staking.unstake(staker);90      await helper.staking.stake(staker, 100n * nominal);91      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.equal(100n * nominal);92    });9394    itSub('should reject transaction if stake amount is more than total free balance minus frozen', async ({helper}) => {95      const staker = accounts.pop()!;9697      // Can't stake full balance because Alice needs to pay some fee98      await expect(helper.staking.stake(staker, 1000n * nominal)).to.be.rejected; // With('Arithmetic')99      await helper.staking.stake(staker, 500n * nominal);100101      // Can't stake 500 tkn because Alice has Less than 500 transferable;102      await expect(helper.staking.stake(staker, 500n * nominal)).to.be.rejected; // With('Arithmetic');103      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(500n * nominal);104    });105106    itSub('for different accounts in one block is possible', async ({helper}) => {107      const crowd = [accounts.pop()!, accounts.pop()!, accounts.pop()!, accounts.pop()!];108109      const crowdStartsToStake = crowd.map(user => helper.staking.stake(user, 100n * nominal));110      await expect(Promise.all(crowdStartsToStake)).to.be.fulfilled;111112      const crowdStakes = await Promise.all(crowd.map(address => helper.staking.getTotalStaked({Substrate: address.address})));113      expect(crowdStakes).to.deep.equal([100n * nominal, 100n * nominal, 100n * nominal, 100n * nominal]);114    });115  });116117  describe('unstake extrinsic', () => {118    itSub('should change balance state from "frozen" to "reserved", add it to "pendingUnstake" map, and subtract it from totalStaked', async ({helper}) => {119      const [staker, recepient] = [accounts.pop()!, accounts.pop()!];120      const totalStakedBefore = await helper.staking.getTotalStaked();121      await helper.staking.stake(staker, 900n * nominal);122      await helper.staking.unstake(staker);123124      // Right after unstake balance is reserved125      // Staker can not transfer126      expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 900n * nominal, feeFrozen: 900n * nominal});127      await expect(helper.balance.transferToSubstrate(staker, recepient.address, 100n * nominal)).to.be.rejectedWith('balances.LiquidityRestrictions');128      expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(900n * nominal);129      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);130      expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore);131    });132133    itSub('should unlock balance after unlocking period ends and remove it from "pendingUnstake"', async ({helper}) => {134      const staker = accounts.pop()!;135      await helper.staking.stake(staker, 100n * nominal);136      await helper.staking.unstake(staker);137      const [pendingUnstake] = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});138139      // Wait for unstaking period. Balance now free ~1000; reserved, frozen, miscFrozeb: 0n140      await helper.wait.forParachainBlockNumber(pendingUnstake.block);141      expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 0n, feeFrozen: 0n});142      expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);143144      // staker can transfer:145      await helper.balance.transferToSubstrate(staker, donor.address, 998n * nominal);146      expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(1n);147    });148149    itSub('should successfully unstake multiple stakes', async ({helper}) => {150      const staker = accounts.pop()!;151      await helper.staking.stake(staker, 100n * nominal);152      await helper.staking.stake(staker, 200n * nominal);153      await helper.staking.stake(staker, 300n * nominal);154155      // staked: [100, 200, 300]; unstaked: 0156      let totalPendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});157      let pendingUnstake = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});158      let stakes = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});159      expect(totalPendingUnstake).to.be.deep.equal(0n);160      expect(pendingUnstake).to.be.deep.equal([]);161      expect(stakes[0].amount).to.equal(100n * nominal);162      expect(stakes[1].amount).to.equal(200n * nominal);163      expect(stakes[2].amount).to.equal(300n * nominal);164165      // Can unstake multiple stakes166      await helper.staking.unstake(staker);167      pendingUnstake = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});168      totalPendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});169      stakes = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});170      expect(totalPendingUnstake).to.be.equal(600n * nominal);171      expect(stakes).to.be.deep.equal([]);172      expect(pendingUnstake[0].amount).to.equal(600n * nominal);173174      expect (await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, feeFrozen: 600n * nominal, miscFrozen: 600n * nominal});175      expect (await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);176      await helper.wait.forParachainBlockNumber(pendingUnstake[0].block);177      expect (await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, feeFrozen: 0n, miscFrozen: 0n});178      expect (await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);179    });180181    itSub('should not have any effects if no active stakes', async ({helper}) => {182      const staker = accounts.pop()!;183184      // unstake has no effect if no stakes at all185      await helper.staking.unstake(staker);186      expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(0n);187      expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n); // TODO bigint closeTo helper188189      // TODO stake() unstake() waitUnstaked() unstake();190191      // can't unstake if there are only pendingUnstakes192      await helper.staking.stake(staker, 100n * nominal);193      await helper.staking.unstake(staker);194      await helper.staking.unstake(staker);195196      expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(100n * nominal);197      expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);198    });199200    itSub('should keep different unlocking block for each unlocking stake', async ({helper}) => {201      const staker = accounts.pop()!;202      await helper.staking.stake(staker, 100n * nominal);203      await helper.staking.unstake(staker);204      await helper.staking.stake(staker, 120n * nominal);205      await helper.staking.unstake(staker);206207      const unstakingPerBlock = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});208      expect(unstakingPerBlock).has.length(2);209      expect(unstakingPerBlock[0].amount).to.equal(100n * nominal);210      expect(unstakingPerBlock[1].amount).to.equal(120n * nominal);211    });212213    itSub('should be possible for different accounts in one block', async ({helper}) => {214      const stakers = [accounts.pop()!, accounts.pop()!, accounts.pop()!];215216      await Promise.all(stakers.map(staker => helper.staking.stake(staker, 100n * nominal)));217      await Promise.all(stakers.map(staker => helper.staking.unstake(staker)));218219      await Promise.all(stakers.map(async (staker) => {220        expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(100n * nominal);221        expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);222      }));223    });224  });225226  describe('collection sponsoring', () => {227    itSub('should actually sponsor transactions', async ({helper}) => {228      const api = helper.getApi();229      const [collectionOwner, tokenSender, receiver] = [accounts.pop()!, accounts.pop()!, accounts.pop()!];230      const collection = await helper.nft.mintCollection(collectionOwner, {name: 'Name', description: 'Description', tokenPrefix: 'Prefix', limits: {sponsorTransferTimeout: 0}});231      const token = await collection.mintToken(collectionOwner, {Substrate: tokenSender.address});232      await helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId));233      const palletBalanceBefore = await helper.balance.getSubstrate(palletAddress);234235      await token.transfer(tokenSender, {Substrate: receiver.address});236      expect (await token.getOwner()).to.be.deep.equal({Substrate: receiver.address});237      const palletBalanceAfter = await helper.balance.getSubstrate(palletAddress);238239      // senders balance the same, transaction has sponsored240      expect (await helper.balance.getSubstrate(tokenSender.address)).to.be.equal(1000n * nominal);241      expect (palletBalanceBefore > palletBalanceAfter).to.be.true;242    });243244    itSub('can not be set by non admin', async ({helper}) => {245      const api = helper.getApi();246      const [collectionOwner, nonAdmin] = [accounts.pop()!, accounts.pop()!];247248      const collection  = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});249250      await expect(helper.signTransaction(nonAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.rejected;251      expect((await collection.getData())?.raw.sponsorship).to.equal('Disabled');252    });253254    itSub('should set pallet address as confirmed admin', async ({helper}) => {255      const api = helper.getApi();256      const [collectionOwner, oldSponsor] = [accounts.pop()!, accounts.pop()!];257258      // Can set sponsoring for collection without sponsor259      const collectionWithoutSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'No-sponsor', description: 'New Collection', tokenPrefix: 'Promotion'});260      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithoutSponsor.collectionId))).to.be.fulfilled;261      expect((await collectionWithoutSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});262263      // Can set sponsoring for collection with unconfirmed sponsor264      const collectionWithUnconfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Unconfirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});265      expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: oldSponsor.address});266      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithUnconfirmedSponsor.collectionId))).to.be.fulfilled;267      expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});268269      // Can set sponsoring for collection with confirmed sponsor270      const collectionWithConfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Confirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});271      await collectionWithConfirmedSponsor.confirmSponsorship(oldSponsor);272      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithConfirmedSponsor.collectionId))).to.be.fulfilled;273      expect((await collectionWithConfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});274    });275276    itSub('can be overwritten by collection owner', async ({helper}) => {277      const api = helper.getApi();278      const [collectionOwner, newSponsor] = [accounts.pop()!, accounts.pop()!];279      const collection  = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});280      const collectionId = collection.collectionId;281282      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionId))).to.be.fulfilled;283284      // Collection limits still can be changed by the owner285      expect(await collection.setLimits(collectionOwner, {sponsorTransferTimeout: 0})).to.be.true;286      expect((await collection.getData())?.raw.limits.sponsorTransferTimeout).to.be.equal(0);287      expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});288289      // Collection sponsor can be changed too290      expect((await collection.setSponsor(collectionOwner, newSponsor.address))).to.be.true;291      expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: newSponsor.address});292    });293294    itSub('should not overwrite collection limits set by the owner earlier', async ({helper}) => {295      const api = helper.getApi();296      const limits = {ownerCanDestroy: true, ownerCanTransfer: true, sponsorTransferTimeout: 0};297      const collectionWithLimits = await helper.nft.mintCollection(accounts.pop()!, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits});298299      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithLimits.collectionId))).to.be.fulfilled;300      expect((await collectionWithLimits.getData())?.raw.limits).to.be.deep.contain(limits);301    });302303    itSub('should reject transaction if collection doesn\'t exist', async ({helper}) => {304      const api = helper.getApi();305      const collectionOwner = accounts.pop()!;306307      // collection has never existed308      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(999999999))).to.be.rejected;309      // collection has been burned310      const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});311      await collection.burn(collectionOwner);312313      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.rejected;314    });315  });316317  describe('stopSponsoringCollection', () => {318    itSub('can not be called by non-admin', async ({helper}) => {319      const api = helper.getApi();320      const [collectionOwner, nonAdmin] = [accounts.pop()!, accounts.pop()!];321      const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});322323      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.fulfilled;324325      await expect(helper.signTransaction(nonAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId))).to.be.rejected;326      expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});327    });328329    itSub('should set sponsoring as disabled', async ({helper}) => {330      const api = helper.getApi();331      const [collectionOwner, recepient] = [accounts.pop()!, accounts.pop()!];332      const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits: {sponsorTransferTimeout: 0}});333      const token = await collection.mintToken(collectionOwner, {Substrate: collectionOwner.address});334335      await helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId));336      await helper.signTransaction(palletAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId));337338      expect((await collection.getData())?.raw.sponsorship).to.be.equal('Disabled');339340      // Transactions are not sponsored anymore:341      const ownerBalanceBefore = await helper.balance.getSubstrate(collectionOwner.address);342      await token.transfer(collectionOwner, {Substrate: recepient.address});343      const ownerBalanceAfter = await helper.balance.getSubstrate(collectionOwner.address);344      expect(ownerBalanceAfter < ownerBalanceBefore).to.be.equal(true);345    });346347    itSub('should not affect collection which is not sponsored by pallete', async ({helper}) => {348      const api = helper.getApi();349      const collectionOwner = accounts.pop()!;350      const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: collectionOwner.address});351      await collection.confirmSponsorship(collectionOwner);352353      await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId))).to.be.rejected;354355      expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: collectionOwner.address});356    });357358    itSub('should reject transaction if collection does not exist', async ({helper}) => {359      const collectionOwner = accounts.pop()!;360      const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});361362      await collection.burn(collectionOwner);363      await expect(helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringCollection', [collection.collectionId], true)).to.be.rejectedWith('common.CollectionNotFound');364      await expect(helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringCollection', [999_999_999], true)).to.be.rejectedWith('common.CollectionNotFound');365    });366  });367368  describe('contract sponsoring', () => {369    itEth('should set palletes address as a sponsor', async ({helper}) => {370      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();371      const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);372      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);373374      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address]);375376      expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;377      expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);378      expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({379        confirmed: {380          substrate: palletAddress,381        },382      });383    });384385    itEth('should overwrite sponsoring mode and existed sponsor', async ({helper}) => {386      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();387      const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);388      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);389390      await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;391392      // Contract is self sponsored393      expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.be.deep.equal({394        confirmed: {395          ethereum: flipper.options.address.toLowerCase(),396        },397      });398399      // set promotion sponsoring400      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);401402      // new sponsor is pallet address403      expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;404      expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);405      expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({406        confirmed: {407          substrate: palletAddress,408        },409      });410    });411412    itEth('can be overwritten by contract owner', async ({helper}) => {413      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();414      const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);415      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);416417      // contract sponsored by pallet418      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);419420      // owner sets self sponsoring421      await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.not.rejected;422423      expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;424      expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);425      expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({426        confirmed: {427          ethereum: flipper.options.address.toLowerCase(),428        },429      });430    });431432    itEth('can not be set by non admin', async ({helper}) => {433      const nonAdmin = accounts.pop()!;434      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();435      const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);436      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);437438      await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;439440      // nonAdmin calls sponsorContract441      await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true)).to.be.rejectedWith('appPromotion.NoPermission');442443      // contract still self-sponsored444      expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({445        confirmed: {446          ethereum: flipper.options.address.toLowerCase(),447        },448      });449    });450451    itEth('should actually sponsor transactions', async ({helper}) => {452      // Contract caller453      const caller = await helper.eth.createAccountWithBalance(donor, 1000n);454      const palletBalanceBefore = await helper.balance.getSubstrate(palletAddress);455456      // Deploy flipper457      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();458      const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);459      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);460461      // Owner sets to sponsor every tx462      await contractHelper.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: contractOwner});463      await contractHelper.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: contractOwner});464      await helper.eth.transferBalanceFromSubstrate(donor, flipper.options.address, 1000n); // transferBalanceToEth(api, alice, flipper.options.address, 1000n);465466      // Set promotion to the Flipper467      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);468469      // Caller calls Flipper470      await flipper.methods.flip().send({from: caller});471      expect(await flipper.methods.getValue().call()).to.be.true;472473      // The contracts and caller balances have not changed474      const callerBalance = await helper.balance.getEthereum(caller);475      const contractBalanceAfter = await helper.balance.getEthereum(flipper.options.address);476      expect(callerBalance).to.be.equal(1000n * nominal);477      expect(1000n * nominal === contractBalanceAfter).to.be.true;478479      // The pallet balance has decreased480      const palletBalanceAfter = await helper.balance.getSubstrate(palletAddress);481      expect(palletBalanceAfter < palletBalanceBefore).to.be.true;482    });483  });484485  describe('stopSponsoringContract', () => {486    itEth('should remove pallet address from contract sponsors', async ({helper}) => {487      const caller = await helper.eth.createAccountWithBalance(donor, 1000n);488      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();489      const flipper = await helper.eth.deployFlipper(contractOwner);490      await helper.eth.transferBalanceFromSubstrate(donor, flipper.options.address);491      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);492493      await contractHelper.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: contractOwner});494      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);495      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address], true);496497      expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.false;498      expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);499      expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({500        disabled: null,501      });502503      await flipper.methods.flip().send({from: caller});504      expect(await flipper.methods.getValue().call()).to.be.true;505506      const callerBalance = await helper.balance.getEthereum(caller);507      const contractBalanceAfter = await helper.balance.getEthereum(flipper.options.address);508509      // caller payed for call510      expect(1000n * nominal > callerBalance).to.be.true;511      expect(contractBalanceAfter).to.be.equal(100n * nominal);512    });513514    itEth('can not be called by non-admin', async ({helper}) => {515      const nonAdmin = accounts.pop()!;516      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();517      const flipper = await helper.eth.deployFlipper(contractOwner);518519      await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address]);520      await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address]))521        .to.be.rejectedWith(/appPromotion\.NoPermission/);522    });523524    itEth('should not affect a contract which is not sponsored by pallete', async ({helper}) => {525      const nonAdmin = accounts.pop()!;526      const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();527      const flipper = await helper.eth.deployFlipper(contractOwner);528      const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);529      await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;530531      await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address], true)).to.be.rejectedWith('appPromotion.NoPermission');532    });533  });534535  describe('payoutStakers', () => {536    itSub('can not be called by non admin', async ({helper}) => {537      const nonAdmin = accounts.pop()!;538      await expect(helper.admin.payoutStakers(nonAdmin, 100)).to.be.rejectedWith('appPromotion.NoPermission');539    });540541    itSub('should increase total staked', async ({helper}) => {542      const staker = accounts.pop()!;543      const totalStakedBefore = await helper.staking.getTotalStaked();544      await helper.staking.stake(staker, 100n * nominal);545546      // Wait for rewards and pay547      const [stakedInBlock] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});548      await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stakedInBlock.block));549      const totalPayout = (await helper.admin.payoutStakers(palletAdmin, 100)).reduce((prev, payout) => prev + payout.payout, 0n);550551      const totalStakedAfter = await helper.staking.getTotalStaked();552      expect(totalStakedAfter).to.equal(totalStakedBefore + (100n * nominal) + totalPayout);553      // staker can unstake554      await helper.staking.unstake(staker);555      expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedAfter - calculateIncome(100n * nominal));556    });557558    itSub('should credit 0.05% for staking period', async ({helper}) => {559      const staker = accounts.pop()!;560561      await waitPromotionPeriodDoesntEnd(helper);562563      await helper.staking.stake(staker, 100n * nominal);564      await helper.staking.stake(staker, 200n * nominal);565566      // wait rewards are available:567      const [_stake1, stake2] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});568      await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake2.block));569570      const payoutToStaker = (await helper.admin.payoutStakers(palletAdmin, 100)).find((payout) => payout.staker === staker.address)?.payout;571      expect(payoutToStaker + 300n * nominal).to.equal(calculateIncome(300n * nominal));572573      const totalStakedPerBlock = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});574      const income1 = calculateIncome(100n * nominal);575      const income2 = calculateIncome(200n * nominal);576      expect(totalStakedPerBlock[0].amount).to.equal(income1);577      expect(totalStakedPerBlock[1].amount).to.equal(income2);578579      const stakerBalance = await helper.balance.getSubstrateFull(staker.address);580      expect(stakerBalance).to.contain({miscFrozen: income1 + income2, feeFrozen: income1 + income2, reserved: 0n});581      expect(stakerBalance.free / nominal).to.eq(999n);582    });583584    itSub('shoud be paid for more than one period if payments was missed', async ({helper}) => {585      const staker = accounts.pop()!;586587      await helper.staking.stake(staker, 100n * nominal);588      // wait for two rewards are available:589      let [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});590      await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);591592      await helper.admin.payoutStakers(palletAdmin, 100);593      [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});594      const frozenBalanceShouldBe = calculateIncome(100n * nominal, 2);595      expect(stake.amount).to.be.equal(frozenBalanceShouldBe);596597      const stakerFullBalance = await helper.balance.getSubstrateFull(staker.address);598599      expect(stakerFullBalance).to.contain({reserved: 0n, feeFrozen: frozenBalanceShouldBe, miscFrozen: frozenBalanceShouldBe});600    });601602    itSub('should not be credited for unstaked (reserved) balance', async ({helper}) => {603      // staker unstakes before rewards has been payed604      const staker = accounts.pop()!;605      await helper.staking.stake(staker, 100n * nominal);606      const [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});607      await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);608      await helper.staking.unstake(staker);609610      // so he did not receive any rewards611      const totalBalanceBefore = await helper.balance.getSubstrate(staker.address);612      await helper.admin.payoutStakers(palletAdmin, 100);613      const totalBalanceAfter = await helper.balance.getSubstrate(staker.address);614615      expect(totalBalanceBefore).to.be.equal(totalBalanceAfter);616    });617618    itSub('should bring compound interest', async ({helper}) => {619      const staker = accounts.pop()!;620621      await helper.staking.stake(staker, 100n * nominal);622623      let [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});624      await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block));625626      await helper.admin.payoutStakers(palletAdmin, 100);627      [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});628      expect(stake.amount).to.equal(calculateIncome(100n * nominal));629630      await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);631      await helper.admin.payoutStakers(palletAdmin, 100);632      [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});633      expect(stake.amount).to.equal(calculateIncome(100n * nominal, 2));634    });635636    itSub.skip('can be paid 1000 rewards in a time', async ({helper}) => {637      // all other stakes should be unstaked638      const oneHundredStakers = await helper.arrange.createCrowd(100, 1050n, donor);639640      // stakers stakes 10 times each641      for (let i = 0; i < 10; i++) {642        await Promise.all(oneHundredStakers.map(staker => helper.staking.stake(staker, 100n * nominal)));643      }644      await helper.wait.newBlocks(40);645      await helper.admin.payoutStakers(palletAdmin, 100);646    });647648    itSub.skip('can handle 40.000 rewards', async ({helper}) => {649      const crowdStakes = async () => {650        // each account in the crowd stakes 2 times651        const crowd = await helper.arrange.createCrowd(500, 300n, donor);652        await Promise.all(crowd.map(account => helper.staking.stake(account, 100n * nominal)));653        await Promise.all(crowd.map(account => helper.staking.stake(account, 100n * nominal)));654        //655      };656657      for (let i = 0; i < 40; i++) {658        await crowdStakes();659      }660661      // TODO pay rewards for some period662    });663  });664});665666function calculateIncome(base: bigint, iter = 0, calcPeriod: bigint = UNLOCKING_PERIOD): bigint {667  const DAY = 7200n;668  const ACCURACY = 1_000_000_000n;669  // 5n / 10_000n = 0.05% p/day670  const income = base + base * (ACCURACY * (calcPeriod * 5n) / (10_000n * DAY)) / ACCURACY ;671672  if (iter > 1) {673    return calculateIncome(income, iter - 1, calcPeriod);674  } else return income;675}676677function rewardAvailableInBlock(stakedInBlock: bigint) {678  if (stakedInBlock % LOCKING_PERIOD === 0n) return stakedInBlock + LOCKING_PERIOD;679  return (stakedInBlock - stakedInBlock % LOCKING_PERIOD) + (LOCKING_PERIOD * 2n);680}681682// Wait while promotion period less than specified block, to avoid boundary cases683// 0 if this should be the beginning of the period.684async function waitPromotionPeriodDoesntEnd(helper: DevUniqueHelper, waitBlockLessThan = LOCKING_PERIOD / 3n) {685  const relayBlockNumber = (await helper.callRpc('api.query.parachainSystem.validationData', [])).value.relayParentNumber.toNumber(); // await helper.chain.getLatestBlockNumber();686  const currentPeriodBlock = BigInt(relayBlockNumber) % LOCKING_PERIOD;687688  if (currentPeriodBlock > waitBlockLessThan) {689    await helper.wait.forRelayBlockNumber(BigInt(relayBlockNumber) + LOCKING_PERIOD - currentPeriodBlock);690  }691}