added unit tests for Datatoken class

test/unit/Datatoken.test.ts

@@ -0,0 +1,209 @@
+import { assert } from 'chai'
+import Web3 from 'web3'
+import ERC721Factory from '@oceanprotocol/contracts/artifacts/contracts/ERC721Factory.sol/ERC721Factory.json'
+import ERC721Template from '@oceanprotocol/contracts/artifacts/contracts/templates/ERC721Template.sol/ERC721Template.json'
+import ERC20Template from '@oceanprotocol/contracts/artifacts/contracts/templates/ERC20Template.sol/ERC20Template.json'
+import PoolTemplate from '@oceanprotocol/contracts/artifacts/contracts/pools/balancer/BPool.sol/BPool.json'
+import { TestContractHandler } from '../TestContractHandler'
+import { NFTFactory } from '../../src/factories/NFTFactory'
+import { Datatoken, NFTDatatoken } from '../../src/datatokens'
+import { AbiItem } from 'web3-utils'
+import { LoggerInstance } from '../../src/utils'
+
+const web3 = new Web3('http://127.0.0.1:8545')
+
+describe('Datatoken', () => {
+  let nftOwner: string
+  let user1: string
+  let user2: string
+  let contractHandler: TestContractHandler
+  let nftDatatoken: NFTDatatoken
+  let datatoken: Datatoken
+  let nftFactory: NFTFactory
+  let nftAddress: string
+  let datatokenAddress: string
+
+  const nftName = 'NFTName'
+  const nftSymbol = 'NFTSymbol'
+  const publishMarketFeeAdress = '0xeE9300b7961e0a01d9f0adb863C7A227A07AaD75'
+  const oceanAddress = '0x967da4048cd07ab37855c090aaf366e4ce1b9f48'
+
+  it('should deploy contracts', async () => {
+    contractHandler = new TestContractHandler(
+      web3,
+      ERC721Template.abi as AbiItem,
+      ERC20Template.abi as AbiItem,
+      PoolTemplate.abi as AbiItem,
+      ERC721Factory.abi as AbiItem
+    )
+    await contractHandler.getAccounts()
+    nftOwner = contractHandler.accounts[0]
+    user1 = contractHandler.accounts[1]
+    user2 = contractHandler.accounts[2]
+    await contractHandler.deployContracts(nftOwner)
+  })
+
+  it('should initialize NFTFactory instance and create a new NFT', async () => {
+    nftFactory = new NFTFactory(
+      contractHandler.factory721Address,
+      web3,
+      ERC721Factory.abi as AbiItem
+    )
+    nftAddress = await nftFactory.createNFT(nftOwner, nftName, nftSymbol, 1)
+    nftDatatoken = new NFTDatatoken(web3, ERC721Template.abi as AbiItem)
+  })
+
+  it('#createERC20 - should create a new ERC20 DT from NFT contract', async () => {
+    await nftDatatoken.addERC20Deployer(nftAddress, nftOwner, nftOwner)
+    datatokenAddress = await nftDatatoken.createERC20(
+      nftAddress,
+      nftOwner,
+      nftOwner,
+      nftOwner,
+      publishMarketFeeAdress,
+      oceanAddress,
+      '0,1',
+      '10000',
+      nftName,
+      nftSymbol,
+      1
+    )
+    assert(datatokenAddress !== null)
+  })
+
+  it('should initialize DT20 Instance', async () => {
+    datatoken = new Datatoken(web3, ERC20Template.abi as AbiItem)
+  })
+
+  it('#mint - should fail to mint DT20, if NOT Minter', async () => {
+    assert((await datatoken.getDTPermissions(datatokenAddress, user1)).minter === false)
+    try {
+      await datatoken.mint(datatokenAddress, user1, '10', user1)
+    } catch (e) {
+      assert(e.message === 'Caller is not Minter')
+    }
+  })
+
+  it('#addMinter - should add user1 as minter, if nftDatatoken has ERC20Deployer permission', async () => {
+    assert(
+      (await nftDatatoken.getNFTPermissions(nftAddress, nftOwner)).deployERC20 === true
+    )
+    assert((await datatoken.getDTPermissions(datatokenAddress, user1)).minter === false)
+
+    await datatoken.addMinter(datatokenAddress, nftOwner, user1)
+
+    assert((await datatoken.getDTPermissions(datatokenAddress, user1)).minter === true)
+  })
+
+  it('#mint - should mint ERC20 datatoken to user1, if Minter', async () => {
+    assert((await datatoken.getDTPermissions(datatokenAddress, nftOwner)).minter === true)
+    await datatoken.mint(datatokenAddress, nftOwner, '10', user1)
+
+    assert((await datatoken.balance(datatokenAddress, user1)) === '10')
+  })
+
+  it('#removeMinter - should remove user1 as minter, if nftDatatoken has ERC20Deployer permission', async () => {
+    assert(
+      (await nftDatatoken.getNFTPermissions(nftAddress, nftOwner)).deployERC20 === true
+    )
+    assert((await datatoken.getDTPermissions(datatokenAddress, user1)).minter === true)
+
+    await datatoken.removeMinter(datatokenAddress, nftOwner, user1)
+
+    assert((await datatoken.getDTPermissions(datatokenAddress, user1)).minter === false)
+  })
+
+  it('#addFeeManager - should add user2 as feeManager, if nftDatatoken has ERC20Deployer permission', async () => {
+    assert(
+      (await nftDatatoken.getNFTPermissions(nftAddress, nftOwner)).deployERC20 === true
+    )
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user2)).feeManager === false
+    )
+
+    await datatoken.addFeeManager(datatokenAddress, nftOwner, user2)
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user2)).feeManager === true
+    )
+  })
+
+  it('#removeFeeManager - should remove user2 as feeManager, if nftDatatoken has ERC20Deployer permission', async () => {
+    assert(
+      (await nftDatatoken.getNFTPermissions(nftAddress, nftOwner)).deployERC20 === true
+    )
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user2)).feeManager === true
+    )
+
+    await datatoken.removeFeeManager(datatokenAddress, nftOwner, user2)
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user2)).feeManager === false
+    )
+  })
+
+  it('#setFeeCollector - should fail to set a new feeCollector, if NOT Fee Manager', async () => {
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user1)).feeManager === false
+    )
+
+    try {
+      await datatoken.setFeeCollector(datatokenAddress, user1, user1)
+    } catch (e) {
+      assert(e.message === 'Caller is not Fee Manager')
+    }
+  })
+
+  it('#setFeeCollector - should set a new feeCollector, if FEE MANAGER', async () => {
+    assert((await datatoken.getFeeCollector(datatokenAddress)) === nftOwner)
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user1)).feeManager === false
+    )
+
+    await datatoken.addFeeManager(datatokenAddress, nftOwner, user1)
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user1)).feeManager === true
+    )
+
+    await datatoken.setFeeCollector(datatokenAddress, user1, user2)
+    assert((await datatoken.getFeeCollector(datatokenAddress)) === user2)
+  })
+
+  it('#cleanPermissions - should clean permissions at ERC20 level', async () => {
+    assert((await datatoken.getDTPermissions(datatokenAddress, nftOwner)).minter === true)
+
+    assert((await datatoken.getFeeCollector(datatokenAddress)) === user2)
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user1)).feeManager === true
+    )
+
+    await datatoken.cleanPermissions(datatokenAddress, nftOwner)
+
+    assert((await datatoken.getFeeCollector(datatokenAddress)) === nftOwner)
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, nftOwner)).minter === false
+    )
+
+    assert(
+      (await datatoken.getDTPermissions(datatokenAddress, user1)).feeManager === false
+    )
+  })
+
+  it('#setData - should set a value into 725Y standard, if nftDatatoken has ERC20Deployer permission', async () => {
+    const data = web3.utils.asciiToHex('SomeData')
+
+    assert(
+      (await nftDatatoken.getNFTPermissions(nftAddress, nftOwner)).deployERC20 === true
+    )
+
+    await datatoken.setData(datatokenAddress, nftOwner, data)
+
+    const key = web3.utils.keccak256(datatokenAddress)
+    assert((await nftDatatoken.getData(nftAddress, key)) === data)
+  })
+})