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

1//2// This file is subject to the terms and conditions defined in3// file 'LICENSE', which is part of this source code package.4//56import chai from 'chai';7import chaiAsPromised from 'chai-as-promised';8import usingApi, { submitTransactionAsync, submitTransactionExpectFailAsync } from './substrate/substrate-api';9import privateKey from './substrate/privateKey';10import {11  deployFlipper,12  getFlipValue,13} from './util/contracthelpers';14import {15  getGenericResult,16} from './util/helpers';1718chai.use(chaiAsPromised);19const expect = chai.expect;2021const value = 0;22const gasLimit = 3000n * 1000000n;2324describe('Integration Test toggleContractWhiteList', () => {2526  it('Enable white list contract mode', async () => {27    await usingApi(async api => {28      const [contract, deployer] = await deployFlipper(api);2930      const enabledBefore = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();31      const enableWhiteListTx = api.tx.nft.toggleContractWhiteList(contract.address, true);32      const enableEvents = await submitTransactionAsync(deployer, enableWhiteListTx);33      const enabled = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();3435      expect(getGenericResult(enableEvents).success).to.be.true;36      expect(enabledBefore).to.be.false;37      expect(enabled).to.be.true;38    });39  });4041  it('Only whitelisted account can call contract', async () => {42    await usingApi(async api => {43      const bob = privateKey('//Bob');4445      const [contract, deployer] = await deployFlipper(api);4647      let flipValueBefore = await getFlipValue(contract, deployer);48      const flip = contract.tx.flip(value, gasLimit);49      await submitTransactionAsync(bob, flip);50      const flipValueAfter = await getFlipValue(contract,deployer);51      expect(flipValueAfter).to.be.eq(!flipValueBefore, 'Anyone can call new contract.');5253      const deployerCanFlip = async () => {54        let flipValueBefore = await getFlipValue(contract, deployer);55        const deployerFlip = contract.tx.flip(value, gasLimit);56        await submitTransactionAsync(deployer, deployerFlip);57        const aliceFlip1Response = await getFlipValue(contract, deployer);58        expect(aliceFlip1Response).to.be.eq(!flipValueBefore, 'Deployer always can flip.');59      };60      await deployerCanFlip();6162      flipValueBefore = await getFlipValue(contract, deployer);63      const enableWhiteListTx = api.tx.nft.toggleContractWhiteList(contract.address, true);64      await submitTransactionAsync(deployer, enableWhiteListTx);65      const flipWithEnabledWhiteList = contract.tx.flip(value, gasLimit);66      await expect(submitTransactionExpectFailAsync(bob, flipWithEnabledWhiteList)).to.be.rejected;67      const flipValueAfterEnableWhiteList = await getFlipValue(contract, deployer);68      expect(flipValueAfterEnableWhiteList).to.be.eq(flipValueBefore, 'Enabling whitelist doesn\'t make it possible to call contract for everyone.');6970      await deployerCanFlip();7172      flipValueBefore = await getFlipValue(contract, deployer);73      const addBobToWhiteListTx = api.tx.nft.addToContractWhiteList(contract.address, bob.address);74      await submitTransactionAsync(deployer, addBobToWhiteListTx);75      const flipWithWhitelistedBob = contract.tx.flip(value, gasLimit);76      await submitTransactionAsync(bob, flipWithWhitelistedBob);77      const flipAfterWhiteListed = await getFlipValue(contract,deployer);78      expect(flipAfterWhiteListed).to.be.eq(!flipValueBefore, 'Bob was whitelisted, now he can flip.');7980      await deployerCanFlip();8182      flipValueBefore = await getFlipValue(contract, deployer);83      const removeBobFromWhiteListTx = api.tx.nft.removeFromContractWhiteList(contract.address, bob.address);84      await submitTransactionAsync(deployer, removeBobFromWhiteListTx);85      const bobRemoved = contract.tx.flip(value, gasLimit);86      await expect(submitTransactionExpectFailAsync(bob, bobRemoved)).to.be.rejected;87      const afterBobRemoved = await getFlipValue(contract, deployer);88      expect(afterBobRemoved).to.be.eq(flipValueBefore, 'Bob can\'t call contract, now when he is removeed from white list.');8990      await deployerCanFlip();9192      flipValueBefore = await getFlipValue(contract, deployer);93      const disableWhiteListTx = api.tx.nft.toggleContractWhiteList(contract.address, false);94      await submitTransactionAsync(deployer, disableWhiteListTx);95      const whiteListDisabledFlip = contract.tx.flip(value, gasLimit);96      await submitTransactionAsync(bob, whiteListDisabledFlip);97      const afterWhiteListDisabled = await getFlipValue(contract,deployer);98      expect(afterWhiteListDisabled).to.be.eq(!flipValueBefore, 'Anyone can call contract with disabled whitelist.');99100    });101  });102103  it('Enabling white list repeatedly should not produce errors', async () => {104    await usingApi(async api => {105      const [contract, deployer] = await deployFlipper(api);106107      const enabledBefore = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();108      const enableWhiteListTx = api.tx.nft.toggleContractWhiteList(contract.address, true);109      const enableEvents = await submitTransactionAsync(deployer, enableWhiteListTx);110      const enabled = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();111      const enableAgainEvents = await submitTransactionAsync(deployer, enableWhiteListTx);112      const enabledAgain = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();113114      expect(getGenericResult(enableEvents).success).to.be.true;115      expect(enabledBefore).to.be.false;116      expect(enabled).to.be.true;117      expect(getGenericResult(enableAgainEvents).success).to.be.true;118      expect(enabledAgain).to.be.true;119    });120  });121122});123124describe('Negative Integration Test toggleContractWhiteList', () => {125126  it('Enable white list for a non-contract', async () => {127    await usingApi(async api => {128      const alice = privateKey('//Alice');129      const bobGuineaPig = privateKey('//Bob');130131      const enabledBefore = (await api.query.nft.contractWhiteListEnabled(bobGuineaPig.address)).toJSON();132      const enableWhiteListTx = api.tx.nft.toggleContractWhiteList(bobGuineaPig.address, true);133      await expect(submitTransactionExpectFailAsync(alice, enableWhiteListTx)).to.be.rejected;134      const enabled = (await api.query.nft.contractWhiteListEnabled(bobGuineaPig.address)).toJSON();135136      expect(enabledBefore).to.be.false;137      expect(enabled).to.be.false;138    });139  });140141  it('Enable white list using a non-owner address', async () => {142    await usingApi(async api => {143      const bob = privateKey('//Bob');144      const [contract] = await deployFlipper(api);145146      const enabledBefore = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();147      const enableWhiteListTx = api.tx.nft.toggleContractWhiteList(contract.address, true);148      await expect(submitTransactionExpectFailAsync(bob, enableWhiteListTx)).to.be.rejected;149      const enabled = (await api.query.nft.contractWhiteListEnabled(contract.address)).toJSON();150151      expect(enabledBefore).to.be.false;152      expect(enabled).to.be.false;153    });154  });155156});