diff options
Diffstat (limited to 'controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ArtifactRepositoryMock.java')
-rw-r--r-- | controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ArtifactRepositoryMock.java | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ArtifactRepositoryMock.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ArtifactRepositoryMock.java index e5e4c9e46fc..1cbda9c165f 100644 --- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ArtifactRepositoryMock.java +++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ArtifactRepositoryMock.java @@ -6,15 +6,32 @@ import com.yahoo.component.Version; import com.yahoo.config.provision.ApplicationId; import com.yahoo.config.provision.zone.ZoneId; import com.yahoo.vespa.hosted.controller.api.integration.deployment.ArtifactRepository; +import com.yahoo.vespa.hosted.controller.api.integration.deployment.StableOsVersion; + +import java.util.HashMap; +import java.util.Map; /** * @author mpolden */ public class ArtifactRepositoryMock extends AbstractComponent implements ArtifactRepository { + private final Map<Integer, StableOsVersion> stableOsVersions = new HashMap<>(); + @Override public byte[] getSystemApplicationPackage(ApplicationId application, ZoneId zone, Version version) { return new byte[0]; } + @Override + public StableOsVersion stableOsVersion(int major) { + StableOsVersion version = stableOsVersions.get(major); + if (version == null) throw new IllegalArgumentException("No version set for major " + major); + return version; + } + + public void promoteOsVersion(StableOsVersion stableOsVersion) { + stableOsVersions.put(stableOsVersion.version().getMajor(), stableOsVersion); + } + } |