Added ability to include container_name in compose file

This commit updates the ParsedDockerComposeFile to pull out the container name for a service if included in the compose file. DockerComposeContainer was then updated with logic to use this specified name if present.

Author: reggiemcdonald

core/src/main/java/org/testcontainers/containers/DockerComposeContainer.java

@@ -74,6 +74,7 @@ public class DockerComposeContainer<SELF extends DockerComposeContainer<SELF>> e
     private final List<File> composeFiles;
     private Set<ParsedDockerComposeFile> parsedComposeFiles;
     private final Map<String, Integer> scalingPreferences = new HashMap<>();
+    private final Map<String, String> customContainerNames = new HashMap<>();
     private DockerClient dockerClient;
     private boolean localCompose;
     private boolean pull = true;
@@ -121,6 +122,8 @@ public DockerComposeContainer(String identifier, List<File> composeFiles) {
 
         this.composeFiles = composeFiles;
         this.parsedComposeFiles = composeFiles.stream().map(ParsedDockerComposeFile::new).collect(Collectors.toSet());
+        this.parsedComposeFiles
+            .forEach(composeFile -> this.customContainerNames.putAll(composeFile.getContainerNames()));
 
         // Use a unique identifier so that containers created for this compose environment can be identified
         this.identifier = identifier;
@@ -285,11 +288,18 @@ private void registerContainersForShutdown() {
     }
 
     private List<Container> listChildContainers() {
+        Set<String> names = customContainerNames.values()
+            .stream()
+            .map(name -> String.format("/%s", name))
+            .collect(Collectors.toSet());
         return dockerClient.listContainersCmd()
                 .withShowAll(true)
                 .exec().stream()
-                .filter(container -> Arrays.stream(container.getNames()).anyMatch(name ->
-                        name.startsWith("/" + project)))
+                .filter(
+                    container -> Arrays.stream(container.getNames()).anyMatch(name ->
+                        names.contains(name) || name.startsWith("/" + project)
+                    )
+                )
                 .collect(toList());
     }
 
@@ -353,7 +363,8 @@ public SELF withExposedService(String serviceName, int servicePort, @NonNull Wai
         int ambassadorPort = nextAmbassadorPort.getAndIncrement();
         ambassadorPortMappings.computeIfAbsent(serviceInstanceName, __ -> new ConcurrentHashMap<>()).put(servicePort, ambassadorPort);
         ambassadorContainer.withTarget(ambassadorPort, serviceInstanceName, servicePort);
-        ambassadorContainer.addLink(new FutureContainer(this.project + "_" + serviceInstanceName), serviceInstanceName);
+        String containerName = getContainerName(serviceInstanceName);
+        ambassadorContainer.addLink(new FutureContainer(containerName), serviceInstanceName);
         addWaitStrategy(serviceInstanceName, waitStrategy);
         return self();
     }
@@ -366,6 +377,17 @@ private String getServiceInstanceName(String serviceName) {
         return serviceInstanceName;
     }
 
+    private String getContainerName(String serviceName) {
+        String serviceInstanceName = serviceName;
+        if (serviceInstanceName.matches(".*_[0-9]+")) {
+            serviceName = serviceInstanceName.substring(0, serviceInstanceName.indexOf('_'));
+            if (customContainerNames.containsKey(serviceName)) {
+                return customContainerNames.get(serviceName);
+            }
+        }
+        return String.format("%s_%s", project, serviceInstanceName);
+    }
+
     /*
      * can have multiple wait strategies for a single container, e.g. if waiting on several ports
      * if no wait strategy is defined, the WaitAllStrategy will return immediately.

core/src/main/java/org/testcontainers/containers/ParsedDockerComposeFile.java

@@ -12,6 +12,7 @@
 import java.io.FileInputStream;
 import java.nio.file.Files;
 import java.nio.file.Path;
+import java.util.HashMap;
 import java.util.HashSet;
 import java.util.Map;
 import java.util.Set;
@@ -30,6 +31,8 @@ class ParsedDockerComposeFile {
 
     @Getter
     private Set<String> dependencyImageNames = new HashSet<>();
+    @Getter
+    private Map<String, String> containerNames = new HashMap<>();
 
     ParsedDockerComposeFile(File composeFile) {
         Yaml yaml = new Yaml();
@@ -86,19 +89,15 @@ private void parseAndValidate() {
 
             final Map serviceDefinitionMap = (Map) serviceDefinition;
 
-            validateNoContainerNameSpecified(serviceName, serviceDefinitionMap);
+            findContainerName(serviceName, serviceDefinitionMap);
             findServiceImageName(serviceDefinitionMap);
             findImageNamesInDockerfile(serviceDefinitionMap);
         }
     }
 
-    private void validateNoContainerNameSpecified(String serviceName, Map serviceDefinitionMap) {
+    private void findContainerName(String serviceName, Map serviceDefinitionMap) {
         if (serviceDefinitionMap.containsKey("container_name")) {
-            throw new IllegalStateException(String.format(
-                "Compose file %s has 'container_name' property set for service '%s' but this property is not supported by Testcontainers, consider removing it",
-                composeFileName,
-                serviceName
-            ));
+            this.containerNames.put(serviceName, (String) serviceDefinitionMap.get("container_name"));
         }
     }