Remove sleeping from LightyModuleTest

Remove sleeping from tests. Instead use dedicated shutdown methods which are designed to accept timeouts. JIRA: LIGHTY-299 Signed-off-by: Ivan Hrasko <[email protected]>
PANTHEONtech · Sep 12, 2024 · 6d3c1ae · 6d3c1ae
1 parent 5721f78
commit 6d3c1ae
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...-core/lighty-controller/src/test/java/io/lighty/core/controller/api/LightyModuleTest.java b/...-core/lighty-controller/src/test/java/io/lighty/core/controller/api/LightyModuleTest.java
@@ -72,10 +72,9 @@ public void testStartStop_whenAlreadyStartedStopped() throws Exception {
             Assert.fail("Init timed out.", e);
         }
         Mockito.verify(executorService, Mockito.times(1)).execute(Mockito.any());
-        this.moduleUnderTest.shutdown();
+        this.moduleUnderTest.shutdown(SLEEP_AFTER_SHUTDOWN_TIMEOUT, TimeUnit.MILLISECONDS);
         Mockito.verify(executorService, Mockito.times(2)).execute(Mockito.any());
-        Thread.sleep(SLEEP_AFTER_SHUTDOWN_TIMEOUT);
-        this.moduleUnderTest.shutdown();
+        this.moduleUnderTest.shutdown(SLEEP_AFTER_SHUTDOWN_TIMEOUT, TimeUnit.MILLISECONDS);
         Mockito.verify(executorService, Mockito.times(2)).execute(Mockito.any());
     }