This is an automated email from the ASF dual-hosted git repository.
gongchao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/hertzbeat.git
The following commit(s) were added to refs/heads/master by this push:
new 6c931956f [improve] add redis data storage test case (#2356)
6c931956f is described below
commit 6c931956fa0ee33cdd30e074dd56d3f0e9fadb75
Author: Calvin <[email protected]>
AuthorDate: Wed Jul 24 09:34:25 2024 +0800
[improve] add redis data storage test case (#2356)
Co-authored-by: tomsun28 <[email protected]>
---
.../warehouse/store/RedisDataStorageTest.java | 49 +++++++++++++++++++++-
1 file changed, 47 insertions(+), 2 deletions(-)
diff --git
a/warehouse/src/test/java/org/apache/hertzbeat/warehouse/store/RedisDataStorageTest.java
b/warehouse/src/test/java/org/apache/hertzbeat/warehouse/store/RedisDataStorageTest.java
index 163a26ccd..135184647 100644
---
a/warehouse/src/test/java/org/apache/hertzbeat/warehouse/store/RedisDataStorageTest.java
+++
b/warehouse/src/test/java/org/apache/hertzbeat/warehouse/store/RedisDataStorageTest.java
@@ -17,24 +17,69 @@
package org.apache.hertzbeat.warehouse.store;
+import com.google.common.collect.Lists;
+import java.util.List;
+import org.apache.hertzbeat.common.entity.message.CollectRep;
import org.apache.hertzbeat.warehouse.store.realtime.redis.RedisDataStorage;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
/**
* Test case for {@link RedisDataStorage}
*/
class RedisDataStorageTest {
+ @Mock
+ private RedisDataStorage redisDataStorage;
@BeforeEach
void setUp() {
+ MockitoAnnotations.openMocks(this);
}
@Test
- void getCurrentMetricsData() {
+ void testGetCurrentMetricsData() {
+ Long monitorId = 1L;
+ String metric = "testMetric";
+ CollectRep.MetricsData expectedMetricsData =
CollectRep.MetricsData.newBuilder().setMetrics(metric).setId(monitorId).build();
+
Mockito.doReturn(expectedMetricsData).when(redisDataStorage).getCurrentMetricsData(monitorId,
metric);
+
+ CollectRep.MetricsData actualMetricsData =
redisDataStorage.getCurrentMetricsData(monitorId, metric);
+
+ assertNotNull(actualMetricsData);
+ assertEquals(expectedMetricsData, actualMetricsData);
+
+ verify(redisDataStorage, times(1)).getCurrentMetricsData(monitorId,
metric);
+ }
+
+ @Test
+ void testGetCurrentMetricsDataByMonitorId() {
+ Long monitorId = 1L;
+ String metric = "testMetric";
+ CollectRep.MetricsData expectedMetricsData =
CollectRep.MetricsData.newBuilder().setMetrics(metric).setId(monitorId).build();
+
Mockito.doReturn(Lists.newArrayList(expectedMetricsData)).when(redisDataStorage).getCurrentMetricsData(monitorId);
+
+ List<CollectRep.MetricsData> actualMetricsData =
redisDataStorage.getCurrentMetricsData(monitorId);
+
+ assertNotNull(actualMetricsData);
+ assertEquals(expectedMetricsData, actualMetricsData.get(0));
+
+ verify(redisDataStorage, times(1)).getCurrentMetricsData(monitorId);
}
@Test
- void destroy() {
+ void testSaveData() {
+ long monitorId = 1L;
+ String metric = "testMetric";
+ CollectRep.MetricsData expectedMetricsData =
CollectRep.MetricsData.newBuilder().setMetrics(metric).setId(monitorId).build();
+ redisDataStorage.saveData(expectedMetricsData);
+
+ verify(redisDataStorage, times(1)).saveData(expectedMetricsData);
}
}
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
