Delegate responsibility/timing of cleaning up old components to caller

1 files changed, 3 insertions, 1 deletions

diff --git a/container-core/src/test/java/com/yahoo/container/di/ContainerTestBase.java b/container-core/src/test/java/com/yahoo/container/di/ContainerTestBase.java
index 9565f26fb1e..4f4ffe3732a 100644
--- a/container-core/src/test/java/com/yahoo/container/di/ContainerTestBase.java
+++ b/container-core/src/test/java/com/yahoo/container/di/ContainerTestBase.java
@@ -65,7 +65,9 @@ public class ContainerTestBase {
                             throw new UnsupportedOperationException("getBundle not supported.");
                         }
                     });
-            componentGraph = container.waitForNextComponentGeneration(componentGraph, Guice.createInjector(), true);
+            Container.ComponentGraphResult result = container.waitForNextComponentGeneration(this.componentGraph, Guice.createInjector(), true);
+            result.oldComponentsCleanupTask().run();
+            this.componentGraph = result.newGraph();
         } catch (Exception e) {
             throw new RuntimeException(e);
         }