Merge branch 'master' into fixCodeStyle

apache · May 26, 2024 · 61c811c · 61c811c
2 parents 7b3ad71 + 49ebdfb
commit 61c811c
Showing 1 changed file with 114 additions and 2 deletions.
diff --git a/collector/src/test/java/org/apache/hertzbeat/collector/collect/icmp/IcmpCollectImplTest.java b/collector/src/test/java/org/apache/hertzbeat/collector/collect/icmp/IcmpCollectImplTest.java
@@ -17,23 +17,135 @@
 
 package org.apache.hertzbeat.collector.collect.icmp;
 
+import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+
+import java.io.IOException;
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+import java.util.ArrayList;
+import java.util.List;
+import org.apache.hertzbeat.common.entity.job.Metrics;
+import org.apache.hertzbeat.common.entity.job.protocol.IcmpProtocol;
+import org.apache.hertzbeat.common.entity.message.CollectRep;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.Mockito;
+import org.mockito.junit.jupiter.MockitoExtension;
 
 /**
  * Test case for {@link IcmpCollectImpl}
  */
+@ExtendWith(MockitoExtension.class)
 class IcmpCollectImplTest {
 
+    @Mock
+    IcmpProtocol icmpProtocol;
+
+    @Mock
+    Metrics metrics;
+
+    @Mock
+    InetAddress inetAddress;
+
+    @Mock
+    CollectRep.MetricsData.Builder builder;
+
+    @InjectMocks
+    private IcmpCollectImpl icmpCollect;
+
     @BeforeEach
     void setUp() {
+        icmpProtocol = IcmpProtocol.builder()
+            .host("127.0.0.1")
+            .timeout("3000")
+            .build();
+        List<String> aliasField = new ArrayList<>();
+        aliasField.add("responseTime");
+        metrics = new Metrics();
+        metrics.setName("test");
+        metrics.setIcmp(icmpProtocol);
+        metrics.setAliasFields(aliasField);
+        builder = CollectRep.MetricsData.newBuilder();
+    }
+
+    @Test
+    void testPreCheck() {
+        assertDoesNotThrow(() -> {
+            icmpCollect.preCheck(metrics);
+        });
+        assertThrows(IllegalArgumentException.class, () -> {
+            icmpCollect.preCheck(null);
+        });
+        metrics.setIcmp(null);
+        assertThrows(IllegalArgumentException.class, () -> {
+            icmpCollect.preCheck(null);
+        });
+    }
+
+    @Test
+    void testCollect() throws Exception {
+        try (MockedStatic<InetAddress> mockedInetAddress = Mockito.mockStatic(InetAddress.class)) {
+            mockedInetAddress.when(() -> InetAddress.getByName(Mockito.anyString())).thenReturn(inetAddress);
+            Mockito.when(inetAddress.isReachable(Mockito.anyInt())).thenReturn(true);
+            assertDoesNotThrow(() -> icmpCollect.collect(builder, 1L, "app", metrics));
+            assertEquals(1, builder.getValuesCount());
+            assertNotNull(builder.getValues(0).getColumns(0));
+
+        }
+    }
+
+    @Test
+    void testUnreachable() throws Exception {
+        try (MockedStatic<InetAddress> mockedInetAddress = Mockito.mockStatic(InetAddress.class)) {
+            mockedInetAddress.when(() -> InetAddress.getByName(Mockito.anyString())).thenReturn(inetAddress);
+            Mockito.when(inetAddress.isReachable(Mockito.anyInt())).thenReturn(false);
+            assertDoesNotThrow(() -> icmpCollect.collect(builder, 1L, "app", metrics));
+            assertEquals(CollectRep.Code.UN_REACHABLE, builder.getCode());
+            assertNotNull(builder.getMsg());
+        }
+    }
+
+    @Test
+    void testUnknownHostException() {
+        try (MockedStatic<InetAddress> mockedInetAddress = Mockito.mockStatic(InetAddress.class)) {
+            mockedInetAddress.when(() -> InetAddress.getByName(Mockito.anyString())).thenThrow(new UnknownHostException("Mocked exception"));
+            assertDoesNotThrow(() -> icmpCollect.collect(builder, 1L, "app", metrics));
+            assertEquals(CollectRep.Code.UN_REACHABLE, builder.getCode());
+            assertNotNull(builder.getMsg());
+        }
+    }
+
+    @Test
+    void testIoException() throws Exception {
+        try (MockedStatic<InetAddress> mockedInetAddress = Mockito.mockStatic(InetAddress.class)) {
+            mockedInetAddress.when(() -> InetAddress.getByName(Mockito.anyString())).thenReturn(inetAddress);
+            Mockito.when(inetAddress.isReachable(Mockito.anyInt())).thenThrow(new IOException("Mocked exception"));
+            assertDoesNotThrow(() -> icmpCollect.collect(builder, 1L, "app", metrics));
+            assertEquals(CollectRep.Code.UN_REACHABLE, builder.getCode());
+            assertNotNull(builder.getMsg());
+        }
     }
 
     @Test
-    void getInstance() {
+    void testException() throws Exception {
+        try (MockedStatic<InetAddress> mockedInetAddress = Mockito.mockStatic(InetAddress.class)) {
+            mockedInetAddress.when(() -> InetAddress.getByName(Mockito.anyString())).thenReturn(inetAddress);
+            Mockito.when(inetAddress.isReachable(Mockito.anyInt())).thenThrow(new RuntimeException("Mocked exception"));
+            assertDoesNotThrow(() -> icmpCollect.collect(builder, 1L, "app", metrics));
+            assertEquals(CollectRep.Code.FAIL, builder.getCode());
+            assertNotNull(builder.getMsg());
+        }
     }
 
     @Test
-    void collect() {
+    void testSupportProtocol() {
+        assertEquals("icmp", icmpCollect.supportProtocol());
     }
 }