difftreelog
Add contract sponsoring tests
in: master
Also removed waitForRecalculationBlock function
1 file changed
tests/src/app-promotion.test.tsdiffbeforeafterboth1// 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 normalizeAccountId,20 getModuleNames,21 Pallets,22} from './util/helpers';2324import chai from 'chai';25import chaiAsPromised from 'chai-as-promised';26import {usingPlaygrounds} from './util/playgrounds';2728import {encodeAddress, mnemonicGenerate} from '@polkadot/util-crypto';29import {stringToU8a} from '@polkadot/util';30import {ApiPromise} from '@polkadot/api';31chai.use(chaiAsPromised);32const expect = chai.expect;3334let alice: IKeyringPair;35let bob: IKeyringPair;36let palletAdmin: IKeyringPair;37let nominal: bigint;38let promotionStartBlock: number | null = null;39const palletAddress = calculatePalleteAddress('appstake');4041before(async function () {42 await usingPlaygrounds(async (helper, privateKeyWrapper) => {43 if (!getModuleNames(helper.api!).includes(Pallets.AppPromotion)) this.skip();44 alice = privateKeyWrapper('//Alice');45 bob = privateKeyWrapper('//Bob');46 palletAdmin = privateKeyWrapper('//palletAdmin');47 nominal = helper.balance.getOneTokenNominal();48 await helper.balance.transferToSubstrate(alice, palletAdmin.address, 100n * nominal);49 await helper.balance.transferToSubstrate(alice, palletAddress, 100n * nominal);50 if (!promotionStartBlock) {51 promotionStartBlock = (await helper.api!.query.parachainSystem.lastRelayChainBlockNumber()).toNumber();52 }53 await helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.startAppPromotion(promotionStartBlock!)));54 });55});5657after(async function () {58 await usingPlaygrounds(async (helper) => {59 await helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.stopAppPromotion()));60 });61});6263describe('app-promotions.stake extrinsic', () => {64 it('should change balance state to "locked", add it to "staked" map, and increase "totalStaked" amount', async () => {65 await usingPlaygrounds(async (helper) => {66 const totalStakedBefore = await helper.staking.getTotalStaked();67 const [staker] = await helper.arrange.creteAccounts([10n], alice);68 69 // Minimum stake amount is 1:70 await expect(helper.staking.stake(staker, nominal - 1n)).to.be.eventually.rejected;71 await helper.staking.stake(staker, nominal);72 expect(await helper.staking.getTotalStakingLocked({Substrate: staker.address})).to.be.equal(nominal);7374 // TODO add helpers to assert bigints. Check balance close to 1075 expect(await helper.balance.getSubstrate(staker.address) - 9n * nominal >= (nominal / 2n)).to.be.true;76 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(nominal);77 // it is potentially flaky test. Promotion can credited some tokens. Maybe we need to use closeTo? 78 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore + nominal); // total tokens amount staked in app-promotion increased 7980 await helper.staking.stake(staker, 2n * nominal);81 expect(await helper.staking.getTotalStakingLocked({Substrate: staker.address})).to.be.equal(3n * nominal);82 83 const stakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map((x) => x[1]);84 expect(stakedPerBlock).to.be.deep.equal([nominal, 2n * nominal]);85 });86 });87 88 it('should reject transaction if stake amount is more than total free balance', async () => {89 await usingPlaygrounds(async helper => { 90 const [staker] = await helper.arrange.creteAccounts([10n], alice);9192 // Can't stake full balance because Alice needs to pay some fee93 await expect(helper.staking.stake(staker, 10n * nominal)).to.be.eventually.rejected;94 await helper.staking.stake(staker, 7n * nominal);9596 // Can't stake 4 tkn because Alice has ~3 free tkn, and 7 locked97 await expect(helper.staking.stake(staker, 4n * nominal)).to.be.eventually.rejected; 98 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(7n * nominal);99 });100 });101 102 it('for different accounts in one block is possible', async () => {103 await usingPlaygrounds(async helper => {104 const crowd = await helper.arrange.creteAccounts([10n, 10n, 10n, 10n], alice);105 106 const crowdStartsToStake = crowd.map(user => helper.staking.stake(user, nominal));107 await expect(Promise.all(crowdStartsToStake)).to.be.eventually.fulfilled;108109 const crowdStakes = await Promise.all(crowd.map(address => helper.staking.getTotalStaked({Substrate: address.address})));110 expect(crowdStakes).to.deep.equal([nominal, nominal, nominal, nominal]);111 });112 });113 // TODO it('Staker stakes 5 times in one block with nonce');114 // TODO it('Staked balance appears as locked in the balance pallet');115 // TODO it('Alice stakes huge amount of tokens');116});117118describe('unstake balance extrinsic', () => { 119 it('should change balance state to "reserved", add it to "pendingUnstake" map, and subtract it from totalStaked', async () => {120 await usingPlaygrounds(async helper => {121 const totalStakedBefore = await helper.staking.getTotalStaked();122 const [staker] = await helper.arrange.creteAccounts([10n], alice);123 await helper.staking.stake(staker, 5n * nominal);124 await helper.staking.unstake(staker, 3n * nominal);125126 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(3n * nominal);127 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(2n * nominal);128 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore + 2n * nominal);129 });130 });131132 it('should remove from the "staked" map starting from the oldest entry', async () => {133 await usingPlaygrounds(async helper => {134 const [staker] = await helper.arrange.creteAccounts([100n], alice);135 await helper.staking.stake(staker, 10n * nominal);136 await helper.staking.stake(staker, 20n * nominal);137 await helper.staking.stake(staker, 30n * nominal);138139 // staked: [10, 20, 30]; unstaked: 0140 let pendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});141 let unstakedPerBlock = (await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address})).map(stake => stake[1]);142 let stakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(stake => stake[1]);143 expect(pendingUnstake).to.be.deep.equal(0n);144 expect(unstakedPerBlock).to.be.deep.equal([]);145 expect(stakedPerBlock).to.be.deep.equal([10n * nominal, 20n * nominal, 30n * nominal]);146 147 // Can unstake the part of a stake148 await helper.staking.unstake(staker, 5n * nominal);149 pendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});150 unstakedPerBlock = (await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address})).map(stake => stake[1]);151 stakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(stake => stake[1]);152 expect(pendingUnstake).to.be.equal(5n * nominal);153 expect(stakedPerBlock).to.be.deep.equal([5n * nominal, 20n * nominal, 30n * nominal]);154 expect(unstakedPerBlock).to.be.deep.equal([5n * nominal]);155156 // Can unstake one stake totally and one more partially157 await helper.staking.unstake(staker, 10n * nominal);158 pendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});159 unstakedPerBlock = (await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address})).map(stake => stake[1]);160 stakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(stake => stake[1]);161 expect(pendingUnstake).to.be.equal(15n * nominal);162 expect(stakedPerBlock).to.be.deep.equal([15n * nominal, 30n * nominal]);163 expect(unstakedPerBlock).to.deep.equal([5n * nominal, 10n * nominal]);164165 // Can totally unstake 2 stakes in one tx166 await helper.staking.unstake(staker, 45n * nominal);167 pendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});168 unstakedPerBlock = (await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address})).map(stake => stake[1]);169 stakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(stake => stake[1]);170 expect(pendingUnstake).to.be.equal(60n * nominal);171 expect(stakedPerBlock).to.deep.equal([]);172 expect(unstakedPerBlock).to.deep.equal([5n * nominal, 10n * nominal, 45n * nominal]);173 });174 });175176 it('should reject transaction if unstake amount is greater than staked', async () => {177 await usingPlaygrounds(async (helper) => {178 const [staker] = await helper.arrange.creteAccounts([10n], alice);179 180 // can't unstsake more than one stake amount181 await helper.staking.stake(staker, 1n * nominal);182 await expect(helper.staking.unstake(staker, 1n * nominal + 1n)).to.be.eventually.rejected;183 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(0n);184 expect(await helper.staking.getTotalStakingLocked({Substrate: staker.address})).to.be.equal(1n * nominal);185186 // can't unstsake more than two stakes amount187 await helper.staking.stake(staker, 1n * nominal);188 await expect(helper.staking.unstake(staker, 2n * nominal + 1n)).to.be.eventually.rejected;189 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(0n);190 expect(await helper.staking.getTotalStakingLocked({Substrate: staker.address})).to.be.equal(2n * nominal);191192 // can't unstake more than have with nonce // TODO not sure we need this assertion193 const nonce1 = await helper.chain.getNonce(staker.address);194 const nonce2 = nonce1 + 1;195 const unstakeMoreThanHaveWithNonce = Promise.all([196 helper.signTransaction(staker, helper.constructApiCall('api.tx.promotion.unstake', [1n * nominal]), 'unstaking 1', {nonce: nonce1}),197 helper.signTransaction(staker, helper.constructApiCall('api.tx.promotion.unstake', [1n * nominal + 1n]), 'unstaking 1+', {nonce: nonce2}),198 ]);199 await expect(unstakeMoreThanHaveWithNonce).to.be.rejected;200 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(1n * nominal);201 });202 });203204 it('should allow to unstake even smallest unit', async () => {205 await usingPlaygrounds(async (helper) => {206 const [staker] = await helper.arrange.creteAccounts([10n], alice);207 await helper.staking.stake(staker, nominal);208 // unstake .000...001 is possible209 await helper.staking.unstake(staker, 1n);210 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(1n);211 });212 });213214 it('should work fine if stake amount is smallest unit', async () => {215 await usingPlaygrounds(async (helper) => {216 const [staker] = await helper.arrange.creteAccounts([10n], alice);217 await helper.staking.stake(staker, nominal);218 await helper.staking.unstake(staker, nominal - 1n);219 await waitForRecalculationBlock(helper.api!);220221 // Everything fine, blockchain alive222 await helper.nft.mintCollection(staker, {name: 'name', description: 'description', tokenPrefix: 'prefix'});223 });224 });225226 // TODO will return balance to "available" state after the period of unstaking is finished, and subtract it from "pendingUnstake227 // TODO for different accounts in one block is possible228});229230describe('Admin adress', () => {231 it('can be set by sudo only', async () => {232 await usingPlaygrounds(async (helper) => {233 // Bob can not set admin not from himself nor as a sudo234 await expect(helper.signTransaction(bob, helper.api!.tx.promotion.setAdminAddress({Substrate: bob.address}))).to.be.eventually.rejected;235 await expect(helper.signTransaction(bob, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress({Substrate: bob.address})))).to.be.eventually.rejected;236237 // Alice can238 await expect(helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress({Substrate: palletAdmin.address})))).to.be.eventually.fulfilled;239 });240 });241 242 it('can be any valid CrossAccountId', async () => {243 // We are not going to set an eth address as a sponsor,244 // but we do want to check, it doesn't break anything;245 await usingPlaygrounds(async (helper) => {246 const [charlie] = await helper.arrange.creteAccounts([10n], alice);247 const ethCharlie = helper.address.substrateToEth(charlie.address); 248 // Alice sets Ethereum address as a sudo. Then Substrate address back...249 await expect(helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress({Ethereum: ethCharlie})))).to.be.eventually.fulfilled;250 await expect(helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress({Substrate: palletAdmin.address})))).to.be.eventually.fulfilled;251 252 // ...It doesn't break anything;253 const collection = await helper.nft.mintCollection(charlie, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});254 await expect(helper.signTransaction(charlie, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.rejected;255 });256 });257258 it('can be reassigned', async () => {259 await usingPlaygrounds(async (helper) => {260 const [oldAdmin, newAdmin, collectionOwner] = await helper.arrange.creteAccounts([10n, 10n, 10n], alice);261 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});262 263 await expect(helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress(normalizeAccountId(oldAdmin))))).to.be.eventually.fulfilled;264 await expect(helper.signTransaction(alice, helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress(normalizeAccountId(newAdmin))))).to.be.eventually.fulfilled;265 await expect(helper.signTransaction(oldAdmin, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.rejected;266 267 await expect(helper.signTransaction(newAdmin, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.fulfilled;268 });269 });270});271272describe('App-promotion collection sponsoring', () => {273 before(async function () {274 await usingPlaygrounds(async (helper) => {275 const tx = helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress({Substrate: palletAdmin.address}));276 await helper.signTransaction(alice, tx);277 });278 });279 280 it('can not be set by non admin', async () => {281 await usingPlaygrounds(async (helper) => {282 const [collectionOwner, nonAdmin] = await helper.arrange.creteAccounts([10n, 10n], alice);283284 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});285 286 await expect(helper.signTransaction(nonAdmin, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.rejected;287 expect((await collection.getData())?.raw.sponsorship).to.equal('Disabled');288 });289 });290291 it('should set pallet address as confirmed admin', async () => {292 await usingPlaygrounds(async (helper) => {293 const [collectionOwner, oldSponsor] = await helper.arrange.creteAccounts([20n, 20n], alice);294 295 // Can set sponsoring for collection without sponsor296 const collectionWithoutSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'No-sponsor', description: 'New Collection', tokenPrefix: 'Promotion'});297 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collectionWithoutSponsor.collectionId))).to.be.eventually.fulfilled;298 expect((await collectionWithoutSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});299300 // Can set sponsoring for collection with unconfirmed sponsor301 const collectionWithUnconfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Unconfirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});302 expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: oldSponsor.address});303 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collectionWithUnconfirmedSponsor.collectionId))).to.be.eventually.fulfilled;304 expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});305306 // Can set sponsoring for collection with confirmed sponsor307 const collectionWithConfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Confirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});308 await collectionWithConfirmedSponsor.confirmSponsorship(oldSponsor);309 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collectionWithConfirmedSponsor.collectionId))).to.be.eventually.fulfilled;310 expect((await collectionWithConfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});311 });312 });313314 it('can be overwritten by collection owner', async () => { 315 await usingPlaygrounds(async (helper) => {316 const [collectionOwner, newSponsor] = await helper.arrange.creteAccounts([20n, 0n], alice);317 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});318 const collectionId = collection.collectionId;319 320 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collectionId))).to.be.eventually.fulfilled;321 322 // Collection limits still can be changed by the owner323 expect(await collection.setLimits(collectionOwner, {sponsorTransferTimeout: 0})).to.be.true;324 expect((await collection.getData())?.raw.limits.sponsorTransferTimeout).to.be.equal(0);325 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});326327 // Collection sponsor can be changed too328 expect((await collection.setSponsor(collectionOwner, newSponsor.address))).to.be.true;329 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: newSponsor.address});330 });331 });332 333 it('should not overwrite collection limits set by the owner earlier', async () => {334 await usingPlaygrounds(async (helper) => {335 const limits = {ownerCanDestroy: true, ownerCanTransfer: true, sponsorTransferTimeout: 0};336 const collectionWithLimits = await helper.nft.mintCollection(alice, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits});337338 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collectionWithLimits.collectionId))).to.be.eventually.fulfilled;339 expect((await collectionWithLimits.getData())?.raw.limits).to.be.deep.contain(limits);340 });341 });342 343 it('should reject transaction if collection doesn\'t exist', async () => {344 await usingPlaygrounds(async (helper) => {345 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(999999999))).to.be.eventually.rejected;346 });347 });348349 it('should reject transaction if collection was burnt', async () => {350 await usingPlaygrounds(async (helper) => {351 const [collectionOwner] = await helper.arrange.creteAccounts([10n], alice);352 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});353 await collection.burn(collectionOwner);354355 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.rejected;356 });357 });358});359360describe('app-promotion stopSponsoringCollection', () => {361 it('can not be called by non-admin', async () => { 362 await usingPlaygrounds(async (helper) => {363 const [collectionOwner, nonAdmin] = await helper.arrange.creteAccounts([10n, 10n], alice);364 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});365 366 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.fulfilled;367 368 await expect(helper.signTransaction(nonAdmin, helper.api!.tx.promotion.stopSponsorignCollection(collection.collectionId))).to.be.eventually.rejected;369 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});370 });371 });372373 it('should set sponsoring as disabled', async () => {374 await usingPlaygrounds(async (helper) => {375 const [collectionOwner] = await helper.arrange.creteAccounts([10n, 10n], alice);376 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});377 378 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.sponsorCollection(collection.collectionId))).to.be.eventually.fulfilled;379 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.stopSponsorignCollection(collection.collectionId))).to.be.eventually.fulfilled;380 381 expect((await collection.getData())?.raw.sponsorship).to.be.equal('Disabled');382 });383 });384385 it('should not affect collection which is not sponsored by pallete', async () => {386 await usingPlaygrounds(async (helper) => {387 const [collectionOwner] = await helper.arrange.creteAccounts([10n, 10n], alice);388 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: collectionOwner.address});389 await collection.confirmSponsorship(collectionOwner);390 391 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.stopSponsorignCollection(collection.collectionId))).to.be.eventually.rejected;392 393 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: collectionOwner.address});394 });395 });396397 it('should reject transaction if collection does not exist', async () => { 398 await usingPlaygrounds(async (helper) => {399 const [collectionOwner] = await helper.arrange.creteAccounts([10n], alice);400 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});401 402 await collection.burn(collectionOwner);403 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.stopSponsorignCollection(collection.collectionId))).to.be.eventually.rejected;404 await expect(helper.signTransaction(palletAdmin, helper.api!.tx.promotion.stopSponsorignCollection(999999999))).to.be.eventually.rejected;405 });406 });407});408409describe('app-promotion contract sponsoring', () => {410 it('will set contract sponsoring mode and set palletes address as a sponsor', async () => {411 // arrange: Alice deploys Flipper412 413 // act: Admin calls appPromotion.sponsorContract(Flipper.address)414415 // assert: contract.sponsoringMode = TODO416 // assert: contract.sponsor to be PalleteAddress417 });418419 it('will overwrite sponsoring mode and existed sponsor', async () => {420 // arrange: Alice deploys Flipper421 // arrange: Alice sets self sponsoring for Flipper422423 // act: Admin calls appPromotion.sponsorContract(Flipper.address)424425 // assert: contract.sponsoringMode = TODO426 // assert: contract.sponsor to be PalleteAddress427 });428429 it('can be overwritten by contract owner', async () => {430 // arrange: Alice deploys Flipper431 // arrange: Admin calls appPromotion.sponsorContract(Flipper.address)432433 // act: Alice sets self sponsoring for Flipper434435 // assert: contract.sponsoringMode = Self436 // assert: contract.sponsor to be contract437 });438439 it('can not be set by non admin', async () => {440 // arrange: Alice deploys Flipper441 // arrange: Alice sets self sponsoring for Flipper442443 // assert: Random calls appPromotion.sponsorContract(Flipper.address) throws444 // assert: contract.sponsoringMode = Self445 // assert: contract.sponsor to be contract446 });447448 it('will return unused gas fee to app-promotion pallete', async () => {449 // arrange: Alice deploys Flipper450 // arrange: Admin calls appPromotion.sponsorContract(Flipper.address)451452 // assert: Bob calls Flipper - expect balances deposit event do not appears for Bob /// Unused gas fee returns to contract453 // assert: Bobs balance the same454 });455456 it('will failed for non contract address', async () => {457 // arrange: web3 creates new address - 0x0458459 // assert: Admin calls appPromotion.sponsorContract(0x0) throws460 // assert: Admin calls appPromotion.sponsorContract(Substrate address) throws461 });462463 it('will actually sponsor transactions', async () => {464 // TODO test it because this is a new way of contract sponsoring465 });466});467468describe('app-promotion stopSponsoringContract', () => {469 before(async function () {470 await usingPlaygrounds(async (helper, privateKeyWrapper) => {471 if (!getModuleNames(helper.api!).includes(Pallets.AppPromotion)) this.skip();472 alice = privateKeyWrapper('//Alice');473 bob = privateKeyWrapper('//Bob');474 palletAdmin = privateKeyWrapper('//palletAdmin');475 await helper.balance.transferToSubstrate(alice, palletAdmin.address, 10n * helper.balance.getOneTokenNominal());476 await helper.balance.transferToSubstrate(alice, palletAddress, 10n * helper.balance.getOneTokenNominal());477 478 const tx = helper.api!.tx.sudo.sudo(helper.api!.tx.promotion.setAdminAddress(normalizeAccountId(palletAdmin)));479 await helper.signTransaction(alice, tx);480 481 nominal = helper.balance.getOneTokenNominal();482 });483 });484 485 it('will set contract sponsoring mode as disabled', async () => {486 // arrange: Alice deploys Flipper487 // arrange: Admin calls appPromotion.sponsorContract(Flipper.address)488 489 // act: Admin calls appPromotion.stopSponsoringContract(Flipper.address)490 // assert: contract sponsoring mode = TODO491492 // act: Bob calls Flipper493494 // assert: PalleteAddress balance did not change495 // assert: Bobs balance less than before /// Bob payed some fee496 });497498 it('can not be called by non-admin', async () => {499 // arrange: Alice deploys Flipper500 // arrange: Admin calls appPromotion.sponsorContract(Flipper.address)501502 // assert: Random calls appPromotion.stopSponsoringContract(Flipper.address) throws503 // assert: contract sponsor is PallereAddress504 });505506 it('will not affect a contract which is not sponsored by pallete', async () => {507 // arrange: Alice deploys Flipper508 // arrange: Alice sets self sponsoring for Flipper509 510 // act: Admin calls appPromotion.stopSponsoringContract(Flipper.address) throws511512 // assert: contract.sponsoringMode = Self513 // assert: contract.sponsor to be contract514 });515516 it('will failed for non contract address', async () => {517 // arrange: web3 creates new address - 0x0518519 // expect stopSponsoringContract(0x0) throws520 });521});522523describe('app-promotion rewards', () => {524 it('should credit 0.05% for staking period', async () => { 525 await usingPlaygrounds(async helper => {526 const [staker] = await helper.arrange.creteAccounts([50n], alice);527 await waitForRecalculationBlock(helper.api!);528 529 await helper.staking.stake(staker, 1n * nominal);530 await helper.staking.stake(staker, 2n * nominal);531 await waitForRelayBlock(helper.api!, 36);532 533 const totalStakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(s => s[1]);534 expect(totalStakedPerBlock).to.be.deep.equal([calculateIncome(nominal, 10n), calculateIncome(2n * nominal, 10n)]);535 });536 });537 538 it('should not be credited for unstaked (reserved) balance', async () => {539 await usingPlaygrounds(async helper => {540 expect.fail('Implement me after unstake method will be fixed');541 });542 });543 544 it('should bring compound interest', async () => {545 await usingPlaygrounds(async helper => {546 const [staker] = await helper.arrange.creteAccounts([80n], alice);547 548 await waitForRecalculationBlock(helper.api!);549 550 await helper.staking.stake(staker, 10n * nominal);551 await helper.staking.stake(staker, 20n * nominal);552 await helper.staking.stake(staker, 30n * nominal);553 554 await waitForRelayBlock(helper.api!, 34);555 let totalStakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(s => s[1]);556 expect(totalStakedPerBlock).to.deep.equal([calculateIncome(10n * nominal, 10n), calculateIncome(20n * nominal, 10n), calculateIncome(30n * nominal, 10n)]);557 558 await waitForRelayBlock(helper.api!, 20);559 totalStakedPerBlock = (await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).map(s => s[1]);560 expect(totalStakedPerBlock).to.deep.equal([calculateIncome(10n * nominal, 10n, 2), calculateIncome(20n * nominal, 10n, 2), calculateIncome(30n * nominal, 10n, 2)]); 561 });562 });563564 // TODO (load test. Can pay reward for 10000 addresses)565});566567568function waitForRecalculationBlock(api: ApiPromise): Promise<void> {569 return new Promise<void>(async (resolve, reject) => {570 const unsubscribe = await api.query.system.events((events) => {571 572 events.forEach((record) => {573 574 const {event, phase} = record;575 const types = event.typeDef;576 577 if (event.section === 'promotion' && event.method === 'StakingRecalculation') {578 unsubscribe();579 resolve();580 }581 });582 });583 });584}585586async function waitForRelayBlock(api: ApiPromise, blocks = 1): Promise<void> {587 const current_block = (await api.query.parachainSystem.lastRelayChainBlockNumber()).toNumber();588 return new Promise<void>(async (resolve, reject) => {589 const unsubscribe = await api.query.parachainSystem.validationData(async (data) => {590 // console.log(`${current_block} || ${data.value.relayParentNumber.toNumber()}`);591 if (data.value.relayParentNumber.toNumber() - current_block >= blocks) {592 unsubscribe();593 resolve();594 }595 });596 });597}598599function calculatePalleteAddress(palletId: any) {600 const address = stringToU8a(('modl' + palletId).padEnd(32, '\0'));601 return encodeAddress(address);602}603604function calculateIncome(base: bigint, calcPeriod: bigint, iter = 0): bigint {605 const DAY = 7200n;606 const ACCURACY = 1_000_000_000n;607 const income = base + base * (ACCURACY * (calcPeriod * 5n) / (10_000n * DAY)) / ACCURACY ;608 609 if (iter > 1) {610 return calculateIncome(income, calcPeriod, iter - 1);611 } else return income;612}