|
17 | 17 |
|
18 | 18 | package org.apache.hertzbeat.warehouse.store;
|
19 | 19 |
|
| 20 | +import java.util.List; |
| 21 | +import org.apache.hertzbeat.common.entity.message.CollectRep; |
20 | 22 | import org.apache.hertzbeat.warehouse.store.realtime.memory.MemoryDataStorage;
|
| 23 | +import org.apache.hertzbeat.warehouse.store.realtime.memory.MemoryProperties; |
21 | 24 | import org.junit.jupiter.api.BeforeEach;
|
22 | 25 | import org.junit.jupiter.api.Test;
|
| 26 | +import org.mockito.InjectMocks; |
| 27 | +import org.mockito.Mock; |
| 28 | +import org.mockito.MockitoAnnotations; |
| 29 | + |
| 30 | +import static org.junit.jupiter.api.Assertions.assertEquals; |
| 31 | +import static org.junit.jupiter.api.Assertions.assertNull; |
| 32 | +import static org.junit.jupiter.api.Assertions.assertTrue; |
| 33 | +import static org.mockito.Mockito.mock; |
| 34 | +import static org.mockito.Mockito.when; |
23 | 35 |
|
24 | 36 | /**
|
25 | 37 | * Test case for {@link MemoryDataStorage}
|
26 | 38 | */
|
| 39 | + |
27 | 40 | class MemoryDataStorageTest {
|
28 | 41 |
|
| 42 | + @Mock |
| 43 | + private MemoryProperties memoryProperties; |
| 44 | + |
| 45 | + @InjectMocks |
| 46 | + private MemoryDataStorage memoryDataStorage; |
| 47 | + |
29 | 48 | @BeforeEach
|
30 | 49 | void setUp() {
|
| 50 | + |
| 51 | + MockitoAnnotations.openMocks(this); |
| 52 | + |
| 53 | + when(memoryProperties.initSize()).thenReturn(null); |
| 54 | + memoryDataStorage = new MemoryDataStorage(memoryProperties); |
31 | 55 | }
|
32 | 56 |
|
33 | 57 | @Test
|
34 |
| - void getCurrentMetricsData() { |
| 58 | + void testGetCurrentMetricsDataByMetric() { |
| 59 | + |
| 60 | + Long monitorId = 1L; |
| 61 | + String metric = "cpuUsage"; |
| 62 | + CollectRep.MetricsData metricsData = mock(CollectRep.MetricsData.class); |
| 63 | + |
| 64 | + memoryDataStorage.saveData(metricsData); |
| 65 | + |
| 66 | + CollectRep.MetricsData result = memoryDataStorage.getCurrentMetricsData(monitorId, metric); |
| 67 | + |
| 68 | + assertNull(result); |
35 | 69 | }
|
36 | 70 |
|
37 | 71 | @Test
|
38 |
| - void destroy() { |
| 72 | + void testGetCurrentMetricsData() { |
| 73 | + |
| 74 | + Long monitorId = 1L; |
| 75 | + CollectRep.MetricsData metricsData1 = mock(CollectRep.MetricsData.class); |
| 76 | + CollectRep.MetricsData metricsData2 = mock(CollectRep.MetricsData.class); |
| 77 | + |
| 78 | + when(metricsData1.getId()).thenReturn(monitorId); |
| 79 | + when(metricsData1.getMetrics()).thenReturn("cpuUsage"); |
| 80 | + when(metricsData1.getCode()).thenReturn(CollectRep.Code.SUCCESS); |
| 81 | + |
| 82 | + when(metricsData2.getId()).thenReturn(monitorId); |
| 83 | + when(metricsData2.getMetrics()).thenReturn("memoryUsage"); |
| 84 | + when(metricsData2.getCode()).thenReturn(CollectRep.Code.SUCCESS); |
| 85 | + |
| 86 | + memoryDataStorage.saveData(metricsData1); |
| 87 | + memoryDataStorage.saveData(metricsData2); |
| 88 | + |
| 89 | + List<CollectRep.MetricsData> result = memoryDataStorage.getCurrentMetricsData(monitorId); |
| 90 | + |
| 91 | + assertEquals(2, result.size()); |
| 92 | + assertTrue(result.contains(metricsData1)); |
| 93 | + assertTrue(result.contains(metricsData2)); |
| 94 | + } |
| 95 | + |
| 96 | + @Test |
| 97 | + void testSaveDataFailure() { |
| 98 | + |
| 99 | + CollectRep.MetricsData metricsData = mock(CollectRep.MetricsData.class); |
| 100 | + when(metricsData.getCode()).thenReturn(CollectRep.Code.FAIL); |
| 101 | + |
| 102 | + memoryDataStorage.saveData(metricsData); |
| 103 | + |
| 104 | + List<CollectRep.MetricsData> result = memoryDataStorage.getCurrentMetricsData(metricsData.getId()); |
| 105 | + assertTrue(result.isEmpty()); |
39 | 106 | }
|
| 107 | + |
| 108 | + @Test |
| 109 | + void testDestroy() { |
| 110 | + |
| 111 | + CollectRep.MetricsData metricsData = mock(CollectRep.MetricsData.class); |
| 112 | + when(metricsData.getId()).thenReturn(1L); |
| 113 | + when(metricsData.getMetrics()).thenReturn("cpuUsage"); |
| 114 | + when(metricsData.getCode()).thenReturn(CollectRep.Code.SUCCESS); |
| 115 | + |
| 116 | + memoryDataStorage.saveData(metricsData); |
| 117 | + memoryDataStorage.destroy(); |
| 118 | + |
| 119 | + List<CollectRep.MetricsData> result = memoryDataStorage.getCurrentMetricsData(1L); |
| 120 | + assertTrue(result.isEmpty()); |
| 121 | + } |
| 122 | + |
40 | 123 | }
|
0 commit comments