git.delta.rocks / unique-network / refs/commits / 9668fc5ff8f9

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 {Pallets} from '../util';19import {itEth, usingEthPlaygrounds, expect} from './util';2021describe('EVM contract allowlist', () => {22  let donor: IKeyringPair;2324  before(async function() {25    await usingEthPlaygrounds(async (_helper, privateKey) => {26      donor = await privateKey({filename: __filename});27    });28  });2930  itEth('Contract allowlist can be toggled', async ({helper}) => {31    const owner = await helper.eth.createAccountWithBalance(donor);32    const flipper = await helper.eth.deployFlipper(owner);33    const helpers = helper.ethNativeContract.contractHelpers(owner);3435    // Any user is allowed by default36    expect(await helpers.methods.allowlistEnabled(flipper.options.address).call()).to.be.false;3738    // Enable39    await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});40    expect(await helpers.methods.allowlistEnabled(flipper.options.address).call()).to.be.true;4142    // Disable43    await helpers.methods.toggleAllowlist(flipper.options.address, false).send({from: owner});44    expect(await helpers.methods.allowlistEnabled(flipper.options.address).call()).to.be.false;45  });4647  itEth('Non-allowlisted user can\'t call contract with allowlist enabled', async ({helper}) => {48    const owner = await helper.eth.createAccountWithBalance(donor);49    const caller = await helper.eth.createAccountWithBalance(donor);50    const flipper = await helper.eth.deployFlipper(owner);51    const helpers = helper.ethNativeContract.contractHelpers(owner);5253    // User can flip with allowlist disabled54    await flipper.methods.flip().send({from: caller});55    expect(await flipper.methods.getValue().call()).to.be.true;5657    // Tx will be reverted if user is not in allowlist58    await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});59    await expect(flipper.methods.flip().send({from: caller})).to.rejected;60    expect(await flipper.methods.getValue().call()).to.be.true;6162    // Adding caller to allowlist will make contract callable again63    await helpers.methods.toggleAllowed(flipper.options.address, caller, true).send({from: owner});64    await flipper.methods.flip().send({from: caller});65    expect(await flipper.methods.getValue().call()).to.be.false;66  });67});6869describe('EVM collection allowlist', () => {70  let donor: IKeyringPair;7172  before(async function() {73    await usingEthPlaygrounds(async (_helper, privateKey) => {74      donor = await privateKey({filename: __filename});75    });76  });7778  // Soft-deprecated79  itEth('Collection allowlist can be added and removed by [eth] address', async ({helper}) => {80    const owner = await helper.eth.createAccountWithBalance(donor);81    const user = helper.eth.createAccount();82    const crossUser = helper.ethCrossAccount.fromAddress(user);83    84    const {collectionAddress} = await helper.eth.createNFTCollection(owner, 'A', 'B', 'C');85    const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner, true);8687    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.false;88    await collectionEvm.methods.addToCollectionAllowList(user).send({from: owner});89    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.true;9091    await collectionEvm.methods.removeFromCollectionAllowList(user).send({from: owner});92    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.false;93  });949596  [97    {mode: 'nft' as const, requiredPallets: []},98    {mode: 'rft' as const, requiredPallets: [Pallets.ReFungible]},99    {mode: 'ft' as const, requiredPallets: []},100  ].map(testCase => 101    itEth(`Collection allowlist can be added and removed by [cross] address fro ${testCase.mode}`, async ({helper}) => {102      const owner = (await helper.eth.createAccountWithBalance(donor)).toLowerCase();103      const [userSub] = await helper.arrange.createAccounts([10n], donor);104      const userEth = await helper.eth.createAccountWithBalance(donor);105      const mintParams = testCase.mode === 'ft' ? [userEth, 100] : [userEth];106      107      const {collectionAddress, collectionId} = await helper.eth.createCollection(testCase.mode, owner, 'A', 'B', 'C');108      const collectionEvm = helper.ethNativeContract.collection(collectionAddress, testCase.mode, owner);109      const userCrossSub = helper.ethCrossAccount.fromKeyringPair(userSub);110      const userCrossEth = helper.ethCrossAccount.fromAddress(userEth);111      const ownerCrossEth = helper.ethCrossAccount.fromAddress(owner);112      113      // Can addToCollectionAllowListCross:114      expect(await helper.collection.allowed(collectionId, {Substrate: userSub.address})).to.be.false;115      await collectionEvm.methods.addToCollectionAllowListCross(userCrossSub).send({from: owner});116      await collectionEvm.methods.addToCollectionAllowListCross(userCrossEth).send({from: owner});117      await collectionEvm.methods.addToCollectionAllowListCross(ownerCrossEth).send({from: owner});118119      // Accounts are in allowed list:120      expect(await helper.collection.allowed(collectionId, {Substrate: userSub.address})).to.be.true;121      expect(await helper.collection.allowed(collectionId, {Ethereum: userEth})).to.be.true;122      expect(await collectionEvm.methods.allowlistedCross(userCrossSub).call({from: owner})).to.be.true;123      expect(await collectionEvm.methods.allowlistedCross(userCrossEth).call({from: owner})).to.be.true;124  125      await collectionEvm.methods.mint(...mintParams).send({from: owner}); // token #1126      await collectionEvm.methods.mint(...mintParams).send({from: owner}); // token #2127      await collectionEvm.methods.setCollectionAccess(1 /* AllowList */).send({from: owner});128      129      // allowlisted account can transfer and transferCross from eth:130      await collectionEvm.methods.transfer(owner, 1).send({from: userEth});131      await collectionEvm.methods.transferCross(userCrossSub, 2).send({from: userEth});132133      if (testCase.mode === 'ft') {134        expect(await helper.ft.getBalance(collectionId, {Ethereum: owner})).to.eq(1n);135        expect(await helper.ft.getBalance(collectionId, {Substrate: userSub.address})).to.eq(2n);136      } else {137        expect(await helper.nft.getTokenOwner(collectionId, 1)).to.deep.eq({Ethereum: owner});138        expect(await helper.nft.getTokenOwner(collectionId, 2)).to.deep.eq({Substrate: userSub.address});139      }140141      // allowlisted cross substrate accounts can transfer from Substrate:142      testCase.mode === 'ft'143        ? await helper.ft.transfer(userSub, collectionId, {Ethereum: userEth}, 2n)144        : await helper.collection.transferToken(userSub, collectionId, 2, {Ethereum: userEth});145      146      // can removeFromCollectionAllowListCross:147      await collectionEvm.methods.removeFromCollectionAllowListCross(userCrossSub).send({from: owner});148      await collectionEvm.methods.removeFromCollectionAllowListCross(userCrossEth).send({from: owner});149      expect(await helper.collection.allowed(collectionId, {Substrate: userSub.address})).to.be.false;150      expect(await helper.collection.allowed(collectionId, {Ethereum: userEth})).to.be.false;151      expect(await collectionEvm.methods.allowlistedCross(userCrossSub).call({from: owner})).to.be.false;152      expect(await collectionEvm.methods.allowlistedCross(userCrossEth).call({from: owner})).to.be.false;153  154      // cannot transfer anymore155      await collectionEvm.methods.mint(...mintParams).send({from: owner});156      await expect(collectionEvm.methods.transfer(owner, 2).send({from: userEth})).to.be.rejectedWith(/Transaction has been reverted/);157    }));158159  // Soft-deprecated160  itEth('Collection allowlist can not be add and remove [eth] address by not owner', async ({helper}) => {161    const owner = await helper.eth.createAccountWithBalance(donor);162    const notOwner = await helper.eth.createAccountWithBalance(donor);163    const user = helper.eth.createAccount();164    const crossUser = helper.ethCrossAccount.fromAddress(user);165166    const {collectionAddress} = await helper.eth.createNFTCollection(owner, 'A', 'B', 'C');167    const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner, true);168169    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.false;170    await expect(collectionEvm.methods.addToCollectionAllowList(user).call({from: notOwner})).to.be.rejectedWith('NoPermission');171    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.false;172    await collectionEvm.methods.addToCollectionAllowList(user).send({from: owner});173    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.true;174    await expect(collectionEvm.methods.removeFromCollectionAllowList(user).call({from: notOwner})).to.be.rejectedWith('NoPermission');175    expect(await collectionEvm.methods.allowlistedCross(crossUser).call({from: owner})).to.be.true;176  });177178  itEth('Collection allowlist can not be add and remove [cross] address by not owner', async ({helper}) => {179    const owner = await helper.eth.createAccountWithBalance(donor);180    const notOwner = await helper.eth.createAccountWithBalance(donor);181    const user = donor;182    183    const {collectionAddress, collectionId} = await helper.eth.createNFTCollection(owner, 'A', 'B', 'C');184    const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);185    186    expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.false;187    const userCross = helper.ethCrossAccount.fromKeyringPair(user);188    await expect(collectionEvm.methods.addToCollectionAllowListCross(userCross).call({from: notOwner})).to.be.rejectedWith('NoPermission');189    expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.false;190    await collectionEvm.methods.addToCollectionAllowListCross(userCross).send({from: owner});191    192    expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.true;193    await expect(collectionEvm.methods.removeFromCollectionAllowListCross(userCross).call({from: notOwner})).to.be.rejectedWith('NoPermission');194    expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.true;195  });196});