From c6b7feddc591e29ecf0f7e1b5517d05223c0fa11 Mon Sep 17 00:00:00 2001
From: ScarletRedMan <terentev.andrey.2002@gmail.com>
Date: Fri, 12 Jan 2024 15:46:58 +0700
Subject: [PATCH] Added timeout for test

---
 .../ru/dragonestia/picker/picker/RoundRobinTests.java     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/app/src/test/java/ru/dragonestia/picker/picker/RoundRobinTests.java b/app/src/test/java/ru/dragonestia/picker/picker/RoundRobinTests.java
index 209c15f..3734db8 100644
--- a/app/src/test/java/ru/dragonestia/picker/picker/RoundRobinTests.java
+++ b/app/src/test/java/ru/dragonestia/picker/picker/RoundRobinTests.java
@@ -12,6 +12,8 @@ import ru.dragonestia.picker.repository.RoomRepository;
 import ru.dragonestia.picker.repository.UserRepository;
 import ru.dragonestia.picker.util.UserFiller;
 
+import java.time.Duration;
+
 @SpringBootTest
 @Import({FillingNodesConfig.class, UserFiller.class})
 public class RoundRobinTests {
@@ -80,8 +82,10 @@ public class RoundRobinTests {
         }
 
         { // fifth iteration. Take 9 users. expected none result
-            var roomOpt = roomRepository.pickFree(node, userFiller.createRandomUsers(9));
-            Assertions.assertTrue(roomOpt.isEmpty());
+            Assertions.assertTimeoutPreemptively(Duration.ofSeconds(1), () -> {
+                var roomOpt = roomRepository.pickFree(node, userFiller.createRandomUsers(9));
+                Assertions.assertTrue(roomOpt.isEmpty());
+            });
         }
     }
 }