squid-js/test/keeper/TestContractHandler.ts

import { Contract } from 'web3-eth-contract'
import ContractHandler from '../../src/keeper/ContractHandler'
import Web3Provider from '../../src/keeper/Web3Provider'
import Logger from '../../src/utils/Logger'
import config from '../config'

interface ContractTest extends Contract {
    testContract?: boolean
    $initialized?: boolean
}

export default class TestContractHandler extends ContractHandler {
    public static async prepareContracts() {
        const web3 = Web3Provider.getWeb3(config)
        const deployerAddress = (await web3.eth.getAccounts())[0]
        this.networkId = await web3.eth.net.getId()

        // deploy contracts
        await TestContractHandler.deployContracts(deployerAddress)
    }

    private static networkId: number

    private static async deployContracts(deployerAddress: string) {
        Logger.log('Trying to deploy contracts')

        // Libraries
        const epochLibrary = await TestContractHandler.deployContract('EpochLibrary', deployerAddress)
        const didRegistryLibrary = await TestContractHandler.deployContract('DIDRegistryLibrary', deployerAddress)

        // Contracts
        const token = await TestContractHandler.deployContract('OceanToken', deployerAddress, [
            deployerAddress,
            deployerAddress
        ])

        const dispenser = await TestContractHandler.deployContract('Dispenser', deployerAddress, [
            token.options.address,
            deployerAddress
        ])

        // Add dispenser as Token minter
        if (!token.$initialized) {
            await token.methods.addMinter(dispenser.options.address).send({ from: deployerAddress })
        }

        const didRegistry = await TestContractHandler.deployContract(
            'DIDRegistry',
            deployerAddress,
            [deployerAddress],
            {
                DIDRegistryLibrary: didRegistryLibrary.options.address
            }
        )

        // Managers
        const templateStoreManager = await TestContractHandler.deployContract('TemplateStoreManager', deployerAddress, [
            deployerAddress
        ])
        const conditionStoreManager = await TestContractHandler.deployContract(
            'ConditionStoreManager',
            deployerAddress,
            [deployerAddress],
            {
                EpochLibrary: epochLibrary.options.address
            }
        )
        const agreementStoreManager = await TestContractHandler.deployContract(
            'AgreementStoreManager',
            deployerAddress,
            [
                deployerAddress,
                conditionStoreManager.options.address,
                templateStoreManager.options.address,
                didRegistry.options.address
            ]
        )

        // Conditions
        const lockRewardCondition = await TestContractHandler.deployContract('LockRewardCondition', deployerAddress, [
            deployerAddress,
            conditionStoreManager.options.address,
            token.options.address
        ])
        const accessSecretStoreCondition = await TestContractHandler.deployContract(
            'AccessSecretStoreCondition',
            deployerAddress,
            [deployerAddress, conditionStoreManager.options.address, agreementStoreManager.options.address]
        )

        // Conditions rewards
        const escrowReward = await TestContractHandler.deployContract('EscrowReward', deployerAddress, [
            deployerAddress,
            conditionStoreManager.options.address,
            token.options.address
        ])

        // Templates
        await TestContractHandler.deployContract('EscrowAccessSecretStoreTemplate', deployerAddress, [
            deployerAddress,
            agreementStoreManager.options.address,
            didRegistry.options.address,
            accessSecretStoreCondition.options.address,
            lockRewardCondition.options.address,
            escrowReward.options.address
        ])
    }

    private static async deployContract(
        name: string,
        from: string,
        args: any[] = [],
        tokens: { [name: string]: string } = {}
    ): Promise<ContractTest> {
        const where = this.networkId

        // dont redeploy if there is already something loaded
        if (TestContractHandler.hasContract(name, where)) {
            const contract: ContractTest = await ContractHandler.getContract(name, where)
            if (contract.testContract) {
                return { ...contract, $initialized: true } as any
            }
        }

        const web3 = Web3Provider.getWeb3(config)

        let contractInstance: ContractTest
        try {
            Logger.log('Deploying', name)
            const sendConfig = {
                from,
                gas: 3000000,
                gasPrice: String(10000000000)
            }
            const artifact = require(`@oceanprotocol/keeper-contracts/artifacts/${name}.development.json`)
            const tempContract = new web3.eth.Contract(artifact.abi, artifact.address)
            const isZos = !!tempContract.methods.initialize

            Logger.debug({
                name,
                from,
                isZos,
                args,
                libraries: artifact.bytecode
                    .replace(/(0x)?[a-f0-9]{8}/gi, '')
                    .replace(/__([^_]+)_*[0-9a-f]{2}/g, '|$1')
                    .split('|')
                    .splice(1)
            })

            contractInstance = await tempContract
                .deploy({
                    data: TestContractHandler.replaceTokens(artifact.bytecode.toString(), tokens),
                    arguments: isZos ? undefined : args
                })
                .send(sendConfig)
            if (isZos) {
                await contractInstance.methods.initialize(...args).send(sendConfig)
            }
            contractInstance.testContract = true
            ContractHandler.setContract(name, where, contractInstance)
            // Logger.log('Deployed', name, 'at', contractInstance.options.address)
        } catch (err) {
            Logger.error('Deployment failed for', name, 'with args', JSON.stringify(args, null, 2), err.message)
            throw err
        }

        return contractInstance
    }

    private static replaceTokens(bytecode: string, tokens: { [name: string]: string }): string {
        return Object.entries(tokens).reduce(
            (acc, [token, address]) => acc.replace(new RegExp(`_+${token}_+`, 'g'), address.substr(2)),
            bytecode
        )
    }
}