test(common): Add unit tests for BufferQueue

Author: codeboyzhou

pom.xml

@@ -58,6 +58,7 @@
         <junit5.version>5.10.2</junit5.version>
         <logback.version>1.5.18</logback.version>
         <mcp-sdk.version>0.10.0</mcp-sdk.version>
+        <mockito.version>5.18.0</mockito.version>
         <reflections.version>0.10.2</reflections.version>
     </properties>
 
@@ -112,6 +113,12 @@
             <version>${junit5.version}</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.mockito</groupId>
+            <artifactId>mockito-core</artifactId>
+            <version>${mockito.version}</version>
+            <scope>test</scope>
+        </dependency>
         <dependency>
             <groupId>org.reflections</groupId>
             <artifactId>reflections</artifactId>

src/main/java/com/github/codeboyzhou/mcp/declarative/exception/McpServerException.java

@@ -2,10 +2,6 @@
 
 public class McpServerException extends RuntimeException {
 
-    public McpServerException(String message) {
-        super(message);
-    }
-
     public McpServerException(String message, Throwable cause) {
         super(message, cause);
     }

src/test/java/com/github/codeboyzhou/mcp/declarative/common/BufferQueueTest.java

@@ -0,0 +1,58 @@
+package com.github.codeboyzhou.mcp.declarative.common;
+
+import org.junit.jupiter.api.Test;
+
+import java.lang.reflect.Field;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.LinkedBlockingQueue;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+
+class BufferQueueTest {
+
+    @Test
+    void testNewInstance() {
+        IllegalArgumentException e = assertThrows(IllegalArgumentException.class, () -> new BufferQueue<>(0));
+        assertEquals("delayMillis must be greater than 0", e.getMessage());
+    }
+
+    @Test
+    @SuppressWarnings("unchecked")
+    void testSubmitCatchException() throws Exception {
+        BlockingQueue<String> queueMock = mock(LinkedBlockingQueue.class);
+        doThrow(InterruptedException.class).when(queueMock).put(anyString());
+
+        BufferQueue<String> bufferQueue = new BufferQueue<>();
+        Field queue = BufferQueue.class.getDeclaredField("queue");
+        queue.setAccessible(true);
+        queue.set(bufferQueue, queueMock);
+
+        bufferQueue.submit("test");
+
+        verify(queueMock).put("test");
+    }
+
+    @Test
+    @SuppressWarnings("unchecked")
+    void testConsumeCatchException() throws Exception {
+        BlockingQueue<String> queueMock = mock(LinkedBlockingQueue.class);
+        doThrow(InterruptedException.class).when(queueMock).take();
+
+        BufferQueue<String> bufferQueue = new BufferQueue<>();
+        Field queue = BufferQueue.class.getDeclaredField("queue");
+        queue.setAccessible(true);
+        queue.set(bufferQueue, queueMock);
+
+        bufferQueue.consume(string -> {
+            // do nothing
+        });
+
+        verify(queueMock).take();
+    }
+
+}