com.ozangunalp.kafka.test.container.ZookeeperNativeContainer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafka-native-test-container Show documentation

There is a newer version: 0.11.1

Show newest version

package com.ozangunalp.kafka.test.container;

import java.util.function.Consumer;
import java.util.function.Function;

import org.testcontainers.containers.GenericContainer;
import org.testcontainers.containers.output.OutputFrame;
import org.testcontainers.containers.wait.strategy.Wait;
import org.testcontainers.utility.DockerImageName;

import com.github.dockerjava.api.command.InspectContainerResponse;

public class ZookeeperNativeContainer extends GenericContainer {

    public static final String DEFAULT_REPOSITORY = System.getProperty("zookeeper-native-container-image", "quay.io/ogunalp/zookeeper-native");
    public static final String DEFAULT_VERSION = System.getProperty("zookeeper-native-container-version", "latest");
    private static final int ZOOKEEPER_PORT = 2181;

    private int exposedPort = -1;
    private Function, Consumer> outputFrameConsumer;

    public static DockerImageName imageName(String version) {
        return DockerImageName.parse(DEFAULT_REPOSITORY + ":" + version);
    }

    public static DockerImageName imageName() {
        return DockerImageName.parse(DEFAULT_REPOSITORY + ":" + DEFAULT_VERSION);
    }

    public ZookeeperNativeContainer() {
        this(imageName());
    }
    
    public ZookeeperNativeContainer(DockerImageName dockerImageName) {
        super(dockerImageName);
        super.addExposedPort(ZOOKEEPER_PORT);
        super.waitingFor(Wait.forLogMessage(".*Zookeeper server started.*", 1));
    }

    @Override
    protected void containerIsStarting(InspectContainerResponse containerInfo, boolean reused) {
        super.containerIsStarting(containerInfo, reused);
        // Set exposed port
        this.exposedPort = getMappedPort(ZOOKEEPER_PORT);
        // follow output
        if (outputFrameConsumer != null) {
            followOutput(outputFrameConsumer.apply(this));
        }
    }
    
    public ZookeeperNativeContainer withFollowOutput(Function, Consumer> outputFrameConsumer) {
        this.outputFrameConsumer = outputFrameConsumer;
        return self();
    }

    public int getExposedZookeeperPort() {
        return exposedPort;
    }

    private void assertNotRunning() {
        if (isRunning()) {
            throw new IllegalStateException("Configuration of the running broker is not permitted.");
        }
    }
    
}