tokenbridge/monitor-e2e/utils.js

const Web3 = require('web3')
const {
  ERC677_BRIDGE_TOKEN_ABI,
  BRIDGE_VALIDATORS_ABI,
  FOREIGN_NATIVE_TO_ERC_ABI,
  FOREIGN_ERC_TO_NATIVE_ABI,
  BOX_ABI
} = require('../commons')
const { validator } = require('../e2e-commons/constants')

const waitUntil = async (predicate, step = 100, timeout = 20000) => {
  const stopTime = Date.now() + timeout
  while (Date.now() <= stopTime) {
    const result = await predicate()
    if (result) {
      return result
    }
    await new Promise(resolve => setTimeout(resolve, step)) // sleep
  }
  throw new Error(`waitUntil timed out after ${timeout} ms`)
}

const sendEther = async (rpcUrl, account, to) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(account.privateKey)

  await web3.eth.sendTransaction({
    from: account.address,
    to,
    gasPrice: '1',
    gas: '50000',
    value: '1000000000000000000'
  })
}

const sendTokens = async (rpcUrl, account, tokenAddress, recipientAddress) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(account.privateKey)
  const erc20Token = new web3.eth.Contract(ERC677_BRIDGE_TOKEN_ABI, tokenAddress)

  await erc20Token.methods.transfer(recipientAddress, web3.utils.toWei('0.01')).send({
    from: account.address,
    gas: '1000000'
  })
}

const sendAMBMessage = async (rpcUrl, account, boxAddress, bridgeAddress, boxOtherSideAddress) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(account.privateKey)
  const homeBox = new web3.eth.Contract(BOX_ABI, boxAddress)

  await homeBox.methods.setValueOnOtherNetwork(3, bridgeAddress, boxOtherSideAddress).send({
    from: account.address,
    gas: '400000'
  })
}

const addValidator = async (rpcUrl, account, bridgeAddress) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(account.privateKey)
  const bridgeContract = new web3.eth.Contract(FOREIGN_NATIVE_TO_ERC_ABI, bridgeAddress)
  const foreignValidatorsAddress = await bridgeContract.methods.validatorContract().call()
  const foreignBridgeValidators = new web3.eth.Contract(BRIDGE_VALIDATORS_ABI, foreignValidatorsAddress)
  await foreignBridgeValidators.methods.addValidator('0xE71FBa5db00172bb0C93d649362B006300000935').send({
    from: account.address,
    gas: '5000000'
  })
}

const migrateToMCD = async (rpcUrl, bridgeAddress) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(validator.privateKey)
  const bridgeContract = new web3.eth.Contract(FOREIGN_ERC_TO_NATIVE_ABI, bridgeAddress)
  await bridgeContract.methods.migrateToMCD().send({
    from: validator.address,
    gas: '4000000'
  })
}

const initializeChaiToken = async (rpcUrl, bridgeAddress) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(validator.privateKey)
  const bridgeContract = new web3.eth.Contract(FOREIGN_ERC_TO_NATIVE_ABI, bridgeAddress)

  await bridgeContract.methods.initializeChaiToken().send({
    from: validator.address,
    gas: '1000000'
  })
}

const setMinDaiTokenBalance = async (rpcUrl, bridgeAddress, limit) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(validator.privateKey)
  const bridgeContract = new web3.eth.Contract(FOREIGN_ERC_TO_NATIVE_ABI, bridgeAddress)

  await bridgeContract.methods.setMinDaiTokenBalance(web3.utils.toWei(limit)).send({
    from: validator.address,
    gas: '1000000'
  })
}

const convertDaiToChai = async (rpcUrl, bridgeAddress) => {
  const web3 = new Web3(new Web3.providers.HttpProvider(rpcUrl))
  web3.eth.accounts.wallet.add(validator.privateKey)
  const bridgeContract = new web3.eth.Contract(FOREIGN_ERC_TO_NATIVE_ABI, bridgeAddress)

  await bridgeContract.methods.convertDaiToChai().send({
    from: validator.address,
    gas: '1000000'
  })
}

module.exports = {
  waitUntil,
  sendEther,
  sendTokens,
  addValidator,
  sendAMBMessage,
  migrateToMCD,
  initializeChaiToken,
  setMinDaiTokenBalance,
  convertDaiToChai
}