MAILET-143 Enhance typing for testcontainer

apache · Dec 21, 2016 · 742598e · 742598e
1 parent 13b00ee
commit 742598e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...ntainer/util-java8/src/test/java/org/apache/james/util/streams/SwarmGenericContainer.java b/...ntainer/util-java8/src/test/java/org/apache/james/util/streams/SwarmGenericContainer.java
@@ -36,15 +36,15 @@ public SwarmGenericContainer(String dockerImageName) {
         super(dockerImageName);
     }
 
-    public GenericContainer<SwarmGenericContainer> withAffinityToContainer() {
+    public SwarmGenericContainer withAffinityToContainer() {
         String container = System.getenv(DOCKER_CONTAINER);
         if (Strings.isNullOrEmpty(container)) {
             LOGGER.warn("'DOCKER_CONTAINER' environment variable not found, dockering without affinity");
-            return this;
+            return self();
         }
         List<String> envVariables = getEnv();
         envVariables.add("affinity:container==" + container);
         setEnv(envVariables);
-        return this;
+        return self();
     }
 }