Fix Flaky throttling test

Move then when then outside the multithreaded execution. This could reduce weird behavior in the mockito mocking storage Signed-off-by: Jasper Kamerling <[email protected]>
OSGP · May 2, 2024 · 578b6bf · 578b6bf
1 parent ebea361
commit 578b6bf
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...vice/src/test/java/org/opensmartgridplatform/throttling/PermitsPerNetworkSegmentTest.java b/...vice/src/test/java/org/opensmartgridplatform/throttling/PermitsPerNetworkSegmentTest.java
@@ -217,12 +217,13 @@ void testHighPrioPool() {
 
     final long start = System.currentTimeMillis();
 
+    when(this.permitReleasedNotifier.waitForAvailablePermit(btsId, cellId, 1000))
+        .thenReturn(false)
+        .thenReturn(true);
+
     final ScheduledThreadPoolExecutor executor = new ScheduledThreadPoolExecutor(2);
     executor.schedule(
         () -> {
-          when(this.permitReleasedNotifier.waitForAvailablePermit(btsId, cellId, 1000))
-              .thenReturn(false)
-              .thenReturn(true);
           this.permitsPerNetworkSegment.releasePermit(
               throttlingConfigId, clientId, btsId, cellId, requestId);