KaneAI - World’s First E2E Software Testing Agent.Signup for Private Beta Join TestMu Conference

How to use AtomicInteger method of org.testcontainers.containers.DockerComposeContainer class

Best Testcontainers-java code snippet using org.testcontainers.containers.DockerComposeContainer.AtomicInteger

Source:AirbyteTestContainer.java

...23import java.util.HashMap;24import java.util.Map;25import java.util.Properties;26import java.util.Scanner;27import java.util.concurrent.atomic.AtomicInteger;28import java.util.concurrent.atomic.AtomicReference;29import java.util.function.Consumer;30import java.util.function.Supplier;31import org.slf4j.Logger;32import org.slf4j.LoggerFactory;33import org.testcontainers.containers.DockerComposeContainer;34import org.testcontainers.containers.SocatContainer;35import org.testcontainers.containers.output.OutputFrame;36/**37 * The goal of this class is to make it easy to run the Airbyte docker-compose configuration from38 * test containers. This helps make it easy to stop the test container without deleting the volumes39 * { @link AirbyteTestContainer#stopRetainVolumes() }. It waits for Airbyte to be ready. It also40 * handles the nuances of configuring the Airbyte docker-compose configuration in test containers.41 */42public class AirbyteTestContainer {43  private static final Logger LOGGER = LoggerFactory.getLogger(AirbyteTestContainer.class);44  private final File dockerComposeFile;45  private final Map<String, String> env;46  private final Map<String, Consumer<String>> customServiceLogListeners;47  private DockerComposeContainer<?> dockerComposeContainer;48  public AirbyteTestContainer(final File dockerComposeFile,49                              final Map<String, String> env,50                              final Map<String, Consumer<String>> customServiceLogListeners) {51    this.dockerComposeFile = dockerComposeFile;52    this.env = env;53    this.customServiceLogListeners = customServiceLogListeners;54  }55  /**56   * Starts Airbyte docker-compose configuration. Will block until the server is reachable or it times57   * outs.58   */59  public void startBlocking() throws IOException, InterruptedException {60    startAsync();61    waitForAirbyte();62  }63  @SuppressWarnings({"unchecked", "rawtypes"})64  public void startAsync() throws IOException, InterruptedException {65    final File cleanedDockerComposeFile = prepareDockerComposeFile(dockerComposeFile);66    dockerComposeContainer = new DockerComposeContainer(cleanedDockerComposeFile).withEnv(env);67    // Only expose logs related to db migrations.68    serviceLogConsumer(dockerComposeContainer, "init");69    serviceLogConsumer(dockerComposeContainer, "bootloader");70    serviceLogConsumer(dockerComposeContainer, "db");71    serviceLogConsumer(dockerComposeContainer, "seed");72    serviceLogConsumer(dockerComposeContainer, "server");73    dockerComposeContainer.start();74  }75  private static Map<String, String> prepareDockerComposeEnvVariables(final File envFile) throws IOException {76    LOGGER.info("Searching for environment in {}", envFile);77    Preconditions.checkArgument(envFile.exists(), "could not find docker compose environment");78    final Properties prop = new Properties();79    prop.load(new FileInputStream(envFile));80    return Maps.fromProperties(prop);81  }82  /**83   * TestContainers docker compose files cannot have container_names, so we filter them.84   */85  private static File prepareDockerComposeFile(final File originalDockerComposeFile) throws IOException {86    final File cleanedDockerComposeFile = Files.createTempFile(Path.of("/tmp"), "docker_compose", "acceptance_test").toFile();87    try (final Scanner scanner = new Scanner(originalDockerComposeFile, StandardCharsets.UTF_8)) {88      try (final FileWriter fileWriter = new FileWriter(cleanedDockerComposeFile, StandardCharsets.UTF_8)) {89        while (scanner.hasNextLine()) {90          final String s = scanner.nextLine();91          if (s.contains("container_name")) {92            continue;93          }94          fileWriter.write(s);95          fileWriter.write('\n');96        }97      }98    }99    return cleanedDockerComposeFile;100  }101  @SuppressWarnings("BusyWait")102  private static void waitForAirbyte() throws InterruptedException {103    // todo (cgardens) - assumes port 8001 which is misleading since we can start airbyte on other104    // ports. need to make this configurable.105    final AirbyteApiClient apiClient = new AirbyteApiClient(106        new ApiClient().setScheme("http")107            .setHost("localhost")108            .setPort(8001)109            .setBasePath("/api"));110    final HealthApi healthApi = apiClient.getHealthApi();111    final AtomicReference<ApiException> lastException = new AtomicReference<>();112    final AtomicInteger attempt = new AtomicInteger();113    final Supplier<Boolean> condition = () -> {114      try {115        healthApi.getHealthCheck();116        return true;117      } catch (final ApiException e) {118        lastException.set(e);119        LOGGER.info("airbyte not ready yet. attempt: {}", attempt.incrementAndGet());120        return false;121      }122    };123    if (!WaitingUtils.waitForCondition(Duration.ofSeconds(5), Duration.ofMinutes(2), condition)) {124      throw new IllegalStateException("Airbyte took too long to start. Including last exception.", lastException.get());125    }126  }...

AtomicInteger

Using AI Code Generation

1import org.testcontainers.containers.DockerComposeContainer2import java.io.File3import java.util.concurrent.atomic.AtomicInteger4public class DockerComposeContainerWithAtomicInteger extends DockerComposeContainer<DockerComposeContainerWithAtomicInteger> {5    private static final AtomicInteger COMPOSE_PROJECT_NAME_COUNTER = new AtomicInteger(1);6    public DockerComposeContainerWithAtomicInteger(File composeFile) {7        super(composeFile);8        withProjectName("compose-test-" + COMPOSE_PROJECT_NAME_COUNTER.getAndIncrement());9    }10}11import org.testcontainers.containers.wait.strategy.Wait12import org.testcontainers.containers.wait.strategy.WaitAllStrategy13import org.testcontainers.containers.wait.strategy.WaitAllStrategy.WaitAllStrategyBuilder14import org.testcontainers.containers.wait.strategy.WaitStrategy15import org.testcontainers.containers.wait.strategy.WaitStrategyTarget16import static java.util.concurrent.TimeUnit.SECONDS17public class DockerComposeContainerWithAtomicIntegerTest extends Specification {18    def 'test docker compose'() {19        def dockerComposeContainer = new DockerComposeContainerWithAtomicInteger(new File("docker-compose.yml"))20                .withExposedService(APP_NAME, APP_PORT, Wait.forHttp("/").forStatusCode(200))21                .withExposedService(DB_NAME, DB_PORT, new WaitAllStrategy())22        dockerComposeContainer.start()23        dockerComposeContainer.getServiceHost(APP_NAME, APP_PORT) != null24        dockerComposeContainer.getServicePort(APP_NAME, APP_PORT) != null25        dockerComposeContainer.getServiceHost(DB_NAME, DB_PORT) != null26        dockerComposeContainer.getServicePort(DB_NAME, DB_PORT) != null27        dockerComposeContainer.stop()28    }29    static class WaitAllStrategy implements WaitStrategy {30        private final WaitAllStrategyBuilder waitAllStrategyBuilder = new WaitAllStrategyBuilder()31        WaitStrategyTarget waitUntilReady(WaitStrategyTarget waitStrategyTarget) {32            waitAllStrategyBuilder.build().waitUntilReady(waitStrategyTarget)33        }34        static class WaitAllStrategyBuilder {35            private final WaitAllStrategy waitAllStrategy = new WaitAllStrategy()36            WaitAllStrategyBuilder() {

AtomicInteger

Using AI Code Generation

1AtomicInteger atomicInteger = new AtomicInteger(0);2DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))3    .withScaledService("service", 2, atomicInteger::incrementAndGet);4DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))5    .withScaledService("service", 2, () -> 1);6DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))7    .withScaledService("service", 2, new AtomicInteger(1)::incrementAndGet);8DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))9    .withScaledService("service", 2, new AtomicInteger(1)::incrementAndGet);10DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))11    .withScaledService("service", 2, new AtomicInteger(1)::incrementAndGet);12DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))13    .withScaledService("service", 2, new AtomicInteger(1)::incrementAndGet);14DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))15    .withScaledService("service", 2, new AtomicInteger(1)::incrementAndGet);16DockerComposeContainer dockerComposeContainer = new DockerComposeContainer(new File("docker-compose.yml"))17    .withScaledService("service", 2, new AtomicInteger(1)::incrementAndGet);

