git.delta.rocks / unique-network / refs/commits / 7aba0b5d7f8d

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 {expect} from 'chai';19import {isAllowlisted, normalizeAccountId} from '../util/helpers';20import {21  contractHelpers,22  createEthAccount,23  createEthAccountWithBalance,24  deployFlipper,25  evmCollection,26  evmCollectionHelpers,27  getCollectionAddressFromResult,28  itWeb3,29} from './util/helpers';30import {itEth, usingEthPlaygrounds} from './util/playgrounds';3132describe('EVM contract allowlist', () => {33  itWeb3('Contract allowlist can be toggled', async ({api, web3, privateKeyWrapper}) => {34    const owner = await createEthAccountWithBalance(api, web3, privateKeyWrapper);35    const flipper = await deployFlipper(web3, owner);3637    const helpers = contractHelpers(web3, owner);3839    // Any user is allowed by default40    expect(await helpers.methods.allowlistEnabled(flipper.options.address).call()).to.be.false;4142    // Enable43    await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});44    expect(await helpers.methods.allowlistEnabled(flipper.options.address).call()).to.be.true;4546    // Disable47    await helpers.methods.toggleAllowlist(flipper.options.address, false).send({from: owner});48    expect(await helpers.methods.allowlistEnabled(flipper.options.address).call()).to.be.false;49  });5051  itWeb3('Non-allowlisted user can\'t call contract with allowlist enabled', async ({api, web3, privateKeyWrapper}) => {52    const owner = await createEthAccountWithBalance(api, web3, privateKeyWrapper);53    const flipper = await deployFlipper(web3, owner);54    const caller = await createEthAccountWithBalance(api, web3, privateKeyWrapper);5556    const helpers = contractHelpers(web3, owner);5758    // User can flip with allowlist disabled59    await flipper.methods.flip().send({from: caller});60    expect(await flipper.methods.getValue().call()).to.be.true;6162    // Tx will be reverted if user is not in allowlist63    await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});64    await expect(flipper.methods.flip().send({from: caller})).to.rejected;65    expect(await flipper.methods.getValue().call()).to.be.true;6667    // Adding caller to allowlist will make contract callable again68    await helpers.methods.toggleAllowed(flipper.options.address, caller, true).send({from: owner});69    await flipper.methods.flip().send({from: caller});70    expect(await flipper.methods.getValue().call()).to.be.false;71  });72});7374describe('EVM collection allowlist', () => {75  let donor: IKeyringPair;7677  before(async function() {78    await usingEthPlaygrounds(async (_helper, privateKey) => {79      donor = privateKey('//Alice');80    });81  });82  83  itEth('Collection allowlist can be added and removed by [eth] address', async ({helper}) => {84    const owner = await helper.eth.createAccountWithBalance(donor);85    const user = helper.eth.createAccount();86    87    const {collectionAddress} = await helper.eth.createNonfungibleCollection(owner, 'A', 'B', 'C');88    const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);89    90    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.false;91    await collectionEvm.methods.addToCollectionAllowList(user).send({from: owner});92    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.true;93    94    await collectionEvm.methods.removeFromCollectionAllowList(user).send({from: owner});95    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.false;96  });9798  // TODO: Temprorary off. Need refactor99  // itEth('Collection allowlist can be added and removed by [sub] address', async ({helper}) => {100  //   const owner = await helper.eth.createAccountWithBalance(donor);101  //   const user = donor;102    103  //   const {collectionAddress, collectionId} = await helper.eth.createNonfungibleCollection(owner, 'A', 'B', 'C');104  //   const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);105    106  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.false;107  //   await collectionEvm.methods.addToCollectionAllowListSubstrate(user.addressRaw).send({from: owner});108  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.true;109    110  //   await collectionEvm.methods.removeFromCollectionAllowListSubstrate(user.addressRaw).send({from: owner});111  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.false;112  // });113114  itEth('Collection allowlist can not be add and remove [eth] address by not owner', async ({helper}) => {115    const owner = await helper.eth.createAccountWithBalance(donor);116    const notOwner = await helper.eth.createAccountWithBalance(donor);117    const user = helper.eth.createAccount();118    119    const {collectionAddress} = await helper.eth.createNonfungibleCollection(owner, 'A', 'B', 'C');120    const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);121    122    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.false;123    await expect(collectionEvm.methods.addToCollectionAllowList(user).call({from: notOwner})).to.be.rejectedWith('NoPermission');124    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.false;125    await collectionEvm.methods.addToCollectionAllowList(user).send({from: owner});126    127    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.true;128    await expect(collectionEvm.methods.removeFromCollectionAllowList(user).call({from: notOwner})).to.be.rejectedWith('NoPermission');129    expect(await collectionEvm.methods.allowed(user).call({from: owner})).to.be.true;130  });131132  // TODO: Temprorary off. Need refactor133  // itEth('Collection allowlist can not be add and remove [sub] address by not owner', async ({helper}) => {134  //   const owner = await helper.eth.createAccountWithBalance(donor);135  //   const notOwner = await helper.eth.createAccountWithBalance(donor);136  //   const user = donor;137    138  //   const {collectionAddress, collectionId} = await helper.eth.createNonfungibleCollection(owner, 'A', 'B', 'C');139  //   const collectionEvm = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);140    141  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.false;142  //   await expect(collectionEvm.methods.addToCollectionAllowListSubstrate(user.addressRaw).call({from: notOwner})).to.be.rejectedWith('NoPermission');143  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.false;144  //   await collectionEvm.methods.addToCollectionAllowListSubstrate(user.addressRaw).send({from: owner});145    146  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.true;147  //   await expect(collectionEvm.methods.removeFromCollectionAllowListSubstrate(user.addressRaw).call({from: notOwner})).to.be.rejectedWith('NoPermission');148  //   expect(await helper.collection.allowed(collectionId, {Substrate: user.address})).to.be.true;149  // });150});