com.infobip.testcontainers.spring.rabbit.RabbitContainerInitializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of infobip-rabbitmq-testcontainers-spring-boot-starter Show documentation

The newest version!

package com.infobip.testcontainers.spring.rabbit;

import static com.infobip.testcontainers.spring.rabbit.RabbitContainerWrapper.PORT;

import java.util.Optional;

import com.infobip.testcontainers.InitializerBase;
import org.springframework.boot.test.util.TestPropertyValues;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.env.Environment;
import org.testcontainers.containers.wait.strategy.Wait;

public class RabbitContainerInitializer extends InitializerBase {

    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        var environment = applicationContext.getEnvironment();
        var wrapper = Optional.ofNullable(environment.getProperty("testcontainers.rabbit.docker.image"))
                              .map(RabbitContainerWrapper::new)
                              .orElseGet(() -> new RabbitContainerWrapper("rabbitmq:3.8.9-alpine"));
        var container = handleReusable(wrapper);

        resolveStaticPort(environment)
                .ifPresent(staticPort -> bindPort(container, staticPort, PORT));

        container.waitingFor(Wait.forListeningPort());
        start(container);

        var values = TestPropertyValues.of(
                String.format("spring.rabbitmq.addresses=%s:%d", container.getHost(), container.getMappedPort(PORT)));
        values.applyTo(applicationContext);

        registerContainerAsBean(applicationContext);
    }

    private Optional resolveStaticPort(Environment environment) {
        return Optional.ofNullable(environment.getProperty("spring.rabbitmq.port"))
                       .map(Integer::valueOf);
    }
}