AtomicInteger

Using AI Code Generation

1AtomicInteger port = new AtomicInteger(8080);2DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))3.withExposedService("app_1", port.getAndIncrement(), Wait.forListeningPort());4DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))5.withService("app_1", Wait.forListeningPort());6DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))7.withScaledService("app_1", 2, Wait.forListeningPort());8DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))9.withScaledService("app_1", 2, Wait.forListeningPort());10DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))11.withScaledService("app_1", 2, Wait.forListeningPort());12DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))13.withScaledService("app_1", 2, Wait.forListeningPort());14DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))15.withScaledService("app_1", 2, Wait.forListeningPort());16DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))17.withScaledService("app_1", 2, Wait.forListeningPort());18DockerComposeContainer environment = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))19.withScaledService("app_1", 2, Wait.forListeningPort());

AtomicInteger

Using AI Code Generation

1public class DockerComposeContainerTest {2    private static final AtomicInteger counter = new AtomicInteger(0);3    private static DockerComposeContainer dockerComposeContainer;4    public static void setUp() {5        dockerComposeContainer = new DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))6                .withLocalCompose(true)7                .withExposedService("elasticsearch_1", 9200, Wait.forListeningPort());8        dockerComposeContainer.start();9    }10    public static void tearDown() {11        dockerComposeContainer.stop();12    }13    public void test1() {14        System.out.println("test1");15    }16    public void test2() {17        System.out.println("test2");18    }19    public void test3() {20        System.out.println("test3");21    }22    public void test4() {23        System.out.println("test4");24    }25    public void test5() {26        System.out.println("test5");27    }28    public void test6() {29        System.out.println("test6");30    }31    public void test7() {32        System.out.println("test7");33    }34    public void test8() {35        System.out.println("test8");36    }37    public void test9() {38        System.out.println("test9");39    }40    public void test10() {41        System.out.println("test10");42    }43    public void test11() {44        System.out.println("test11");45    }46    public void test12() {47        System.out.println("test12");48    }49    public void test13() {50        System.out.println("test13");51    }52    public void test14() {53        System.out.println("test14");54    }55    public void test15() {56        System.out.println("test15");57    }58    public void test16() {59        System.out.println("test16");60    }61    public void test17() {62        System.out.println("test17");63    }64    public void test18() {65        System.out.println("test18");66    }

