[Improve] add some unit test (#2334)

Signed-off-by: yuluo-yx <[email protected]>
Co-authored-by: tomsun28 <[email protected]>

collector/src/test/java/org/apache/hertzbeat/collector/dispatch/MetricsCollectorQueueTest.java

@@ -17,28 +17,39 @@
 
 package org.apache.hertzbeat.collector.dispatch;
 
-import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.mockito.Mockito.mock;
 
 /**
  * Test case for {@link MetricsCollectorQueue}
  */
 class MetricsCollectorQueueTest {
 
+    private MetricsCollectorQueue metricsCollectorQueue;
+
+    private MetricsCollect mockJob;
+
     @BeforeEach
     void setUp() {
-    }
 
-    @AfterEach
-    void tearDown() {
+        metricsCollectorQueue = new MetricsCollectorQueue();
+        mockJob = mock(MetricsCollect.class);
     }
 
     @Test
-    void addJob() {
+    void testAddJob() throws InterruptedException {
+
+        metricsCollectorQueue.addJob(mockJob);
+        assertEquals(mockJob, metricsCollectorQueue.getJob());
     }
 
     @Test
-    void getJob() {
+    void testGetJobTimeout() throws InterruptedException {
+
+        assertNull(metricsCollectorQueue.getJob());
     }
+
 }

collector/src/test/java/org/apache/hertzbeat/collector/dispatch/WorkerPoolTest.java

@@ -17,28 +17,48 @@
 
 package org.apache.hertzbeat.collector.dispatch;
 
-import org.junit.jupiter.api.AfterEach;
+import java.util.concurrent.RejectedExecutionException;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
 
 /**
  * Test case for {@link WorkerPool}
  */
 class WorkerPoolTest {
 
+    private WorkerPool workerPool;
+
+    private Runnable mockTask;
+
     @BeforeEach
     void setUp() {
+
+        workerPool = new WorkerPool();
+        mockTask = mock(Runnable.class);
     }
 
-    @AfterEach
-    void tearDown() {
+    @Test
+    void testExecuteJob() {
+
+        assertDoesNotThrow(() -> workerPool.executeJob(mockTask));
     }
 
     @Test
-    void executeJob() {
+    void testExecuteJobThrowsException() {
+
+        workerPool = mock(WorkerPool.class);
+        doThrow(new RejectedExecutionException()).when(workerPool).executeJob(mockTask);
+
+        assertThrows(RejectedExecutionException.class, () -> workerPool.executeJob(mockTask));
     }
 
     @Test
-    void destroy() {
+    void testDestroy() {
+        assertDoesNotThrow(() -> workerPool.destroy());
     }
+
 }