diff options
Diffstat (limited to 'clustercontroller-core/src/test/java/com')
2 files changed, 2 insertions, 14 deletions
diff --git a/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterFixture.java b/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterFixture.java index f8d41405e85..66669e8b481 100644 --- a/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterFixture.java +++ b/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterFixture.java @@ -9,13 +9,10 @@ import com.yahoo.vdslib.state.NodeState; import com.yahoo.vdslib.state.NodeType; import com.yahoo.vdslib.state.State; import com.yahoo.vespa.clustercontroller.core.listeners.NodeListener; - import java.util.Collection; import java.util.HashSet; import java.util.List; -import java.util.Map; import java.util.Set; -import java.util.TreeMap; import static org.mockito.Mockito.mock; @@ -24,7 +21,6 @@ public class ClusterFixture { public final ContentCluster cluster; public final Distribution distribution; public final FakeTimer timer; - private final EventLogInterface eventLog; final StateChangeHandler nodeStateChangeHandler; private final ClusterStateGenerator.Params params = new ClusterStateGenerator.Params(); @@ -32,9 +28,8 @@ public class ClusterFixture { this.cluster = cluster; this.distribution = distribution; this.timer = new FakeTimer(); - this.eventLog = mock(EventLogInterface.class); var context = new FleetControllerContextImpl(new FleetControllerId(cluster.getName(), 0)); - this.nodeStateChangeHandler = new StateChangeHandler(context, timer, eventLog); + this.nodeStateChangeHandler = new StateChangeHandler(context, timer, mock(EventLogInterface.class)); this.params.cluster(this.cluster); } @@ -150,13 +145,6 @@ public class ClusterFixture { return this; } - static Map<NodeType, Integer> buildTransitionTimeMap(int distributorTransitionTime, int storageTransitionTime) { - Map<NodeType, Integer> maxTransitionTime = new TreeMap<>(); - maxTransitionTime.put(NodeType.DISTRIBUTOR, distributorTransitionTime); - maxTransitionTime.put(NodeType.STORAGE, storageTransitionTime); - return maxTransitionTime; - } - void disableTransientMaintenanceModeOnDown() { this.params.transitionTimes(0); } diff --git a/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterStateGeneratorTest.java b/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterStateGeneratorTest.java index 78911c414a2..f786f86eaa6 100644 --- a/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterStateGeneratorTest.java +++ b/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/ClusterStateGeneratorTest.java @@ -24,7 +24,7 @@ public class ClusterStateGeneratorTest { private static AnnotatedClusterState generateFromFixtureWithDefaultParams(ClusterFixture fixture) { final ClusterStateGenerator.Params params = new ClusterStateGenerator.Params(); params.cluster = fixture.cluster; - params.transitionTimes = ClusterFixture.buildTransitionTimeMap(0, 0); + params.transitionTimes = ClusterStateGenerator.Params.buildTransitionTimeMap(0, 0); params.currentTimeInMillis = 0; return ClusterStateGenerator.generatedStateFrom(params); } |