AtomicInteger

Using AI Code Generation

1AtomicInteger containerCount = new AtomicInteger(0);2DockerComposeContainer container = new DockerComposeContainer(new File("docker-compose.yml"))3.withScaledService("web", 2, (service) -> {4    containerCount.incrementAndGet();5    return service.withAffinityToContainer();6});7container.start();8assertEquals(2, containerCount.get());9AtomicInteger containerCount = new AtomicInteger(0);10GenericContainer container = new GenericContainer().withScaledService("web", 2, (service) -> {11    containerCount.incrementAndGet();12    return service.withAffinityToContainer();13});14container.start();15assertEquals(2, containerCount.get());16AtomicInteger containerCount = new AtomicInteger(0);17FixedHostPortGenericContainer container = new FixedHostPortGenericContainer().withScaledService("web", 2, (service) -> {18    containerCount.incrementAndGet();19    return service.withAffinityToContainer();20});21container.start();22assertEquals(2, containerCount.get());23AtomicInteger containerCount = new AtomicInteger(0);24GenericContainer container = new GenericContainer().withScaledService("web", 2, (service) -> {25    containerCount.incrementAndGet();26    return service.withAffinityToContainer();27});28container.start();29assertEquals(2, containerCount.get());30AtomicInteger containerCount = new AtomicInteger(0);31WaitAllStrategy container = new WaitAllStrategy().withScaledService("web", 2, (service) -> {32    containerCount.incrementAndGet();33    return service.withAffinityToContainer();34});35container.start();36assertEquals(2, containerCount.get());37AtomicInteger containerCount = new AtomicInteger(0);38WaitAllStrategy container = new WaitAllStrategy().withScaledService("web", 2, (service) -> {39    containerCount.incrementAndGet();40    return service.withAffinityToContainer();41});42container.start();43assertEquals(2, containerCount.get());44AtomicInteger containerCount = new AtomicInteger(0);

AtomicInteger

Using AI Code Generation

1AtomicInteger counter = new AtomicInteger(1);2DockerComposeContainer composeContainer = new DockerComposeContainer(new File("src/test/resources/compose.yml"))3    .withScaledService("selenium-hub", 1, counter::incrementAndGet)4    .withScaledService("chrome", 1, counter::incrementAndGet)5    .withScaledService("firefox", 1, counter::incrementAndGet);6composeContainer.start();7AtomicInteger counter = new AtomicInteger(1);8DockerComposeContainer composeContainer = new DockerComposeContainer(new File("src/test/resources/compose.yml"))9    .withScaledService("selenium-hub", 1, counter::incrementAndGet)10    .withScaledService("chrome", 1, counter::incrementAndGet)11    .withScaledService("firefox", 1, counter::incrementAndGet);12composeContainer.start();13AtomicInteger counter = new AtomicInteger(1);14DockerComposeContainer composeContainer = new DockerComposeContainer(new File("src/test/resources/compose.yml"))15    .withScaledService("selenium-hub", 1, counter::incrementAndGet)16    .withScaledService("chrome", 1, counter::incrementAndGet)17    .withScaledService("firefox", 1, counter::incrementAndGet);18composeContainer.start();19AtomicInteger counter = new AtomicInteger(1);20DockerComposeContainer composeContainer = new DockerComposeContainer(new File("src/test/resources/compose.yml"))21    .withScaledService("selenium-hub", 1, counter::incrementAndGet)22    .withScaledService("chrome", 1, counter::incrementAndGet)23    .withScaledService("firefox", 1, counter::incrementAndGet);24composeContainer.start();25AtomicInteger counter = new AtomicInteger(1);26DockerComposeContainer composeContainer = new DockerComposeContainer(new File("src/test/resources/compose.yml"))27    .withScaledService("selenium-hub", 1, counter::incrementAndGet)28    .withScaledService("chrome", 1, counter::incrementAndGet)29    .withScaledService("firefox", 1, counter::incrementAndGet);

