1234567891011121314151617import privateKey from '../substrate/privateKey';18import {expect} from 'chai';19import {20 contractHelpers,21 createEthAccountWithBalance,22 transferBalanceToEth,23 deployFlipper,24 itWeb3,25 SponsoringMode,26 createEthAccount,27} from './util/helpers';2829describe('Sponsoring EVM contracts', () => {30 itWeb3('Sponsoring can be set by the address that has deployed the contract', async ({api, web3}) => {31 const owner = await createEthAccountWithBalance(api, web3);32 const flipper = await deployFlipper(web3, owner);33 const helpers = contractHelpers(web3, owner);34 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;35 await helpers.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Allowlisted).send({from: owner});36 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.true;37 });3839 itWeb3('Sponsoring cannot be set by the address that did not deployed the contract', async ({api, web3}) => {40 const owner = await createEthAccountWithBalance(api, web3);41 const notOwner = await createEthAccountWithBalance(api, web3);42 const flipper = await deployFlipper(web3, owner);43 const helpers = contractHelpers(web3, owner);44 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;45 await expect(helpers.methods.setSponsoringMode(notOwner, SponsoringMode.Allowlisted).send({from: notOwner})).to.rejected;46 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;47 });4849 itWeb3('In generous mode, non-allowlisted user transaction will be sponsored', async ({api, web3}) => {50 const alice = privateKey('//Alice');5152 const owner = await createEthAccountWithBalance(api, web3);53 const caller = await createEthAccountWithBalance(api, web3);5455 const flipper = await deployFlipper(web3, owner);5657 const helpers = contractHelpers(web3, owner);5859 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;60 await helpers.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: owner});61 await helpers.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: owner});62 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.true;6364 await transferBalanceToEth(api, alice, flipper.options.address);6566 const originalFlipperBalance = await web3.eth.getBalance(flipper.options.address);67 expect(originalFlipperBalance).to.be.not.equal('0');6869 await flipper.methods.flip().send({from: caller});70 expect(await flipper.methods.getValue().call()).to.be.true;7172 73 const balanceAfter = await web3.eth.getBalance(flipper.options.address);74 expect(+balanceAfter).to.be.lessThan(+originalFlipperBalance);75 });7677 itWeb3('Sponsoring is set, an address that has no UNQ can send a transaction and it works. Sponsor balance should decrease (allowlisted)', async ({api, web3}) => {78 const alice = privateKey('//Alice');7980 const owner = await createEthAccountWithBalance(api, web3);81 const caller = createEthAccount(web3);8283 const flipper = await deployFlipper(web3, owner);8485 const helpers = contractHelpers(web3, owner);86 await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});87 await helpers.methods.toggleAllowed(flipper.options.address, caller, true).send({from: owner});8889 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;90 await helpers.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Allowlisted).send({from: owner});91 await helpers.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: owner});92 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.true;9394 await transferBalanceToEth(api, alice, flipper.options.address);9596 const originalFlipperBalance = await web3.eth.getBalance(flipper.options.address);97 expect(originalFlipperBalance).to.be.not.equal('0');9899 await flipper.methods.flip().send({from: caller});100 expect(await flipper.methods.getValue().call()).to.be.true;101102 103 const balanceAfter = await web3.eth.getBalance(flipper.options.address);104 expect(+balanceAfter).to.be.lessThan(+originalFlipperBalance);105 });106107 itWeb3('Sponsoring is set, an address that has no UNQ can send a transaction and it works. Sponsor balance should not decrease (non-allowlisted)', async ({api, web3}) => {108 const alice = privateKey('//Alice');109110 const owner = await createEthAccountWithBalance(api, web3);111 const caller = createEthAccount(web3);112113 const flipper = await deployFlipper(web3, owner);114115 const helpers = contractHelpers(web3, owner);116117 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;118 await helpers.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Allowlisted).send({from: owner});119 await helpers.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: owner});120 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.true;121122 await transferBalanceToEth(api, alice, flipper.options.address);123124 const originalFlipperBalance = await web3.eth.getBalance(flipper.options.address);125 expect(originalFlipperBalance).to.be.not.equal('0');126127 await expect(flipper.methods.flip().send({from: caller})).to.be.rejectedWith(/InvalidTransaction::Payment/);128 expect(await flipper.methods.getValue().call()).to.be.false;129130 131 const balanceAfter = await web3.eth.getBalance(flipper.options.address);132 expect(+balanceAfter).to.be.equals(+originalFlipperBalance);133 });134135 itWeb3('Sponsoring is set, an address that has UNQ can send a transaction and it works. User balance should not change', async ({api, web3}) => {136 const alice = privateKey('//Alice');137138 const owner = await createEthAccountWithBalance(api, web3);139 const caller = await createEthAccountWithBalance(api, web3);140 const originalCallerBalance = await web3.eth.getBalance(caller);141142 const flipper = await deployFlipper(web3, owner);143144 const helpers = contractHelpers(web3, owner);145 await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});146 await helpers.methods.toggleAllowed(flipper.options.address, caller, true).send({from: owner});147148 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;149 await helpers.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Allowlisted).send({from: owner});150 await helpers.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: owner});151 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.true;152153 await transferBalanceToEth(api, alice, flipper.options.address);154155 const originalFlipperBalance = await web3.eth.getBalance(flipper.options.address);156 expect(originalFlipperBalance).to.be.not.equal('0');157158 await flipper.methods.flip().send({from: caller});159 expect(await flipper.methods.getValue().call()).to.be.true;160161 expect(await web3.eth.getBalance(caller)).to.be.equals(originalCallerBalance);162 });163164 itWeb3('Sponsoring is limited, with setContractRateLimit. The limitation is working if transactions are sent more often, the sender pays the commission.', async ({api, web3}) => {165 const alice = privateKey('//Alice');166167 const owner = await createEthAccountWithBalance(api, web3);168 const caller = await createEthAccountWithBalance(api, web3);169 const originalCallerBalance = await web3.eth.getBalance(caller);170171 const flipper = await deployFlipper(web3, owner);172173 const helpers = contractHelpers(web3, owner);174 await helpers.methods.toggleAllowlist(flipper.options.address, true).send({from: owner});175 await helpers.methods.toggleAllowed(flipper.options.address, caller, true).send({from: owner});176177 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.false;178 await helpers.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Allowlisted).send({from: owner});179 await helpers.methods.setSponsoringRateLimit(flipper.options.address, 10).send({from: owner});180 expect(await helpers.methods.sponsoringEnabled(flipper.options.address).call()).to.be.true;181182 await transferBalanceToEth(api, alice, flipper.options.address);183184 const originalFlipperBalance = await web3.eth.getBalance(flipper.options.address);185 expect(originalFlipperBalance).to.be.not.equal('0');186187 await flipper.methods.flip().send({from: caller});188 expect(await flipper.methods.getValue().call()).to.be.true;189 expect(await web3.eth.getBalance(caller)).to.be.equals(originalCallerBalance);190191 await flipper.methods.flip().send({from: caller});192 expect(await web3.eth.getBalance(caller)).to.be.not.equals(originalCallerBalance);193 });194195 196 itWeb3('Default rate limit equals 7200', async ({api, web3}) => {197 const owner = await createEthAccountWithBalance(api, web3);198 const flipper = await deployFlipper(web3, owner);199 const helpers = contractHelpers(web3, owner);200 expect(await helpers.methods.getSponsoringRateLimit(flipper.options.address).call()).to.be.equals('7200');201 });202});