ocean.js/test/BalancerContractHandler.ts

import Web3 from 'web3'
import { AbiItem } from 'web3-utils/types'
import { Contract } from 'web3-eth-contract'

export class BalancerContractHandler {
  public factory: Contract
  public pool: Contract
  public accounts: string[]
  public poolBytecode: string
  public factoryBytecode: string
  public factoryAddress: string
  public poolAddress: string
  public web3: Web3

  constructor(
    factoryABI: AbiItem | AbiItem[],
    factoryBytecode: string,
    poolABI: AbiItem | AbiItem[],
    poolBytecode: string,
    web3: Web3
  ) {
    this.web3 = web3
    this.factory = new this.web3.eth.Contract(factoryABI)
    this.factoryBytecode = factoryBytecode
    this.pool = new this.web3.eth.Contract(poolABI)
    this.poolBytecode = poolBytecode
  }

  public async getAccounts() {
    this.accounts = await this.web3.eth.getAccounts()
  }

  public async deployContracts(minter: string) {
    const estGas = await this.factory
      .deploy({
        data: this.factoryBytecode,
        arguments: []
      })
      .estimateGas(function (err, estGas) {
        if (err) console.log('DeployContracts: ' + err)
        return estGas
      })
    // deploy the contract and get it's address
    this.factoryAddress = await this.factory
      .deploy({
        data: this.factoryBytecode,
        arguments: []
      })
      .send({
        from: minter,
        gas: estGas + 1,
        gasPrice: '3000000000'
      })
      .then(function (contract) {
        return contract.options.address
      })
  }

  public async SdeployContracts(minter: string) {
    let estGas
    estGas = await this.pool
      .deploy({
        data: this.poolBytecode,
        arguments: []
      })
      .estimateGas(function (err, estGas) {
        if (err) console.log('Pool deploy estimate gas: ' + err)
        return estGas
      })
    // deploy the contract and get it's address
    this.poolAddress = await this.pool
      .deploy({
        data: this.poolBytecode,
        arguments: []
      })
      .send({
        from: minter,
        gas: estGas + 1,
        gasPrice: '3000000000'
      })
      .then(function (contract) {
        return contract.options.address
      })
    estGas = await this.factory
      .deploy({
        data: this.factoryBytecode,
        arguments: [this.poolAddress]
      })
      .estimateGas(function (err, estGas) {
        if (err) console.log('DeployContracts: ' + err)
        return estGas
      })
    // deploy the contract and get it's address
    this.factoryAddress = await this.factory
      .deploy({
        data: this.factoryBytecode,
        arguments: [this.poolAddress]
      })
      .send({
        from: minter,
        gas: estGas + 1,
        gasPrice: '3000000000'
      })
      .then(function (contract) {
        return contract.options.address
      })
  }
}