`e0ec9ae1d490`

source

tests/src/maintenanceMode.seqtest.ts12.5 KiBsource history
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, usingPlaygrounds} from './util';20import {itEth} from './eth/util';2122async function maintenanceEnabled(api: ApiPromise): Promise<boolean> {23  return (await api.query.maintenance.enabled()).toJSON() as boolean;24}2526describe('Integration Test: Maintenance Mode', () => {27  let superuser: IKeyringPair;28  let donor: IKeyringPair;29  let bob: IKeyringPair;3031  before(async () => {32    await usingPlaygrounds(async (helper, privateKey) => {33      superuser = await privateKey('//Alice');34      donor = await privateKey({filename: __filename});35      [bob] = await helper.arrange.createAccounts([100n], donor);3637      if (await maintenanceEnabled(helper.getApi())) {38        console.warn('\tMaintenance mode was left enabled BEFORE the test suite! Disabling it now.');39        await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', [])).to.be.fulfilled;40      }41    });42  });4344  itSub('Allows superuser to enable and disable maintenance mode - and disallows anyone else', async ({helper}) => {45    // Make sure non-sudo can't enable maintenance mode46    await expect(helper.executeExtrinsic(superuser, 'api.tx.maintenance.enable', []), 'on commoner enabling MM')47      .to.be.rejectedWith(/BadOrigin/);4849    // Set maintenance mode50    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);51    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;5253    // Make sure non-sudo can't disable maintenance mode54    await expect(helper.executeExtrinsic(bob, 'api.tx.maintenance.disable', []), 'on commoner disabling MM')55      .to.be.rejectedWith(/BadOrigin/);5657    // Disable maintenance mode58    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);59    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;60  });6162  itSub('MM blocks unique pallet calls', async ({helper}) => {63    // Can create an NFT collection before enabling the MM64    const nftCollection = await helper.nft.mintCollection(bob, {65      tokenPropertyPermissions: [{key: 'test', permission: {66        collectionAdmin: true,67        tokenOwner: true,68        mutable: true,69      }}],70    });7172    // Can mint an NFT before enabling the MM73    const nft = await nftCollection.mintToken(bob);7475    // Can create an FT collection before enabling the MM76    const ftCollection = await helper.ft.mintCollection(superuser);7778    // Can mint an FT before enabling the MM79    await expect(ftCollection.mint(superuser)).to.be.fulfilled;8081    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);82    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;8384    // Unable to create a collection when the MM is enabled85    await expect(helper.nft.mintCollection(superuser), 'cudo forbidden stuff')86      .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);8788    // Unable to set token properties when the MM is enabled89    await expect(nft.setProperties(90      bob,91      [{key: 'test', value: 'test-val'}],92    )).to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);9394    // Unable to mint an NFT when the MM is enabled95    await expect(nftCollection.mintToken(superuser))96      .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);9798    // Unable to mint an FT when the MM is enabled99    await expect(ftCollection.mint(superuser))100      .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);101102    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);103    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;104105    // Can create a collection after disabling the MM106    await expect(helper.nft.mintCollection(bob), 'MM is disabled, the collection should be created').to.be.fulfilled;107108    // Can set token properties after disabling the MM109    await nft.setProperties(bob, [{key: 'test', value: 'test-val'}]);110111    // Can mint an NFT after disabling the MM112    await nftCollection.mintToken(bob);113114    // Can mint an FT after disabling the MM115    await ftCollection.mint(superuser);116  });117118  itSub.ifWithPallets('MM blocks unique pallet calls (Re-Fungible)', [Pallets.ReFungible], async ({helper}) => {119    // Can create an RFT collection before enabling the MM120    const rftCollection = await helper.rft.mintCollection(superuser);121122    // Can mint an RFT before enabling the MM123    await expect(rftCollection.mintToken(superuser)).to.be.fulfilled;124125    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);126    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;127128    // Unable to mint an RFT when the MM is enabled129    await expect(rftCollection.mintToken(superuser))130      .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);131    132    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);133    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;134135    // Can mint an RFT after disabling the MM136    await rftCollection.mintToken(superuser);137  });138139  itSub('MM allows native token transfers and RPC calls', async ({helper}) => {140    // We can use RPC before the MM is enabled141    const totalCount = await helper.collection.getTotalCount();142143    // We can transfer funds before the MM is enabled144    await expect(helper.balance.transferToSubstrate(superuser, bob.address, 2n)).to.be.fulfilled;145146    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);147    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;148149    // RPCs work while in maintenance150    expect(await helper.collection.getTotalCount()).to.be.deep.equal(totalCount);151152    // We still able to transfer funds153    await expect(helper.balance.transferToSubstrate(bob, superuser.address, 1n)).to.be.fulfilled;154155    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);156    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;157158    // RPCs work after maintenance159    expect(await helper.collection.getTotalCount()).to.be.deep.equal(totalCount);160161    // Transfers work after maintenance162    await expect(helper.balance.transferToSubstrate(bob, superuser.address, 1n)).to.be.fulfilled;163  });164165  itSched.ifWithPallets('MM blocks scheduled calls and the scheduler itself', [Pallets.Scheduler], async (scheduleKind, {helper}) => {166    const collection = await helper.nft.mintCollection(bob);167168    const nftBeforeMM = await collection.mintToken(bob);169    const nftDuringMM = await collection.mintToken(bob);170    const nftAfterMM = await collection.mintToken(bob);171172    const [173      scheduledIdBeforeMM,174      scheduledIdDuringMM,175      scheduledIdBunkerThroughMM,176      scheduledIdAttemptDuringMM,177      scheduledIdAfterMM,178    ] = scheduleKind == 'named'179      ? helper.arrange.makeScheduledIds(5)180      : new Array(5);181182    const blocksToWait = 6;183184    // Scheduling works before the maintenance185    await nftBeforeMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdBeforeMM})186      .transfer(bob, {Substrate: superuser.address});187188    await helper.wait.newBlocks(blocksToWait + 1);189    expect(await nftBeforeMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});190191    // Schedule a transaction that should occur *during* the maintenance192    await nftDuringMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdDuringMM})193      .transfer(bob, {Substrate: superuser.address});194    195    // Schedule a transaction that should occur *after* the maintenance196    await nftDuringMM.scheduleAfter(blocksToWait * 2, {scheduledId: scheduledIdBunkerThroughMM})197      .transfer(bob, {Substrate: superuser.address});198199    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);200    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;201202    await helper.wait.newBlocks(blocksToWait + 1);203    // The owner should NOT change since the scheduled transaction should be rejected204    expect(await nftDuringMM.getOwner()).to.be.deep.equal({Substrate: bob.address});205206    // Any attempts to schedule a tx during the MM should be rejected207    await expect(nftDuringMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdAttemptDuringMM})208      .transfer(bob, {Substrate: superuser.address}))209      .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);210211    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);212    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;213214    // Scheduling works after the maintenance215    await nftAfterMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdAfterMM})216      .transfer(bob, {Substrate: superuser.address});217    218    await helper.wait.newBlocks(blocksToWait + 1);219220    expect(await nftAfterMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});221    // The owner of the token scheduled for transaction *before* maintenance should now change *after* maintenance222    expect(await nftDuringMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});223  });224225  itEth('Disallows Ethereum transactions to execute while in maintenance', async ({helper}) => {226    const owner = await helper.eth.createAccountWithBalance(donor);227    const receiver = helper.eth.createAccount();228    229    const {collectionAddress} = await helper.eth.createERC721MetadataCompatibleNFTCollection(owner, 'A', 'B', 'C', '');230231    // Set maintenance mode232    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);233    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;234235    const contract = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);236    const tokenId = await contract.methods.nextTokenId().call();237    expect(tokenId).to.be.equal('1');238239    await expect(contract.methods.mintWithTokenURI(receiver, 'Test URI').send())240      .to.be.rejectedWith(/submit transaction to pool failed: Pool\(InvalidTransaction\(InvalidTransaction::Call\)\)/);241242    await expect(contract.methods.ownerOf(tokenId).call()).rejectedWith(/token not found/);243244    // Disable maintenance mode245    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);246    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;247  });248249  itSub('Allows to enable and disable MM repeatedly', async ({helper}) => {250    // Set maintenance mode251    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);252    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);253    expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;254255    // Disable maintenance mode256    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);257    await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);258    expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;259  });260261  afterEach(async () => {262    await usingPlaygrounds(async helper => {263      if (await maintenanceEnabled(helper.getApi())) {264        console.warn('\tMaintenance mode was left enabled AFTER a test has finished! Be careful. Disabling it now.');265        await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);266      }267      expect(await maintenanceEnabled(helper.getApi()), 'Disastrous! Exited the test suite with maintenance mode on.').to.be.false;268    });269  });270});