diff options
author | Håkon Hallingstad <hakon@yahooinc.com> | 2022-05-09 11:17:46 +0200 |
---|---|---|
committer | Håkon Hallingstad <hakon@yahooinc.com> | 2022-05-09 11:17:46 +0200 |
commit | 7feccb4136704ee8c7e31c6e17672b32ebb92b45 (patch) | |
tree | a278c69c9464ae9630d43ff35ca4905b7caf5753 /controller-server/src/test/java/com/yahoo | |
parent | 623c97dddb66baa91d305dcfa81e37395747684d (diff) |
Getting system as zone
Diffstat (limited to 'controller-server/src/test/java/com/yahoo')
3 files changed, 23 insertions, 9 deletions
diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneApiMock.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneApiMock.java index 9e08e774744..d173fcb0e18 100644 --- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneApiMock.java +++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneApiMock.java @@ -35,15 +35,15 @@ public class ZoneApiMock implements ZoneApi { } public static ZoneApiMock fromId(String id) { - return newBuilder().withId(id).build(); + return from(ZoneId.from(id)); } public static ZoneApiMock from(Environment environment, RegionName region) { - return newBuilder().with(ZoneId.from(environment, region)).build(); + return from(ZoneId.from(environment, region)); } - public static ZoneApiMock from(ZoneId zone) { - return newBuilder().with(zone).build(); + public static ZoneApiMock from(ZoneId id) { + return newBuilder().with(id).build(); } @Override @@ -80,7 +80,7 @@ public class ZoneApiMock implements ZoneApi { private SystemName systemName = SystemName.defaultSystem(); private ZoneId id = ZoneId.defaultId(); - private ZoneId virtualId ; + private ZoneId virtualId = null; private CloudName cloudName = CloudName.defaultName(); private String cloudNativeRegionName = id.region().value(); diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneRegistryMock.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneRegistryMock.java index 5b8e25cbfe8..849503ae8d1 100644 --- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneRegistryMock.java +++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ZoneRegistryMock.java @@ -25,6 +25,7 @@ import com.yahoo.vespa.hosted.controller.api.integration.zone.ZoneRegistry; import java.net.URI; import java.time.Duration; +import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; import java.util.List; @@ -51,7 +52,7 @@ public class ZoneRegistryMock extends AbstractComponent implements ZoneRegistry /** * This sets the default list of zones contained in this. If your test need a particular set of zones, use - * {@link #setZones(List)} instead of changing the default set.} + * {@link #setZones(List)} instead of changing the default set. */ public ZoneRegistryMock(SystemName system) { this.system = system; @@ -156,6 +157,21 @@ public class ZoneRegistryMock extends AbstractComponent implements ZoneRegistry } @Override + public ZoneFilter zonesIncludingSystem() { + var fullZones = new ArrayList<ZoneApi>(1 + zones.size()); + fullZones.add(systemAsZone()); + fullZones.addAll(zones); + return ZoneFilterMock.from(fullZones, zoneRoutingMethods, reprovisionToUpgradeOs); + } + + private ZoneApiMock systemAsZone() { + return ZoneApiMock.newBuilder() + .with(ZoneId.from("prod.us-east-1")) + .withVirtualId(ZoneId.from("prod.controller")) + .build(); + } + + @Override public AthenzService getConfigServerHttpsIdentity(ZoneId zone) { return new AthenzService("vespadomain", "provider-" + zone.environment().value() + "-" + zone.region().value()); } diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/maintenance/ArchiveAccessMaintainerTest.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/maintenance/ArchiveAccessMaintainerTest.java index e73d66b0329..5571f957e83 100644 --- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/maintenance/ArchiveAccessMaintainerTest.java +++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/maintenance/ArchiveAccessMaintainerTest.java @@ -17,7 +17,6 @@ import java.util.Map; import java.util.Optional; import java.util.Set; import java.util.stream.Collectors; -import java.util.stream.Stream; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNull; @@ -49,8 +48,7 @@ public class ArchiveAccessMaintainerTest { assertEquals(Set.of(tenant1role), archiveService.authorizedIamRolesForKey.get(testBucket.keyArn())); var expected = Map.of("archive.bucketCount", - Stream.concat(Stream.of(tester.controller().zoneRegistry().systemZone().getVirtualId()), - tester.controller().zoneRegistry().zones().all().ids().stream()) + tester.controller().zoneRegistry().zonesIncludingSystem().all().ids().stream() .collect(Collectors.toMap( zone -> Map.of("zone", zone.value()), zone -> zone.equals(testZone) ? 1d : 0d))); |