ConversationHandler tests

Now testing that the threading parts of ConversationHandler works
miniwolf · Nov 24, 2024 · 75b6e59 · 75b6e59
1 parent affab03
commit 75b6e59
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/fagiServer/src/main/java/com/fagi/handler/ConversationHandler.java b/fagiServer/src/main/java/com/fagi/handler/ConversationHandler.java
@@ -43,6 +43,7 @@ public void tick() {
             conversation.addMessage(message);
             data.storeConversation(conversation);
         } catch (InterruptedException ie) {
+            Thread.currentThread().interrupt();
             ie.printStackTrace();
         }
     }

diff --git a/fagiServer/src/test/java/TextMessageIntegrationTests.java b/fagiServer/src/test/java/TextMessageIntegrationTests.java
@@ -78,4 +78,28 @@ void sendingAMessageToConversation_ShouldResultInConversationBeingStored() {
                 .verify(data, times(1))
                 .storeConversation(conversation);
     }
+
+    @Test
+    void callingRun_ShouldCallTickAndSendMesasage() throws InterruptedException {
+        when(data.getConversation(Mockito.anyLong())).thenReturn(conversation);
+
+        inputHandler.handleInput(message);
+
+        var conversationThread = new Thread(conversationHandler);
+        conversationThread.setDaemon(true);
+        conversationThread.start();
+
+        while (conversationHandler.queueSize() > 0 && conversationThread.isAlive()) {
+            Thread.sleep(10); // Minimal sleep to avoid busy-waiting
+        }
+
+        conversationThread.interrupt();
+
+        conversationThread.join(1000);
+
+        Assertions.assertAll(
+                () -> Assertions.assertTrue(conversationThread.isInterrupted()),
+                () -> Assertions.assertTrue(conversation.getMessages().contains(message))
+        );
+    }
 }