Revert unit tests changes.

Rollczi · Nov 15, 2024 · 8b441d5 · 8b441d5
1 parent dbd4910
commit 8b441d5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...ev/rollczi/litecommands/argument/resolver/completeable/CompletableFutureArgumentTest.java b/...ev/rollczi/litecommands/argument/resolver/completeable/CompletableFutureArgumentTest.java
@@ -4,6 +4,7 @@
 import dev.rollczi.litecommands.annotations.command.Command;
 import dev.rollczi.litecommands.annotations.execute.Execute;
 import dev.rollczi.litecommands.reflect.type.TypeToken;
+import dev.rollczi.litecommands.scheduler.SchedulerExecutorPoolImpl;
 import dev.rollczi.litecommands.unit.annotations.LiteTestSpec;
 import dev.rollczi.litecommands.unit.blocking.Blocking;
 import dev.rollczi.litecommands.unit.blocking.BlockingArgument;
@@ -15,6 +16,9 @@
 
 class CompletableFutureArgumentTest extends LiteTestSpec {
 
+    static LiteTestConfig config = builder -> builder
+        .scheduler(new SchedulerExecutorPoolImpl("litecommands-test"));
+
     public static final Class<CompletableFuture<String>> FUTURE_TYPE = new TypeToken<CompletableFuture<String>>() {}.getRawType();
 
     @Command(name = "async")
@@ -37,7 +41,7 @@ void test() {
 
         await()
             .atLeast(400, TimeUnit.MILLISECONDS)
-            .atMost(1500, TimeUnit.MILLISECONDS)
+            .atMost(900, TimeUnit.MILLISECONDS)
             .until(() -> completableFuture.isDone());
 
         assertThat(completableFuture.join())