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

difftreelog

source

tests/src/maintenance.seqtest.ts20.4 KiBsourcehistory
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 {ApiPromise} from '@polkadot/api';19import {expect, itSched, itSub, Pallets, requirePalletsOrSkip, usingPlaygrounds} from './util';20import {itEth} from './eth/util';21import {UniqueHelper} from './util/playgrounds/unique';22import {main as correctState} from './migrations/correctStateAfterMaintenance';2324async function maintenanceEnabled(api: ApiPromise): Promise<boolean> {25  return (await api.query.maintenance.enabled()).toJSON() as boolean;26}2728describe('Integration Test: Maintenance Functionality', () => {29  let superuser: IKeyringPair;30  let donor: IKeyringPair;31  let bob: IKeyringPair;3233  before(async function() {34    await usingPlaygrounds(async (helper, privateKey) => {35      requirePalletsOrSkip(this, helper, [Pallets.Maintenance]);36      superuser = await privateKey('//Alice');37      donor = await privateKey({url: import.meta.url});38      [bob] = await helper.arrange.createAccounts([10000n], donor);3940    });41  });4243  describe('Maintenance Mode', () => {44    before(async function() {45      await usingPlaygrounds(async (helper) => {46        if(await maintenanceEnabled(helper.getApi())) {47          console.warn('\tMaintenance mode was left enabled BEFORE the test suite! Disabling it now.');48          await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', [])).to.be.fulfilled;49        }50      });51    });5253    itSub('Allows superuser to enable and disable maintenance mode - and disallows anyone else', async ({helper}) => {54      // Make sure non-sudo can't enable maintenance mode55      await expect(helper.executeExtrinsic(superuser, 'api.tx.maintenance.enable', []), 'on commoner enabling MM')56        .to.be.rejectedWith(/BadOrigin/);5758      // Set maintenance mode59      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);60      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;6162      // Make sure non-sudo can't disable maintenance mode63      await expect(helper.executeExtrinsic(bob, 'api.tx.maintenance.disable', []), 'on commoner disabling MM')64        .to.be.rejectedWith(/BadOrigin/);6566      // Disable maintenance mode67      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);68      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;69    });7071    itSub('MM blocks unique pallet calls', async ({helper}) => {72      // Can create an NFT collection before enabling the MM73      const nftCollection = await helper.nft.mintCollection(bob, {74        tokenPropertyPermissions: [{75          key: 'test', permission: {76            collectionAdmin: true,77            tokenOwner: true,78            mutable: true,79          },80        }],81      });8283      // Can mint an NFT before enabling the MM84      const nft = await nftCollection.mintToken(bob);8586      // Can create an FT collection before enabling the MM87      const ftCollection = await helper.ft.mintCollection(superuser);8889      // Can mint an FT before enabling the MM90      await expect(ftCollection.mint(superuser)).to.be.fulfilled;9192      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);93      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;9495      // Unable to create a collection when the MM is enabled96      await expect(helper.nft.mintCollection(superuser), 'cudo forbidden stuff')97        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);9899      // Unable to set token properties when the MM is enabled100      await expect(nft.setProperties(101        bob,102        [{key: 'test', value: 'test-val'}],103      )).to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);104105      // Unable to mint an NFT when the MM is enabled106      await expect(nftCollection.mintToken(superuser))107        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);108109      // Unable to mint an FT when the MM is enabled110      await expect(ftCollection.mint(superuser))111        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);112113      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);114      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;115116      // Can create a collection after disabling the MM117      await expect(helper.nft.mintCollection(bob), 'MM is disabled, the collection should be created').to.be.fulfilled;118119      // Can set token properties after disabling the MM120      await nft.setProperties(bob, [{key: 'test', value: 'test-val'}]);121122      // Can mint an NFT after disabling the MM123      await nftCollection.mintToken(bob);124125      // Can mint an FT after disabling the MM126      await ftCollection.mint(superuser);127    });128129    itSub.ifWithPallets('MM blocks unique pallet calls (Re-Fungible)', [Pallets.ReFungible], async ({helper}) => {130      // Can create an RFT collection before enabling the MM131      const rftCollection = await helper.rft.mintCollection(superuser);132133      // Can mint an RFT before enabling the MM134      await expect(rftCollection.mintToken(superuser)).to.be.fulfilled;135136      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);137      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;138139      // Unable to mint an RFT when the MM is enabled140      await expect(rftCollection.mintToken(superuser))141        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);142143      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);144      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;145146      // Can mint an RFT after disabling the MM147      await rftCollection.mintToken(superuser);148    });149150    itSub('MM allows native token transfers and RPC calls', async ({helper}) => {151      // We can use RPC before the MM is enabled152      const totalCount = await helper.collection.getTotalCount();153154      // We can transfer funds before the MM is enabled155      await expect(helper.balance.transferToSubstrate(superuser, bob.address, 2n)).to.be.fulfilled;156157      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);158      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;159160      // RPCs work while in maintenance161      expect(await helper.collection.getTotalCount()).to.be.deep.equal(totalCount);162163      // We still able to transfer funds164      await expect(helper.balance.transferToSubstrate(bob, superuser.address, 1n)).to.be.fulfilled;165166      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);167      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;168169      // RPCs work after maintenance170      expect(await helper.collection.getTotalCount()).to.be.deep.equal(totalCount);171172      // Transfers work after maintenance173      await expect(helper.balance.transferToSubstrate(bob, superuser.address, 1n)).to.be.fulfilled;174    });175176    itSched.ifWithPallets('MM blocks scheduled calls and the scheduler itself', [Pallets.Scheduler], async (scheduleKind, {helper}) => {177      const collection = await helper.nft.mintCollection(bob);178179      const nftBeforeMM = await collection.mintToken(bob);180      const nftDuringMM = await collection.mintToken(bob);181      const nftAfterMM = await collection.mintToken(bob);182183      const [184        scheduledIdBeforeMM,185        scheduledIdDuringMM,186        scheduledIdBunkerThroughMM,187        scheduledIdAttemptDuringMM,188        scheduledIdAfterMM,189      ] = scheduleKind == 'named'190        ? helper.arrange.makeScheduledIds(5)191        : new Array(5);192193      const blocksToWait = 6;194195      // Scheduling works before the maintenance196      await nftBeforeMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdBeforeMM})197        .transfer(bob, {Substrate: superuser.address});198199      await helper.wait.newBlocks(blocksToWait + 1);200      expect(await nftBeforeMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});201202      // Schedule a transaction that should occur *during* the maintenance203      await nftDuringMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdDuringMM})204        .transfer(bob, {Substrate: superuser.address});205206      // Schedule a transaction that should occur *after* the maintenance207      await nftDuringMM.scheduleAfter(blocksToWait * 2, {scheduledId: scheduledIdBunkerThroughMM})208        .transfer(bob, {Substrate: superuser.address});209210      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);211      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;212213      await helper.wait.newBlocks(blocksToWait + 1);214      // The owner should NOT change since the scheduled transaction should be rejected215      expect(await nftDuringMM.getOwner()).to.be.deep.equal({Substrate: bob.address});216217      // Any attempts to schedule a tx during the MM should be rejected218      await expect(nftDuringMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdAttemptDuringMM})219        .transfer(bob, {Substrate: superuser.address}))220        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);221222      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);223      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;224225      // Scheduling works after the maintenance226      await nftAfterMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdAfterMM})227        .transfer(bob, {Substrate: superuser.address});228229      await helper.wait.newBlocks(blocksToWait + 1);230231      expect(await nftAfterMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});232      // The owner of the token scheduled for transaction *before* maintenance should now change *after* maintenance233      expect(await nftDuringMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});234    });235236    itEth('Disallows Ethereum transactions to execute while in maintenance', async ({helper}) => {237      const owner = await helper.eth.createAccountWithBalance(donor);238      const receiver = helper.eth.createAccount();239240      const {collectionAddress} = await helper.eth.createERC721MetadataCompatibleNFTCollection(owner, 'A', 'B', 'C', '');241242      // Set maintenance mode243      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);244      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;245246      const contract = await helper.ethNativeContract.collection(collectionAddress, 'nft', owner);247      const tokenId = await contract.methods.nextTokenId().call();248      expect(tokenId).to.be.equal('1');249250      await expect(contract.methods.mintWithTokenURI(receiver, 'Test URI').send())251        .to.be.rejectedWith(/Returned error: unknown error/);252253      await expect(contract.methods.ownerOf(tokenId).call()).rejectedWith(/token not found/);254255      // Disable maintenance mode256      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);257      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;258    });259260    itSub('Allows to enable and disable MM repeatedly', async ({helper}) => {261      // Set maintenance mode262      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);263      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);264      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;265266      // Disable maintenance mode267      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);268      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);269      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;270    });271272    afterEach(async () => {273      await usingPlaygrounds(async helper => {274        if(helper.fetchMissingPalletNames([Pallets.Maintenance]).length != 0) return;275        if(await maintenanceEnabled(helper.getApi())) {276          console.warn('\tMaintenance mode was left enabled AFTER a test has finished! Be careful. Disabling it now.');277          await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);278        }279        expect(await maintenanceEnabled(helper.getApi()), 'Disastrous! Exited the test suite with maintenance mode on.').to.be.false;280      });281    });282  });283284  describe('Preimage Execution', () => {285    const preimageHashes: string[] = [];286287    async function notePreimage(helper: UniqueHelper, preimage: any): Promise<string> {288      const result = await helper.preimage.notePreimage(bob, preimage);289      const events = result.result.events.filter(x => x.event.method === 'Noted' && x.event.section === 'preimage');290      const preimageHash = events[0].event.data[0].toHuman();291      return preimageHash;292    }293294    before(async function() {295      await usingPlaygrounds(async (helper) => {296        requirePalletsOrSkip(this, helper, [Pallets.Preimage, Pallets.Maintenance]);297298        // create a preimage to be operated with in the tests299        const randomAccounts = await helper.arrange.createCrowd(10, 0n, superuser);300        const randomIdentities = randomAccounts.map((acc, i) => [301          acc.address, {302            deposit: 0n,303            judgements: [],304            info: {305              display: {306                raw: `Random Account #${i}`,307              },308            },309          },310        ]);311        const preimage = helper.constructApiCall('api.tx.identity.forceInsertIdentities', [randomIdentities]).method.toHex();312        preimageHashes.push(await notePreimage(helper, preimage));313      });314    });315316    itSub('Successfully executes call in a preimage', async ({helper}) => {317      const result = await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [318        preimageHashes[0], {refTime: 10000000000, proofSize: 10000},319      ])).to.be.fulfilled;320321      // preimage is executed, and an appropriate event is present322      const events = result.result.events.filter((x: any) => x.event.method === 'IdentitiesInserted' && x.event.section === 'identity');323      expect(events.length).to.be.equal(1);324325      // the preimage goes back to being unrequested326      expect(await helper.preimage.getPreimageInfo(preimageHashes[0])).to.have.property('unrequested');327    });328329    itSub('Does not allow execution of a preimage that would fail', async ({helper}) => {330      const [zeroAccount] = await helper.arrange.createAccounts([0n], superuser);331332      const preimage = helper.constructApiCall('api.tx.balances.forceTransfer', [333        {Id: zeroAccount.address}, {Id: superuser.address}, 1000n,334      ]).method.toHex();335      const preimageHash = await notePreimage(helper, preimage);336      preimageHashes.push(preimageHash);337338      await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [339        preimageHash, {refTime: 10000000000, proofSize: 10000},340      ])).to.be.rejectedWith(/^Token: FundsUnavailable$/);341    });342343    itSub('Does not allow preimage execution with non-root', async ({helper}) => {344      await expect(helper.executeExtrinsic(bob, 'api.tx.maintenance.executePreimage', [345        preimageHashes[0], {refTime: 10000000000, proofSize: 10000},346      ])).to.be.rejectedWith(/^Misc: BadOrigin$/);347    });348349    itSub('Does not allow execution of non-existent preimages', async ({helper}) => {350      await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [351        '0x1010101010101010101010101010101010101010101010101010101010101010', {refTime: 10000000000, proofSize: 10000},352      ])).to.be.rejectedWith(/^Misc: Unavailable$/);353    });354355    itSub('Does not allow preimage execution with less than minimum weights', async ({helper}) => {356      await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [357        preimageHashes[0], {refTime: 1000, proofSize: 100},358      ])).to.be.rejectedWith(/^Misc: Exhausted$/);359    });360361    after(async function() {362      await usingPlaygrounds(async (helper) => {363        if(helper.fetchMissingPalletNames([Pallets.Preimage, Pallets.Maintenance]).length != 0) return;364365        for(const hash of preimageHashes) {366          await helper.preimage.unnotePreimage(bob, hash);367        }368      });369    });370  });371372  describe('Integration Test: Maintenance mode & App Promo', () => {373    let superuser: IKeyringPair;374375    before(async function() {376      await usingPlaygrounds(async (helper, privateKey) => {377        requirePalletsOrSkip(this, helper, [Pallets.Maintenance]);378        superuser = await privateKey('//Alice');379      });380    });381382    describe('Test AppPromo script for check state after Maintenance mode', () => {383      before(async function () {384        await usingPlaygrounds(async (helper) => {385          if(await maintenanceEnabled(helper.getApi())) {386            console.warn('\tMaintenance mode was left enabled BEFORE the test suite! Disabling it now.');387            await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', [])).to.be.fulfilled;388          }389        });390      });391      itSub('Can find and fix inconsistent state', async ({helper}) => {392        const api = helper.getApi();393394        await helper.executeExtrinsic(superuser, 'api.tx.sudo.sudo', [api.tx.system.setStorage([395          // pendingUnstake(1 -> [superuser.address, 100UNQ])396          ['0x42b67acb8bd223c60d0c8f621ffefc0ae280fa2db99bd3827aac976de75af95f5153cb1f00942ff401000000',397            '0x04d43593c715fdd31c61141abd04a99fd6822c8558854ccde39a5684e7a56da27d000010632d5ec76b0500000000000000'],398          // pendingUnstake(2 -> [superuser.address, 100UNQ])399          ['0x42b67acb8bd223c60d0c8f621ffefc0ae280fa2db99bd3827aac976de75af95f9eb2dcce60f37a2702000000',400            '0x04d43593c715fdd31c61141abd04a99fd6822c8558854ccde39a5684e7a56da27d000010632d5ec76b0500000000000000'],401          // Balances.freezes(superuser.address -> freeze with app promo id and 200 UNQ )402          ['0xc2261276cc9d1f8598ea4b6a74b15c2fb1c0eb12e038e5c7f91e120ed4b7ebf1de1e86a9a8c739864cf3cc5ec2bea59fd43593c715fdd31c61141abd04a99fd6822c8558854ccde39a5684e7a56da27d',403            '0x046170707374616b656170707374616b65000020c65abc8ed70a00000000000000'],404        ])]);405406        expect((await api.query.appPromotion.pendingUnstake(1)).toJSON()).to.be.deep.equal([[superuser.address, '0x00000000000000056bc75e2d63100000']]);407        expect((await api.query.appPromotion.pendingUnstake(2)).toJSON()).to.be.deep.equal([[superuser.address, '0x00000000000000056bc75e2d63100000']]);408        expect((await api.query.balances.freezes(superuser.address))409          .map(lock => ({id: lock.id.toUtf8(), amount: lock.amount.toBigInt()})))410          .to.be.deep.equal([{id: 'appstakeappstake', amount: 200000000000000000000n}]);411        await correctState();412413        expect((await api.query.appPromotion.pendingUnstake(1)).toJSON()).to.be.deep.equal([]);414        expect((await api.query.appPromotion.pendingUnstake(2)).toJSON()).to.be.deep.equal([]);415        expect((await api.query.balances.freezes(superuser.address)).toJSON()).to.be.deep.equal([]);416417      });418419      itSub('(!negative test!) Only works when Maintenance mode is disabled', async({helper}) => {420        await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', [])).to.be.fulfilled;421        await expect(correctState()).to.be.rejectedWith('The network is still in maintenance mode');422        await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', [])).to.be.fulfilled;423      });424    });425  });426});