io.neow3j.test.DeployContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of devpack-test Show documentation
neow3j: test framework
The newest version!
package io.neow3j.test;

import io.neow3j.contract.SmartContract;
import io.neow3j.types.Hash256;
import org.junit.jupiter.api.extension.ExtensionConfigurationException;

import java.util.HashMap;
import java.util.Map;

import static java.lang.String.format;

/**
 * Holds information from the deployment of contracts that are under test.
 */
public class DeployContext {

    private Map, SmartContract> deployedContracts = new HashMap<>();

    private Map, Hash256> deployTxHashes = new HashMap<>();

    /**
     * Gets the {@code SmartContract} instance of the given contract class. This contains, e.g.,
     * the deployed contract's script hash.
     *
     * @param contractClass The class of the deployed contract.
     * @return the {@code SmartContract} instance of the deployed contract.
     */
    public SmartContract getDeployedContract(Class contractClass) {
        if (deployedContracts.containsKey(contractClass)) {
            return deployedContracts.get(contractClass);
        }
        throw new ExtensionConfigurationException(
                format("The contract %s is not (yet) deployed.", contractClass.getSimpleName()));
    }

    protected void addDeployedContract(Class contractClass, SmartContract contract) {
        deployedContracts.put(contractClass, contract);
    }

    /**
     * Gets the hash of the transaction in which the given contract was deployed.
     *
     * @param contractClass the class of the deployed contract.
     * @return the transaction hash.
     */
    public Hash256 getDeployTxHash(Class contractClass) {
        if (deployTxHashes.containsKey(contractClass)) {
            return deployTxHashes.get(contractClass);
        }
        throw new ExtensionConfigurationException(
                format("The contract %s is not (yet) deployed.", contractClass.getSimpleName()));
    }

    protected void addDeployTxHash(Class contractClass, Hash256 tx) {
        deployTxHashes.put(contractClass, tx);
    }

}