org.web3j.container.openethereum.OpenEthereumContainer.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of web3j-unit Show documentation

Web3j-unit extension

There is a newer version: 4.12.2

/*
 * Copyright 2019 Web3 Labs Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package org.web3j.container.openethereum

import org.testcontainers.containers.wait.strategy.Wait
import org.testcontainers.containers.wait.strategy.WaitStrategy
import org.web3j.container.KGenericContainer

class OpenEthereumContainer(
    version: String?,
    resourceFiles: HashMap,
    hostFiles: HashMap,
    genesisPath: String,
    rpcPort: Int
) :
    KGenericContainer(
        "openethereum/openethereum",
        version,
        addResourceFiles(resourceFiles),
        addHostFiles(hostFiles),
        "openethereum/openethereum_start.sh",
        if (genesisPath == "dev") "openethereum/$genesisPath" else genesisPath,
        rpcPort
    ) {
    override fun withWaitStrategy(): WaitStrategy {
        return Wait.forHttp("/").withMethod("OPTIONS").forStatusCode(200).forPort(8545)
    }
}

fun addResourceFiles(resourceFiles: java.util.HashMap): java.util.HashMap {
    return resourceFiles.let {
        it["openethereum/key.txt"] = "/key"
        it["openethereum/password.txt"] = "/password.txt"
        it["openethereum/config.toml"] = "/config.toml"
        it["openethereum/dev.json"] = "/dev.json"
        it
    }
}

fun addHostFiles(hostFiles: java.util.HashMap): java.util.HashMap {
    return hostFiles.let {
        it["openethereum/key.txt"] = "/home/openethereum/.local/share/openethereum/key.txt"
        it["openethereum/password.txt"] = "/home/openethereum/.local/share/openethereum/password.txt"
        it["openethereum/config.toml"] = "/home/openethereum/.local/share/openethereum/config.toml"
        it["openethereum/dev.json"] = "/home/openethereum/.local/share/openethereum/dev.json"
        it
    }
}