|
|
@@ -1,291 +1,175 @@
|
|
1
|
1
|
package com.water.data_engine.controller;
|
|
2
|
2
|
|
|
3
|
|
-import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
4
|
|
-import com.water.data_engine.entity.HistoricalQuery;
|
|
5
|
|
-import com.water.data_engine.entity.WaterQuantity;
|
|
6
|
|
-import com.water.data_engine.entity.WaterQuality;
|
|
7
|
|
-import com.water.data_engine.service.HistoryDataService;
|
|
8
|
|
-import org.junit.jupiter.api.BeforeEach;
|
|
9
|
|
-import org.junit.jupiter.api.DisplayName;
|
|
|
3
|
+import com.water.common.core.result.R;
|
|
|
4
|
+import com.water.data_engine.entity.DataReport;
|
|
|
5
|
+import com.water.data_engine.service.DataStatisticsService;
|
|
|
6
|
+import com.water.data_engine.service.ReportService;
|
|
10
|
7
|
import org.junit.jupiter.api.Test;
|
|
11
|
8
|
import org.junit.jupiter.api.extension.ExtendWith;
|
|
12
|
9
|
import org.mockito.InjectMocks;
|
|
13
|
10
|
import org.mockito.Mock;
|
|
14
|
11
|
import org.mockito.junit.jupiter.MockitoExtension;
|
|
15
|
|
-import org.springframework.http.HttpStatus;
|
|
16
|
|
-import org.springframework.http.ResponseEntity;
|
|
17
|
12
|
|
|
18
|
|
-import java.math.BigDecimal;
|
|
19
|
|
-import java.time.LocalDateTime;
|
|
|
13
|
+import java.time.LocalDate;
|
|
20
|
14
|
import java.util.List;
|
|
|
15
|
+import java.util.Map;
|
|
21
|
16
|
|
|
22
|
17
|
import static org.junit.jupiter.api.Assertions.*;
|
|
23
|
|
-import static org.mockito.ArgumentMatchers.any;
|
|
|
18
|
+import static org.mockito.ArgumentMatchers.*;
|
|
24
|
19
|
import static org.mockito.Mockito.*;
|
|
25
|
20
|
|
|
26
|
21
|
/**
|
|
27
|
|
- * 历史数据控制器测试
|
|
|
22
|
+ * HistoryDataController 单元测试（对应 Issue #71）。
|
|
|
23
|
+ *
|
|
|
24
|
+ * 覆盖历史数据回溯、水量/水质/报警报表、报表生成/列表/详情/发布等 API。
|
|
|
25
|
+ * 真正匹配 HistoryDataController 的实际 API（而非旧版脱节的 HistoryDataService）。
|
|
28
|
26
|
*/
|
|
29
|
27
|
@ExtendWith(MockitoExtension.class)
|
|
30
|
28
|
class HistoryDataControllerTest {
|
|
31
|
29
|
|
|
32
|
|
- @Mock
|
|
33
|
|
- private HistoryDataService historyDataService;
|
|
|
30
|
+ @Mock DataStatisticsService dataStatisticsService;
|
|
|
31
|
+ @Mock ReportService reportService;
|
|
|
32
|
+
|
|
|
33
|
+ @InjectMocks HistoryDataController controller;
|
|
|
34
|
+
|
|
|
35
|
+ private DataReport sampleReport() {
|
|
|
36
|
+ DataReport r = new DataReport();
|
|
|
37
|
+ r.setId(1L);
|
|
|
38
|
+ r.setReportCode("WATER-VOL-1");
|
|
|
39
|
+ r.setReportType("water_volume");
|
|
|
40
|
+ r.setReportName("水量汇总报表");
|
|
|
41
|
+ r.setStatus("GENERATED");
|
|
|
42
|
+ r.setPeriodStart(LocalDate.of(2026, 6, 1));
|
|
|
43
|
+ r.setPeriodEnd(LocalDate.of(2026, 6, 7));
|
|
|
44
|
+ return r;
|
|
|
45
|
+ }
|
|
34
|
46
|
|
|
35
|
|
- @InjectMocks
|
|
36
|
|
- private HistoryDataController historyDataController;
|
|
|
47
|
+ // ---------- 历史数据回溯 ----------
|
|
37
|
48
|
|
|
38
|
|
- @BeforeEach
|
|
39
|
|
- void setUp() {
|
|
40
|
|
- // 重置mock
|
|
41
|
|
- reset(historyDataService);
|
|
|
49
|
+ @Test
|
|
|
50
|
+ void queryHistoricalData_returnsOk() {
|
|
|
51
|
+ when(dataStatisticsService.queryHistoricalData(eq("water_volume"), any(), any(), any()))
|
|
|
52
|
+ .thenReturn(Map.of("totalRecords", 100, "records", List.of()));
|
|
|
53
|
+ R<Map<String, Object>> r = controller.queryHistoricalData("water_volume", null, null, null);
|
|
|
54
|
+ assertEquals(200, r.getCode());
|
|
|
55
|
+ assertEquals(100, r.getData().get("totalRecords"));
|
|
42
|
56
|
}
|
|
43
|
57
|
|
|
44
|
58
|
@Test
|
|
45
|
|
- @DisplayName("查询水量历史数据-正常响应")
|
|
46
|
|
- void testQueryQuantityHistory_Success() {
|
|
47
|
|
- // Given
|
|
48
|
|
- HistoricalQuery query = new HistoricalQuery();
|
|
49
|
|
- query.setArea("城东");
|
|
50
|
|
- query.setStartTime(LocalDateTime.of(2026, 1, 1, 0, 0));
|
|
51
|
|
- query.setEndTime(LocalDateTime.of(2026, 1, 31, 23, 59));
|
|
52
|
|
- query.setPageNum(1);
|
|
53
|
|
- query.setPageSize(10);
|
|
54
|
|
-
|
|
55
|
|
- WaterQuantity wq = new WaterQuantity();
|
|
56
|
|
- wq.setId(1L);
|
|
57
|
|
- wq.setMonitorPoint("城东水厂出口");
|
|
58
|
|
- wq.setArea("城东");
|
|
59
|
|
- wq.setFlowRate(new BigDecimal("120.5"));
|
|
60
|
|
- wq.setPressure(new BigDecimal("0.35"));
|
|
61
|
|
- wq.setCollectTime(LocalDateTime.of(2026, 1, 15, 10, 0));
|
|
62
|
|
-
|
|
63
|
|
- Page<WaterQuantity> mockPage = new Page<>(1, 10);
|
|
64
|
|
- mockPage.setRecords(List.of(wq));
|
|
65
|
|
- mockPage.setTotal(1);
|
|
66
|
|
-
|
|
67
|
|
- when(historyDataService.queryQuantityHistory(query)).thenReturn(mockPage);
|
|
|
59
|
+ void queryHistoricalData_exceptionReturnsFail() {
|
|
|
60
|
+ when(dataStatisticsService.queryHistoricalData(any(), any(), any(), any()))
|
|
|
61
|
+ .thenThrow(new RuntimeException("db error"));
|
|
|
62
|
+ R<Map<String, Object>> r = controller.queryHistoricalData(null, null, null, null);
|
|
|
63
|
+ assertEquals(500, r.getCode());
|
|
|
64
|
+ assertTrue(r.getMessage().contains("历史数据查询失败"));
|
|
|
65
|
+ }
|
|
68
|
66
|
|
|
69
|
|
- // When
|
|
70
|
|
- ResponseEntity<Page<WaterQuantity>> response = historyDataController.queryQuantityHistory(query);
|
|
|
67
|
+ // ---------- 报表查询（get）----------
|
|
71
|
68
|
|
|
72
|
|
- // Then
|
|
73
|
|
- assertNotNull(response);
|
|
74
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
75
|
|
- assertNotNull(response.getBody());
|
|
76
|
|
- assertEquals(1, response.getBody().getRecords().size());
|
|
77
|
|
- assertEquals("城东", response.getBody().getRecords().get(0).getArea());
|
|
78
|
|
- verify(historyDataService, times(1)).queryQuantityHistory(query);
|
|
|
69
|
+ @Test
|
|
|
70
|
+ void getWaterVolumeSummary_delegates() {
|
|
|
71
|
+ when(dataStatisticsService.getWaterVolumeSummary(any(), any()))
|
|
|
72
|
+ .thenReturn(Map.of("totalSupply", 5000.0));
|
|
|
73
|
+ R<Map<String, Object>> r = controller.getWaterVolumeSummary(null, null);
|
|
|
74
|
+ assertEquals(200, r.getCode());
|
|
|
75
|
+ assertEquals(5000.0, r.getData().get("totalSupply"));
|
|
79
|
76
|
}
|
|
80
|
77
|
|
|
81
|
78
|
@Test
|
|
82
|
|
- @DisplayName("查询水量历史数据-分页参数为空")
|
|
83
|
|
- void testQueryQuantityHistory_EmptyParams() {
|
|
84
|
|
- // Given
|
|
85
|
|
- HistoricalQuery query = new HistoricalQuery();
|
|
86
|
|
- query.setPageNum(1);
|
|
87
|
|
- query.setPageSize(20);
|
|
88
|
|
-
|
|
89
|
|
- Page<WaterQuantity> mockPage = new Page<>(1, 20);
|
|
90
|
|
- mockPage.setRecords(List.of());
|
|
91
|
|
- mockPage.setTotal(0);
|
|
92
|
|
-
|
|
93
|
|
- when(historyDataService.queryQuantityHistory(query)).thenReturn(mockPage);
|
|
94
|
|
-
|
|
95
|
|
- // When
|
|
96
|
|
- ResponseEntity<Page<WaterQuantity>> response = historyDataController.queryQuantityHistory(query);
|
|
97
|
|
-
|
|
98
|
|
- // Then
|
|
99
|
|
- assertNotNull(response);
|
|
100
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
101
|
|
- assertNotNull(response.getBody());
|
|
102
|
|
- assertEquals(0, response.getBody().getRecords().size());
|
|
103
|
|
- verify(historyDataService, times(1)).queryQuantityHistory(query);
|
|
|
79
|
+ void getWaterQualityRate_delegates() {
|
|
|
80
|
+ when(dataStatisticsService.getWaterQualityRate(any(), any()))
|
|
|
81
|
+ .thenReturn(Map.of("passRate", 98.5));
|
|
|
82
|
+ R<Map<String, Object>> r = controller.getWaterQualityRate(null, null);
|
|
|
83
|
+ assertEquals(200, r.getCode());
|
|
104
|
84
|
}
|
|
105
|
85
|
|
|
106
|
86
|
@Test
|
|
107
|
|
- @DisplayName("查询水质历史数据-正常响应")
|
|
108
|
|
- void testQueryQualityHistory_Success() {
|
|
109
|
|
- // Given
|
|
110
|
|
- HistoricalQuery query = new HistoricalQuery();
|
|
111
|
|
- query.setPointCode("WQ001");
|
|
112
|
|
- query.setStartTime(LocalDateTime.of(2026, 1, 1, 0, 0));
|
|
113
|
|
- query.setEndTime(LocalDateTime.of(2026, 1, 31, 23, 59));
|
|
114
|
|
-
|
|
115
|
|
- WaterQuality wq = new WaterQuality();
|
|
116
|
|
- wq.setId(1L);
|
|
117
|
|
- wq.setMonitorPoint("水厂出口");
|
|
118
|
|
- wq.setPointCode("WQ001");
|
|
119
|
|
- wq.setPh(new BigDecimal("7.2"));
|
|
120
|
|
- wq.setTurbidity(new BigDecimal("0.5"));
|
|
121
|
|
- wq.setIsQualified(1);
|
|
122
|
|
- wq.setCollectTime(LocalDateTime.of(2026, 1, 10, 8, 0));
|
|
123
|
|
-
|
|
124
|
|
- Page<WaterQuality> mockPage = new Page<>(1, 20);
|
|
125
|
|
- mockPage.setRecords(List.of(wq));
|
|
126
|
|
- mockPage.setTotal(1);
|
|
127
|
|
-
|
|
128
|
|
- when(historyDataService.queryQualityHistory(query)).thenReturn(mockPage);
|
|
|
87
|
+ void getAlarmStatistics_delegates() {
|
|
|
88
|
+ when(dataStatisticsService.getAlarmStatistics(any(), any()))
|
|
|
89
|
+ .thenReturn(Map.of("totalAlarms", 12));
|
|
|
90
|
+ R<Map<String, Object>> r = controller.getAlarmStatistics(null, null);
|
|
|
91
|
+ assertEquals(200, r.getCode());
|
|
|
92
|
+ }
|
|
129
|
93
|
|
|
130
|
|
- // When
|
|
131
|
|
- ResponseEntity<Page<WaterQuality>> response = historyDataController.queryQualityHistory(query);
|
|
|
94
|
+ // ---------- 报表生成（generate，校验 toReportMap 字段映射）----------
|
|
132
|
95
|
|
|
133
|
|
- // Then
|
|
134
|
|
- assertNotNull(response);
|
|
135
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
136
|
|
- assertNotNull(response.getBody());
|
|
137
|
|
- assertEquals(1, response.getBody().getRecords().size());
|
|
138
|
|
- assertEquals("WQ001", response.getBody().getRecords().get(0).getPointCode());
|
|
139
|
|
- verify(historyDataService, times(1)).queryQualityHistory(query);
|
|
|
96
|
+ @Test
|
|
|
97
|
+ void generateWaterVolumeReport_mapsReportFields() {
|
|
|
98
|
+ when(reportService.generateWaterVolumeReport(any(), any(), any())).thenReturn(sampleReport());
|
|
|
99
|
+ R<Map<String, Object>> r = controller.generateWaterVolumeReport("周报", null, null);
|
|
|
100
|
+ assertEquals(200, r.getCode());
|
|
|
101
|
+ Map<String, Object> data = r.getData();
|
|
|
102
|
+ // 验证字段映射：reportNo <- reportCode, title <- reportName
|
|
|
103
|
+ assertEquals("WATER-VOL-1", data.get("reportNo"));
|
|
|
104
|
+ assertEquals("water_volume", data.get("reportType"));
|
|
|
105
|
+ assertEquals("水量汇总报表", data.get("title"));
|
|
|
106
|
+ assertEquals("GENERATED", data.get("status"));
|
|
|
107
|
+ assertEquals("2026-06-01~2026-06-07", data.get("period"));
|
|
140
|
108
|
}
|
|
141
|
109
|
|
|
142
|
110
|
@Test
|
|
143
|
|
- @DisplayName("查询水质历史数据-无结果返回空列表")
|
|
144
|
|
- void testQueryQualityHistory_EmptyResult() {
|
|
145
|
|
- // Given
|
|
146
|
|
- HistoricalQuery query = new HistoricalQuery();
|
|
147
|
|
- query.setStartTime(LocalDateTime.of(2026, 1, 1, 0, 0));
|
|
148
|
|
- query.setEndTime(LocalDateTime.of(2026, 1, 31, 23, 59));
|
|
149
|
|
-
|
|
150
|
|
- Page<WaterQuality> mockPage = new Page<>(1, 20);
|
|
151
|
|
- mockPage.setRecords(List.of());
|
|
152
|
|
- mockPage.setTotal(0);
|
|
153
|
|
-
|
|
154
|
|
- when(historyDataService.queryQualityHistory(query)).thenReturn(mockPage);
|
|
155
|
|
-
|
|
156
|
|
- // When
|
|
157
|
|
- ResponseEntity<Page<WaterQuality>> response = historyDataController.queryQualityHistory(query);
|
|
158
|
|
-
|
|
159
|
|
- // Then
|
|
160
|
|
- assertNotNull(response);
|
|
161
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
162
|
|
- assertNotNull(response.getBody());
|
|
163
|
|
- assertEquals(0, response.getBody().getRecords().size());
|
|
164
|
|
- verify(historyDataService, times(1)).queryQualityHistory(query);
|
|
|
111
|
+ void generateWaterQualityReport_mapsReportFields() {
|
|
|
112
|
+ when(reportService.generateWaterQualityReport(any(), any(), any())).thenReturn(sampleReport());
|
|
|
113
|
+ R<Map<String, Object>> r = controller.generateWaterQualityReport("周报", null, null);
|
|
|
114
|
+ assertEquals(200, r.getCode());
|
|
|
115
|
+ assertEquals("WATER-VOL-1", r.getData().get("reportNo"));
|
|
165
|
116
|
}
|
|
166
|
117
|
|
|
167
|
118
|
@Test
|
|
168
|
|
- @DisplayName("区域水量聚合统计-正常响应")
|
|
169
|
|
- void testAggregateQuantityByArea_Success() {
|
|
170
|
|
- // Given
|
|
171
|
|
- LocalDateTime start = LocalDateTime.of(2026, 1, 1, 0, 0);
|
|
172
|
|
- LocalDateTime end = LocalDateTime.of(2026, 1, 31, 23, 59);
|
|
173
|
|
-
|
|
174
|
|
- // Mock返回聚合数据
|
|
175
|
|
- when(historyDataService.aggregateQuantityByArea(start, end, null))
|
|
176
|
|
- .thenReturn(List.of(
|
|
177
|
|
- Map.of("area", "城东", "avg_flow_rate", new BigDecimal("120.5"), "record_count", 1000L),
|
|
178
|
|
- Map.of("area", "城西", "avg_flow_rate", new BigDecimal("95.8"), "record_count", 850L)
|
|
179
|
|
- ));
|
|
180
|
|
-
|
|
181
|
|
- // When
|
|
182
|
|
- ResponseEntity<List<Map<String, Object>>> response = historyDataController.aggregateQuantityByArea(start, end, null);
|
|
183
|
|
-
|
|
184
|
|
- // Then
|
|
185
|
|
- assertNotNull(response);
|
|
186
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
187
|
|
- assertNotNull(response.getBody());
|
|
188
|
|
- assertEquals(2, response.getBody().size());
|
|
189
|
|
- assertEquals("城东", response.getBody().get(0).get("area"));
|
|
190
|
|
- assertEquals(1000L, response.getBody().get(0).get("record_count"));
|
|
191
|
|
- verify(historyDataService, times(1)).aggregateQuantityByArea(start, end, null);
|
|
|
119
|
+ void generateAlarmStatisticsReport_mapsReportFields() {
|
|
|
120
|
+ when(reportService.generateAlarmStatisticsReport(any(), any(), any())).thenReturn(sampleReport());
|
|
|
121
|
+ R<Map<String, Object>> r = controller.generateAlarmStatisticsReport("周报", null, null);
|
|
|
122
|
+ assertEquals(200, r.getCode());
|
|
192
|
123
|
}
|
|
193
|
124
|
|
|
194
|
125
|
@Test
|
|
195
|
|
- @DisplayName("区域水量聚合统计-指定区域")
|
|
196
|
|
- void testAggregateQuantityByArea_WithSpecificArea() {
|
|
197
|
|
- // Given
|
|
198
|
|
- LocalDateTime start = LocalDateTime.of(2026, 1, 1, 0, 0);
|
|
199
|
|
- LocalDateTime end = LocalDateTime.of(2026, 1, 31, 23, 59);
|
|
200
|
|
- String area = "城东";
|
|
201
|
|
-
|
|
202
|
|
- when(historyDataService.aggregateQuantityByArea(start, end, area))
|
|
203
|
|
- .thenReturn(List.of(
|
|
204
|
|
- Map.of("area", "城东", "avg_flow_rate", new BigDecimal("120.5"), "record_count", 1000L)
|
|
205
|
|
- ));
|
|
206
|
|
-
|
|
207
|
|
- // When
|
|
208
|
|
- ResponseEntity<List<Map<String, Object>>> response = historyDataController.aggregateQuantityByArea(start, end, area);
|
|
209
|
|
-
|
|
210
|
|
- // Then
|
|
211
|
|
- assertNotNull(response);
|
|
212
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
213
|
|
- assertNotNull(response.getBody());
|
|
214
|
|
- assertEquals(1, response.getBody().size());
|
|
215
|
|
- assertEquals("城东", response.getBody().get(0).get("area"));
|
|
216
|
|
- verify(historyDataService, times(1)).aggregateQuantityByArea(start, end, area);
|
|
|
126
|
+ void generateComprehensiveReport_mapsReportFields() {
|
|
|
127
|
+ when(reportService.generateComprehensiveReport(any(), any(), any())).thenReturn(sampleReport());
|
|
|
128
|
+ R<Map<String, Object>> r = controller.generateComprehensiveReport("周报", null, null);
|
|
|
129
|
+ assertEquals(200, r.getCode());
|
|
217
|
130
|
}
|
|
218
|
131
|
|
|
219
|
132
|
@Test
|
|
220
|
|
- @DisplayName("导出水量数据-正常响应")
|
|
221
|
|
- void testExportQuantityData_Success() {
|
|
222
|
|
- // Given
|
|
223
|
|
- HistoricalQuery query = new HistoricalQuery();
|
|
224
|
|
- query.setDataType("quantity");
|
|
225
|
|
- query.setArea("城东");
|
|
226
|
|
- query.setStartTime(LocalDateTime.of(2026, 1, 1, 0, 0));
|
|
227
|
|
- query.setEndTime(LocalDateTime.of(2026, 1, 31, 23, 59));
|
|
228
|
|
-
|
|
229
|
|
- // Mock返回导出结构
|
|
230
|
|
- Map<String, Object> exportData = Map.of(
|
|
231
|
|
- "dataType", "quantity",
|
|
232
|
|
- "area", "城东",
|
|
233
|
|
- "headers", List.of("监测点", "区域", "流量(m³/h)", "压力(MPa)", "采集时间"),
|
|
234
|
|
- "data", List.of(
|
|
235
|
|
- List.of("城东水厂", "城东", "120.5", "0.35", "2026-01-15 10:00:00")
|
|
236
|
|
- ),
|
|
237
|
|
- "fileName", "水量数据_20260101_20260131.xlsx"
|
|
238
|
|
- );
|
|
239
|
|
-
|
|
240
|
|
- when(historyDataService.queryForExport(query)).thenReturn(exportData);
|
|
|
133
|
+ void generateReport_exceptionReturnsFail() {
|
|
|
134
|
+ when(reportService.generateWaterVolumeReport(any(), any(), any()))
|
|
|
135
|
+ .thenThrow(new RuntimeException("gen error"));
|
|
|
136
|
+ R<Map<String, Object>> r = controller.generateWaterVolumeReport("周报", null, null);
|
|
|
137
|
+ assertEquals(500, r.getCode());
|
|
|
138
|
+ assertTrue(r.getMessage().contains("水量汇总报表生成失败"));
|
|
|
139
|
+ }
|
|
241
|
140
|
|
|
242
|
|
- // When
|
|
243
|
|
- ResponseEntity<Map<String, Object>> response = historyDataController.exportQuantityData(query);
|
|
|
141
|
+ // ---------- 报表列表/详情/发布 ----------
|
|
244
|
142
|
|
|
245
|
|
- // Then
|
|
246
|
|
- assertNotNull(response);
|
|
247
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
248
|
|
- assertNotNull(response.getBody());
|
|
249
|
|
- assertEquals("quantity", response.getBody().get("dataType"));
|
|
250
|
|
- assertEquals("城东", response.getBody().get("area"));
|
|
251
|
|
- assertNotNull(response.getBody().get("headers"));
|
|
252
|
|
- assertNotNull(response.getBody().get("data"));
|
|
253
|
|
- assertNotNull(response.getBody().get("fileName"));
|
|
254
|
|
- verify(historyDataService, times(1)).queryForExport(query);
|
|
|
143
|
+ @Test
|
|
|
144
|
+ void getReports_mapsList() {
|
|
|
145
|
+ when(reportService.listReports(any(), any())).thenReturn(List.of(sampleReport(), sampleReport()));
|
|
|
146
|
+ R<List<Map<String, Object>>> r = controller.getReports("water_volume", null);
|
|
|
147
|
+ assertEquals(200, r.getCode());
|
|
|
148
|
+ assertEquals(2, r.getData().size());
|
|
|
149
|
+ assertEquals("WATER-VOL-1", r.getData().get(0).get("reportNo"));
|
|
255
|
150
|
}
|
|
256
|
151
|
|
|
257
|
152
|
@Test
|
|
258
|
|
- @DisplayName("导出水质数据-正常响应")
|
|
259
|
|
- void testExportQualityData_Success() {
|
|
260
|
|
- // Given
|
|
261
|
|
- HistoricalQuery query = new HistoricalQuery();
|
|
262
|
|
- query.setDataType("quality");
|
|
263
|
|
- query.setStartTime(LocalDateTime.of(2026, 1, 1, 0, 0));
|
|
264
|
|
- query.setEndTime(LocalDateTime.of(2026, 1, 31, 23, 59));
|
|
265
|
|
-
|
|
266
|
|
- // Mock返回导出结构
|
|
267
|
|
- Map<String, Object> exportData = Map.of(
|
|
268
|
|
- "dataType", "quality",
|
|
269
|
|
- "headers", List.of("监测点", "pH值", "浊度(NTU)", "是否合格", "采集时间"),
|
|
270
|
|
- "data", List.of(
|
|
271
|
|
- List.of("水厂出口", "7.2", "0.5", "合格", "2026-01-15 10:00:00")
|
|
272
|
|
- ),
|
|
273
|
|
- "fileName", "水质数据_20260101_20260131.xlsx"
|
|
274
|
|
- );
|
|
275
|
|
-
|
|
276
|
|
- when(historyDataService.queryForExport(query)).thenReturn(exportData);
|
|
|
153
|
+ void getReportDetail_found() {
|
|
|
154
|
+ when(reportService.getReport(1L)).thenReturn(sampleReport());
|
|
|
155
|
+ R<Map<String, Object>> r = controller.getReportDetail(1L);
|
|
|
156
|
+ assertEquals(200, r.getCode());
|
|
|
157
|
+ assertEquals("水量汇总报表", r.getData().get("title"));
|
|
|
158
|
+ }
|
|
277
|
159
|
|
|
278
|
|
- // When
|
|
279
|
|
- ResponseEntity<Map<String, Object>> response = historyDataController.exportQualityData(query);
|
|
|
160
|
+ @Test
|
|
|
161
|
+ void getReportDetail_notFound() {
|
|
|
162
|
+ when(reportService.getReport(999L)).thenReturn(null);
|
|
|
163
|
+ R<Map<String, Object>> r = controller.getReportDetail(999L);
|
|
|
164
|
+ assertEquals(500, r.getCode());
|
|
|
165
|
+ assertTrue(r.getMessage().contains("报表不存在"));
|
|
|
166
|
+ }
|
|
280
|
167
|
|
|
281
|
|
- // Then
|
|
282
|
|
- assertNotNull(response);
|
|
283
|
|
- assertEquals(HttpStatus.OK, response.getStatusCode());
|
|
284
|
|
- assertNotNull(response.getBody());
|
|
285
|
|
- assertEquals("quality", response.getBody().get("dataType"));
|
|
286
|
|
- assertNotNull(response.getBody().get("headers"));
|
|
287
|
|
- assertNotNull(response.getBody().get("data"));
|
|
288
|
|
- assertNotNull(response.getBody().get("fileName"));
|
|
289
|
|
- verify(historyDataService, times(1)).queryForExport(query);
|
|
|
168
|
+ @Test
|
|
|
169
|
+ void publishReport_invokesService() {
|
|
|
170
|
+ doNothing().when(reportService).publishReport(1L);
|
|
|
171
|
+ R<String> r = controller.publishReport(1L);
|
|
|
172
|
+ assertEquals(200, r.getCode());
|
|
|
173
|
+ verify(reportService).publishReport(1L);
|
|
290
|
174
|
}
|
|
291
|
|
-}
|
|
|
175
|
+}
|
bot_dev3
bot_dev1