1 files changed, 12 insertions, 0 deletions
diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/NodeRepositoryMock.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/NodeRepositoryMock.java
index 1e535e7e9de..76a30c289b8 100644
--- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/NodeRepositoryMock.java
+++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/NodeRepositoryMock.java
@@ -8,6 +8,7 @@ import com.yahoo.config.provision.NodeResources;
 import com.yahoo.config.provision.NodeType;
 import com.yahoo.config.provision.zone.ZoneId;
 import com.yahoo.vespa.hosted.controller.api.identifiers.DeploymentId;
+import com.yahoo.vespa.hosted.controller.api.integration.configserver.Application;
 import com.yahoo.vespa.hosted.controller.api.integration.configserver.Node;
 import com.yahoo.vespa.hosted.controller.api.integration.configserver.NodeRepository;
 import com.yahoo.vespa.hosted.controller.api.integration.configserver.TargetVersions;
@@ -34,6 +35,7 @@ import java.util.stream.Collectors;
 public class NodeRepositoryMock implements NodeRepository {
 
     private final Map<ZoneId, Map<HostName, Node>> nodeRepository = new HashMap<>();
+    private final Map<ZoneId, Map<ApplicationId, Application>> applications = new HashMap<>();
     private final Map<ZoneId, TargetVersions> targetVersions = new HashMap<>();
 
     /** Add or update given nodes in zone */
@@ -43,6 +45,11 @@ public class NodeRepositoryMock implements NodeRepository {
                                                                                 Function.identity())));
     }
 
+    public void putApplication(ZoneId zone, Application application) {
+        applications.putIfAbsent(zone, new HashMap<>());
+        applications.get(zone).put(application.id(), application);
+    }
+
     /** Add or update given node in zone */
     public void putNodes(ZoneId zone, Node node) {
         putNodes(zone, Collections.singletonList(node));
@@ -162,6 +169,11 @@ public class NodeRepositoryMock implements NodeRepository {
     }
 
     @Override
+    public Application getApplication(ZoneId zone, ApplicationId applicationId) {
+        return applications.get(zone).get(applicationId);
+    }
+
+    @Override
     public void upgrade(ZoneId zone, NodeType type, Version version) {
         this.targetVersions.compute(zone, (ignored, targetVersions) -> {
             if (targetVersions == null) {