diff options
author | Eirik Nygaard <eirik.nygaard@yahooinc.com> | 2022-11-22 12:39:33 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-22 12:39:33 +0100 |
commit | e24e240f01de0c291eb62ab03718fbda6abdb1b2 (patch) | |
tree | 07a3a0f73ef3d7f878ac3f371ac83b23989c2d54 /node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java | |
parent | 11690f693f82c0ca5b6bb4df49ebf9d5958560cf (diff) | |
parent | 6af8feb54aabce61e264328bf572d02a63443f6c (diff) |
Merge pull request #24959 from vespa-engine/bratseth/revert-exclusive-cleanupv8.89.6
Bratseth/revert exclusive cleanup MERGEOK
Diffstat (limited to 'node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java')
-rw-r--r-- | node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java b/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java index 00797cd6305..7969864c063 100644 --- a/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java +++ b/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/autoscale/AutoscalingTester.java @@ -13,6 +13,7 @@ import com.yahoo.config.provision.NodeType; import com.yahoo.config.provision.Zone; import com.yahoo.test.ManualClock; import com.yahoo.transaction.Mutex; +import com.yahoo.vespa.flags.InMemoryFlagSource; import com.yahoo.vespa.hosted.provision.Node; import com.yahoo.vespa.hosted.provision.NodeRepository; import com.yahoo.vespa.hosted.provision.Nodelike; @@ -43,16 +44,24 @@ class AutoscalingTester { private final HostResourcesCalculator hostResourcesCalculator; private final CapacityPolicies capacityPolicies; - public AutoscalingTester(Zone zone, HostResourcesCalculator resourcesCalculator, List<Flavor> hostFlavors, int hostCount) { - this(zone, hostFlavors, resourcesCalculator); + public AutoscalingTester(Zone zone, + HostResourcesCalculator resourcesCalculator, + List<Flavor> hostFlavors, + InMemoryFlagSource flagSource, + int hostCount) { + this(zone, hostFlavors, resourcesCalculator, flagSource); for (Flavor flavor : hostFlavors) provisioningTester.makeReadyNodes(hostCount, flavor.name(), NodeType.host, 8); provisioningTester.activateTenantHosts(); } - private AutoscalingTester(Zone zone, List<Flavor> flavors, HostResourcesCalculator resourcesCalculator) { + private AutoscalingTester(Zone zone, + List<Flavor> flavors, + HostResourcesCalculator resourcesCalculator, + InMemoryFlagSource flagSource) { provisioningTester = new ProvisioningTester.Builder().zone(zone) .flavors(flavors) + .flagSource(flagSource) .resourcesCalculator(resourcesCalculator) .hostProvisioner(zone.cloud().dynamicProvisioning() ? new MockHostProvisioner(flavors, zone.cloud()) : null) .build(); @@ -142,7 +151,7 @@ class AutoscalingTester { } public Autoscaler.Advice autoscale(ApplicationId applicationId, ClusterSpec cluster, Capacity capacity) { - capacity = capacityPolicies.applyOn(capacity, applicationId, capacityPolicies.decideExclusivity(capacity, cluster).isExclusive()); + capacity = capacityPolicies.applyOn(capacity, applicationId, capacityPolicies.decideExclusivity(capacity, cluster.isExclusive())); Application application = nodeRepository().applications().get(applicationId).orElse(Application.empty(applicationId)) .withCluster(cluster.id(), false, capacity); try (Mutex lock = nodeRepository().applications().lock(applicationId)) { |