@@ -174,11 +174,13 @@ public void should_evict_down_node_metrics_when_timeout_fires() throws Exception
174
174
// trigger node1 UP -> DOWN
175
175
eventBus .fire (NodeStateEvent .changed (NodeState .UP , NodeState .DOWN , node1 ));
176
176
177
- Thread .sleep (expireAfter .toMillis ());
177
+ Thread .sleep (expireAfter .toMillis () + 100 );
178
178
179
179
// then node-level metrics should be evicted from node1, but
180
180
// node2 and node3 metrics should not have been evicted
181
- await ().untilAsserted (() -> assertNodeMetricsEvicted (session , node1 ));
181
+ await ()
182
+ .atMost (Duration .ofSeconds (15 ))
183
+ .untilAsserted (() -> assertNodeMetricsEvicted (session , node1 ));
182
184
assertNodeMetricsNotEvicted (session , node2 );
183
185
assertNodeMetricsNotEvicted (session , node3 );
184
186
@@ -226,7 +228,8 @@ public void should_not_evict_down_node_metrics_when_node_is_back_up_before_timeo
226
228
eventBus .fire (NodeStateEvent .changed (NodeState .FORCED_DOWN , NodeState .UP , node2 ));
227
229
eventBus .fire (NodeStateEvent .added (node3 ));
228
230
229
- Thread .sleep (expireAfter .toMillis ());
231
+ // Add a small buffer to ensure the timeout would have fired if it wasn't cancelled
232
+ Thread .sleep (expireAfter .toMillis () + 100 );
230
233
231
234
// then no node-level metrics should be evicted
232
235
assertNodeMetricsNotEvicted (session , node1 );
0 commit comments