ocean-subgraph/test/integration/DFRewards.test.ts

import { NftFactory, sleep, Datatoken, DfRewards } from '@oceanprotocol/lib'
import { assert } from 'chai'
import Web3 from 'web3'
import { homedir } from 'os'
import fs from 'fs'
import { fetch } from 'cross-fetch'

const data = JSON.parse(
  fs.readFileSync(
    process.env.ADDRESS_FILE ||
      `${homedir}/.ocean/ocean-contracts/artifacts/address.json`,
    'utf8'
  )
)

const addresses = data.development
// const aquarius = new Aquarius('http://127.0.0.1:5000')
const web3 = new Web3('http://127.0.0.1:8545')

const subgraphUrl =
  'http://127.0.0.1:9000/subgraphs/name/oceanprotocol/ocean-subgraph'

describe('DFRewards tests', async () => {
  const nftName = 'testNFT'
  const nftSymbol = 'TST'
  const marketPlaceFeeAddress = '0x1230000000000000000000000000000000000000'
  const feeToken = '0x3210000000000000000000000000000000000000'
  const publishMarketFeeAmount = '0.1'
  const cap = '10000'
  const templateIndex = 1
  let datatokenAddress1: string
  let datatokenAddress2: string
  let dataToken: Datatoken
  let Factory: NftFactory
  let factoryAddress: string
  let accounts: string[]
  let publisher: string
  let dfRewards: DfRewards
  let user1: string
  let user2: string

  before(async () => {
    factoryAddress = addresses.ERC721Factory.toLowerCase()
    Factory = new NftFactory(factoryAddress, web3)
    accounts = await web3.eth.getAccounts()
    dataToken = new Datatoken(web3)
    dfRewards = new DfRewards(addresses.DFRewards, web3)
    publisher = accounts[0].toLowerCase()
    user1 = accounts[1].toLowerCase()
    user2 = accounts[2].toLowerCase()
  })

  it('should publish two datatokens', async () => {
    let result = await Factory.createNftWithDatatoken(
      publisher,
      {
        name: nftName,
        symbol: nftSymbol,
        templateIndex,
        tokenURI: '',
        transferable: true,
        owner: publisher
      },
      {
        templateIndex,
        cap,
        feeAmount: publishMarketFeeAmount,
        paymentCollector: '0x0000000000000000000000000000000000000000',
        feeToken,
        minter: publisher,
        mpFeeAddress: marketPlaceFeeAddress,
        name: 'DT1',
        symbol: 'DT1'
      }
    )
    datatokenAddress1 = result.events.TokenCreated.returnValues[0].toLowerCase()

    result = await Factory.createNftWithDatatoken(
      publisher,
      {
        name: nftName,
        symbol: nftSymbol,
        templateIndex,
        tokenURI: '',
        transferable: true,
        owner: publisher
      },
      {
        templateIndex,
        cap,
        feeAmount: publishMarketFeeAmount,
        paymentCollector: '0x0000000000000000000000000000000000000000',
        feeToken,
        minter: publisher,
        mpFeeAddress: marketPlaceFeeAddress,
        name: 'DT2',
        symbol: 'DT2'
      }
    )
    datatokenAddress2 = result.events.TokenCreated.returnValues[0].toLowerCase()
  })

  it('should top-up DF Rewards', async () => {
    // mint tokens
    await dataToken.mint(datatokenAddress1, publisher, '1000')
    await dataToken.mint(datatokenAddress2, publisher, '1000')
    // approve
    await dataToken.approve(
      datatokenAddress1,
      addresses.DFRewards,
      '1000',
      publisher
    )
    await dataToken.approve(
      datatokenAddress2,
      addresses.DFRewards,
      '1000',
      publisher
    )

    // top-up DF Rewards
    const tx = await dfRewards.allocateRewards(
      publisher,
      [user1, user2],
      ['100', '200'],
      datatokenAddress1
    )
    const user1Balance = await dfRewards.getAvailableRewards(
      user1,
      datatokenAddress1
    )
    // check subgraph
    await sleep(2000)
    const initialQuery = {
      query: `query {
        dfrewards(where: {id:"${user1.toLowerCase()}"}){
            id
            receiver {
              id
            }
            availableClaims(where: {token:"${datatokenAddress1.toLowerCase()}"}){
              id
              receiver {
                id
              }
              amount
              token {
                id
              }
            }
            history(orderBy:timestamp,orderDirection:desc){
              id
              receiver {
                id
              }
              amount
              token {
                id
              }
              type
              tx
            }
          }        
                }`
    }
    const initialResponse = await fetch(subgraphUrl, {
      method: 'POST',
      body: JSON.stringify(initialQuery)
    })
    const info = (await initialResponse.json()).data.dfrewards
    assert(info[0].receiver.id === user1.toLowerCase())
    assert(String(info[0].availableClaims[0].amount) === user1Balance)
    assert(
      info[0].availableClaims[0].token.id === datatokenAddress1.toLowerCase()
    )
    assert(info[0].history[0].amount === '100')
    assert(info[0].history[0].tx === tx.transactionHash.toLowerCase())
    assert(info[0].history[0].type === 'Allocated')
  })

  it('user2 claims some rewards', async () => {
    const expectedReward = await dfRewards.getAvailableRewards(
      user2,
      datatokenAddress1
    )
    await dfRewards.claimRewards(user2, user2, datatokenAddress1)

    const user2Balance = await dfRewards.getAvailableRewards(
      user2,
      datatokenAddress1
    )
    // check subgraph
    await sleep(2000)
    const initialQuery = {
      query: `query {
        dfrewards(where: {id:"${user2.toLowerCase()}"}){
            id
            receiver {
              id
            }
            availableClaims(where: {token:"${datatokenAddress1.toLowerCase()}"}){
              id
              receiver {
                id
              }
              amount
              token {
                id
              }
            }
            history(orderBy:timestamp,orderDirection:desc){
              id
              receiver {
                id
              }
              amount
              token {
                id
              }
              type
              tx
            }
          }        
                }`
    }
    const initialResponse = await fetch(subgraphUrl, {
      method: 'POST',
      body: JSON.stringify(initialQuery)
    })
    const info = (await initialResponse.json()).data.dfrewards
    assert(info[0].receiver.id === user2.toLowerCase())
    assert(String(info[0].availableClaims[0].amount) === user2Balance)
    assert(
      info[0].availableClaims[0].token.id === datatokenAddress1.toLowerCase()
    )
    assert(info[0].history[0].amount === expectedReward)
    assert(info[0].history[0].type === 'Claimed')
  })
})