Expand ApplicationStore with dev package storage and retrieval

1 files changed, 11 insertions, 0 deletions

diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ApplicationStoreMock.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ApplicationStoreMock.java
index bb9602903f4..cb307a731b0 100644
--- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ApplicationStoreMock.java
+++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ApplicationStoreMock.java
@@ -20,6 +20,7 @@ import static org.junit.Assert.assertFalse;
 public class ApplicationStoreMock implements ApplicationStore {
 
     private final Map<ApplicationId, Map<ApplicationVersion, byte[]>> store = new ConcurrentHashMap<>();
+    private final Map<ApplicationId, byte[]> devStore = new ConcurrentHashMap<>();
 
     @Override
     public byte[] get(ApplicationId application, ApplicationVersion applicationVersion) {
@@ -68,4 +69,14 @@ public class ApplicationStoreMock implements ApplicationStore {
         store.remove(tester.id());
     }
 
+    @Override
+    public void putDev(ApplicationId application, byte[] applicationPackage) {
+        devStore.put(application, applicationPackage);
+    }
+
+    @Override
+    public byte[] getDev(ApplicationId application) {
+        return requireNonNull(devStore.get(application));
+    }
+
 }