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

difftreelog

chore add Fractionalizer contract documentation, prevent QTZ/UNQ transfers from nonowners, tests for TransfersNotAllowed

Grigoriy Simonov2022-08-11parent: #6d585f9.patch.diff
in: master

3 files changed

modifiedpallets/refungible/src/erc_token.rsdiffbeforeafterboth
--- a/pallets/refungible/src/erc_token.rs
+++ b/pallets/refungible/src/erc_token.rs
@@ -186,7 +186,7 @@
 			.weight_calls_budget(<StructureWeight<T>>::find_parent());
 
 		<Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)
-			.map_err(|_| "transfer error")?;
+			.map_err(dispatch_to_evm::<T>)?;
 		Ok(true)
 	}
 
modifiedtests/src/eth/fractionalizer/Fractionalizer.soldiffbeforeafterboth
6import {UniqueRefungible} from "../api/UniqueRefungible.sol";6import {UniqueRefungible} from "../api/UniqueRefungible.sol";
7import {UniqueNFT} from "../api/UniqueNFT.sol";7import {UniqueNFT} from "../api/UniqueNFT.sol";
88
9/// @dev Fractionalization contract. It stores mappings between NFT and RFT tokens,
10/// stores allowlist of NFT tokens available for fractionalization, has methods
11/// for fractionalization and defractionalization of NFT tokens.
9contract Fractionalizer {12contract Fractionalizer {
10 struct Token {13 struct Token {
11 address _collection;14 address _collection;
17 mapping(address => Token) rft2nftMapping;20 mapping(address => Token) rft2nftMapping;
18 bytes32 refungibleCollectionType = keccak256(bytes("ReFungible"));21 bytes32 refungibleCollectionType = keccak256(bytes("ReFungible"));
1922
23 //TODO: add nonPayable modifier after Solidity updates to 0.9.
20 constructor() {24 receive() external payable onlyOwner {}
21 }25
2226 /// @dev Method modifier to only allow contract owner to call it.
23 modifier onlyOwner() {27 modifier onlyOwner() {
24 address contracthelpersAddress = 0x842899ECF380553E8a4de75bF534cdf6fBF64049;28 address contracthelpersAddress = 0x842899ECF380553E8a4de75bF534cdf6fBF64049;
25 ContractHelpers contractHelpers = ContractHelpers(contracthelpersAddress);29 ContractHelpers contractHelpers = ContractHelpers(contracthelpersAddress);
28 _;32 _;
29 }33 }
3034
35 /// @dev This emits when RFT collection setting is changed.
31 event RFTCollectionSet(address _collection);36 event RFTCollectionSet(address _collection);
37
38 /// @dev This emits when NFT collection is allowed or disallowed.
32 event AllowListSet(address _collection, bool _status);39 event AllowListSet(address _collection, bool _status);
40
41 /// @dev This emits when NFT token is fractionalized by contract.
33 event Fractionalized(address _collection, uint256 _tokenId, address _rftToken, uint128 _amount);42 event Fractionalized(address _collection, uint256 _tokenId, address _rftToken, uint128 _amount);
43
44 /// @dev This emits when NFT token is defractionalized by contract.
34 event Defractionalized(address _rftToken, address _nftCollection, uint256 _nftTokenId);45 event Defractionalized(address _rftToken, address _nftCollection, uint256 _nftTokenId);
3546
47 /// Set RFT collection that contract will work with. RFT tokens for fractionalized NFT tokens
48 /// would be created in this collection.
49 /// @dev Throws if RFT collection is already configured for this contract.
50 /// Throws if collection of wrong type (NFT, Fungible) is provided instead
51 /// of RFT collection.
52 /// Throws if `msg.sender` is not owner or admin of provided RFT collection.
53 /// Can only be called by contract owner.
54 /// @param _collection address of RFT collection.
36 function setRFTCollection(address _collection) public onlyOwner {55 function setRFTCollection(address _collection) public onlyOwner {
37 require(56 require(
38 rftCollection == address(0),57 rftCollection == address(0),
53 emit RFTCollectionSet(rftCollection);72 emit RFTCollectionSet(rftCollection);
54 }73 }
5574
75 /// Creates and sets RFT collection that contract will work with. RFT tokens for fractionalized NFT tokens
76 /// would be created in this collection.
77 /// @dev Throws if RFT collection is already configured for this contract.
78 /// Can only be called by contract owner.
79 /// @param _name name for created RFT collection.
80 /// @param _description description for created RFT collection.
81 /// @param _tokenPrefix token prefix for created RFT collection.
56 function createAndSetRFTCollection(string calldata _name, string calldata _description, string calldata _tokenPrefix) public onlyOwner {82 function createAndSetRFTCollection(string calldata _name, string calldata _description, string calldata _tokenPrefix) public onlyOwner {
57 require(83 require(
58 rftCollection == address(0),84 rftCollection == address(0),
63 emit RFTCollectionSet(rftCollection);89 emit RFTCollectionSet(rftCollection);
64 }90 }
6591
92 /// Allow or disallow NFT collection tokens from being fractionalized by this contract.
93 /// @dev Can only be called by contract owner.
94 /// @param collection NFT token address.
95 /// @param status `true` to allow and `false` to disallow NFT token.
66 function setNftCollectionIsAllowed(address collection, bool status) public onlyOwner {96 function setNftCollectionIsAllowed(address collection, bool status) public onlyOwner {
67 nftCollectionAllowList[collection] = status;97 nftCollectionAllowList[collection] = status;
68 emit AllowListSet(collection, status);98 emit AllowListSet(collection, status);
69 }99 }
70100
101 /// Fractionilize NFT token.
102 /// @dev Takes NFT token from `msg.sender` and transfers RFT token to `msg.sender`
103 /// instead. Creates new RFT token if provided NFT token never was fractionalized
104 /// by this contract or existing RFT token if it was.
105 /// Throws if RFT collection isn't configured for this contract.
106 /// Throws if fractionalization of provided NFT token is not allowed
107 /// Throws if `msg.sender` is not owner of provided NFT token
108 /// @param _collection NFT collection address
109 /// @param _token id of NFT token to be fractionalized
110 /// @param _pieces number of pieces new RFT token would have
71 function nft2rft(address _collection, uint256 _token, uint128 _pieces) public {111 function nft2rft(address _collection, uint256 _token, uint128 _pieces) public {
72 require(112 require(
73 rftCollection != address(0),113 rftCollection != address(0),
109 emit Fractionalized(_collection, _token, rftTokenAddress, _pieces);149 emit Fractionalized(_collection, _token, rftTokenAddress, _pieces);
110 }150 }
111151
152 /// Defrationalize NFT token.
153 /// @dev Takes RFT token from `msg.sender` and transfers corresponding NFT token
154 /// to `msg.sender` instead.
155 /// Throws if RFT collection isn't configured for this contract.
156 /// Throws if provided RFT token is no from configured RFT collection.
157 /// Throws if RFT token was not created by this contract.
158 /// Throws if `msg.sender` isn't owner of all RFT token pieces.
159 /// @param _collection RFT collection address
160 /// @param _token id of RFT token
112 function rft2nft(address _collection, uint256 _token) public {161 function rft2nft(address _collection, uint256 _token) public {
113 require(162 require(
114 rftCollection != address(0),163 rftCollection != address(0),
modifiedtests/src/eth/fractionalizer/fractionalizer.test.tsdiffbeforeafterboth
--- a/tests/src/eth/fractionalizer/fractionalizer.test.ts
+++ b/tests/src/eth/fractionalizer/fractionalizer.test.ts
@@ -19,17 +19,15 @@
 import {ApiPromise} from '@polkadot/api';
 import {evmToAddress} from '@polkadot/util-crypto';
 import {readFile} from 'fs/promises';
-import {submitTransactionAsync} from '../../substrate/substrate-api';
-import {UNIQUE} from '../../util/helpers';
+import {executeTransaction, submitTransactionAsync} from '../../substrate/substrate-api';
+import {getCreateCollectionResult, getCreateItemResult, UNIQUE} from '../../util/helpers';
 import {collectionIdToAddress, CompiledContract, createEthAccountWithBalance, createNonfungibleCollection, createRefungibleCollection, GAS_ARGS, itWeb3, tokenIdFromAddress, uniqueNFT, uniqueRefungible, uniqueRefungibleToken} from '../util/helpers';
 import {Contract} from 'web3-eth-contract';
 import * as solc from 'solc';
 
 import chai from 'chai';
-import chaiAsPromised from 'chai-as-promised';
 import chaiLike from 'chai-like';
 import {IKeyringPair} from '@polkadot/types/types';
-chai.use(chaiAsPromised);
 chai.use(chaiLike);
 const expect = chai.expect;
 let fractionalizer: CompiledContract;
@@ -93,9 +91,8 @@
 
 async function initFractionalizer(api: ApiPromise, web3: Web3, privateKeyWrapper: (account: string) => IKeyringPair, owner: string) {
   const fractionalizer = await deployFractionalizer(web3, owner);
-  const tx = api.tx.balances.transfer(evmToAddress(fractionalizer.options.address), 10n * UNIQUE);
-  const alice = privateKeyWrapper('//Alice');
-  await submitTransactionAsync(alice, tx);
+  const amount = 10n * UNIQUE;
+  await web3.eth.sendTransaction({from: owner, to: fractionalizer.options.address, value: `${amount}`, ...GAS_ARGS});
   const result = await fractionalizer.methods.createAndSetRFTCollection('A', 'B', 'C').send();
   const rftCollectionAddress = result.events.RFTCollectionSet.returnValues._collection;
   return {fractionalizer, rftCollectionAddress};
@@ -151,8 +148,7 @@
 
   itWeb3('Set Allowlist', async ({api, web3, privateKeyWrapper}) => {
     const owner = await createEthAccountWithBalance(api, web3, privateKeyWrapper);
-    const {fractionalizer} = await initFractionalizer(api, web3, privateKeyWrapper, owner);    
-
+    const {fractionalizer} = await initFractionalizer(api, web3, privateKeyWrapper, owner);
     const {collectionIdAddress: nftCollectionAddress} = await createNonfungibleCollection(api, web3, owner);
     const result1 = await fractionalizer.methods.setNftCollectionIsAllowed(nftCollectionAddress, true).send({from: owner});
     expect(result1.events).to.be.like({
@@ -238,7 +234,7 @@
     await fractionalizer.methods.setRFTCollection(collectionIdAddress).send();
 
     await expect(fractionalizer.methods.setRFTCollection(collectionIdAddress).call())
-      .to.eventually.be.rejectedWith(/RFT collection is already set$/g);
+      .to.be.rejectedWith(/RFT collection is already set$/g);
   });
 
   itWeb3('call setRFTCollection with NFT collection', async ({api, web3, privateKeyWrapper}) => {
@@ -250,7 +246,7 @@
     await nftContract.methods.addCollectionAdmin(fractionalizer.options.address).send();
 
     await expect(fractionalizer.methods.setRFTCollection(collectionIdAddress).call())
-      .to.eventually.be.rejectedWith(/Wrong collection type. Collection is not refungible.$/g);
+      .to.be.rejectedWith(/Wrong collection type. Collection is not refungible.$/g);
   });
 
   itWeb3('call setRFTCollection while not collection admin', async ({api, web3, privateKeyWrapper}) => {
@@ -259,7 +255,7 @@
     const {collectionIdAddress} = await createRefungibleCollection(api, web3, owner);
 
     await expect(fractionalizer.methods.setRFTCollection(collectionIdAddress).call())
-      .to.eventually.be.rejectedWith(/Fractionalizer contract should be an admin of the collection$/g);
+      .to.be.rejectedWith(/Fractionalizer contract should be an admin of the collection$/g);
   });
 
   itWeb3('call setRFTCollection after createAndSetRFTCollection', async ({api, web3, privateKeyWrapper}) => {
@@ -273,7 +269,7 @@
     const collectionIdAddress = result.events.RFTCollectionSet.returnValues._collection;
 
     await expect(fractionalizer.methods.setRFTCollection(collectionIdAddress).call())
-      .to.eventually.be.rejectedWith(/RFT collection is already set$/g);
+      .to.be.rejectedWith(/RFT collection is already set$/g);
   });
 
   itWeb3('call nft2rft without setting RFT collection for contract', async ({api, web3, privateKeyWrapper}) => {
@@ -287,7 +283,7 @@
     const fractionalizer = await deployFractionalizer(web3, owner);
 
     await expect(fractionalizer.methods.nft2rft(nftCollectionAddress, nftTokenId, 100).call())
-      .to.eventually.be.rejectedWith(/RFT collection is not set$/g);
+      .to.be.rejectedWith(/RFT collection is not set$/g);
   });
 
   itWeb3('call nft2rft while not owner of NFT token', async ({api, web3, privateKeyWrapper}) => {
@@ -305,7 +301,7 @@
     await fractionalizer.methods.setNftCollectionIsAllowed(nftCollectionAddress, true).send();
 
     await expect(fractionalizer.methods.nft2rft(nftCollectionAddress, nftTokenId, 100).call())
-      .to.eventually.be.rejectedWith(/Only token owner could fractionalize it$/g);
+      .to.be.rejectedWith(/Only token owner could fractionalize it$/g);
   });
 
   itWeb3('call nft2rft while not in list of allowed accounts', async ({api, web3, privateKeyWrapper}) => {
@@ -320,7 +316,7 @@
 
     await nftContract.methods.approve(fractionalizer.options.address, nftTokenId).send();
     await expect(fractionalizer.methods.nft2rft(nftCollectionAddress, nftTokenId, 100).call())
-      .to.eventually.be.rejectedWith(/Fractionalization of this collection is not allowed by admin$/g);
+      .to.be.rejectedWith(/Fractionalization of this collection is not allowed by admin$/g);
   });
 
   itWeb3('call nft2rft while fractionalizer doesnt have approval for nft token', async ({api, web3, privateKeyWrapper}) => {
@@ -335,7 +331,7 @@
 
     await fractionalizer.methods.setNftCollectionIsAllowed(nftCollectionAddress, true).send();
     await expect(fractionalizer.methods.nft2rft(nftCollectionAddress, nftTokenId, 100).call())
-      .to.eventually.be.rejectedWith(/ApprovedValueTooLow$/g);
+      .to.be.rejectedWith(/ApprovedValueTooLow$/g);
   });
 
   itWeb3('call rft2nft without setting RFT collection for contract', async ({api, web3, privateKeyWrapper}) => {
@@ -348,7 +344,7 @@
     await refungibleContract.methods.mint(owner, rftTokenId).send();
     
     await expect(fractionalizer.methods.rft2nft(rftCollectionAddress, rftTokenId).call())
-      .to.eventually.be.rejectedWith(/RFT collection is not set$/g);
+      .to.be.rejectedWith(/RFT collection is not set$/g);
   });
 
   itWeb3('call rft2nft for RFT token that is not from configured RFT collection', async ({api, web3, privateKeyWrapper}) => {
@@ -361,7 +357,7 @@
     await refungibleContract.methods.mint(owner, rftTokenId).send();
     
     await expect(fractionalizer.methods.rft2nft(rftCollectionAddress, rftTokenId).call())
-      .to.eventually.be.rejectedWith(/Wrong RFT collection$/g);
+      .to.be.rejectedWith(/Wrong RFT collection$/g);
   });
 
   itWeb3('call rft2nft for RFT token that was not minted by fractionalizer contract', async ({api, web3, privateKeyWrapper}) => {
@@ -378,7 +374,7 @@
     await refungibleContract.methods.mint(owner, rftTokenId).send();
     
     await expect(fractionalizer.methods.rft2nft(rftCollectionAddress, rftTokenId).call())
-      .to.eventually.be.rejectedWith(/No corresponding NFT token found$/g);
+      .to.be.rejectedWith(/No corresponding NFT token found$/g);
   });
 
   itWeb3('call rft2nft without owning all RFT pieces', async ({api, web3, privateKeyWrapper}) => {
@@ -393,6 +389,82 @@
     await refungibleTokenContract.methods.transfer(receiver, 50).send();
     await refungibleTokenContract.methods.approve(fractionalizer.options.address, 50).send();
     await expect(fractionalizer.methods.rft2nft(rftCollectionAddress, tokenId).call())
-      .to.eventually.be.rejectedWith(/Not all pieces are owned by the caller$/g);
+      .to.be.rejectedWith(/Not all pieces are owned by the caller$/g);
+  });
+
+  itWeb3('send QTZ/UNQ to contract from non owner', async ({api, web3, privateKeyWrapper}) => {
+    const owner = await createEthAccountWithBalance(api, web3, privateKeyWrapper);
+    const payer = await createEthAccountWithBalance(api, web3, privateKeyWrapper);
+
+    const fractionalizer = await deployFractionalizer(web3, owner);
+    const amount = 10n * UNIQUE;
+    await expect(web3.eth.sendTransaction({from: payer, to: fractionalizer.options.address, value: `${amount}`, ...GAS_ARGS})).to.be.rejected;
+  });
+
+  itWeb3('fractionalize NFT with NFT transfers disallowed', async ({api, web3, privateKeyWrapper}) => {
+    const alice = privateKeyWrapper('//Alice');
+    let collectionId;
+    {
+      const tx = api.tx.unique.createCollectionEx({name: 'A', description: 'B', tokenPrefix: 'C', mode: 'NFT'});
+      const events = await submitTransactionAsync(alice, tx);
+      const result = getCreateCollectionResult(events);
+      collectionId = result.collectionId;
+    }
+    const owner = await createEthAccountWithBalance(api, web3, privateKeyWrapper);
+    let nftTokenId;
+    {
+      const createData = {nft: {}};
+      const tx = api.tx.unique.createItem(collectionId, {Ethereum: owner}, createData as any);
+      const events = await executeTransaction(api, alice, tx);
+      const result = getCreateItemResult(events);
+      nftTokenId = result.itemId;
+    }
+    {
+      const tx = api.tx.unique.setTransfersEnabledFlag(collectionId, false);
+      await executeTransaction(api, alice, tx);
+    }
+    const nftCollectionAddress = collectionIdToAddress(collectionId);
+    const {fractionalizer} = await initFractionalizer(api, web3, privateKeyWrapper, owner);
+    await fractionalizer.methods.setNftCollectionIsAllowed(nftCollectionAddress, true).send();
+
+    const nftContract = uniqueNFT(web3, nftCollectionAddress, owner);
+    await nftContract.methods.approve(fractionalizer.options.address, nftTokenId).send();
+    await expect(fractionalizer.methods.nft2rft(nftCollectionAddress, nftTokenId, 100).call())
+      .to.be.rejectedWith(/TransferNotAllowed$/g);
+  });
+  
+  itWeb3('fractionalize NFT with RFT transfers disallowed', async ({api, web3, privateKeyWrapper}) => {
+    const owner = await createEthAccountWithBalance(api, web3, privateKeyWrapper);
+    const alice = privateKeyWrapper('//Alice');
+
+    let collectionId;
+    {
+      const tx = api.tx.unique.createCollectionEx({name: 'A', description: 'B', tokenPrefix: 'C', mode: 'ReFungible'});
+      const events = await submitTransactionAsync(alice, tx);
+      const result = getCreateCollectionResult(events);
+      collectionId = result.collectionId;
+    }
+    const rftCollectionAddress = collectionIdToAddress(collectionId);
+    const fractionalizer = await deployFractionalizer(web3, owner);
+    {
+      const changeAdminTx = api.tx.unique.addCollectionAdmin(collectionId, {Ethereum: fractionalizer.options.address});
+      await submitTransactionAsync(alice, changeAdminTx);
+    }
+    await fractionalizer.methods.setRFTCollection(rftCollectionAddress).send();
+    {
+      const tx = api.tx.unique.setTransfersEnabledFlag(collectionId, false);
+      await executeTransaction(api, alice, tx);
+    }
+
+    const {collectionIdAddress: nftCollectionAddress} = await createNonfungibleCollection(api, web3, owner);
+    const nftContract = uniqueNFT(web3, nftCollectionAddress, owner);
+    const nftTokenId = await nftContract.methods.nextTokenId().call();
+    await nftContract.methods.mint(owner, nftTokenId).send();
+
+    await fractionalizer.methods.setNftCollectionIsAllowed(nftCollectionAddress, true).send();
+    await nftContract.methods.approve(fractionalizer.options.address, nftTokenId).send();
+
+    await expect(fractionalizer.methods.nft2rft(nftCollectionAddress, nftTokenId, 100n).call())
+      .to.be.rejectedWith(/TransferNotAllowed$/g);
   });
 });
\ No newline at end of file