git.delta.rocks / unique-network / refs/commits / 1ae086c66ec9

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';2223async function maintenanceEnabled(api: ApiPromise): Promise<boolean> {24  return (await api.query.maintenance.enabled()).toJSON() as boolean;25}2627describe('Integration Test: Maintenance Functionality', () => {28  let superuser: IKeyringPair;29  let donor: IKeyringPair;30  let bob: IKeyringPair;3132  before(async function() {33    await usingPlaygrounds(async (helper, privateKey) => {34      requirePalletsOrSkip(this, helper, [Pallets.Maintenance]);35      superuser = await privateKey('//Alice');36      donor = await privateKey({filename: __filename});37      [bob] = await helper.arrange.createAccounts([10000n], donor);3839    });40  });4142  describe('Maintenance Mode', () => {43    before(async function() {44      await usingPlaygrounds(async (helper) => {45        if (await maintenanceEnabled(helper.getApi())) {46          console.warn('\tMaintenance mode was left enabled BEFORE the test suite! Disabling it now.');47          await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', [])).to.be.fulfilled;48        }49      });50    });5152    itSub('Allows superuser to enable and disable maintenance mode - and disallows anyone else', async ({helper}) => {53      // Make sure non-sudo can't enable maintenance mode54      await expect(helper.executeExtrinsic(superuser, 'api.tx.maintenance.enable', []), 'on commoner enabling MM')55        .to.be.rejectedWith(/BadOrigin/);5657      // Set maintenance mode58      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);59      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;6061      // Make sure non-sudo can't disable maintenance mode62      await expect(helper.executeExtrinsic(bob, 'api.tx.maintenance.disable', []), 'on commoner disabling MM')63        .to.be.rejectedWith(/BadOrigin/);6465      // Disable maintenance mode66      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);67      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;68    });6970    itSub('MM blocks unique pallet calls', async ({helper}) => {71      // Can create an NFT collection before enabling the MM72      const nftCollection = await helper.nft.mintCollection(bob, {73        tokenPropertyPermissions: [{key: 'test', permission: {74          collectionAdmin: true,75          tokenOwner: true,76          mutable: true,77        }}],78      });7980      // Can mint an NFT before enabling the MM81      const nft = await nftCollection.mintToken(bob);8283      // Can create an FT collection before enabling the MM84      const ftCollection = await helper.ft.mintCollection(superuser);8586      // Can mint an FT before enabling the MM87      await expect(ftCollection.mint(superuser)).to.be.fulfilled;8889      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);90      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;9192      // Unable to create a collection when the MM is enabled93      await expect(helper.nft.mintCollection(superuser), 'cudo forbidden stuff')94        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);9596      // Unable to set token properties when the MM is enabled97      await expect(nft.setProperties(98        bob,99        [{key: 'test', value: 'test-val'}],100      )).to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);101102      // Unable to mint an NFT when the MM is enabled103      await expect(nftCollection.mintToken(superuser))104        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);105106      // Unable to mint an FT when the MM is enabled107      await expect(ftCollection.mint(superuser))108        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);109110      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);111      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;112113      // Can create a collection after disabling the MM114      await expect(helper.nft.mintCollection(bob), 'MM is disabled, the collection should be created').to.be.fulfilled;115116      // Can set token properties after disabling the MM117      await nft.setProperties(bob, [{key: 'test', value: 'test-val'}]);118119      // Can mint an NFT after disabling the MM120      await nftCollection.mintToken(bob);121122      // Can mint an FT after disabling the MM123      await ftCollection.mint(superuser);124    });125126    itSub.ifWithPallets('MM blocks unique pallet calls (Re-Fungible)', [Pallets.ReFungible], async ({helper}) => {127      // Can create an RFT collection before enabling the MM128      const rftCollection = await helper.rft.mintCollection(superuser);129130      // Can mint an RFT before enabling the MM131      await expect(rftCollection.mintToken(superuser)).to.be.fulfilled;132133      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);134      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;135136      // Unable to mint an RFT when the MM is enabled137      await expect(rftCollection.mintToken(superuser))138        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);139140      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);141      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;142143      // Can mint an RFT after disabling the MM144      await rftCollection.mintToken(superuser);145    });146147    itSub('MM allows native token transfers and RPC calls', async ({helper}) => {148      // We can use RPC before the MM is enabled149      const totalCount = await helper.collection.getTotalCount();150151      // We can transfer funds before the MM is enabled152      await expect(helper.balance.transferToSubstrate(superuser, bob.address, 2n)).to.be.fulfilled;153154      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);155      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;156157      // RPCs work while in maintenance158      expect(await helper.collection.getTotalCount()).to.be.deep.equal(totalCount);159160      // We still able to transfer funds161      await expect(helper.balance.transferToSubstrate(bob, superuser.address, 1n)).to.be.fulfilled;162163      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);164      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;165166      // RPCs work after maintenance167      expect(await helper.collection.getTotalCount()).to.be.deep.equal(totalCount);168169      // Transfers work after maintenance170      await expect(helper.balance.transferToSubstrate(bob, superuser.address, 1n)).to.be.fulfilled;171    });172173    itSched.ifWithPallets('MM blocks scheduled calls and the scheduler itself', [Pallets.Scheduler], async (scheduleKind, {helper}) => {174      const collection = await helper.nft.mintCollection(bob);175176      const nftBeforeMM = await collection.mintToken(bob);177      const nftDuringMM = await collection.mintToken(bob);178      const nftAfterMM = await collection.mintToken(bob);179180      const [181        scheduledIdBeforeMM,182        scheduledIdDuringMM,183        scheduledIdBunkerThroughMM,184        scheduledIdAttemptDuringMM,185        scheduledIdAfterMM,186      ] = scheduleKind == 'named'187        ? helper.arrange.makeScheduledIds(5)188        : new Array(5);189190      const blocksToWait = 6;191192      // Scheduling works before the maintenance193      await nftBeforeMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdBeforeMM})194        .transfer(bob, {Substrate: superuser.address});195196      await helper.wait.newBlocks(blocksToWait + 1);197      expect(await nftBeforeMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});198199      // Schedule a transaction that should occur *during* the maintenance200      await nftDuringMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdDuringMM})201        .transfer(bob, {Substrate: superuser.address});202203      // Schedule a transaction that should occur *after* the maintenance204      await nftDuringMM.scheduleAfter(blocksToWait * 2, {scheduledId: scheduledIdBunkerThroughMM})205        .transfer(bob, {Substrate: superuser.address});206207      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);208      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;209210      await helper.wait.newBlocks(blocksToWait + 1);211      // The owner should NOT change since the scheduled transaction should be rejected212      expect(await nftDuringMM.getOwner()).to.be.deep.equal({Substrate: bob.address});213214      // Any attempts to schedule a tx during the MM should be rejected215      await expect(nftDuringMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdAttemptDuringMM})216        .transfer(bob, {Substrate: superuser.address}))217        .to.be.rejectedWith(/Invalid Transaction: Transaction call is not expected/);218219      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);220      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;221222      // Scheduling works after the maintenance223      await nftAfterMM.scheduleAfter(blocksToWait, {scheduledId: scheduledIdAfterMM})224        .transfer(bob, {Substrate: superuser.address});225226      await helper.wait.newBlocks(blocksToWait + 1);227228      expect(await nftAfterMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});229      // The owner of the token scheduled for transaction *before* maintenance should now change *after* maintenance230      expect(await nftDuringMM.getOwner()).to.be.deep.equal({Substrate: superuser.address});231    });232233    itEth('Disallows Ethereum transactions to execute while in maintenance', async ({helper}) => {234      const owner = await helper.eth.createAccountWithBalance(donor);235      const receiver = helper.eth.createAccount();236237      const {collectionAddress} = await helper.eth.createERC721MetadataCompatibleNFTCollection(owner, 'A', 'B', 'C', '');238239      // Set maintenance mode240      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);241      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;242243      const contract = await helper.ethNativeContract.collection(collectionAddress, 'nft', owner);244      const tokenId = await contract.methods.nextTokenId().call();245      expect(tokenId).to.be.equal('1');246247      await expect(contract.methods.mintWithTokenURI(receiver, 'Test URI').send())248        .to.be.rejectedWith(/Returned error: unknown error/);249250      await expect(contract.methods.ownerOf(tokenId).call()).rejectedWith(/token not found/);251252      // Disable maintenance mode253      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);254      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;255    });256257    itSub('Allows to enable and disable MM repeatedly', async ({helper}) => {258      // Set maintenance mode259      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);260      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.enable', []);261      expect(await maintenanceEnabled(helper.getApi()), 'MM is OFF when it should be ON').to.be.true;262263      // Disable maintenance mode264      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);265      await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);266      expect(await maintenanceEnabled(helper.getApi()), 'MM is ON when it should be OFF').to.be.false;267    });268269    afterEach(async () => {270      await usingPlaygrounds(async helper => {271        if (helper.fetchMissingPalletNames([Pallets.Maintenance]).length != 0) return;272        if (await maintenanceEnabled(helper.getApi())) {273          console.warn('\tMaintenance mode was left enabled AFTER a test has finished! Be careful. Disabling it now.');274          await helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.disable', []);275        }276        expect(await maintenanceEnabled(helper.getApi()), 'Disastrous! Exited the test suite with maintenance mode on.').to.be.false;277      });278    });279  });280281  describe('Preimage Execution', () => {282    let preimageHash: string;283284    async function notePreimage(helper: UniqueHelper, preimage: any): Promise<string> {285      const result = await helper.preimage.notePreimage(bob, preimage);286      const events = result.result.events.filter(x => x.event.method === 'Noted' && x.event.section === 'preimage');287      const preimageHash = events[0].event.data[0].toHuman();288      return preimageHash;289    }290291    before(async function() {292      await usingPlaygrounds(async (helper) => {293        requirePalletsOrSkip(this, helper, [Pallets.Preimage, Pallets.Maintenance]);294295        // create a preimage to be operated with in the tests296        const randomAccounts = await helper.arrange.createCrowd(10, 0n, superuser);297        const randomIdentities = randomAccounts.map((acc, i) => [298          acc.address, {299            deposit: 0n,300            judgements: [],301            info: {302              display: {303                raw: `Random Account #${i}`,304              },305            },306          },307        ]);308        const preimage = helper.constructApiCall('api.tx.identity.forceInsertIdentities', [randomIdentities]).method.toHex();309        preimageHash = await notePreimage(helper, preimage);310      });311    });312313    itSub('Successfully executes call in a preimage', async ({helper}) => {314      const result = await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [315        preimageHash, null, {refTime: 10000000000, proofSize: 10000000000},316      ])).to.be.fulfilled;317318      // preimage is executed, and an appropriate event is present319      const events = result.result.events.filter((x: any) => x.event.method === 'IdentitiesInserted' && x.event.section === 'identity');320      expect(events.length).to.be.equal(1);321322      // the preimage goes back to being unrequested323      expect(await helper.preimage.getPreimageInfo(preimageHash)).to.have.property('unrequested');324    });325326    itSub('Does not allow execution of a preimage that would fail', async ({helper}) => {327      const [zeroAccount] = await helper.arrange.createAccounts([0n], superuser);328329      const preimage = helper.constructApiCall('api.tx.balances.forceTransfer', [330        {Id: zeroAccount.address}, {Id: superuser.address}, 1000n,331      ]).method.toHex();332      const preimageHash = await notePreimage(helper, preimage);333334      await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [335        preimageHash, null, {refTime: 100000000000, proofSize: 100000000000},336      ])).to.be.rejectedWith(/balances\.InsufficientBalance/);337    });338339    itSub('Does not allow preimage execution with non-root', async ({helper}) => {340      await expect(helper.executeExtrinsic(bob, 'api.tx.maintenance.executePreimage', [341        preimageHash, null, {refTime: 100000000000, proofSize: 100000000000},342      ])).to.be.rejectedWith(/BadOrigin/);343    });344345    itSub('Does not allow execution of non-existent preimages', async ({helper}) => {346      await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [347        '0x1010101010101010101010101010101010101010101010101010101010101010', null, {refTime: 100000000000, proofSize: 100000000000},348      ])).to.be.rejectedWith(/Unavailable/);349    });350351    itSub('Does not allow preimage execution with less than minimum weights', async ({helper}) => {352      await expect(helper.getSudo().executeExtrinsic(superuser, 'api.tx.maintenance.executePreimage', [353        preimageHash, null, {refTime: 1000, proofSize: 1000},354      ])).to.be.rejectedWith(/Exhausted/);355    });356357    after(async function() {358      await usingPlaygrounds(async (helper) => {359        if (helper.fetchMissingPalletNames([Pallets.Preimage, Pallets.Maintenance]).length != 0) return;360361        await helper.preimage.unnotePreimage(bob, preimageHash);362      });363    });364  });365});