git.delta.rocks / unique-network / refs/commits / 74c60c798ed6

difftreelog

CORE-300 Add tests

Trubnikov Sergey2022-03-18parent: #d030d8a.patch.diff
in: master

3 files changed

modifiedpallets/unique/src/tests.rsdiffbeforeafterboth
--- a/pallets/unique/src/tests.rs
+++ b/pallets/unique/src/tests.rs
@@ -18,6 +18,7 @@
 use super::*;
 use crate::mock::*;
 use crate::{AccessMode, CollectionMode};
+use sp_runtime::AccountId32;
 use up_data_structs::{
 	COLLECTION_NUMBER_LIMIT, CollectionId, CreateItemData, CreateFungibleData, CreateNftData,
 	CreateReFungibleData, MAX_DECIMAL_POINTS, COLLECTION_ADMINS_LIMIT, MetaUpdatePermission,
@@ -2854,3 +2855,34 @@
 		);
 	});
 }
+
+#[test]
+fn collection_sponsoring() {
+	new_test_ext().execute_with(|| {
+		// default_limits();
+		let user1 = 1_u64;
+		let user2 = 777_u64;
+		let origin1 = Origin::signed(user1);
+		let origin2 = Origin::signed(user2);
+		let account2 = account(user2);
+		
+		let collection_id =
+		create_test_collection_for_owner(&CollectionMode::NFT, user1, CollectionId(1));
+		assert_ok!(TemplateModule::set_collection_sponsor(origin1.clone(), collection_id, user1));
+		assert_ok!(TemplateModule::confirm_sponsorship(origin1.clone(), collection_id));
+
+		// Expect error while have no permissions
+		assert!(TemplateModule::create_item(origin2.clone(), collection_id, account2.clone(), default_nft_data().into()).is_err());
+
+		assert_ok!(TemplateModule::set_public_access_mode(origin1.clone(), collection_id, AccessMode::AllowList));
+		assert_ok!(TemplateModule::add_to_allow_list(origin1.clone(), collection_id, account2.clone()));
+		assert_ok!(TemplateModule::set_mint_permission(origin1.clone(), collection_id, true));
+
+		assert_eq!(<pallet_balances::Pallet<Test>>::free_balance(user2), 0);
+		let balance_before = <pallet_balances::Pallet<Test>>::free_balance(user1);
+		
+		assert_ok!(TemplateModule::create_item(origin2, collection_id, account2, default_nft_data().into()));
+		let balance_after = <pallet_balances::Pallet<Test>>::free_balance(user1);
+		assert_ne!(balance_before, balance_after);
+	});
+}
\ No newline at end of file
modifiedtests/src/eth/contractSponsoring.test.tsdiffbeforeafterboth
before · tests/src/eth/contractSponsoring.test.ts
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 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    // Balance should be taken from flipper instead of caller73    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    // Balance should be taken from flipper instead of caller103    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    // Balance should be taken from flipper instead of caller131    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  // TODO: Find a way to calculate default rate limit196  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});
modifiedtests/src/eth/util/helpers.tsdiffbeforeafterboth
--- a/tests/src/eth/util/helpers.ts
+++ b/tests/src/eth/util/helpers.ts
@@ -241,6 +241,12 @@
   return collector;
 }
 
+/** 
+ * pallet evm_contract_helpers
+ * @param web3 
+ * @param caller - eth address
+ * @returns 
+ */
 export function contractHelpers(web3: Web3, caller: string) {
   return new web3.eth.Contract(contractHelpersAbi as any, '0x842899ECF380553E8a4de75bF534cdf6fBF64049', {from: caller, ...GAS_ARGS});
 }