@@ -45,6 +45,7 @@ public class AtomicRateLimiterTest {
45
45
public static final String LIMITER_NAME = "test" ;
46
46
public static final long CYCLE_IN_NANOS = 500_000_000L ;
47
47
public static final long POLL_INTERVAL_IN_NANOS = 2_000_000L ;
48
+ public static final int TIMEOUT = 100 ;
48
49
private RateLimiterConfig rateLimiterConfig ;
49
50
private AtomicRateLimiter rateLimiter ;
50
51
private AtomicRateLimiter .AtomicRateLimiterMetrics metrics ;
@@ -109,7 +110,7 @@ public void reserveAndRefresh() throws Exception {
109
110
caller .setDaemon (true );
110
111
caller .start ();
111
112
awaitImpatiently ()
112
- .atMost (10 , MILLISECONDS )
113
+ .atMost (TIMEOUT , MILLISECONDS )
113
114
.until (caller ::getState , equalTo (Thread .State .TIMED_WAITING ));
114
115
then (metrics .getAvailablePermissions ()).isEqualTo (-1 );
115
116
then (metrics .getNanosToWait ()).isEqualTo (CYCLE_IN_NANOS + CYCLE_IN_NANOS );
@@ -140,7 +141,7 @@ public void reserveFewThenSkipCyclesBeforeRefresh() throws Exception {
140
141
firstCaller .setDaemon (true );
141
142
firstCaller .start ();
142
143
awaitImpatiently ()
143
- .atMost (50 , MILLISECONDS )
144
+ .atMost (TIMEOUT , MILLISECONDS )
144
145
.until (firstCaller ::getState , equalTo (Thread .State .TIMED_WAITING ));
145
146
then (metrics .getAvailablePermissions ()).isEqualTo (-1 );
146
147
then (metrics .getNanosToWait ()).isEqualTo (CYCLE_IN_NANOS * 2 );
@@ -153,7 +154,7 @@ public void reserveFewThenSkipCyclesBeforeRefresh() throws Exception {
153
154
secondCaller .setDaemon (true );
154
155
secondCaller .start ();
155
156
awaitImpatiently ()
156
- .atMost (50 , MILLISECONDS )
157
+ .atMost (TIMEOUT , MILLISECONDS )
157
158
.until (secondCaller ::getState , equalTo (Thread .State .TIMED_WAITING ));
158
159
then (metrics .getAvailablePermissions ()).isEqualTo (-2 );
159
160
then (metrics .getNanosToWait ()).isEqualTo (CYCLE_IN_NANOS * 3 );
@@ -187,7 +188,7 @@ public void rejectedByTimeout() throws Exception {
187
188
caller .start ();
188
189
189
190
awaitImpatiently ()
190
- .atMost (100 , MILLISECONDS )
191
+ .atMost (TIMEOUT , MILLISECONDS )
191
192
.until (caller ::getState , equalTo (Thread .State .TIMED_WAITING ));
192
193
then (metrics .getAvailablePermissions ()).isEqualTo (0 );
193
194
then (metrics .getNanosToWait ()).isEqualTo (CYCLE_IN_NANOS );
@@ -223,7 +224,7 @@ public void waitingThreadIsInterrupted() throws Exception {
223
224
caller .start ();
224
225
225
226
awaitImpatiently ()
226
- .atMost (100 , MILLISECONDS )
227
+ .atMost (TIMEOUT , MILLISECONDS )
227
228
.until (caller ::getState , equalTo (Thread .State .TIMED_WAITING ));
228
229
then (metrics .getAvailablePermissions ()).isEqualTo (0 );
229
230
then (metrics .getNanosToWait ()).isEqualTo (CYCLE_IN_NANOS );
0 commit comments