Skip to content

Commit 7da57c6

Browse files
committed
fix the test and checkstyle
1 parent 85d29ca commit 7da57c6

File tree

3 files changed

+7
-11
lines changed

3 files changed

+7
-11
lines changed

flink-autoscaler/src/main/java/org/apache/flink/autoscaler/DelayedScaleDown.java

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
/** All delayed scale down requests. */
3939
public class DelayedScaleDown {
4040

41+
/** Details of the recommended parallelism. */
4142
@Data
4243
public static class RecommendedParallelism {
4344
@Nonnull private final Instant triggerTime;
@@ -132,12 +133,6 @@ public DelayedScaleDown() {
132133
this.delayedVertices = new HashMap<>();
133134
}
134135

135-
// TODO : remove this and refactor tests;
136-
public VertexDelayedScaleDownInfo triggerScaleDown(
137-
JobVertexID vertex, Instant triggerTime, int parallelism) {
138-
return triggerScaleDown(vertex, triggerTime, parallelism, false);
139-
}
140-
141136
/** Trigger a scale down, and return the corresponding {@link VertexDelayedScaleDownInfo}. */
142137
@Nonnull
143138
public VertexDelayedScaleDownInfo triggerScaleDown(

flink-autoscaler/src/test/java/org/apache/flink/autoscaler/DelayedScaleDownTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -131,19 +131,19 @@ void testMaxRecommendedParallelismForSlidingWindow() {
131131
instant);
132132
// 5, [8, 6, 4, 3] -> 8
133133
assertVertexDelayedScaleDownInfo(
134-
delayedScaleDown.triggerScaleDown(vertex, instant.plusSeconds(4), 3),
134+
delayedScaleDown.triggerScaleDown(vertex, instant.plusSeconds(4), 3, true),
135135
instant,
136136
new DelayedScaleDown.RecommendedParallelism(instant.plusSeconds(1), 8, false),
137137
instant.plusSeconds(1));
138138
// 5, 8, [6, 4, 3, 3] -> 6
139139
assertVertexDelayedScaleDownInfo(
140-
delayedScaleDown.triggerScaleDown(vertex, instant.plusSeconds(5), 3),
140+
delayedScaleDown.triggerScaleDown(vertex, instant.plusSeconds(5), 3, true),
141141
instant,
142142
new DelayedScaleDown.RecommendedParallelism(instant.plusSeconds(2), 6, true),
143143
instant.plusSeconds(2));
144144
// 5, 8, 6, [4, 3, 3, 3] -> 4
145145
assertVertexDelayedScaleDownInfo(
146-
delayedScaleDown.triggerScaleDown(vertex, instant.plusSeconds(6), 3),
146+
delayedScaleDown.triggerScaleDown(vertex, instant.plusSeconds(6), 3, true),
147147
instant,
148148
new DelayedScaleDown.RecommendedParallelism(instant.plusSeconds(3), 4, true),
149149
instant.plusSeconds(3));

flink-autoscaler/src/test/java/org/apache/flink/autoscaler/state/AbstractAutoScalerStateStoreTest.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,8 +207,9 @@ protected void testDiscardAllState() throws Exception {
207207
stateStore.storeScalingTracking(ctx, scalingTracking);
208208

209209
var delayedScaleDown = new DelayedScaleDown();
210-
delayedScaleDown.triggerScaleDown(new JobVertexID(), Instant.now(), 10);
211-
delayedScaleDown.triggerScaleDown(new JobVertexID(), Instant.now().plusSeconds(10), 12);
210+
delayedScaleDown.triggerScaleDown(new JobVertexID(), Instant.now(), 10, true);
211+
delayedScaleDown.triggerScaleDown(
212+
new JobVertexID(), Instant.now().plusSeconds(10), 12, true);
212213

213214
stateStore.storeDelayedScaleDown(ctx, delayedScaleDown);
214215

0 commit comments

Comments
 (0)