Actually test new codepath

1 files changed, 3 insertions, 2 deletions

diff --git a/orchestrator/src/test/java/com/yahoo/vespa/orchestrator/policy/HostedVespaClusterPolicyTest.java b/orchestrator/src/test/java/com/yahoo/vespa/orchestrator/policy/HostedVespaClusterPolicyTest.java
index ea5f1784265..97c1c7ab5de 100644
--- a/orchestrator/src/test/java/com/yahoo/vespa/orchestrator/policy/HostedVespaClusterPolicyTest.java
+++ b/orchestrator/src/test/java/com/yahoo/vespa/orchestrator/policy/HostedVespaClusterPolicyTest.java
@@ -51,10 +51,11 @@ public class HostedVespaClusterPolicyTest {
 
     @Test
     public void testStorageSuspensionLimit() {
+        when(clusterApi.serviceType()).thenReturn(ServiceType.STORAGE);
         when(clusterApi.clusterId()).thenReturn(new ClusterId("some-cluster-id"));
         when(clusterApi.isStorageCluster()).thenReturn(true);
-        assertEquals(ConcurrentSuspensionLimitForCluster.ONE_NODE,
-                policy.getConcurrentSuspensionLimit(clusterApi, false));
+        assertEquals(ConcurrentSuspensionLimitForCluster.ALL_NODES,
+                policy.getConcurrentSuspensionLimit(clusterApi, true));
     }
 
     @Test