AtomicInteger

Using AI Code Generation

1org.testcontainers.containers.DockerComposeContainer container = new org.testcontainers.containers.DockerComposeContainer(new File("src/test/resources/docker-compose.yml"))2    .withLocalCompose(true)3    .withExposedService("redis_1", 6379)4    .withExposedService("redis_2", 6379)5    .withExposedService("redis_3", 6379)6    .withExposedService("redis_4", 6379)7    .withExposedService("redis_5", 6379)8    .withExposedService("redis_6", 6379)9    .withExposedService("redis_7", 6379)10    .withExposedService("redis_8", 6379)11    .withExposedService("redis_9", 6379)12    .withExposedService("redis_10", 6379)13    .withExposedService("redis_11", 6379)14    .withExposedService("redis_12", 6379)15    .withExposedService("redis_13", 6379)16    .withExposedService("redis_14", 6379)17    .withExposedService("redis_15", 6379)18    .withExposedService("redis_16", 6379)19    .withExposedService("redis_17", 6379)20    .withExposedService("redis_18", 6379)21    .withExposedService("redis_19", 6379)22    .withExposedService("redis_20", 6379)23    .withExposedService("redis_21", 6379)24    .withExposedService("redis_22", 6379)25    .withExposedService("redis_23", 6379)26    .withExposedService("redis_24", 6379)27    .withExposedService("redis_25", 6379)28    .withExposedService("redis_26", 6379)29    .withExposedService("redis_27", 6379)30    .withExposedService("redis_28", 637

AtomicInteger

Using AI Code Generation

1public class DockerComposeContainer extends GenericContainer<DockerComposeContainer> {2    private static final Logger logger = LoggerFactory.getLogger(DockerComposeContainer.class);3    private static final String COMPOSE_CONTAINER_NAME = "docker-compose";4    private static final String COMPOSE_FILE_NAME = "docker-compose.yml";5    private static final int DOCKER_COMPOSE_EXIT_CODE = 0;6    private static final int DOCKER_COMPOSE_HEALTHCHECK_EXIT_CODE = 1;7    private final String composeFile;8    private final String identifier;9    private final List<String> env = new ArrayList<>();10    private final List<String> command = new ArrayList<>();11    private final List<String> exposedServicePorts = new ArrayList<>();12    private final List<String> links = new ArrayList<>();13    private final List<String> volumes = new ArrayList<>();14    private final List<String> extraHosts = new ArrayList<>();15    private final Map<String, String> serviceNames = new HashMap<>();16    private final Map<String, String> servicePorts = new HashMap<>();17    private final Map<String, String> serviceLabels = new HashMap<>();18    private final Map<String, String> serviceExtraHosts = new HashMap<>();19    private final Map<String, String> serviceVolumes = new HashMap<>();20    private final Map<String, String> serviceLinks = new HashMap<>();21    private final Map<String, String> serviceEnv = new HashMap<>();22    private final Map<String, String> serviceCommand = new HashMap<>();23    private final Map<String, String> serviceNetworks = new HashMap<>();24    private final Map<String, String> serviceNetworkAliases = new HashMap<>();25    private final Map<String, String> serviceHealthcheck = new HashMap<>();26    private final Map<String, String> serviceHealthcheckTest = new HashMap<>();27    private final Map<String, String> serviceHealthcheckInterval = new HashMap<>();28    private final Map<String, String> serviceHealthcheckTimeout = new HashMap<>();29    private final Map<String, String> serviceHealthcheckRetries = new HashMap<>();30    private final Map<String, String> serviceHealthcheckStartPeriod = new HashMap<>();31    private final Map<String, String> serviceHealthcheckCmd = new HashMap<>();32    private final Map<String, String> serviceHealthcheckTestCmd = new HashMap<>();33    private final Map<String, String> serviceHealthcheckIntervalCmd = new HashMap<>();34    private final Map<String, String> serviceHealthcheckTimeoutCmd = new HashMap<>();

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.